postproc_setup.py

#!usr/bin/env python
"""
Configuration for post processors
"""

import postproc_config

# Import your processor as private here:
from ABB.RAPID.rapid \
    import SimpleRAPIDProcessor \
    as __SimpleRAPIDProcessor
from KUKA.EntertainTech.entertaintech \
    import SimpleEntertainTechProcessor \
    as __SimpleEntertainTechProcessor
from KUKA.KRL.krl \
    import SimpleKRLProcessor \
    as __SimpleKRLProcessor
from GENERAL.CSV.comma_separated_vals \
    import SimpleCSVProcessor \
    as __SimpleCSVProcessor


# Add your processor to private list here:
__supported_processors = [
    __SimpleRAPIDProcessor,
    __SimpleKRLProcessor,
    __SimpleEntertainTechProcessor,
    __SimpleCSVProcessor
]


def construct_processor_name(type_robot, type_processor):
    """
    Construct the name of a single Post Processor.
    :param type_robot: Type of robot.
    :param type_processor: Type of processor.
    :return:
    """
    return '{} {}'.format(type_robot, type_processor)


def get_processor_name(processor):
    """
    Get the name from a given Post Processor
    :param processor: Processor itself.
    :return:
    """
    return construct_processor_name(processor.type_robot, processor.type_processor)


def get_processor_names():
    """
    Get the names of all processors using existing POST_PROCESSORS dict.
    Sorts names alphabetically by default. User-accessible function.
    :return: 'ROBOT_TYPE POSTPROCESSOR_TYPE'
    """
    names = POST_PROCESSORS.keys()
    names.sort()
    # Get default from config
    default = postproc_config.DEFAULT_POST_PROCESSOR
    if default in names:
        names.remove(default)
        names.insert(0, default)
    return names


# All post processors dict!
POST_PROCESSORS = {}

# Configure the dict (don't touch)
for processor in __supported_processors:
    p = processor()
    name = construct_processor_name(p.type_robot, p.type_processor)
    POST_PROCESSORS[name] = processor