import cv2
import numpy as np
import matplotlib.pyplot as plt
# 配置中文字體顯示（可選）
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

# 讀取圖像（支持多種格式）
img = cv2.imread('example.jpg')
print(f"圖像尺寸: {img.shape}")  # 輸出：(高度, 寬度, 通道數(shù))
# 顯示圖像
cv2.imshow('OpenCV Window', img)
cv2.waitKey(0)  # 等待按鍵
cv2.destroyAllWindows()
# 保存圖像
cv2.imwrite('output.jpg', img)  # 自動(dòng)保存為JPG格式

# 讀取并顯示圖像
plt.figure(figsize=(10,5))
img_plt = plt.imread('example.png')  # 自動(dòng)歸一化到[0,1]
plt.subplot(121)
plt.imshow(img_plt)
plt.title('Matplotlib顯示')
# OpenCV與Matplotlib顏色空間差異
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)  # BGR轉(zhuǎn)RGB
plt.subplot(122)
plt.imshow(img_rgb)
plt.title('OpenCV轉(zhuǎn)RGB顯示')
plt.show()

# 訪問(wèn)特定位置像素值
px = img[100, 50]  # 獲取坐標(biāo)(50,100)處像素值
print(f"Blue={px[0]}, Green={px[1]}, Red={px[2]}")
# 修改像素值
img[100, 50] = [0, 255, 0]  # 將該位置改為綠色

# 提取ROI區(qū)域（感興趣區(qū)域）
roi = img[50:150, 100:200]
# 批量修改像素值
img[200:300, 300:400] = [0, 0, 255]  # 填充紅色矩形

print(f"最大像素值: {img.max()}")
print(f"最小像素值: {img.min()}")
print(f"平均像素值: {img.mean()}")

# 分離通道（OpenCV方式）
b, g, r = cv2.split(img)
cv2.imshow('Blue Channel', b)
# 合并通道
merged = cv2.merge([b, g, r])
# 直接操作特定通道（Numpy方式）
img_copy = img.copy()
img_copy[:, :, 0] = 0  # 清空藍(lán)色通道

# 顯示各通道灰度圖
plt.figure(figsize=(12, 3))
for i, ch in enumerate(['Blue', 'Green', 'Red']):
    plt.subplot(1,3,i+1)
    plt.imshow(cv2.split(img)[i], cmap='gray')
    plt.title(f'{ch}通道')
    plt.axis('off')
plt.show()

def image_processing_pipeline(path):
    # 1. 圖像讀取
    src = cv2.imread(path)
    if src is None:
        print("錯(cuò)誤：無(wú)法讀取圖像！")
        return
    # 2. 通道分析
    b, g, r = cv2.split(src)
    # 3. 創(chuàng)建特殊效果
    special_effect = src.copy()
    special_effect[:, :, 1] = cv2.add(special_effect[:, :, 1], 50)  # 增強(qiáng)綠色通道
    # 4. 結(jié)果展示
    plt.figure(figsize=(15, 5))
    plt.subplot(141)
    plt.imshow(cv2.cvtColor(src, cv2.COLOR_BGR2RGB))
    plt.title('原圖')
    plt.subplot(142)
    plt.imshow(r, cmap='gray')
    plt.title('紅色通道')
    plt.subplot(143)
    plt.imshow(special_effect[:, :, [2,1,0]])
    plt.title('增強(qiáng)綠色通道')
    plt.subplot(144)
    plt.imshow(b, cmap='viridis')  # 使用不同色圖
    plt.title('藍(lán)色通道（Viridis色圖）')
    plt.tight_layout()
    plt.show()
# 執(zhí)行處理流程
image_processing_pipeline('test_image.jpg')

# 推薦方式（Numpy向量化）
start_time = cv2.getTickCount()
brighter = cv2.add(img, np.array([30]))
print(f"耗時(shí)：{(cv2.getTickCount() - start_time)/cv2.getTickFrequency():.4f}s")
# 不推薦方式（雙重循環(huán)）
start_time = cv2.getTickCount()
for y in range(img.shape[0]):
    for x in range(img.shape[1]):
        img[y,x] = np.clip(img[y,x] + 30, 0, 255)
print(f"耗時(shí)：{(cv2.getTickCount() - start_time)/cv2.getTickFrequency():.4f}s")

# 檢查數(shù)組內(nèi)存布局
if not img.flags.c_contiguous:
    img = np.ascontiguousarray(img)

# 統(tǒng)一顏色空間轉(zhuǎn)換
def show_image統(tǒng)一(img_bgr, title='Image'):
    plt.imshow(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB))
    plt.title(title)
    plt.axis('off')
    plt.show()

# 金字塔下采樣
pyramid = cv2.pyrDown(img)
print(f"下采樣后尺寸：{pyramid.shape}")