LuatOS/olddemo/camera/spi_cam/main.lua

PROJECT = "camerademo"
VERSION = "1.0.0"
-- 实际使用时选1个就行
require "bf30a2"
require "gc032a"
require "gc0310"
require "gc2145"
sys = require("sys")
log.style(1)

local SCAN_MODE = 0 -- 写1演示扫码
local scan_pause = true
local getRawStart = false
local RAW_MODE = 0 -- 写1演示获取原始图像
local DONE_WITH_CLOSE = false
local MIPI_MODE = false
if hmeta.chip() == "UIS8910" then   --Air722测试要把VLCD电压拉高到3.2V
    MIPI_MODE = true
    pm.ioVol(pm.IOVOL_LCD, 3200)
end
-- SCAN_MODE和RAW_MODE都没有写1就是拍照

-- 根据不同的BSP返回不同的值
-- spi_id,pin_reset,pin_dc,pin_cs,bl
local function lcd_pin()
    local rtos_bsp = rtos.bsp()
    local chip_type = hmeta.chip()
    if string.find(rtos_bsp,"EC718") or string.find(chip_type,"EC718") then
        return lcd.HWID_0, 36, 0xff, 0xff, 0xff -- 注意:EC718P有硬件lcd驱动接口, 无需使用spi,当然spi驱动也支持
    elseif string.find(chip_type,"UIS8910") then
        return lcd.HWID_0, 0xff, 0xff, 0xff, 0xff
    else
        log.info("main", "bsp not support")
        return
    end
end
local spi_id, pin_reset, pin_dc, pin_cs, bl = lcd_pin()
if spi_id ~= lcd.HWID_0 then
    spi_lcd = spi.deviceSetup(spi_id, pin_cs, 0, 0, 8, 20 * 1000 * 1000, spi.MSB, 1, 0)
    port = "device"
else
    port = spi_id
end

--[[ 此为合宙售卖的1.8寸TFT LCD LCD 分辨率:128X160 屏幕ic:st7735 购买地址:https://item.taobao.com/item.htm?spm=a1z10.5-c.w4002-24045920841.19.6c2275a1Pa8F9o&id=560176729178]]
-- lcd.init("st7735",{port = port,pin_dc = pin_dc, pin_pwr = bl, pin_rst = pin_reset,direction = 0,w = 128,h = 160,xoffset = 0,yoffset = 0},spi_lcd)

--[[ 此为合宙售卖的1.54寸TFT LCD LCD 分辨率:240X240 屏幕ic:st7789 购买地址:https://item.taobao.com/item.htm?spm=a1z10.5-c.w4002-24045920841.20.391445d5Ql4uJl&id=659456700222]]
-- lcd.init("st7789",{port = port,pin_dc = pin_dc, pin_pwr = bl, pin_rst = pin_reset,direction = 0,w = 240,h = 320,xoffset = 0,yoffset = 0},spi_lcd,true)

--[[ 此为合宙售卖的0.96寸TFT LCD LCD 分辨率:160X80 屏幕ic:st7735s 购买地址:https://item.taobao.com/item.htm?id=661054472686]]
-- lcd.init("st7735v",{port = port,pin_dc = pin_dc, pin_pwr = bl, pin_rst = pin_reset,direction = 1,w = 160,h = 80,xoffset = 0,yoffset = 24},spi_lcd)
-- 如果显示颜色相反，请解开下面一行的注释，关闭反色
-- lcd.invoff()
-- 如果显示依旧不正常，可以尝试老版本的板子的驱动
-- lcd.init("st7735s",{port = port,pin_dc = pin_dc, pin_pwr = bl, pin_rst = pin_reset,direction = 2,w = 160,h = 80,xoffset = 0,yoffset = 0},spi_lcd)
if MIPI_MODE then
    lcd.init("st7796", {
        port = port,
        pin_dc = pin_dc,
        pin_pwr = bl,
        pin_rst = pin_reset,
        direction = 0,
        w = 320,
        h = 480,
        xoffset = 0,
        yoffset = 0
    }, spi_lcd, true)
    log.info("init st7796")
else
    lcd.init("gc9306x", {
        port = port,
        pin_dc = pin_dc,
        pin_pwr = bl,
        pin_rst = pin_reset,
        direction = 0,
        w = 240,
        h = 320,
        xoffset = 0,
        yoffset = 0
    }, spi_lcd, true)
end

local uartid = uart.VUART_0 -- 根据实际设备选取不同的uartid
-- 初始化
local result = uart.setup(uartid, -- 串口id
    115200, -- 波特率
    8, -- 数据位
    1 -- 停止位
)

local function cb(id, str)
    if type(str) == 'string' then
        log.info("扫码结果", str)
    elseif str == false then
        log.error("摄像头没有数据")
    else
        log.info("摄像头数据", str)
        sys.publish("capture done", true)
    end
end


local function press_key()
    log.info("boot press")
    sys.publish("PRESS", true)
end
if hmeta.chip() == "UIS8910" then
    gpio.setup(9, press_key, gpio.PULLDOWN, gpio.RISING)
    gpio.debounce(9, 100, 1)
else
    gpio.setup(0, press_key, gpio.PULLDOWN, gpio.RISING)
    gpio.debounce(0, 100, 1)
end

local rawbuff, err
if RAW_MODE ~= 1 then
    if MIPI_MODE then
        rawbuff, err = zbuff.create(400 * 1024, 0, zbuff.HEAP_AUTO) -- gc2145
    else
        rawbuff, err = zbuff.create(60 * 1024, 0, zbuff.HEAP_AUTO)
    end

else
    rawbuff, err = zbuff.create(640 * 480 * 2, 0, zbuff.HEAP_AUTO) -- gc032a
    -- local rawbuff = zbuff.create(240 * 320 * 2, zbuff.HEAP_AUTO)  --bf302a
end
if rawbuff == nil then
    log.info(err)
end

local function device_init()
    local cspiId, i2cId
    if hmeta.chip() == "UIS8910" then
        cspiId, i2cId = 0,1
        i2c.setup(i2cId, i2c.FAST)
        if MIPI_MODE then
            gpio.setup(11, 1) -- 722 mipi camera reset ctrl by gpio11
            return gc2145Init(cspiId, i2cId, 24000000, SCAN_MODE, SCAN_MODE)
        else
            return gc0310Init(cspiId, i2cId, 25000000, SCAN_MODE, SCAN_MODE)
        end
    else
        cspiId, i2cId = 1,0
        i2c.setup(i2cId, i2c.FAST)
        -- return bf30a2Init(cspiId,i2cId,25500000,SCAN_MODE,SCAN_MODE)
        -- return gc032aInit(cspiId,i2cId,24000000,SCAN_MODE,SCAN_MODE)
        return gc0310Init(cspiId, i2cId, 25500000, SCAN_MODE, SCAN_MODE)
    end
    


end

local function pdwn(id, level)
    if hmeta.chip() == "UIS8910" then
        camera.pwdn_pin(id, level)
    else
        gpio.setup(5, level) -- PD拉低
    end
end
local tx_buff = zbuff.create(16384)
local function send_to_pc(uart_id, buff)
    
    local pos = 0
    log.info("需要发送", buff:used(), "byte")
    while pos < buff:used() do
        if buff:used() - pos > 16384 then
            tx_buff:copy(0, buff, pos, 16384)
            pos = pos + 16384
        else
            tx_buff:copy(0, buff, pos, buff:used() - pos)
            pos = buff:used()
        end
        log.info("发送", tx_buff:used(), "byte")
        uart.tx(uart_id, tx_buff)
        tx_buff:del()
        sys.wait(5)
    end
end

sys.taskInit(function()
    log.info("摄像头启动")


    local camera_id = device_init()
	camera.on(camera_id, "scanned", cb)
    pdwn(camera_id, 0)
    if DONE_WITH_CLOSE then
        camera.close(camera_id)
    else
        camera.stop(camera_id)
        camera.preview(camera_id, true)
    end
    log.info("按下boot开始测试")
    log.info(rtos.meminfo("sys"))
    log.info(rtos.meminfo("psram"))
    while 1 do
        result, data = sys.waitUntil("PRESS", 30000)
        if result == true and data == true then
            if SCAN_MODE == 1 then
                if scan_pause then
                    log.info("启动扫码")
                    if DONE_WITH_CLOSE then
                        camera_id = device_init()
                    end
                    camera.start(camera_id)
                    scan_pause = false
                    sys.wait(200)
                    log.info(rtos.meminfo("sys"))
                    log.info(rtos.meminfo("psram"))
                else
                    log.info("停止扫码")
                    if DONE_WITH_CLOSE then
                        camera.close(camera_id)
                    else
                        camera.stop(camera_id)
                    end
                    scan_pause = true
                    sys.wait(200)
                    log.info(rtos.meminfo("sys"))
                    log.info(rtos.meminfo("psram"))
                end
            elseif RAW_MODE == 1 then
                if getRawStart == false then
                    getRawStart = true
                    log.debug("摄像头首次捕获原始图像")
                    if DONE_WITH_CLOSE then
                        camera_id = device_init()
                    end
                    camera.startRaw(camera_id, 640, 480, rawbuff) -- gc032a
                    -- camera.startRaw(camera_id,240,320,rawbuff) --bf302a
                else
                    log.debug("摄像头继续捕获原始图像")
                    camera.getRaw(camera_id)
                end
                result, data = sys.waitUntil("capture done", 30000)
                if DONE_WITH_CLOSE then
                    camera.close(camera_id)
                else
                    camera.stop(camera_id)
                end
                log.info("摄像头捕获原始图像完成")
                log.info(rtos.meminfo("sys"))
                log.info(rtos.meminfo("psram"))
                send_to_pc(uartid, rawbuff) --找个能保存数据的串口工具保存成文件就能在电脑上看了, 格式为JPG    
                rawbuff:del()
            else
                log.debug("摄像头拍照")
                if DONE_WITH_CLOSE then
                    camera_id = device_init()
                end
                camera.capture(camera_id, rawbuff, 1) -- 2和3需要非常多非常多的psram,尽量不要用
                result, data = sys.waitUntil("capture done", 30000)
                log.info(rawbuff:used())
                if DONE_WITH_CLOSE then
                    camera.close(camera_id)
                else
                    camera.stop(camera_id)
                end
                send_to_pc(uartid, rawbuff) -- 找个能保存数据的串口工具保存成文件就能在电脑上看了, 格式为JPG
                rawbuff:del()
                log.info(rtos.meminfo("sys"))
                log.info(rtos.meminfo("psram"))
            end

        end
    end

end)

-- 用户代码已结束---------------------------------------------
-- 结尾总是这一句
sys.run()
-- sys.run()之后后面不要加任何语句!!!!!