python/gtsam/utils/__init__.py

Go to the documentation of this file.

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