from PIL import Image

import numpy as np

#封装一个图像处理类

class TestNumpy(object):

    def photo2paint(self,img_url):

        # 读取图片

        my_photo = np.asarray(Image.open(img_url).convert("L")).astype("float")

        # print(my_photo.shape)

        #设置灰度阈值(范围0~100)

        depth = 40

        # 将灰度系数映射到numpy

        grad = np.gradient(my_photo)

        #获取坐标

        grad_x,grad_y = grad

        #分别处理

        grad_x = grad_x * depth/100

        grad_y = grad_y * depth/100

        #设置阿尔法值  设置坐标范围

        A = np.sqrt(grad_x**2 + grad_y**2 + 1)

        #重新赋值

        uni_x = grad_x/A

        uni_y = grad_y/A

        uni_z = 1/A

        #灰度光源设置 角度

        vec_el = np.pi/2.2

        vec_az = np.pi/4

        #设置x轴和y轴

        dx = np.cos(vec_el)* np.cos(vec_az)

        dy = np.cos(vec_el)* np.sin(vec_az)

        #设置z轴

        dz = np.sin(vec_el)

        #设置byte值

        b = 255*(dx*uni_x+dy*uni_y+dz*uni_z)

        #光源归一化

        b = b.clip(0,255)

        #声明图像类图像

        im = Image.fromarray(b.astype("uint8"))

        #保存图像

        im.save("./test_new.jpg")

if __name__ == "__main__":

    testnumpy = TestNumpy()

    testnumpy.photo2paint('./test_numpy.jpg')