CameraMachine/python/service/image_pic_deal.py

"""
"""

import cv2
import numpy as np
from PIL import Image, ImageDraw, ImageFont
from module.view_control.auto_deal_pics.data import DataModeAutoDealPics


class OnePicDeal(object):
    def __init__(self):
        # 数据模型
        self.data_mode_auto_deal_pics = DataModeAutoDealPics()

    def check_shoe_is_right(self, im=None, image_path=None):
        # 先进行API识别左右脚
        if im is None:
            im = Image.open(image_path)
        try:
            r_data = self.data_mode_auto_deal_pics.check_is_right_foot_by_api(image=im)
        except BaseException as e:
            r_data = None
            print("20", e)

        flag = self.check_shoe_is_right_by_pixel(im=im)
        if r_data:
            if "拖鞋" in r_data:
                flag = flag is not True
        if flag:
            print("自动识别----->这是左脚")
        else:
            print("自动识别----->这是右脚")
        return flag

    def check_shoe_is_right_by_pixel(self, im=None, image_path=None):
        if im is None:
            im = Image.open(image_path)
        # 注意，只支持透明图
        # 打开图像文件
        im = im.crop(im.getbbox())
        # image.show()
        # 获取图像第一行的像素数据
        pixel_data = im.load()
        pix_list = []
        h = int(im.height / 20)
        for i in range(im.width):
            _r, _g, _b, _a = pixel_data[i, h]
            if _a > 10:
                pix_list.append(i)

        left_f_num = 0
        middle_w = int(im.width / 2)
        for i in pix_list:
            if i < middle_w:
                left_f_num += 1
            else:
                left_f_num -= 1
        if left_f_num > 0:
            return True
        else:
            return False

    def how_to_use(self):
        # 单图缩放处理
        a = {"command": "resize",
             "plugins_mode": "relative",  # pixel 相对（宽度、高度、其他参考图），或绝对像素
             "base_im": {"im": "im"},  # 参考基于其他图 PIL格式
             "base": "width",  # base:pixel,width,height,by_im 基于长边、基于短边  （基于短边时，则缩放到指定尺寸）by_im确保能塞进参考图内
             "value": 649,  # 固定值，如果为百分比，则为0
             "percentage": 0, }  # 百分比

        # 单图圆角处理
        a = {"command": "radius",  # radius
             "plugins_mode": "relative",  # pixel 相对（短边），或绝对像素
             "circular_pos": (0, 1, 0, 1),  # 从左上角顺时针，记录圆角数量
             "value": 649,  # 固定值，如果为百分比，则为0
             "percentage": 0, }  # 百分比

        # 单图处理成圆形
        a = {"command": "circular",  # circular
             }

        # 单图旋转处理
        a = {"command": "rotate",
             "plugins_mode": "",
             "value": 649,  # 固定值 顺时针
             }

        # 图片粘贴处理
        a = {
            "command": "paste_img",
            "img": {"im": "im"},
            "pos": {"plugins_mode": "relative",  # pixel
                    "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
                    "value": (100, 100),
                    "percentage": (0.5, 0.5),
                    },
            "margins": (0, 0, 0, 0),  # 上下左右边距
        }

        # 图片剪裁处理
        a = {
            "command": "crop_img",
            "img": {"im": "im"},
            "pos": {"plugins_mode": "relative",  # pixel
                    "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
                    "value": (100, 100, 10, 10),
                    },
            "color_fill": (255, 255, 255)
        }

        # 图片添加文字
        a = {
            "command": "add_text",
            "pos": {"plugins_mode": "relative",  # pixel
                    "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
                    "value": (100, 100),
                    "percentage": 0, },
            "font": "",
            "text": "",
            "anchor": "",  # mm 为居中 ma 为左右居中，上面居顶
            "align": "对齐方式",
            "direction": "文本的方向",
            "max_len_one_line": "单行长度",
            "spacing": 10,
            "fill": "文字颜色",
        }

    def add_text(self, img: Image, command):
        draw_1 = ImageDraw.Draw(img)
        # 定义字体，你需要有一个.ttf字体文件
        font = command["font"]
        text = command["text"]
        spacing = 4 if not command["spacing"] else command["spacing"]
        fill = command["fill"]
        anchor = None if not command["anchor"] else command["anchor"]
        align = "left" if not command["align"] else command["align"]  # left, center 或 right
        _, _, text_width, text_height = draw_1.textbbox((0, 0), text, font=font)

        xy = (0, 0)
        if command["pos"]["plugins_mode"] == "pixel":
            value = command["pos"]["value"]
            xy = value

        draw_1.multiline_text(xy, text,
                              fill=fill,
                              font=font,
                              anchor=anchor,
                              spacing=spacing,
                              align=align,
                              direction=None,
                              features=None,
                              language=None,
                              stroke_width=0,
                              stroke_fill=None,
                              embedded_color=False)
        return img

    def resize(self, img: Image, command):
        if command["plugins_mode"] == "pixel":
            if command["base"] == "width":
                img = self.to_resize(img, width=command["value"])
            if command["base"] == "high":
                img = self.to_resize(img, height=command["value"])
        # 相对值
        if command["plugins_mode"] == "relative":
            base_im = command["base_im"]["im"]
            if command["base"] == "width":
                img = self.to_resize(img, width=img.width * command["percentage"] if not base_im else int(
                    base_im.width * command["percentage"]))
            if command["base"] == "height":
                img = self.to_resize(img, width=img.height * command["percentage"] if not base_im else int(
                    base_im.height * command["percentage"]))
            # by_im确保能塞进参考图内
            if command["base"] == "by_im":
                percentage = 1 if not command["percentage"] else command["percentage"]
                box_width, box_height = int(base_im.width * percentage), int(base_im.height * percentage)
                width, height = img.width, img.height
                if box_width / box_height < width / height:
                    scale = box_width / width
                else:
                    scale = box_height / height
                img = img.resize((int(width * scale), int(height * scale)))
        # img.show()
        return img

    def paste_img(self, img: Image, command):
        # 粘贴绝对像素
        base = "nw" if not command["pos"]["base"] else command["pos"]["base"]
        value = command["pos"]["value"]
        percentage = (0, 0) if not command["pos"]["percentage"] else command["pos"]["percentage"]

        if command["margins"]:
            top, down, left, right = command["margins"]
        else:
            top, down, left, right = 0, 0, 0, 0

        if percentage != (0, 0):  # percentage 不按占比模式
            if base in ("nw", "wn", "wc", "cw", "nc", "cn", "center"):
                value = (int(img.width), int(img.height))
            if base in ("sw", "ws", "sc", "cs", "center"):
                value = (int(img.width), -1 * int(img.height))
            if base in ("ec", "ce"):
                value = (int(img.width), int(img.height))

        img_1 = command["img"]["im"]
        if command["pos"]["plugins_mode"] == "pixel":
            if base == "ec" or "ce":
                p_x = int(img.width - img_1.width) + value[0]
                p_y = value[1]
            if base == "nw" or "wn":
                deviation_x, deviation_y = 0, 0
                p_x, p_y = value

            if base == "cs" or "sc":
                p_x, p_y = value
            if base == "center":
                deviation_x, deviation_y = int((img.width - img_1.width) / 2), int((img.height - img_1.height) / 2)
                p_x = deviation_x + value[0] + left
                p_y = deviation_y + value[1] + top
            if base == "sw" or base == "ws":
                # deviation_x, deviation_y = 0, int((img.height - img_1.height))
                p_x = value[0] + left
                p_y = img.height - (img_1.height + value[1] + down)
            if base == "wc" or base == "cw":
                p_x = value[0] + left
                p_y = int((img.height - img_1.height) / 2) + value[1] + top

            try:
                img.paste(img_1, (p_x, p_y), img_1)
            except:
                img.paste(img_1, (p_x, p_y), img_1.convert("RGBA"))

        return img

    def crop_img(self, img: Image, command):
        base = "nw" if not command["pos"]["base"] else command["pos"]["base"]
        # print(base)
        value = command["pos"]["value"]
        percentage = command["pos"]["percentage"]

        if command["margins"]:
            top, down, left, right = command["margins"]
        else:
            top, down, left, right = 0, 0, 0, 0

        out_img_size = (value[2], value[3])
        # 默认填充色
        color_fill = command["color_fill"]
        if not color_fill:
            color_fill = (0, 0, 0)

        if command["pos"]["plugins_mode"] == "pixel":
            if base == "nw" or "wn":
                box = value
            if base == "sw" or base == "ws":
                # deviation_x, deviation_y = 0, int((img.height - img_1.height))
                box = (value[0], img.height - (value[1] + value[3]), value[2], value[3])
                print(box)

            if base == "se" or base == "es":
                box = (img.width - (value[0] + value[2]), img.height - (value[1] + value[3]), value[2], value[3])
                print(box)
            box = [box[0], box[1], box[0] + box[2], box[1] + box[3]]

        print("box", box)
        print("img.width", img.width)
        print("img.height", img.height)

        out_img = img.crop(box=box)
        print(out_img.size)
        # out_img.show()
        if box[0] < 0:
            out_img.paste(Image.new("RGB", (-1 * box[0], out_img.height), color_fill), (0, 0))
        # print(img.width, box)

        if box[2] > img.width:
            # print(box[2] - img.width, img.height)
            i = Image.new("RGB", (box[2] - img.width, out_img.height), color_fill)
            out_img.paste(i, (img.width - box[0], 0))
        if box[1] < 0:
            out_img.paste(Image.new("RGB", (img.width, -1 * box[1]), color_fill), (0, 0))

        if box[3] > img.height:
            out_img.paste(Image.new("RGB", (out_img.width, box[3] - img.height), color_fill),
                          (0, img.height - box[1]))

        # bg_img = Image.new("RGB", out_img_size, color_fill)
        # if box[0] < 0:
        #     x = -1 * box[0]
        # elif box[2] > img.width:
        #     x = img.width - box[2]
        # else:
        #     x = 0
        #
        # if box[1] < 0:
        #     y = -1 * box[1]
        # elif box[2] > img.height:
        #     y = img.height - box[1]
        # else:
        #     y = 0
        # bg_img.paste(img, box=(x, y), )

        # print(box)
        # img = img.crop(box=box)
        # out_img.show()
        return out_img

    def radius(self, img: Image, command):
        # 单图圆角处理

        radii = command["value"]
        if radii > img.width / 2:
            radii = int(img.width / 2)
        if radii > img.height / 2:
            radii = int(img.height / 2)

        # 画圆（用于分离4个角）
        circle = Image.new('L', (radii * 2, radii * 2), 0)  # 创建一个黑色背景的画布
        draw = ImageDraw.Draw(circle)
        draw.ellipse((0, 0, radii * 2, radii * 2), fill=255)  # 画白色圆形

        # 原图
        img = img.convert("RGBA")
        w, h = img.size

        # 画4个角（将整圆分离为4个部分）
        alpha = Image.new('L', img.size, 255)
        _pos = command["circular_pos"]
        if not _pos:
            _pos = (1, 1, 1, 1)
        for index, i in enumerate(_pos):
            if index == 0 and i == 1:
                alpha.paste(circle.crop((0, 0, radii, radii)), (0, 0))  # 左上角
            if index == 1 and i == 1:
                alpha.paste(circle.crop((radii, 0, radii * 2, radii)), (w - radii, 0))  # 右上角
            if index == 2 and i == 1:
                alpha.paste(circle.crop((radii, radii, radii * 2, radii * 2)), (w - radii, h - radii))  # 右下角
            if index == 3 and i == 1:
                alpha.paste(circle.crop((0, radii, radii, radii * 2)), (0, h - radii))  # 左下角
        # alpha.show()
        img.putalpha(alpha)  # 白色区域透明可见，黑色区域不可见
        return img

    def to_resize(self, _im, width=None, height=None):
        _im_x, _im_y = _im.size
        if width and height:
            if _im_x >= _im_y:
                height = None
            else:
                width = None
        if width:
            re_x = int(width)
            re_y = int(_im_y * re_x / _im_x)
        else:
            re_y = int(height)
            re_x = int(_im_x * re_y / _im_y)
        _im = _im.resize((re_x, re_y))
        return _im

    def add_pic(self, detailed_images):
        if not detailed_images:
            return
        page_len = 0
        for index, im in enumerate(detailed_images):
            page_len += im.height
        bg_im = Image.new("RGB", (im.width, page_len), (255, 255, 255))

        n = 0
        for index, im in enumerate(detailed_images):
            bg_im.paste(im, (0, n))
            n += im.height
        # bg_im.show()
        return bg_im

    def get_goods_pos(self, im, cut_image):
        # 保留多余内容
        old_x, old_y = im.size
        x1, y1, x2, y2 = cut_image.getbbox()
        goods_w, goods_h = x2 - x1, y2 - y1
        _w, _h = int(goods_w / 10), int(goods_h / 10)  # 上下左右扩展位置
        new_x1, new_y1, new_x2, new_y2 = x1 - _w, y1 - _h, x2 + _w, y2 + _h  # 防止超限
        new_x1 = 0 if new_x1 < 0 else new_x1
        new_y1 = 0 if new_y1 < 0 else new_y1
        new_x2 = old_x if new_x2 > old_x else new_x2
        new_y2 = old_y if new_y2 > old_y else new_y2
        # 剪切掉多余的内容,保留阴影
        im = im.crop((new_x1, new_y1, new_x2, new_y2))  # 切图
        return im

    def deal_one_pic(self, data=None, is_show=False):
        """
        通用图片处理器
        1、图片位置处理
        输出拼接后的图片，以及拼接后商品所属位置
        """
        for command in data:
            if command["command"] == "paste_img":
                img1 = command["img1"]["im"]
                img2 = command["img2"]["im"]
                if "resize" in command["img2"]:
                    if "width" in command["img2"]["resize"]:
                        img2 = self.to_resize(img2, width=command["img2"]["resize"]["width"])
                        if is_show:
                            img2.show()
                img1.paste(img2, command["img2"]["pos"])
            if command["command"] == "image_crop":
                img1 = img1.crop(command["image_crop"])
        img = img1
        return img

    def deal_one_pic_2(self, orign_im, data=None):
        """
        通用图片处理器
        1、基于某原始图，进行加工，包括粘贴、单图处理等逻辑
        """
        # 单图缩放处理
        a = {"command": "resize",
             "plugins_mode": "relative",  # pixel 相对（宽度、高度、其他参考图），或绝对像素
             "base_im": "im",  # 参考基于其他图 PIL格式
             "base": "width",  # base:pixel,width,height,by_long_side,by_short_side基于长边、基于短边  （基于短边时，则缩放到指定尺寸）
             "value": 649,  # 固定值，如果为百分比，则为0
             "percentage": 0, }  # 百分比

        # 单图圆角处理
        a = {"command": "radius",  # radius
             "plugins_mode": "relative",  # pixel 相对（短边），或绝对像素
             "circular_pos": (0, 1, 0, 1),  # 从左上角顺时针，记录圆角数量
             "value": 649,  # 固定值，如果为百分比，则为0
             "percentage": 0, }  # 百分比

        # 单图处理成圆形
        a = {"command": "circular",  # circular
             }

        # 单图旋转处理
        a = {"command": "rotate",
             "plugins_mode": "",
             "value": 649,  # 固定值 顺时针
             }

        # 图片粘贴处理
        a = {
            "command": "paste_img",
            "img": {"im": "im"},
            "pos": {"plugins_mode": "relative",  # pixel
                    "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
                    "value": (100, 100),
                    "percentage": 0, },
            "margins": (0, 0, 0, 0),  # 上下左右边距
        }

        # 图片剪裁处理
        a = {
            "command": "crop_img",
            "img": {"im": "im"},
            "plugins_mode": "relative",  # pixel
            "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
            "box": (0, 0, 0, 0),
        }

        # 图片添加文字
        a = {
            "command": "add_text",
            "pos": {"plugins_mode": "relative",  # pixel
                    "base": "center",  # nw,nc,ne,ec ... 各个方向参考点
                    "value": (100, 100),
                    "percentage": 0, },
            "font": "",
            "text": "",
            "align": "对齐方式",
            "direction": "文本的方向",
            "fill": "文字颜色",
        }

        # 图片渐变处理
        # https://blog.csdn.net/skying159/article/details/119532479

        def get_value(data_dict, key):
            return 1

        r_dict = {"nw": 0,
                  }

        _r = 0
        for command in data:
            if command["command"] == "paste_img":
                img1 = command["img1"]["im"]
                img2 = command["img2"]["im"]
                if "resize" in command["img2"]:
                    if "width" in command["img2"]["resize"]:
                        img2 = self.to_resize(img2, width=command["img2"]["resize"]["width"])
                if "pos" in command["img2"]:
                    base = command["img2"]["base"]
                    value = command["img2"]["value"]
                    if command["img2"]["plugins_mode"] == "relative":
                        # 相对位置处理，居中比较特殊
                        # 其他的先进行旋转与镜像，处理后，进行反向操作
                        x, y = 0, 0
                        if base == "center":
                            x, y = int((img1.width - img2.width) / 2), int((img1.height - img2.height) / 2)

                        if base == "nw" or base == "nw":
                            pass

                        w, h = img1.width * value[0], img1.height * value[0]

                img1.paste(img2, command["img2"]["pos"])
                if _r != 0:
                    img1 = img1.rotate(_r * -1)
                continue
            if command["command"] == "image_crop":
                img1 = img1.crop(command["image_crop"])
                continue
        img = img1
        return img

    def get_overlay_pic(self, pil_img_1, pil_img_2, color):
        im_w, im_h = pil_img_1.size
        cv_im = cv2.cvtColor(np.asarray(pil_img_1), cv2.COLOR_RGB2BGR)

        # 创建一张纯色底图
        image_white = Image.new("RGB", (im_w, im_h), color)

        cv_image_white = cv2.cvtColor(np.asarray(image_white), cv2.COLOR_RGB2BGR)
        new_im = self.to_color_2(cv_image_white, cv_im)
        # new_im = cv2.addWeighted(new_im, 0.7, cv_im_2, 0.3, 0)
        # new_im = cv2.add(new_im, cv_im_2)

        new_im = Image.fromarray(cv2.cvtColor(new_im, cv2.COLOR_BGR2RGB))
        new_im.paste(pil_img_2, (0, 0), pil_img_2)
        return new_im

    def to_color_2(self, target, blend):  # 正片叠底
        return np.array(np.multiply(target / 256, blend / 256) * 256, dtype=np.uint8)

    def to_color_1(self, img1, img2):  # PS颜色模式
        img1 = cv2.cvtColor(img1, cv2.COLOR_BGR2HSV)
        img2 = cv2.cvtColor(img2, cv2.COLOR_BGR2HSV)
        img2[:, :, 0] = img1[:, :, 0]
        img2[:, :, 1] = img1[:, :, 1]
        res = cv2.cvtColor(img2, cv2.COLOR_HSV2BGR)
        return res


if __name__ == '__main__':
    from collections import defaultdict


    def set_dict(one_dict):
        _ = defaultdict(str)
        for i, v in one_dict.items():
            if isinstance(v, dict):
                v = set_dict(v)
            _[i] = v
        return _


    def deal_one_pic(img: Image, data=None):
        """
        通用图片处理器
        1、基于某原始图，进行加工，包括粘贴、单图处理等逻辑
        """
        data = [set_dict(x) for x in data]

        for command in data:
            if command["command"] == "resize":
                img = OnePicDeal().resize(img, command)
                continue
            if command["command"] == "paste_img":
                img = OnePicDeal().paste_img(img, command)
                continue
            if command["command"] == "crop_img":
                img = OnePicDeal().crop_img(img, command)
                continue
            if command["command"] == "radius":
                img = OnePicDeal().radius(img, command)
                continue
            if command["command"] == "add_text":
                img = OnePicDeal().add_text(img, command)
                continue

        return img


    image_path = r"D:\MyDocuments\PythonCode\MyPython\red_dragonfly\deal_pics\auto_capture_V2\IPC\output\-秋季订货会夏季补充给葛明明\NUM24106316\阴影图处理\NUM24106316（3）_后跟_阴影.png"
    to_paste_img = Image.open(image_path)
    data = []
    bg_color = (246, 246, 246)
    # 单图缩放处理
    view = "后跟"
    data.append({"command": "resize",
                 "plugins_mode": "pixel",  # pixel 相对（宽度、高度、其他参考图），或绝对像素
                 "base_im": {"im": ""},  # 参考基于其他图 PIL格式
                 "base": "width",  # base:pixel,width,height,by_im 基于长边、基于短边  （基于短边时，则缩放到指定尺寸）by_im确保能塞进参考图内
                 "value": 1300 if view == "后跟" else 2200,  # 固定值，如果为百分比，则为0
                 "percentage": 0, })  # 百分比

    view_dict = {"俯视": (0, 0, 1100, 1200),
                 "内里": (-1, -100, 1100, 1200),
                 "后跟": (0, -100, 1100, 1200),
                 "鞋底": (0, -100, 1100, 1200),
                 }

    data.append({
        "command": "crop_img",
        "img": "",
        "pos": {"plugins_mode": "pixel",  # pixel
                "base": "sw",  # nw,nc,ne,ec ... 各个方向参考点
                "value": view_dict[view],
                },
        "color_fill": bg_color,
    })

    # 处理圆角
    data.append({"command": "radius",  # radius
                 "plugins_mode": "relative",  # pixel 相对（短边），或绝对像素
                 "circular_pos": (1, 0, 0, 1),  # 从左上角顺时针，记录圆角数量
                 "value": 100,  # 固定值，如果为百分比，则为0
                 "percentage": 0, })  # 百分比

    to_paste_img = deal_one_pic(to_paste_img, data)
    # print(to_paste_img.size)

    # 粘贴到白底图上
    img = Image.new("RGB", (1200, 1316), (255, 255, 255))
    data = []
    data.append({
        "command": "paste_img",
        "img": {"im": to_paste_img},
        "pos": {"plugins_mode": "pixel",  # pixel  relative
                "base": "wc",  # nw,nc,ne,ec,... 各个方向参考点
                "value": (100, 0),
                "percentage": "",
                },
        "margins": (0, 0, 0, 0),  # 上下左右边距
    })
    img = deal_one_pic(img, data)
    img.show()