generators.py

Go to the documentation of this file.

from ros_introspection.ros_generator import PRIMITIVES
 
from .util import TRAILING_PATTERN, roscompile
 
STANDARD = {
    'Header': 'std_msgs'
}
 
 
@roscompile
def fill_in_msg_package_names(package):
    all_names = set()
    gens = list(package.get_all_generators())
    for gen in gens:
        all_names.add(gen.base_name)
    for gen in gens:
        for section in gen.sections:
            for field in section.fields:
                if '/' in field.type or field.type in PRIMITIVES:
                    continue
 
                if field.type in STANDARD:
                    field.type = STANDARD[field.type] + '/' + field.type
                    gen.changed = True
                elif field.type in all_names:
                    field.type = package.name + '/' + field.type
                    gen.changed = True
 
 
@roscompile
def remove_trailing_whitespace_from_generators(package):
    for gen in package.get_all_generators():
        for i, content in enumerate(gen.contents):
            if not isinstance(content, str):
                continue
            m = TRAILING_PATTERN.match(content)
            if m:
                gen.contents[i] = m.group(1) + '\n'