zfbox/core/ModelNodeManager.py

import multiprocessing

from threading import Lock
from core.ModelNode import ModelNode
from myutils.ConfigManager import myCongif

class ModelNodeManger:
    def __init__(self,device_id,model_id,model_path,threshold=0.5,iou_thres=0.5):
        self.ch_count = 0       #当前模型关联的通道数量
        self.channel_list = {}  #channel_id modelNodle
        self.clist_Lock = Lock()  # count的维护锁
        self.modelNode_list = []  # model_Node
        self.brun = multiprocessing.Value('b',True) #brun.value = False,brun.value = True
        self.m_p_status = multiprocessing.Value('i',0)
        self.model = None  # 模型对象
        #-------
        self.max_count = myCongif.get_data('maxCount')
        self.device_id = device_id
        self.model_id = model_id
        self.model_path = model_path
        self.threshold = threshold
        self.iou_thres = iou_thres


    # #添加数据
    # def pro_add_data(self,channel_id,data):
    #     pass
    #     # #加锁？  -- 对于多线程这里的耗时会不会有点大。。。 model_nodel 反馈出去？？？？？getModle_Nodel
    #     # with self.clist_Lock:
    #     #     if channel_id in self.channel_list:
    #     #         self.modelNode_list[channel_id].pro_add_data(data)

    #暴露modelNodel -- 避免pro_add_data锁竞争
    def getModle_Nodel(self,channel_id):
        model_nodel = None
        if channel_id in self.channel_list:
            model_nodel = self.channel_list[channel_id]
        return model_nodel

    def start_model_th(self, channel_id, out_mq):
        with self.clist_Lock:
            if channel_id in self.channel_list: #已经在List 说明已经启动过了，但不应该重复调用
                print("不应该重复调用start_model_th")
                return True
            else:  # 还没入list ，查找可以插入的model_node
                for model_node in self.modelNode_list:
                    if model_node.ch_count < self.max_count:  # 没满可以插入
                        model_node.start_model_th(channel_id, out_mq)   #+1
                        self.channel_list[channel_id] = model_node      #新增 channel_di 与model_node的对应关系
                        return True
                # 执行到这说明没有空的model_nodel -- 需要新建
                model_node = ModelNode(self.device_id,self.model_path,self.max_count)
                model_node.start_model_th(channel_id, out_mq)  # 启动线程，并+1
                #两个数据结构，新增节点
                self.modelNode_list.append(model_node)      #新增一个model_node节点
                self.channel_list[channel_id] = model_node  #新增 channel_di 与model_node的对应关系
                #行动完更新数量
                self.ch_count +=1
                return True

    def stop_model_th(self, channel_id):
        with self.clist_Lock:
            if channel_id in self.channel_list:
                model_node = self.channel_list[channel_id]
                model_node.stop_model_th(channel_id)    #减-，根据情况停止工作进程、
                # 需要考虑下model_node是否要删除，还是保留后重复利用-
                if model_node.ch_count ==0:
                    self.modelNode_list.remove(model_node)  #如何没有关联通道了，侧删除该modelnodel
                    del model_node  #删除对象
                del self.channel_list[channel_id]   #删除channel_id 与 model_node的对应关心节点
                #行动完更新数量
                self.ch_count -= 1
                return True
            else:
                print("不在list里面，不应该调用stop_model_th")
                return False