import cv2

# 读
img = cv2.imread('img.jpg')
img = cv2.imread('img.jpeg', flags=cv2.IMREAD_UNCHANGED)


# 显
cv2.imshow(winname='title', mat=img)
cv2.waitKey(delay=0) # 单位是毫秒，0代表无限等待。若在等待时间内按下任意键则返回按键的ASCII码，程序继续运行。若没有按下任何键，超时后返回-1
cv2.destroyAllWindows()
# cv2.destroyWindow('title') # 也可以


# 写
cv2.imwrite('filename',img)
# 可以指定图像质量
cv2.imwrite(filename='img.jpg', img=img, params=[cv2.IMWRITE_JPEG_QUALITY, i]) # i是0～100的整数，表示图像质量，默认95

cv2.imwrite(filename='img.png', img=img, params=[cv2.IMWRITE_PNG_COMPRESSION, i]) # i是0～9，表示压缩级别，默认3

# 灰度化，并且变成单通道
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 灰度转RGB（实际仍然是灰度）
img_rgb = cv2.cvtColor(img_grey, code=cv2.COLOR_GRAY2RGB)
# 可以转来转去：
# cv2.COLOR_X2Y，其中X,Y = RGB, BGR, GRAY, HSV, YCrCb, XYZ, Lab, Luv, HLS
# BGR

# 二值化
retval, dst = cv2.threshold(img_gray, thresh=127, maxval=255, type=0)
# retval 是域值，dst 是二值化后的图
# 可以输入单通道/3通道，对应返回单通道/3通道

img_flip = cv2.flip(img, flipCode=0)
# flipCode>0 左右翻转
# flipCode=0 上下翻转
# flipCode<0 上下左右同时翻转

# 入参分别是图片、文本、像素、字体、字体大小、颜色、字体粗细
imgzi = cv2.putText(img=img, text="guofei", org=(500, 500),
                    fontFace=font, fontScale=2.5, color=(0, 0, 0), thickness=2)

import numpy as np
import cv2

np.set_printoptions(threshold='nan')
# 创建一个宽512高512的黑色画布，RGB(0,0,0)即黑色
img=np.zeros((512,512,3),np.uint8)


# 画直线,图片对象，起始坐标(x轴,y轴)，结束坐标，颜色，宽度
cv2.line(img,(0,0),(311,511),(255,0,0),10)
# 画矩形，图片对象，左上角坐标，右下角坐标，颜色，宽度
cv2.rectangle(img,(30,166),(130,266),(0,255,0),3)
# 画圆形，图片对象，中心点坐标，半径大小，颜色，宽度
cv2.circle(img,(222,222),50,(255.111,111),-1)
# 画椭圆形，图片对象，中心点坐标，长短轴，顺时针旋转度数，开始角度(右长轴表0度，上短轴表270度)，颜色，宽度
cv2.ellipse(img,(333,333),(50,20),0,0,150,(255,222,222),-1)

# 画多边形，指定各个点坐标,array必须是int32类型
pts=np.array([[10,5],[20,30],[70,20],[50,10]], np.int32)
# -1表示该纬度靠后面的纬度自动计算出来，实际上是4

pts = pts.reshape((-1,1,2,))
# print(pts)
# 画多条线，False表不闭合，True表示闭合，闭合即多边形
cv2.polylines(img,[pts],True,(255,255,0),5)

#写字,字体选择
font=cv2.FONT_HERSHEY_SCRIPT_COMPLEX

# 图片对象，要写的内容，左边距，字的底部到画布上端的距离，字体，大小，颜色，粗细
cv2.putText(img,"OpenCV",(10,400),font,3.5,(255,255,255),2)

a=cv2.imwrite("picture.jpg",img)
cv2.imshow("picture",img)
cv2.waitKey(0)

cv2.destroyAllWindows()

import numpy as np
import cv2

# 定义一块宽600，高400的画布，初始化为白色
canvas = np.zeros((400, 600, 3), dtype=np.uint8) + 255

# 画一条纵向的正中央的黑色分界线
cv2.line(canvas, (300, 0), (300, 399), (0, 0, 0), 2)

# 画一条右半部份画面以150为界的横向分界线
cv2.line(canvas, (300, 149), (599, 149), (0, 0, 0), 2)

# 左半部分的右下角画个红色的圆
cv2.circle(canvas, (200, 300), 75, (0, 0, 255), 5)

# 左半部分的左下角画个蓝色的矩形
cv2.rectangle(canvas, (20, 240), (100, 360), (255, 0, 0), thickness=3)

# 定义两个三角形，并执行内部绿色填充
triangles = np.array([
    [(200, 240), (145, 333), (255, 333)],
    [(60, 180), (20, 237), (100, 237)]])
cv2.fillPoly(canvas, triangles, (0, 255, 0))

# 画一个黄色五角星
# 第一步通过旋转角度的办法求出五个顶点
phi = 4 * np.pi / 5
rotations = [[[np.cos(i * phi), -np.sin(i * phi)], [i * np.sin(phi), np.cos(i * phi)]] for i in range(1, 5)]
pentagram = np.array([[[[0, -1]] + [np.dot(m, (0, -1)) for m in rotations]]], dtype=np.float)

# 定义缩放倍数和平移向量把五角星画在左半部分画面的上方
pentagram = np.round(pentagram * 80 + np.array([160, 120])).astype(np.int)

# 将5个顶点作为多边形顶点连线，得到五角星
cv2.polylines(canvas, pentagram, True, (0, 255, 255), 9)

# 按像素为间隔从左至右在画面右半部份的上方画出HSV空间的色调连续变化
for x in range(302, 600):
    color_pixel = np.array([[[round(180*float(x-302)/298), 255, 255]]], dtype=np.uint8)
    line_color = [int(c) for c in cv2.cvtColor(color_pixel, cv2.COLOR_HSV2BGR)[0][0]]
    cv2.line(canvas, (x, 0), (x, 147), line_color)

# 如果定义圆的线宽大于半斤，则等效于画圆点，随机在画面右下角的框内生成坐标
np.random.seed(42)
n_pts = 30
pts_x = np.random.randint(310, 590, n_pts)
pts_y = np.random.randint(160, 390, n_pts)
pts = zip(pts_x, pts_y)

# 画出每个点，颜色随机
for pt in pts:
    pt_color = [int(c) for c in np.random.randint(0, 255, 3)]
    cv2.circle(canvas, pt, 3, pt_color, 5)

# 在左半部分最上方打印文字
cv2.putText(canvas,
            'Python-OpenCV Drawing Example',
            (5, 15),
            cv2.FONT_HERSHEY_SIMPLEX,
            0.5,
            (0, 0, 0),
            1)

cv2.imshow('Example of basic drawing functions', canvas)
cv2.waitKey()

from PIL import Image

# 读
img = Image.open(fp='img.png')
# 存
img.save('img_new.png')

# 显
img.show()

# 看格式
img.mode  # 'RGBA'

img.convert("RGB")
# '1': 把图像转为二值图
# 'L'：将图像转换为灰度图像。
# 'RGB'：将图像转换为 RGB 彩色图像。
# 'RGBA'：将图像转换为带有透明通道的 RGBA 图像。
# 'CMYK'：将图像转换为 CMYK 颜色模式。
# 'HSV'：将图像转换为 HSV 颜色模式。
# 'YCbCr'：将图像转换为 YCbCr 颜色模式。


img.size # 图像大小
img.info # 图像信息



# ???img.load()

img.getpixel(xy=(6, 10)) # 返回指定位置的像素值
img.putpixel(xy=(6, 10), value=(5,6,255)) # 设定像素值
img.getchannel(0)  # 某一个通道作为单独的图片返回


img.tobytes()  # 返回二进制形式
img.frombytes(my_bytes) # 二进制转图片

np.array(img) # 转 ndarray

import base64
from PIL import Image
import io
import cv2
import numpy as np

# base64 格式的图片
with open('测试图片.png', 'rb') as f:
    s_b64 = base64.b64encode(f.read()).decode('utf-8')

# 转回为 bytes 类型
image_data = base64.b64decode(s_b64)

# PIL.Image 需要借助 io.BytesIO 来读取
image = Image.open(io.BytesIO(image_data))

# opencv 需要借助 np.frombuffer 来读取
image = cv2.imdecode(
    np.frombuffer(image_data, np.uint8)
    , cv2.IMREAD_COLOR
)

from PIL import ImageEnhance

enhancer = ImageEnhance.Brightness(image=img)
img1 = enhancer.enhance(factor=0.5)  # 亮度减半

img1.show()

enhancer = ImageEnhance.Color(image=img)  # 颜色增强
ImageEnhance.Contrast(image=img)
ImageEnhance.Sharpness(image=img)