# -*- coding: utf-8 -*-

"""
图像处理
"""

import re
import sys
import math
import time
from .base import AipBase
from .base import base64
from .base import json
from .base import urlencode
from .base import quote

class AipImageProcess(AipBase):

    """
    图像处理
    """

    __imageQualityEnhanceUrl = 'https://aip.baidubce.com/rest/2.0/image-process/v1/image_quality_enhance'

    __dehazeUrl = 'https://aip.baidubce.com/rest/2.0/image-process/v1/dehaze'

    __contrastEnhanceUrl = 'https://aip.baidubce.com/rest/2.0/image-process/v1/contrast_enhance'

    __colourizeUrl = 'https://aip.baidubce.com/rest/2.0/image-process/v1/colourize'

    __stretchRestoreUrl = 'https://aip.baidubce.com/rest/2.0/image-process/v1/stretch_restore'

    __styleTrans = "https://aip.baidubce.com/rest/2.0/image-process/v1/style_trans"

    __inpainting = "https://aip.baidubce.com/rest/2.0/image-process/v1/inpainting"

    __imageDefinitionEnhance = "https://aip.baidubce.com/rest/2.0/image-process/v1/image_definition_enhance"

    __selfieAnime = "https://aip.baidubce.com/rest/2.0/image-process/v1/selfie_anime"

    __skySeg = "https://aip.baidubce.com/rest/2.0/image-process/v1/sky_seg"

    __colorEnhances = "https://aip.baidubce.com/rest/2.0/image-process/v1/color_enhance"

    __removeMoireV1Url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/remove_moire'
    __customizeStylizationV1Url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/customize_stylization'
    __docRepairV1Url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/doc_repair'
    __denoiseV1Url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/denoise'



    def imageQualityEnhance(self, image, options=None):
        """
            图像无损放大
        """
        options = options or {}

        data = {}
        data['image'] = base64.b64encode(image).decode()

        data.update(options)

        return self._request(self.__imageQualityEnhanceUrl, data)
    
    def dehaze(self, image, options=None):
        """
            图像去雾
        """
        options = options or {}

        data = {}
        data['image'] = base64.b64encode(image).decode()

        data.update(options)

        return self._request(self.__dehazeUrl, data)
    
    def contrastEnhance(self, image, options=None):
        """
            图像对比度增强
        """
        options = options or {}

        data = {}
        data['image'] = base64.b64encode(image).decode()

        data.update(options)

        return self._request(self.__contrastEnhanceUrl, data)
    
    def colourize(self, image, options=None):
        """
            黑白图像上色
        """
        options = options or {}

        data = {}
        data['image'] = base64.b64encode(image).decode()

        data.update(options)

        return self._request(self.__colourizeUrl, data)
    
    def stretchRestore(self, image, options=None):
        """
            拉伸图像恢复
        """
        options = options or {}

        data = {}
        data['image'] = base64.b64encode(image).decode()

        data.update(options)

        return self._request(self.__stretchRestoreUrl, data)


    def selfieAnime(self, image, options=None):
        """
            人像动漫化
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__selfieAnime, data)

    def imageDefinitionEnhance(self, image, options=None):
        """
            图像清晰度增强
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__imageDefinitionEnhance, data)

    def styleTrans(self, image, options=None):
        """
            图像风格转换
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__styleTrans, data)


    def skySeg(self, image, options=None):
        """
            天空分割
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__skySeg, data)

    def inpaintingByMask(self, image, rectangle, options=None):
        """
            图像修复
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data['rectangle'] = rectangle
        data.update(options)
        return self._request(self.__inpainting, data)


    def removeMoireV1(self, image, options=None):
        """
            图片去摩尔纹
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/ql4wdlnc0
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__removeMoireV1Url, data)

    def removeMoireV1Url(self, url, options=None):
        """
            图片去摩尔纹 - url
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/ql4wdlnc0
        """
        options = options or {}
        data = {}
        data['url'] = url
        data.update(options)
        return self._request(self.__removeMoireV1Url, data)

    def removeMoireV1Pdf(self, pdf, options=None):
        """
            图片去摩尔纹 - pdf
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/ql4wdlnc0
        """
        options = options or {}
        data = {}
        data['pdf_file'] = base64.b64encode(pdf).decode()
        data.update(options)
        return self._request(self.__removeMoireV1Url, data)

    def customizeStylizationV1(self, image, options=None):
        """
            图像风格自定义
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/al50vf6bq
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__customizeStylizationV1Url, data)

    def customizeStylizationV1Url(self, url, options=None):
        """
            图像风格自定义 - url
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/al50vf6bq
        """
        options = options or {}
        data = {}
        data['url'] = url
        data.update(options)
        return self._request(self.__customizeStylizationV1Url, data)

    def docRepairV1(self, image, options=None):
        """
            文档图片去底纹
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/Nl6os53ab
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data.update(options)
        return self._request(self.__docRepairV1Url, data)

    def docRepairV1Url(self, url, options=None):
        """
            文档图片去底纹 - url
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/Nl6os53ab
        """
        options = options or {}
        data = {}
        data['url'] = url
        data.update(options)
        return self._request(self.__docRepairV1Url, data)

    def denoiseV1(self, image, option, options=None):
        """
            图像去噪
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/Tl78sby7g
        """
        options = options or {}
        data = {}
        data['image'] = base64.b64encode(image).decode()
        data['option'] = option
        data.update(options)
        return self._request(self.__denoiseV1Url, data)

    def denoiseV1Url(self, url, option, options=None):
        """
            图像去噪 - url
            接口使用说明文档: https://ai.baidu.com/ai-doc/IMAGEPROCESS/Tl78sby7g
        """
        options = options or {}
        data = {}
        data['url'] = url
        data['option'] = option
        data.update(options)
        return self._request(self.__denoiseV1Url, data)