"""
AI-Camera (Inherit C module)

MicroPython library for the AI-Camera(Inherit C module)
=======================================================
@dahanzimin From the Mixly Team
"""

import esp_ai
from micropython import const

CODE_DETECTION      = const(0)
COLOR_DETECTION     = const(1)
MOTION_DEECTION     = const(2)
CAT_FACE_DETECTION  = const(3)
FACE_DETECTION      = const(4)
FACE_RECOGNITION    = const(5)

class AI:
    def __init__(self, function):
        self._func = function
        self._ai = None
        self._once = True

    def _init(self, *args):
        if self._func == CODE_DETECTION:
            self._ai = esp_ai.code_recognition()
        elif self._func == COLOR_DETECTION:
            self._ai = esp_ai.color_detection(color=args[0])
        elif self._func == MOTION_DEECTION:
            self._ai = esp_ai.motion_recognition(threshold=args[0])
        elif self._func == CAT_FACE_DETECTION:
            self._ai = esp_ai.cat_detection()
        elif self._func == FACE_DETECTION:
            self._ai = esp_ai.face_detection()
        elif self._func == FACE_RECOGNITION:
            self._ai = esp_ai.face_recognition()
        else:
            raise AttributeError('AI model is not supported')
        self._ai.start()    #启动检测，可以通过LCD观察结果
        self._once = False

    def _result(self, res, _t, _s=0, _n=0): #_s:第几个, _n:细分第几个
        if not res: return None
        if _t == 'len':
            return  res[0]
        elif _t == 'pos':
            if len(res) >= (5 + _s * 4):
                return res[(1 + _s * 4):(5 + _s * 4)]
        elif _t == 'keypoint':
            if len(res) >= (7 + _s * 14 + _n * 2):
                return res[(5 + _s * 14 + _n * 2):(7 + _s * 14 + _n * 2)]

    def code_recognition(self):
        if self._func == CODE_DETECTION:
            if self._once: self._init()
            return self._ai.read()
        else:
            raise AttributeError('This model can only run QR code detection')

    def color_detection(self, color=0, event='pos', num=0):
        if self._func == COLOR_DETECTION:
            if self._once: self._init(color)
            return self._result(self._ai.read(), event, num)
        else:
            raise AttributeError('This model can only run color detection')

    def motion_recognition(self, threshold=50):
        if self._func == MOTION_DEECTION:
            if self._once: self._init(threshold)
            return bool(self._ai.read() >= threshold)
        else:
            raise AttributeError('This model can only run motion recognition')

    def cat_detection(self, event='pos', num=0):
        if self._func == CAT_FACE_DETECTION:
            if self._once: self._init()
            return self._result(self._ai.read(), event, num)
        else:
            raise AttributeError('This model can only run cat face detection')

    def face_detection(self, event='pos', num=0, point=0):
        if self._func == FACE_DETECTION:
            if self._once: self._init()
            return self._result(self._ai.read(), event, num, point)
        else:
            raise AttributeError('This model can only run face detection')

    def face_recognition(self, event='pos', num=0, point=0):
        if self._func == FACE_RECOGNITION:
            if self._once: self._init()
            return self._result(self._ai.recognize(), event, num, point)
        else:
            raise AttributeError('This model can only run face recognition')

    def face_enroll(self):
        if self._func == FACE_RECOGNITION:
            return self._ai.enroll()    
        else:
            raise AttributeError('This model can only run face recognition')

    def face_delete(self, _id):
        if self._func == FACE_RECOGNITION:
            self._ai.delete(_id)  
        else:
            raise AttributeError('This model can only run face recognition')

    def stop(self):
        if self._ai is not None:
            self._ai.stop()