23 lines
678 B
Python
23 lines
678 B
Python
import glob
|
|
import os.path as osp
|
|
|
|
|
|
def findExampleDataFile(name):
|
|
"""
|
|
Find the example data file specified by `name`.
|
|
"""
|
|
# This is okay since gtsam will already be loaded
|
|
# before this function is accessed. Thus no effect.
|
|
import gtsam
|
|
|
|
site_package_path = gtsam.__path__[0]
|
|
# add the * at the end to glob the entire directory
|
|
data_path = osp.join(site_package_path, "Data", "*")
|
|
|
|
extensions = ("", ".graph", ".txt", ".out", ".xml", ".g2o")
|
|
|
|
for data_file_path in glob.glob(data_path, recursive=True):
|
|
for ext in extensions:
|
|
if (name + ext) == osp.basename(data_file_path):
|
|
return data_file_path
|