'''
实验名称：颜色识别
版本： v1.0
实验目的：实现单个颜色识别
'''

import sensor,lcd,time

lcd.init(freq=15000000)                 #初始化LCD
sensor.reset()                          #复位和初始化摄像头
sensor.set_vflip(1)                     #将摄像头设置成后置方式（所见即所得）
sensor.set_pixformat(sensor.RGB565)     # 设置像素格式为彩色 RGB565
#sensor.set_pixformat(sensor.GRAYSCALE)  # 设置像素格式为灰色
sensor.set_framesize(sensor.QVGA)       # 设置帧大小为 QVGA (320x240)
sensor.set_auto_gain(False)
sensor.set_auto_whitebal(False)
sensor.set_auto_gain(0,0)
sensor.skip_frames(time = 200)          # 等待设置生效
clock = time.clock()                    # 创建一个时钟来追踪 FPS（每秒拍摄帧数）

#lcd初始化
lcd.init()

clock=time.clock()

# 颜色识别阈值 (L Min, L Max, A Min, A Max, B Min, B Max) LAB模型
# 下列阈值元组用来识别 红、绿、蓝三种颜色，可通过调整数据阈值完成更多颜色的识别。
# thresholds = [(59, 100, 40, 127, 5, 127), # 红色阈值
#               (87, 100, -59, 127, -10, 127), # 绿色阈值
#               (0, 30, 0, 64, -128, -20)] # 蓝色阈值
thresholds = [(35, 100, 6, 127, 0, 127), # 红色阈值
              (25, 100, -6, 127, 5, 43), # 黄色阈值
              (87, 100, -59, 127, -10, 127)] # 绿色阈值

while True:

    clock.tick()

    img=sensor.snapshot()
    img = img.resize(224, 224)

    blobs = img.find_blobs([thresholds[1]]) # 0,1,2分别表示红，黄，绿色。
    if blobs:
        print(blobs)
        for b in blobs:
            print(b)
            tmp=img.draw_rectangle(b[0:4])
            tmp=img.draw_cross(b[5], b[6])

    lcd.display(img)     #LCD显示图片
    print(clock.fps())   #打印FPS