class Face_Comp(object):
def __init__(self):
self.tcp = Comunicacion()
self.fdetect = Face_Detect2()
self.th_detect = ThreadAdmin()
self.host = ''
self.port = ''
self.ob_imagen = '' # type: Imagen
self.ob_label_fps = '' # type: Label
self.cola_imagen = queue.LifoQueue()
self.conexion = False # Estado de la conexion
self.tiempo = Timer()
self.evento_conexion = '' # Evento que devuelve el estado de la conexion
self.fun_envio = '' # Evento para movimiento de la vista
def config(self, host, puerto, ob_imagen, ob_label_fps, fun_envio):
# type: (str, int, Imagen, Label, object)->None
""" Se requiere IP del Servidor y Objeto Imagen
Devuelve la imagen y los fps en un objeto label
fun envio = fun(modulo, comando, valor)
"""
self.host = host
self.port = puerto
self.ob_imagen = ob_imagen
self.ob_label_fps = ob_label_fps
self.tcp.config(host,
puerto,
binario=True,
callback=self.__call_conexion)
self.fdetect.config(Callback_Imagen=self.__call_imagen)
self.fdetect.config_callback(Func_Unica=self.__call_posdetect)
self.fun_envio = fun_envio
def config_eventos(self, evento_conexion=''):
"""evento_conexion: Funcion que devuelve True o False cuando conecta o desconecta
Ej: def evento(Estado)
"""
self.evento_conexion = evento_conexion
def iniciar(self):
self.tiempo.iniciar()
self.tcp.iniciar()
def desconectar(self):
self.tcp.desconectar()
self.th_detect.close()
def __call_conexion(self, codigo, mensaje):
""" recepcion de datos binarios del servidor tcp"""
# Conectado
if codigo == 2:
self.conexion = True
self.th_detect.start(self.__th_detector,
'',
'DETECTOR',
enviar_ejecucion=True)
self.evento_conexion(True) # Devolvemos conectado
# Desconecado
elif codigo == 0:
self.conexion = False
self.evento_conexion(False) # Devolvemos desconectado
# Recepcion de datos
elif codigo == 4:
self.cola_imagen.put(mensaje) # encolamos la imgen recibida
# Errores
elif codigo == -1:
self.conexion = False
self.evento_conexion(False) # Devolvemos desconectado
# Otros mensajes
else:
if codigo != 3: # 3 es envio de mensajes
print("MEN: " + str(codigo) + " " + str(mensaje))
def __call_imagen(self, cv_imagen):
""" recepcion de imagen procesada por el detector para insertar en el objeto imagen"""
self.ob_imagen.imagen_cv(cv_imagen)
def __th_detector(self, run):
""" envia a analizar las imagenes que entran en cola """
while run.value:
if self.cola_imagen.qsize() > 0:
# print(tiempo.fps())
self.ob_label_fps.set_text("FPS: " +
str(round(self.tiempo.fps(), 2)))
imag = self.cola_imagen.get()
self.fdetect.imagen(imag)
time.sleep(0.01)
def __call_posdetect(self, x, y):
modulo = "FACE"
comando = "CENTRAR"
valor = str(x) + "/" + str(y) # separador de valores /
self.fun_envio(modulo, comando, valor)
class Eventos(object):
def __init__(self, c_objetos, c_log):
# type: (Form_Principal, Logg)->None # utilizado para visualizar ayuda
self.objetos = c_objetos
self.log = c_log
# Configuracion de objetos sin graficos
self.tcp_cli = Comunicacion()
self.face_comp = Face_Comp()
self.tcp_cli.config(config.SERVIDOR,
config.PUERTO,
callback=self.evento_conexion)
self.face_comp.config(config.SERVIDOR, config.PUERTO_IMG,
self.objetos.box_imagen, self.objetos.label_fps,
self.fun_envio)
# Metodos de los objetos graficos
self.objetos.bot_conectar.accion(self.click_conectar)
self.objetos.bot_conec_img.accion(self.click_conectar_img)
self.objetos.bot_guardar.accion(self.click_guardar)
self.objetos.chbox_sens_cent.accion(self.click_chbox_sensor_cent)
self.objetos.chbox_sens_izq.accion(self.click_chbox_sensor_izq)
self.objetos.chbox_sens_der.accion(self.click_chbox_sensor_der)
self.objetos.chbox_sens_bru.accion(self.click_chbox_brujula)
self.objetos.bot_update_cent.accion(self.up_sonic_cent_time)
self.objetos.bot_update_bru.accion(self.up_brujula_time)
# Callback a eventos
self.face_comp.config_eventos(self.evento_im_conexion)
###########################################################
### METODOS (ACCIONES DE LOS BOTONES) ###
###########################################################
def click_conectar(self):
"""evento cuando se presiona boton de conexion """
if self.tcp_cli.conexion:
self.objetos.bot_conectar.set_text("Desconectando...")
self.tcp_cli.desconectar()
else:
self.objetos.bot_conectar.set_text("Conectando...")
self.tcp_cli.iniciar()
def click_conectar_img(self):
"""evento cuando se presiona boton de conexion de imagen"""
if self.face_comp.conexion:
self.objetos.bot_conec_img.set_text("Desconectando...")
self.face_comp.desconectar()
else:
self.objetos.bot_conec_img.set_text("Conectando...")
self.face_comp.iniciar()
def click_guardar(self):
"""evento cuando se presiona boton de guardar"""
config.CONFIGURACION.set('CONEXION', 'SERVIDOR',
self.objetos.text_ip.text)
config.CONFIGURACION.set('CONEXION', 'PUERTO',
self.objetos.text_port.text)
config.CONFIGURACION.set('CONEXION', 'PUERTO_IMG',
self.objetos.text_port2.text)
config.CONFIGURACION.set('SONICO', 'UPDATE',
self.objetos.text_speed_cent.text)
with open(config.ARCHIVO, 'w') as archivo:
config.CONFIGURACION.write(archivo) # salvamos el archivo
def click_chbox_sensor_cent(self, estado):
"""evento cuando se presiona checkbox de sensor central"""
if estado:
self.tcp_cli.enviar("SONICO|ENCENDIDO|1")
else:
self.tcp_cli.enviar("SONICO|ENCENDIDO|0")
def click_chbox_brujula(self, estado):
"""evento cuando se presiona checkbox de la brujula"""
if estado:
self.tcp_cli.enviar("COMPAS|ENCENDIDO|1")
else:
self.tcp_cli.enviar("COMPAS|ENCENDIDO|0")
def click_chbox_sensor_izq(self, estado):
"""evento cuando se presiona checkbox de sensor izquierdo"""
if estado:
self.tcp_cli.enviar("[SONICO_IZQ-ON]")
else:
self.tcp_cli.enviar("[SONICO_IZQ-OFF]")
def click_chbox_sensor_der(self, estado):
"""evento cuando se presiona checkbox de sensor derecho"""
if estado:
self.tcp_cli.enviar("[SONICO_DER-ON]")
else:
self.tcp_cli.enviar("[SONICO_DER-OFF]")
def up_sonic_cent_time(self):
"""evento cuando se presiona el boton de actualizacion de tiempo"""
tiempo = self.objetos.text_speed_cent.text
self.tcp_cli.enviar("SONICO|SPEED|" + tiempo)
def up_brujula_time(self):
"""evento cuando se presiona el boton de actualizacion de tiempo"""
tiempo = self.objetos.text_speed_bru.text
self.tcp_cli.enviar("COMPAS|SPEED|" + tiempo)
###########################################################
### EVENTOS (CALLBACK DE CONEXION) ###
###########################################################
def evento_conexion(self, codigo, mensaje):
if codigo == 2: # Conectado
self.objetos.bot_conectar.set_text("Desconectar")
elif codigo == 0: # Desconectado
self.objetos.bot_conectar.set_text("Conectar")
elif codigo == -1: # Error
self.objetos.bot_conectar.set_text("Conectar")
elif codigo == 4: # Recepcion de datos
self.recepcion_datos(mensaje)
else:
self.log.log(str(codigo) + " " + mensaje, "EVENTOS")
###########################################################
### PROCESAMIENTO DE DATOS RECIBIDOS ###
###########################################################
def recepcion_datos(self, datos):
datos = datos.replace("[", "", 1)
datos = datos.replace("]", "", 1)
try:
self.log.log("RECP: " + datos)
mensaje_split = str(datos).split(":", 1)
# recepcion de 2 valores (parametros)
if len(mensaje_split) > 1:
parametro = mensaje_split[0] # Parametro
valor = mensaje_split[1] # valor como string
valor_f = float(valor) # valor como float
valor_int = int(valor_f) # valor como integer
print(valor_int)
if parametro == "COMPAS":
self.objetos.compbox.rotar(int(valor_int))
if parametro == "SONICO-CENT":
self.objetos.prog_cent.set_text(str(valor_int))
if parametro == "SONICO-IZQ":
self.objetos.prog_izq.set_text(str(valor_int))
if parametro == "SONICO-DER":
self.objetos.prog_der.set_text(str(valor_int))
except Exception as e:
self.log.log("ERR: " + str(e))
# EVENTOS
def evento_im_conexion(self, Estado):
if Estado:
self.objetos.bot_conec_img.set_text("Desconectar")
else:
self.objetos.bot_conec_img.set_text("Conectar")
# ENVIOS
def fun_envio(self, modulo, comando, valor):
mensaje = modulo + "|" + comando + "|" + valor
self.tcp_cli.enviar(mensaje)
class Gestor_cli(object):
def __init__(self, c_objetos):
# type: (Form_Principal)->None # ejemplo ty_ (int, int)
self.conexion = False # Estado de la conexion
self.objeto = c_objetos
self.tcp_gestor = Comunicacion()
self.host = "127.0.0.1"
self.log = self.__log_default # Para configurar log
# Elementos de gestion
self.face_analis = Face_Comp()
self.fun_conexion = '' # devulve ele estado de la conexion
def config(self, host, evento_conexion):
""" host: IP del servidor
evento_conexion devuelve el estado de la conexion
"""
self.host = host
self.tcp_gestor.config(host, callback=self.__call_conexion)
# Modulos de gestion
self.face_analis.config(host, 50002, self.objeto.box_imagen, self.objeto.label_fps, self.enviar)
self.face_analis.config_eventos(self.__evento_face)
self.fun_conexion = evento_conexion
def config_log(self, Log):
"""posibilidad de configurar clase Log(Texto, Modulo)"""
self.log = Log.log
def conectar(self):
self.log("INICIANDO GESTOR CLIENTE", "GESTOR-CLI")
# Gestor
self.tcp_gestor.iniciar()
def desconectar(self):
self.log("DESCONECTAR GESTOR CLIENTE", "GESTOR-CLI")
# Gestor
self.tcp_gestor.desconectar()
def conectar_face(self):
#self.log("INICIANDO GESTOR CLIENTE", "GESTOR-CLI")
self.face_analis.iniciar()
def desconectar_face(self):
#self.log("DESCONECTAR GESTOR CLIENTE", "GESTOR-CLI")
self.face_analis.desconectar()
############################
### ENVIO DE DATOS ###
############################
def enviar(self, modulo, comando, valor):
mensaje = modulo + "|" + comando + "|" + valor
self.tcp_gestor.enviar(mensaje)
############################
### RECEPCION DE DATOS ###
############################
def __call_conexion(self, codigo, mensaje):
if codigo == 2: # Conectado
self.log("CONECTADO A GESTOR SERVIDOR", "GESTOR-CLI")
self.conexion = True
self.fun_conexion(True)
if codigo == 0: # Desconectado
self.conexion = False
self.fun_conexion(False)
def __evento_face(self, estado):
# Log por defecto
def __log_default(self, Texto, Modulo):
print(Texto)