Source code for hutts_verification.image_preprocessing.pipeline_builder

"""
The class that is responsible for creating a Pipeline. This class is the Builder
of the Builder design pattern.
"""

from hutts_verification.image_preprocessing.pipeline import Pipeline

__author__ = "Nicolai van Niekerk"
__copyright__ = "Copyright 2017, Java the Hutts"
__license__ = "BSD"
__maintainer__ = "Nicolai van Niekerk"
__email__ = "nicvaniek@gmail.com"
__status__ = "Development"


[docs]class PipelineBuilder: """ The PipelineBuilder will assemble all the parts of the Pipeline. """ def __init__(self): """ Initialize Builder. """ self.pipeline = Pipeline()
[docs] def set_blur_manager(self, value): """ This function adds the specified blur manager to the pipeline. :param value (BlurManager): BlurManager object to be added. """ self.pipeline.blur_manager = value
[docs] def set_color_manager(self, value): """ This function adds the specified color manager to the pipeline. :param value (ColorManager): ColorManager object to be added. """ self.pipeline.color_manager = value
[docs] def set_threshold_manager(self, value): """ This function adds the specified threshold manager to the pipeline. :param value (ThresholdManager): ThresholdManager object to be added. """ self.pipeline.threshold_manager = value
[docs] def set_face_detector(self, value): """ This function adds the specified face detector to the pipeline. :param value (FaceDetector): FaceDetector object to be added. """ self.pipeline.face_detector = value
[docs] def get_result(self): """ This function returns the fully-assembled pipeline. Returns: (obj): The Pipeline. """ return self.pipeline