AiDA_Python/app/service/design_fast/item.py

from app.service.design_fast.pipeline import LoadImage, KeyPoint, Segmentation, Color, PrintPainting, Scaling, Split, LoadBodyImage, ContourDetection, NoSegPrintPainting


class BaseItem:
    def __init__(self, data, basic):
        self.result = data.copy()
        self.result['name'] = data['type'].lower()
        self.result.pop("type")
        self.result.update(basic)
        self.result['design_type'] = basic.get('design_type', None)


class OthersItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.Others_pipeline = [
            LoadImage(minio_client),
            Segmentation(minio_client),
            Color(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.Others_pipeline:
            self.result = item(self.result)
        return self.result


class TopItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.top_pipeline = [
            LoadImage(minio_client),
            KeyPoint(),
            Segmentation(minio_client),
            # BackPerspective(minio_client),
            Color(minio_client),
            NoSegPrintPainting(minio_client),
            PrintPainting(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.top_pipeline:
            self.result = item(self.result)
        return self.result


class BottomItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.bottom_pipeline = [
            LoadImage(minio_client),
            KeyPoint(),
            ContourDetection(),
            Segmentation(minio_client),
            # BackPerspective(minio_client),
            Color(minio_client),
            NoSegPrintPainting(minio_client),
            PrintPainting(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.bottom_pipeline:
            self.result = item(self.result)
        return self.result


"""merge"""


class OthersMergeItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.Others_pipeline = [
            LoadImage(minio_client),
            # KeyPoint(),
            # ContourDetection(),
            Segmentation(minio_client),
            # BackPerspective(minio_client),
            Color(minio_client),
            # NoSegPrintPainting(minio_client),
            # PrintPainting(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.Others_pipeline:
            self.result = item(self.result)
        return self.result


class TopMergeItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.top_pipeline = [
            LoadImage(minio_client),
            KeyPoint(),
            Segmentation(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.top_pipeline:
            self.result = item(self.result)
        return self.result


class BottomMergeItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.bottom_pipeline = [
            LoadImage(minio_client),
            KeyPoint(),
            Segmentation(minio_client),
            Scaling(),
            Split(minio_client)
        ]

    def process(self):
        for item in self.bottom_pipeline:
            self.result = item(self.result)
        return self.result


class BodyItem(BaseItem):
    def __init__(self, data, basic, minio_client):
        super().__init__(data, basic)
        self.top_pipeline = [
            LoadBodyImage(minio_client),
        ]

    def process(self):
        for item in self.top_pipeline:
            self.result = item(self.result)
        return self.result