import cv2
import numpy as np


class Photo:
    def __init__(self, filepath):
        self.img = cv2.imread(filepath)

    def image(self):
        return self.img

    # 逆时针旋转90度, img的shape会变
    def rotate(self):
        rows, cols, depth = self.img.shape
        newImage = np.zeros((cols, rows, 3), np.uint8)
        for r in range(0, rows):
            for c in range(0, cols):
                newImage[cols - 1 - c, r] = self.img[r, c]
        self.img = newImage

    # 水平镜像
    def horizontal_mirror(self):
        self.img = cv2.flip(self.img, 1, dst=None)

    # 水平镜像
    def vertical_mirror(self):
        self.img = cv2.flip(self.img, 0, dst=None)

    # ratio: [1, 500]
    def resize(self, ratio):
        self.img = cv2.resize(self.img, dsize=None, fx=ratio / 100.0, fy=ratio / 100.0, interpolation=cv2.INTER_LINEAR)

    # value: [0,100]
    def lighting(self, value):
        # 具体做法先归一化到1，然后gamma作为指数值求出新的像素值再还原
        gamma_table = [np.power(x / 255.0, value) * 255.0 for x in range(256)]
        gamma_table = np.round(np.array(gamma_table)).astype(np.uint8)
        # 实现映射用的是Opencv的查表函数
        self.img = cv2.LUT(self.img, gamma_table)

    # value: [-100,100]
    def contrast(self, value):
        lst = value * self.img
        lst += np.mean(self.img - lst)
        np.clip(lst, 0, 255, self.img)


# value: [-100,100]

def saturate(self, value):
    pass


# value: [-100,100]
def hue(self, value):
    pass


# value: [0,100]
def sharpen(self, value):
    pass


# value: [0,100]
def blur(self, value):
    pass


def toGray(self):
    pass


def invert(self):
    pass


def toBinary(self):
    pass