import picamera2
from libcamera import controls
from PIL import Image
import cv2
import time

picam2 = picamera2.Picamera2()
picam2.configure(picam2.create_still_configuration(
    main={"size": (2048, 2048),
          "format": "RGB888"}
    # main={"size": (1800, 1800)}
))
picam2.start()  # 开始相机流
# 设置自动对焦，如果摄像头支持
try:
    picam2.set_controls({"AfMode": controls.AfModeEnum.Continuous})
except Exception as e:
    print('该摄像头无法自动对焦, 请手动对焦: {}'.format(e))

# 设置自动白平衡
try:
    picam2.set_controls({"AwbMode": controls.AwbModeEnum.Auto})  # 自动白平衡
except Exception as e:
    print('设置自动白平衡失败: {}'.format(e))

time.sleep(0.1)  # 等待相机初始化

frame = picam2.capture_array()  # 捕获图像
# 打印图像信息
print(frame)
print(frame.shape)  # 这里使用 shape 来查看图像的尺寸

# 保存图片查看结果
cv2.imwrite("test.jpg", frame)