def _find_nodes(self, settings):
pkgs = {pkg.name: pkg for pkg in self.project.packages}
ws = settings.workspace if settings else None
extractor = NodeExtractor(pkgs,
self.environment,
ws=ws,
node_cache=self.node_cache,
parse_nodes=self.parse_nodes)
if self.parse_nodes and not CppAstParser is None:
if settings is None:
CppAstParser.set_library_path()
db_dir = os.path.join(extractor.workspace, "build")
if os.path.isfile(os.path.join(db_dir,
"compile_commands.json")):
CppAstParser.set_database(db_dir)
else:
CppAstParser.set_library_path(settings.cpp_parser_lib)
CppAstParser.set_standard_includes(settings.cpp_includes)
db_dir = settings.cpp_compile_db
if (db_dir and os.path.isfile(
os.path.join(db_dir, "compile_commands.json"))):
CppAstParser.set_database(settings.cpp_compile_db)
for pkg in self.project.packages:
if not pkg.name in self.package_cache:
extractor.find_nodes(pkg)
def launch(self):
self.parse_arg()
#FIND WORKSPACE:
#Fixme: the env variable ROS_PACKAGE_PATH is not the best way to extract the workspace path
ros_pkg_path = os.environ.get("ROS_PACKAGE_PATH")
ws = ros_pkg_path[:ros_pkg_path.find("/src:")]
#BONSAI PARSER
parser = CppAstParser(workspace=ws)
parser.set_library_path("/usr/lib/llvm-4.0/lib")
parser.set_standard_includes(
"/usr/lib/llvm-4.0/lib/clang/4.0.0/include")
db_dir = os.path.join(ws, "build")
if os.path.isfile(os.path.join(db_dir, "compile_commands.json")):
parser.set_database(db_dir)
if (self.args.node):
self.extract_node(self.args.name, self.args.name,
self.args.package_name, None, ws, None)
if (self.args.launch):
pkg = Package(self.args.package_name)
rospack = rospkg.RosPack()
pkg.path = rospack.get_path(self.args.package_name)
relative_path = "launch/"
for root, dirs, files in os.walk(pkg.path):
for file in files:
if file.endswith(self.args.name):
relative_path = root.replace("/" + pkg.path, "")
source = SourceFile(self.args.name, relative_path, pkg)
pkgs = {pkg.id: pkg}
launch_parser = LaunchParser(pkgs=pkgs)
rossystem = ros_system(self.args.name.replace(".launch", ""))
try:
source.tree = launch_parser.parse(source.path)
for node in source.tree.children:
if isinstance(node, NodeTag):
name = node.attributes["name"]
node_name = node.attributes["type"]
node_pkg = node.attributes["pkg"]
try:
ns = node.attributes["ns"]
except:
ns = None
self.extract_node(name, node_name, node_pkg, ns, ws,
rossystem)
except LaunchParserError as e:
print("Parsing error in %s:\n%s", source.path, str(e))
system_str = rossystem.save_model(self.args.model_path)
if self.args.output:
print system_str
def launch(self):
self.parse_arg()
if self.args.node:
#FIND WORKSPACE:
#Fixme: the env variable ROS_PACKAGE_PATH is not the best way to extract the workspace path
ros_pkg_path = os.environ.get("ROS_PACKAGE_PATH")
ws = ros_pkg_path[:ros_pkg_path.find("/src:")]
#FIND PACKAGE PATH
pkg = Package(self.args.package_name)
rospack = rospkg.RosPack()
pkg.path = rospack.get_path(self.args.package_name)
#BONSAI PARSER
parser = CppAstParser(workspace=ws)
parser.set_library_path("/usr/lib/llvm-3.8/lib")
parser.set_standard_includes(
"/usr/lib/llvm-3.8/lib/clang/3.8.0/include")
db_dir = os.path.join(ws, "build")
if os.path.isfile(os.path.join(db_dir, "compile_commands.json")):
parser.set_database(db_dir)
#HAROS NODE EXTRACTOR
analysis = NodeExtractor(pkg,
env=dict(os.environ),
ws=ws,
node_cache=False,
parse_nodes=True)
node = Node(self.args.name, pkg, rosname=RosName(self.args.name))
srcdir = pkg.path[len(ws):]
srcdir = os.path.join(ws, srcdir.split(os.sep, 1)[0])
bindir = os.path.join(ws, "build")
#HAROS CMAKE PARSER
parser = RosCMakeParser(srcdir, bindir, pkgs=[pkg])
parser.parse(os.path.join(pkg.path, "CMakeLists.txt"))
for target in parser.executables.itervalues():
if target.output_name == self.args.name:
for file_in in target.files:
full_path = file_in
relative_path = full_path.replace(
pkg.path + "/", "").rpartition("/")[0]
file_name = full_path.rsplit('/', 1)[-1]
source_file = SourceFile(file_name, relative_path, pkg)
node.source_files.append(source_file)
parser = CppAstParser(workspace=ws)
node.source_tree = parser.global_scope
model_str = ""
for sf in node.source_files:
if parser.parse(sf.path) is None:
print "File not found"
else:
# ROS MODEL EXTRACT PRIMITIVES
rosmodel = ros_model(self.args.package_name,
self.args.name, self.args.name)
self.extract_primitives(node, parser.global_scope,
analysis, rosmodel)
# SAVE ROS MODEL
model_str = rosmodel.save_model()
if self.args.output:
print model_str
else:
text_file = open(self.args.name + ".ros", "w")
text_file.write(model_str)
text_file.close()