# -*- coding: utf-8 -*- 
""" 
@File : morphologyService.py 
@Author: csc
@Date : 2022/7/19
形态学操作
"""
import cv2
import numpy as np

structure = [cv2.MORPH_RECT, cv2.MORPH_CROSS, cv2.MORPH_ELLIPSE]


def erode(imgs, args):
    """
    腐蚀: 消除物体边界点，使目标缩小，可以消除小于结构元素的噪声点
    kernelType: int | 结构元类型 0: MORPH_RECT, 1: MORPH_CROSS, 2: MORPH_ELLIPSE
    kernelX, kernelY: int, int | (x, y) 结构元大小
    """
    kernel = cv2.getStructuringElement(structure[args['kernelType']],
                                       (int(args['kernelX']), int(args['kernelY'])))
    res = cv2.erode(imgs[0], kernel)
    return res


def dilate(imgs, args):
    """
    膨胀: 将与物体接触的所有背景点合并到物体中，使目标增大，可添补目标中的空洞
    kernelType: int | 结构元类型 0: MORPH_RECT, 1: MORPH_CROSS, 2: MORPH_ELLIPSE
    kernelSize: Tuple<int, int> | (x, y) 结构元大小
    """
    kernel = cv2.getStructuringElement(structure[int(args['kernelType'])],
                                       (int(args['kernelX']), int(args['kernelY'])))
    res = cv2.dilate(imgs[0], kernel)
    return res


def morphOpen(imgs, args):
    """
    开运算: 先腐蚀后膨胀，消除图像上细小的噪声，并平滑物体边界
    kernelType: int | 结构元类型 0: MORPH_RECT, 1: MORPH_CROSS, 2: MORPH_ELLIPSE
    kernelSize: Tuple<int, int> | (x, y) 结构元大小
    """
    kernel = cv2.getStructuringElement(structure[int(args['kernelType'])],
                                       (int(args['kernelX']), int(args['kernelY'])))
    res = cv2.morphologyEx(imgs[0], cv2.MORPH_OPEN, kernel)
    return res


def morphClose(imgs, args):
    """
    闭运算: 先膨胀后腐蚀，填充物体内细小的空洞，并平滑物体边界
    kernelType: int | 结构元类型 0: MORPH_RECT, 1: MORPH_CROSS, 2: MORPH_ELLIPSE
    kernelSize: Tuple<int, int> | (x, y) 结构元大小
    """
    kernel = cv2.getStructuringElement(structure[int(args['kernelType'])],
                                       (int(args['kernelX']), int(args['kernelY'])))
    res = cv2.morphologyEx(imgs[0], cv2.MORPH_CLOSE, kernel)
    return res