Skip to content
Snippets Groups Projects
Commit b40df3a6 authored by Guilhem Saurel's avatar Guilhem Saurel
Browse files

[Python] add more paths to getModelPath, fix #37

and document their purpose
parent a00409a3
No related branches found
No related tags found
No related merge requests found
Pipeline #10445 passed
EXAMPLE_ROBOT_DATA_MODEL_DIR = "${CMAKE_INSTALL_PREFIX}/share/${PROJECT_NAME}/robots" EXAMPLE_ROBOT_DATA_MODEL_DIR = "${CMAKE_INSTALL_PREFIX}/share/${PROJECT_NAME}/robots"
EXAMPLE_ROBOT_DATA_SOURCE_DIR = "${PROJECT_SOURCE_DIR}/robots"
...@@ -3,7 +3,6 @@ import warnings ...@@ -3,7 +3,6 @@ import warnings
from os.path import dirname, exists, join from os.path import dirname, exists, join
import numpy as np import numpy as np
import pinocchio as pin import pinocchio as pin
from pinocchio.robot_wrapper import RobotWrapper from pinocchio.robot_wrapper import RobotWrapper
...@@ -11,13 +10,16 @@ pin.switchToNumpyArray() ...@@ -11,13 +10,16 @@ pin.switchToNumpyArray()
def getModelPath(subpath, printmsg=False): def getModelPath(subpath, printmsg=False):
source = dirname(dirname(dirname(__file__))) # top level source directory
paths = [ paths = [
join(dirname(dirname(dirname(dirname(__file__)))), 'robots'), join(dirname(dirname(dirname(source))), 'robots'), # function called from "make release" in build/ dir
join(dirname(dirname(dirname(__file__))), 'robots') join(dirname(source), 'robots'), # function called from a build/ dir inside top level source
join(source, 'robots') # function called from top level source dir
] ]
try: try:
from .path import EXAMPLE_ROBOT_DATA_MODEL_DIR from .path import EXAMPLE_ROBOT_DATA_MODEL_DIR, EXAMPLE_ROBOT_DATA_SOURCE_DIR
paths.append(EXAMPLE_ROBOT_DATA_MODEL_DIR) paths.append(EXAMPLE_ROBOT_DATA_MODEL_DIR) # function called from installed project
paths.append(EXAMPLE_ROBOT_DATA_SOURCE_DIR) # function called from off-tree build dir
except ImportError: except ImportError:
pass pass
paths += [join(p, '../../../share/example-robot-data/robots') for p in sys.path] paths += [join(p, '../../../share/example-robot-data/robots') for p in sys.path]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment