img = resize(img, (256, 256))[..., :3]
avatars.append(img)
log('load checkpoints..')
generator, kp_detector = load_checkpoints(config_path=opt.config,
checkpoint_path=opt.checkpoint,
device=device)
fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D,
flip_input=True,
device=device)
# cap = cv2.VideoCapture(opt.cam)
cap = VideoCaptureAsync(opt.cam)
if not cap.isOpened():
log("Cannot open camera. Try to choose other CAMID in './scripts/settings.sh'"
)
exit()
cap.start()
ret, frame = cap.read()
if not ret:
log("Cannot read from camera")
exit()
if _streaming:
stream = pyfakewebcam.FakeWebcam(f'/dev/video{opt.virt_cam}',
frame.shape[1], frame.shape[0])
cur_ava = 0
class Interfaz:
def __init__(self):
self.Interfaz = Tk()
self.cap = None
self.hilo2 = None
self.sincronizacionAnterior = [[255, 255, 255], [255, 255, 255],
[255, 255, 255]]
self.tramasRecibidas = np.array([], dtype=int)
self.cargaUtil = []
self.datos = np.array([], dtype=int)
self.frames = []
self.errores = []
self.cont = 0
self.numTramas = 0
self.tramasValidas = 0
self.tramasInvalidas = 0
self.tiempoInicial = 0
self.tiempoFinal = 0
self.tiempoTotal = 0
self.TramasFaltantes = 0
self.tramaAnterior = 0
self.tramasBitsErroneos = 0
self.FER = 0
self.BER = 0
self.FPS = 0
self.TramasTransmitidas = 0
self.tramaAnteriorValida = 0
self.TramaInicial = 0
self.ciclos = 0
self.TramaInv = 0
self.bitsTrama = 0
self.errores_totales = 0
self.tramasTotales = 0
self.valor = 0
self.mostrarRecibidas = ""
#Caracteristicas de la ventana
self.Interfaz.title("Receptor")
self.Interfaz.geometry("1093x500+0+0")
self.Interfaz.resizable(width=False, height=False)
self.Interfaz.configure(bg='#F1F1F1')
#TAMAÑO DE LA MATRIZ
self.etiqueta2 = Label(text="Tamaño de la matriz").place(relx=0.04,
rely=0.1)
self.tamanoMatriz = ttk.Combobox(state="readonly",
values=[8, 10, 12, 14, 16])
self.tamanoMatriz.place(relx=0.155, rely=0.1, width=82)
self.tamanoMatriz.current(0)
#NUMERO DE COLORES
self.etiqueta3 = Label(text="Número de colores").place(relx=0.04,
rely=0.17)
self.numeroColores = ttk.Combobox(state="readonly", values=[2, 4, 8])
self.numeroColores.place(relx=0.155, rely=0.17, width=82)
self.numeroColores.current(0)
#SEGUNDOS
self.etiqueta3 = Label(text="Segundos capturados").place(relx=0.04,
rely=0.24)
self.segundos = ttk.Combobox(state="readonly",
values=[
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14,
16, 18, 20, 22, 24, 26, 28, 30
])
self.segundos.place(relx=0.155, rely=0.24, width=82)
self.segundos.current(0)
#Botones
self.transmitir = Button(self.Interfaz,
text="Comenzar Captura",
command=self.IniciarProcesamiento).place(
relx=0.04, rely=0.31)
self.transmitir = Button(self.Interfaz,
text="Parar Captura",
command=self.PararCaptura).place(relx=0.155,
rely=0.31)
#self.scrolledtext1 = st.ScrolledText(self.Interfaz, width=30, height=20)
#self.scrolledtext1.place(relx=0.1, rely=0.35)
#PRUEBA DE IMAGEN EN LABEL
self.l1 = tkk.Label(self.Interfaz,
text=" ",
borderwidth=2,
relief="groove")
self.l1.place(relx=0.25, rely=0.08, width=350, height=350)
#TRAMAS
self.TR = StringVar()
self.TF = StringVar()
self.etiquetaR = Label(self.Interfaz,
textvariable=self.TR).place(relx=0.59, rely=0.1)
self.etiquetaFa = Label(self.Interfaz,
textvariable=self.TF).place(relx=0.59,
rely=0.28)
self.TR.set("Tramas Recibidas: ")
self.TF.set("Tramas Faltantes: ")
self.TTF = Text(self.Interfaz, height=3, state="normal", width=50)
self.TTF.place(relx=0.59, rely=0.15)
self.TTF.configure(state="normal")
self.TTF.insert(INSERT, "")
#FPS
self.TFPS = StringVar()
self.etiquetaFPS = Label(self.Interfaz,
textvariable=self.TFPS).place(relx=0.04,
rely=0.42)
self.TFPS.set("FPS: ")
#FER
self.TFER = StringVar()
self.etiquetaF = Label(self.Interfaz,
textvariable=self.TFER).place(relx=0.04,
rely=0.49)
self.TFER.set("FER: ")
#BER
self.TBER = StringVar()
self.etiquetaB = Label(self.Interfaz,
textvariable=self.TBER).place(relx=0.04,
rely=0.56)
self.TBER.set("BER: ")
#Consola
self.etiquetaConsola = Label(self.Interfaz,
text="Estado de procesamiento: ").place(
relx=0.59, rely=0.34)
self.Consola = st.ScrolledText(self.Interfaz,
width=50,
height=12,
wrap=WORD,
background='White')
self.Consola.grid(row=0, column=1)
self.Consola.place(relx=0.59, rely=0.39)
self.borrarImagenes()
self.Interfaz.mainloop()
def borrarImagenes(self):
for file in os.listdir("."):
if os.path.isfile(file) and file.startswith("Nuevo16-"):
os.remove(file)
for file in os.listdir("."):
if os.path.isfile(file) and file.startswith("Nuevo_"):
os.remove(file)
for file in os.listdir("."):
if os.path.isfile(file) and file.startswith("NuevoNegro"):
os.remove(file)
def frombits(self, bits):
chars = []
fin = math.ceil(bits.shape[0] / 8)
bits = bits[:fin * 8]
for b in range(int(len(bits) / 8)):
byte = bits[b * 8:(b + 1) * 8]
chars.append(chr(int(''.join([str(bit) for bit in byte]), 2)))
return ''.join(chars)
def leerTramas(self, dst2, c):
#Verificar si la primera trama valida fue recibida
if self.numTramas == 0:
print("Primera trama: ", c)
#Calculo de los colores referencia, obtencion de los bits y campos de la trama
imagen = coloresReferencia(dst2, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
coloresR = imagen.obtenerColoresReferencia()
print("Muestra de color")
matriz = muestraDeColor(dst2, int(self.tamanoMatriz.get()),
coloresR, int(self.numeroColores.get()))
print("Mapeo a bit")
bits = matriz.mapeoaBit()
trama = Trama(bits, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
trama.obtenerCampos()
#si la trama es valida, almacena el número total de tramas e incrementa las tramas validas en 1
if trama.numeroDeTrama in self.tramasRecibidas:
print("La trama ya esta")
else:
if trama.tramaValida == True and self.ciclos == 0:
print("primera trama recibida correctamente: ", c)
self.numTramas = trama.numeroTramas
self.TramaInicial = trama.numeroDeTrama
self.tramaAnterior = trama.numeroDeTrama
self.tramaAnteriorValida = trama.tramaValida
print("Trama anterior: ", self.tramaAnterior)
self.errores = (-1) * np.ones(self.numTramas)
self.bitsTrama = len(bits)
if not self.cargaUtil:
self.cargaUtil = np.ones((trama.numeroTramas, 1),
dtype=int)
self.cargaUtil = self.cargaUtil.tolist()
self.tramasRecibidas = np.concatenate(
(self.tramasRecibidas, trama.numeroDeTrama), axis=None)
self.TTF.insert(END, format(trama.numeroDeTrama) + ",")
print("Tramas recibidas: ", self.tramasRecibidas)
self.cargaUtil[trama.numeroDeTrama - 1] = trama.cargaUtil
print("valida")
self.TramasTransmitidas = TramaBER(
self.numTramas, int(self.numeroColores.get()),
int(self.tamanoMatriz.get()))
self.TramasTransmitidas.generarTramas()
errores_BER = self.TramasTransmitidas.compararTrama(
trama.numeroDeTrama, bits)
self.errores[trama.numeroDeTrama - 1] = errores_BER
self.Consola.insert(
INSERT, 'Primera Trama recibida correctamente\n')
#self.TR.set("Tramas Recibidas: "+format(self.tramasRecibidas))
self.TF.set("Tramas Faltantes: " +
format(self.numTramas - 1))
self.Consola.insert(END, "")
self.Consola.see(END)
#Si es la primera trama valida, obtiene el numero de tramas y el numero de trama
else:
imagen = coloresReferencia(dst2, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
coloresR = imagen.obtenerColoresReferencia()
matriz = muestraDeColor(dst2, int(self.tamanoMatriz.get()),
coloresR, int(self.numeroColores.get()))
bits = matriz.indicadores()
trama = Trama(bits, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
trama.obtenerIndicadores()
#self.tramaAnteriorValida = trama.tramaValida
print("Trama anterior: ", self.tramaAnterior)
print("Trama actual: ", trama.numeroDeTrama)
print("Trama validez: ", self.tramaAnteriorValida)
print("Trama numeroTRamas: ", trama.numeroTramas)
self.Consola.insert(
INSERT, '*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*\n')
#self.Consola.insert(INSERT,'Trama anterior: '+format(self.tramaAnterior)+'\n')
self.Consola.insert(
INSERT, 'Trama actual: ' + format(trama.numeroDeTrama) + '\n')
self.Consola.insert(END, "")
self.Consola.see(END)
BER_actual = 0
if trama.numeroDeTrama != 0 and trama.numeroDeTrama <= self.numTramas and trama.numeroDeTrama not in self.tramasRecibidas:
self.bitsAnterior = matriz.mapeoaBit()
errores_BER = self.TramasTransmitidas.compararTrama(
trama.numeroDeTrama, self.bitsAnterior)
self.errores[trama.numeroDeTrama - 1] = errores_BER
#Si el número de trama es igual al anterior, y la trama anterior fue invalida, procesa la siguiente
if self.tramaAnterior == trama.numeroDeTrama and self.tramaAnteriorValida == False:
print("Trama anterior invalida, leyendo siguiente trama...")
print(
"-----------------------------------------------------------Entro a while"
)
self.Consola.insert(
INSERT, 'No es una trama valida, se leeran más: \n')
c += 1
nextF = cv2.imread('Nuevo16-' + str(c) + '.png')
while (nextF is not None
) and self.tramaAnterior == trama.numeroDeTrama:
print("******************************************")
print("Procesando: Nuevo_16-" + str(c))
dst = self.transformacionPerspectiva(nextF, c)
nombreImagen = 'Nuevo_16-' + str(c) + '.png'
imgl = Image.open(nombreImagen)
imgl.thumbnail((350, 350), Image.ANTIALIAS)
imgScreen = ImageTk.PhotoImage(imgl)
self.l1.configure(image=imgScreen)
imagen = coloresReferencia(dst,
int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
coloresR = imagen.obtenerColoresReferencia()
matriz = muestraDeColor(dst, int(self.tamanoMatriz.get()),
coloresR,
int(self.numeroColores.get()))
bits = matriz.mapeoaBit()
trama = Trama(bits, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
trama.obtenerCampos()
self.Consola.insert(
INSERT,
'Trama actual: ' + format(trama.numeroDeTrama) + '\n')
self.Interfaz.update()
self.tramaAnteriorValida = trama.tramaValida
if self.tramaAnterior == trama.numeroDeTrama:
if trama.tramaValida == True:
if trama.numeroDeTrama in self.tramasRecibidas:
print("Ya esta")
else:
BER_actual = 0
self.tramasRecibidas = np.concatenate(
(self.tramasRecibidas,
trama.numeroDeTrama),
axis=None)
print("Tramas recibidas: ",
self.tramasRecibidas)
self.TTF.insert(
END,
format(trama.numeroDeTrama) + ",")
self.TramasFaltantes = self.numTramas - self.tramasRecibidas.shape[
0]
print("Tramas faltantes: ",
self.TramasFaltantes)
self.cargaUtil[trama.numeroDeTrama -
1] = trama.cargaUtil
#self.TR.set("Tramas Recibidas: "+format(self.tramasRecibidas))
self.TF.set("Tramas Faltantes: " +
format(self.TramasFaltantes))
self.errores[trama.numeroDeTrama - 1] = 0
self.Consola.insert(
INSERT, 'Ya se encontró una trama : ' +
format(trama.numeroDeTrama) + '\n')
self.Interfaz.update()
if self.tramasRecibidas.shape[
0] == trama.numeroTramas:
for h in range(trama.numeroTramas):
self.datos = np.concatenate(
(self.datos, self.cargaUtil[h]),
axis=None)
print("Se recibieron todas las tramas")
self.tiempoFinal = time.time()
self.tiempoTotal = self.tiempoFinal - self.tiempoInicial
print("Bits recibidos: ",
self.datos.shape[0])
print("Tiempo: ", self.tiempoTotal)
mb.showinfo(
message=
"Se recibieron todas las tramas",
title="Título")
file = open('texto.txt',
'w',
encoding="utf-8")
file.write(self.frombits(self.datos))
file.close()
subprocess.run(["notepad", "texto.txt"])
break
else:
print("Trama no valida")
BER_actual = self.TramasTransmitidas.compararTrama(
trama.numeroDeTrama, bits)
if trama.numeroDeTrama <= self.numTramas and trama.numeroDeTrama != 0:
self.tramaAnterior = trama.numeroDeTrama
else:
if trama.numeroDeTrama <= self.numTramas and trama.numeroDeTrama != 0:
self.Consola.insert(
INSERT, 'No se encontraron tramas: ' +
format(self.tramaAnterior) + '\n')
self.tramaAnterior = trama.numeroDeTrama
self.errores[self.tramaAnterior - 1] = BER_actual
break
c += 1
nextF = cv2.imread('Nuevo16-' + str(c) + '.png')
#if BER_actual !=0:
# self.tramasInvalidas += 1
# print("Se sumo trama invalida: ",trama.numeroDeTrama)
print("numero de tramas: ", trama.numeroTramas)
print("numero de trama: ", trama.numeroDeTrama)
#En caso de ser diferente
else:
#Verifica que el número de trama sea válido y guarda el numero de trama
self.Consola.insert(
INSERT,
'Trama actual: ' + format(trama.numeroDeTrama) + '\n')
if trama.numeroTramas == self.numTramas and trama.numeroDeTrama <= self.numTramas and trama.numeroDeTrama > 0:
self.tramaAnterior = trama.numeroDeTrama
self.tramaAnteriorValida = trama.tramaValida
bits = matriz.mapeoaBit()
trama = Trama(bits, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
trama.obtenerCampos()
#Verifica si la trama ya fue recibida
if trama.numeroDeTrama in self.tramasRecibidas:
print("La trama ya esta")
self.Consola.insert(INSERT,
'Celda de sincronización igual\n')
self.errores[trama.numeroDeTrama - 1] = 0
if trama.tramaValida == True:
print("trama valida")
self.Consola.insert(INSERT, 'trama valida\n')
else:
print("trama invalida")
self.Consola.insert(INSERT, 'trama invalida\n')
self.Consola.insert(END, "")
self.Consola.see(END)
else:
#self.Consola.insert(INSERT,'Entro al else: \n')
if trama.tramaValida == True and trama.numeroDeTrama > 0 and trama.numeroDeTrama < self.numTramas:
self.tramasRecibidas = np.concatenate(
(self.tramasRecibidas, trama.numeroDeTrama),
axis=None)
self.TTF.insert(END,
format(trama.numeroDeTrama) + ",")
print("Tramas recibidas: ", self.tramasRecibidas)
self.TramasFaltantes = self.numTramas - self.tramasRecibidas.shape[
0]
print("Tramas faltantes: ", self.TramasFaltantes)
self.cargaUtil[trama.numeroDeTrama -
1] = trama.cargaUtil
#self.TR.set("Tramas Recibidas: "+format(self.tramasRecibidas))
self.TF.set("Tramas Faltantes: " +
format(self.TramasFaltantes))
#self.Consola.insert(INSERT,'Entro al if \n')
self.Interfaz.update()
#if self.tramasRecibidas.shape[0] == trama.numeroTramas:
# for h in range(trama.numeroTramas):
# self.datos = np.concatenate((self.datos,self.cargaUtil[h]),axis=None)
# print("Se recibieron todas las tramas")
# self.tiempoFinal = time.time()
# self.tiempoTotal = self.tiempoFinal - self.tiempoInicial
# print("Bits recibidos: ",self.datos.shape[0])
# print("Tiempo: ",self.tiempoTotal)
# self.tramasValidas = self.tramasRecibidas.shape[0]
# self.tramasInvalidas=sum(1 for item in self.errores if item!=0 and item!=-1)
# self.errores_totales=sum(item for item in self.errores if item!=0 and item!=-1)
# self.tramasTotales = self.tramasValidas + self.tramasInvalidas
# bitTotales=self.tramasTotales*self.bitsTrama
# self.FER = self.tramasInvalidas / self.tramasTotales
# self.TFER.set("FER: " + format(self.FER))
# self.BER=self.errores_totales/bitTotales
# self.TBER.set("BER: " + format(self.BER))
# self.Consola.insert(INSERT,'Bits erroneos:'+format(self.errores_totales)+' \n')
# self.Consola.insert(INSERT,'Bits Recibidoss:'+format(bitTotales)+' \n')
# self.Consola.insert(INSERT,'BER:'+format(self.BER)+' \n')
# self.Consola.insert(INSERT,'Frames erroneos:'+format(self.tramasInvalidas)+' \n')
# self.Consola.insert(INSERT,'Frames Recibidos:'+format(self.tramasTotales)+' \n')
# self.Consola.insert(INSERT,'FER:'+format(self.FER)+' \n')
# mb.showinfo(message="Se recibieron todas las tramas", title="Título")
# file = open('texto.txt','a')
# file.write(self.frombits(self.datos))
# file.close()
# subprocess.run(["notepad","texto.txt"])
else:
#print("invalida")
#BER_calculado=self.TramasTransmitidas.compararTrama(trama.numeroDeTrama,bits)
#self.Consola.insert(INSERT,'BER de la trama: '+format(BER_calculado)+'\n')
#self.Consola.insert(INSERT,'Entro al segundo else \n')
self.TramaInv += 1
self.Consola.insert(INSERT, 'trama invalida\n')
self.Consola.insert(END, "")
self.Consola.see(END)
self.tramaAnteriorValida = False
elif len(self.tramasRecibidas) != 0 and (
(int(self.numeroColores.get()) == 2 and trama.numeroDeTrama
== 255 and trama.numeroTramas == 255) or
(int(self.numeroColores.get()) == 4 and trama.numeroDeTrama
== 0 and trama.numeroTramas == 0) or
(int(self.numeroColores.get()) == 8 and trama.numeroDeTrama
== 146 and trama.numeroTramas == 36)):
#Tramas que llegaron
#self.tramasValidas = self.tramasRecibidas.shape[0]
#Tramas totales
#self.self.tramasTotales = self.tramasValidas + self.tramasInvalidas
#Calculo de FER
#self.FER = self.tramasInvalidas / self.self.tramasTotales
self.Consola.insert(INSERT, 'Entro a Elif\n')
print("*******************************>>>> Entro",
len(self.tramasRecibidas))
self.Consola.insert(END, "")
self.Consola.see(END)
self.tramasValidas = self.tramasValidas + self.tramasRecibidas.shape[
0]
self.tramasInvalidas = self.tramasInvalidas + sum(
1 for item in self.errores if item != 0 and item != -1)
self.errores_totales = self.errores_totales + sum(
item
for item in self.errores if item != 0 and item != -1)
self.Consola.insert(
INSERT, 'Bits ERRORES:' + format(self.errores) + ' \n')
tramasTotalCiclo = self.tramasValidas + self.tramasInvalidas
bitTotales = tramasTotalCiclo * self.bitsTrama
self.FER = self.tramasInvalidas / tramasTotalCiclo
self.TFER.set("FER: " + format(self.FER))
self.BER = self.errores_totales / bitTotales
self.TBER.set("BER: " + format(self.BER))
self.Consola.insert(
INSERT, 'Bits erroneos:' +
format(self.errores_totales) + ' \n')
self.Consola.insert(
INSERT,
'Bits Recibidoss:' + format(bitTotales) + ' \n')
self.Consola.insert(INSERT,
'BER:' + format(self.BER) + ' \n')
self.Consola.insert(
INSERT, 'Frames erroneos:' +
format(self.tramasInvalidas) + ' \n')
self.Consola.insert(
INSERT,
'Frames Recibidos:' + format(tramasTotalCiclo) + ' \n')
self.Consola.insert(INSERT,
'FER:' + format(self.FER) + ' \n')
self.Consola.insert(END, "")
self.Consola.see(END)
self.errores = (-1) * np.ones(self.numTramas)
#self.mostrarRecibidas = str(self.mostrarRecibidas) + str(self.tramasRecibidas) + ","
#self.TR.set("Tramas Recibidas: "+format(self.tramasRecibidas))
#self.TTF.insert(END,format(trama.numeroDeTrama) + ",")
self.tramasRecibidas = np.array([], dtype=int)
print("Longitud del array: ", len(self.tramasRecibidas))
self.Interfaz.update()
#self.BER = 0
#self.FER = 0
#self.numTramas = 0
#self.ciclos = 1
#self.TramaInicial = 0
return c
def leerFrames(self):
frame = cv2.imread('Nuevo16-0.png')
nextFrame = cv2.imread('Nuevo16-1.png')
c = 0
while (frame is not None) and (nextFrame is not None):
print("******************************************")
print("Procesando: Nuevo16-" + str(c))
dst2 = self.transformacionPerspectiva(frame, c)
nombreImagen = 'Nuevo_16-' + str(c) + '.png'
#try:
imgl = Image.open(nombreImagen)
imgl.thumbnail((350, 350), Image.ANTIALIAS)
imgScreen = ImageTk.PhotoImage(imgl)
self.l1.configure(image=imgScreen)
self.Interfaz.update()
c = self.leerTramas(dst2, c)
#except:
# print('No leyo la imagen')
c = c + 1
frame = cv2.imread('Nuevo16-' + str(c) + '.png')
self.tramasValidas = self.tramasValidas + self.tramasRecibidas.shape[0]
self.tramasInvalidas = self.tramasInvalidas + sum(
1 for item in self.errores if item != 0 and item != -1)
self.errores_totales = self.errores_totales + sum(
item for item in self.errores if item != 0 and item != -1)
self.tramasTotales = self.tramasValidas + self.tramasInvalidas
bitTotales = self.tramasTotales * self.bitsTrama
self.FER = self.tramasInvalidas / self.tramasTotales
self.TFER.set("FER: " + format(self.FER))
self.BER = self.errores_totales / bitTotales
self.TBER.set("BER: " + format(self.BER))
self.Consola.insert(INSERT, '*****************************\n')
mb.showinfo(message="Se recibieron todas las tramas", title="Título")
if self.errores[0] != -1:
###################################### IMPRIMIR ARCHIVO
for h in range(self.numTramas):
self.datos = np.concatenate((self.datos, self.cargaUtil[h]),
axis=None)
file = open('texto.txt', 'w', encoding="utf-8")
print("Aquiiii", self.datos.shape)
file.write(self.frombits(self.datos))
file.close()
subprocess.run(["notepad", "texto.txt"])
self.Consola.insert(INSERT, 'Se acabaron las imagenes\n')
self.Consola.insert(
INSERT, 'Bits erroneos:' + format(self.errores_totales) + ' \n')
self.Consola.insert(INSERT,
'Bits Recibidoss:' + format(bitTotales) + ' \n')
self.Consola.insert(INSERT, 'BER:' + format(self.BER) + ' \n')
self.Consola.insert(
INSERT, 'Frames erroneos:' + format(self.tramasInvalidas) + ' \n')
self.Consola.insert(
INSERT, 'Frames Recibidos:' + format(self.tramasTotales) + ' \n')
self.Consola.insert(INSERT, 'FER:' + format(self.FER) + ' \n')
self.Consola.insert(END, "")
self.Consola.see(END)
print("Tramas validas: ", self.tramasValidas)
print("Tramas invalidas: ", self.tramasInvalidas)
print("Tramas totales: ", self.tramasTotales)
def transformacionPerspectiva(self, frame, c):
bilFilter = cv2.bilateralFilter(frame, 9, 75, 75)
gray = cv2.cvtColor(bilFilter, cv2.COLOR_BGR2GRAY)
with Lock():
#cv2.imshow('gry',gray)
cv2.waitKey(25)
cv2.imwrite("NuevoGris.png", gray)
ret, thresh = cv2.threshold(gray, 200, 255, 1)
contours, h = cv2.findContours(thresh, 1, 2)
#cv2.imshow('vhf',thresh)
cv2.imwrite("NuevoNegro.png" + str(c) + ".png", thresh)
for i, cnt in enumerate(contours):
approx = cv2.approxPolyDP(cnt, 0.05 * cv2.arcLength(cnt, True),
True)
area = cv2.contourArea(cnt)
auxT = int(self.tamanoMatriz.get()) + 4
tamanoFinal = auxT * 20
#if len(approx)==4 and area > 9000 and area<270000:
if len(approx) == 4 and area > 5000 and area < 270000:
approx1 = [approx[0], approx[1], approx[3], approx[2]]
pts1 = np.float32(approx1)
pts2 = np.float32([[0, 0], [0, tamanoFinal], [tamanoFinal, 0],
[tamanoFinal, tamanoFinal]])
M = cv2.getPerspectiveTransform(pts1, pts2)
dst = cv2.warpPerspective(frame, M, (tamanoFinal, tamanoFinal))
img = dst
gray2 = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh2 = cv2.threshold(gray2, 155, 255, 1)
#cv2.imshow("GRIS",thresh2)
#cv2.imshow("gris2",gray2)
# Remove some small noise if any.
dilate = cv2.dilate(thresh2, None)
erode = cv2.erode(dilate, None)
# Find contours with cv2.RETR_CCOMP
contours, hierarchy = cv2.findContours(erode, cv2.RETR_CCOMP,
cv2.CHAIN_APPROX_SIMPLE)
#cv2.drawContours(img,contours,-1,(255,0,255),3)
e1x = 300
e2x = 0
e1y = 300
e2y = 0
for p, cnt in enumerate(contours):
# Check if it is an external contour and its area is more than 100
if hierarchy[0, p, 3] == -1 and cv2.contourArea(cnt) > 300:
x, y, w, h = cv2.boundingRect(cnt)
#cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255),1)
#cv2.circle(img,(x,y), 2, (0,0,255), -1)
#print(format(x)+'--'+format(y))
#cv2.imshow('Segundo',img)
#cv2.imwrite("N-"+str(c)+".png", img)
if x <= e1x + 8 and y <= e1y:
e1x = x
e1y = y
#print(format(x)+'-2-'+format(y))
if x + w > e2x and y + h > e2y:
e2x = x + w
e2y = y + h
cv2.circle(img, (x + w, y + h), 2, (0, 255, 0), -1)
#cv2.rectangle(img,(e1x,e1y),(e2x,e2y),(0,255,0),2)
#cv2.imshow('Segundo',img)
pts1 = np.float32([[e1x, e1y], [e2x, e1y], [e1x, e2y],
[e2x, e2y]])
pts2 = np.float32([[0, 0], [0, tamanoFinal], [tamanoFinal, 0],
[tamanoFinal, tamanoFinal]])
M = cv2.getPerspectiveTransform(pts1, pts2)
dst2 = cv2.warpPerspective(img, M, (tamanoFinal, tamanoFinal))
rows, cols = dst2.shape[:2]
r = rotarImagen(dst2, int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
dst2 = r.rotar()
nlineas = int(self.tamanoMatriz.get()) + 3
tcuadrado = round(tamanoFinal / (nlineas + 1))
for x in range(nlineas):
#lineas verticales
pt1 = ((x + 1) * tcuadrado, 0)
pt2 = ((x + 1) * tcuadrado, tamanoFinal)
cv2.line(dst2, pt1, pt2, (0, 255, 0), 1)
#lineas horizontales
pt1 = (0, (x + 1) * tcuadrado)
pt2 = (tamanoFinal, (x + 1) * tcuadrado)
cv2.line(dst2, pt1, pt2, (0, 255, 0), 1)
#cv2.imshow("Transformacion2", dst2)
#coloresR = coloresReferencia(dst2,self.tamanoMatriz.get(),self.numeroColores.get())
#print(coloresR.obtenerColoresReferencia())
#sincronizacion=CeldaSincronizacion(self.tamanoMatriz.get())
#celdaSincronizacion=sincronizacion.LeerCeldas(dst2)
#FuncionSincronizacion=Sincronizacion(celdaSincronizacion,self.sincronizacionAnterior)
#comparar=FuncionSincronizacion.CompararCeldas()
#self.sincronizacionAnterior = celdaSincronizacion
#if comparar==1:
# print('sincronización igual')
#else:
#print('sincronización nuevo')
cv2.imwrite("Nuevo_16-" + str(c) + ".png", dst2)
return dst2
def IniciarProcesamiento(self):
self.hilo = threading.Thread(target=self.IniciarCaptura2,
args=(int(self.segundos.get()), ))
self.hilo.start()
def IniciarCaptura2(self, segundos):
self.cap = None
self.hilo2 = None
self.sincronizacionAnterior = [[255, 255, 255], [255, 255, 255],
[255, 255, 255]]
self.tramasRecibidas = np.array([], dtype=int)
self.cargaUtil = []
self.datos = np.array([], dtype=int)
self.frames = []
self.errores = []
self.cont = 0
self.numTramas = 0
self.tramasValidas = 0
self.tramasInvalidas = 0
self.tiempoInicial = 0
self.tiempoFinal = 0
self.tiempoTotal = 0
self.TramasFaltantes = 0
self.tramaAnterior = 0
self.tramasBitsErroneos = 0
self.FER = 0
self.BER = 0
self.FPS = 0
self.TramasTransmitidas = 0
self.tramaAnteriorValida = 0
self.TramaInicial = 0
self.ciclos = 0
self.TramaInv = 0
self.bitsTrama = 0
self.errores_totales = 0
self.tramasTotales = 0
self.valor = 0
self.mostrarRecibidas = ""
self.borrarImagenes()
top = Toplevel()
top.title("Información")
msg = Message(
top,
width=300,
text=" Iniciando camara... ")
msg.pack()
self.cap = VideoCaptureAsync(1, 1920, 1080, 30)
self.cap.start()
contador = 0
inicial = time.time()
limite = inicial + segundos
while self.cap.isOpened():
top.destroy()
top = Toplevel()
top.title("Información")
msg = Message(
top,
width=300,
text=" Capturando... ")
msg.pack()
while inicial <= limite:
inicial = time.time()
try:
ret, frame = self.cap.read()
if ret:
cv2.imwrite("Nuevo16-" + str(contador) + ".png", frame)
print("Nuevo16-", str(contador))
contador += 1
except KeyboardInterrupt:
#pass
exit()
break
self.FPS = contador / segundos
self.TFPS.set("FPS: " + format(self.FPS))
top.destroy()
self.cap.stop()
hilo2 = threading.Thread(target=self.leerFrames)
hilo2.start()
def IniciarCaptura(self, segundos):
self.tiempoInicial = time.time()
if self.patronesPorSegundo.get() == 30:
self.cap = VideoCaptureAsync('http://192.168.1.68:4747/video',
1920, 1080, 30)
else:
self.cap = VideoCaptureAsync('http://192.168.1.68:4747/video',
1280, 720, 60)
self.cap.start()
contador = 0
inicial = time.time()
limite = inicial + segundos
while self.cap.isOpened():
while inicial <= limite:
inicial = time.time()
try:
ret, frame = self.cap.read()
if ret:
bilFilter = cv2.bilateralFilter(frame, 9, 75, 75)
gray = cv2.cvtColor(bilFilter, cv2.COLOR_BGR2GRAY)
with Lock():
cv2.imshow('gry', gray)
cv2.waitKey(25)
cv2.imwrite("NuevoGris.png", gray)
ret, thresh = cv2.threshold(gray, 200, 255, 1)
contours, h = cv2.findContours(thresh, 1, 2)
cv2.imshow('vhf', thresh)
cv2.imwrite("NuevoNegro.png", thresh)
for i, cnt in enumerate(contours):
approx = cv2.approxPolyDP(
cnt, 0.05 * cv2.arcLength(cnt, True), True)
area = cv2.contourArea(cnt)
auxT = int(self.tamanoMatriz.get()) + 4
tamanoFinal = auxT * 20
if len(approx
) == 4 and area > 9000 and area < 270000:
approx1 = [
approx[0], approx[1], approx[3], approx[2]
]
pts1 = np.float32(approx1)
pts2 = np.float32([[0, 0], [0, tamanoFinal],
[tamanoFinal, 0],
[tamanoFinal, tamanoFinal]])
M = cv2.getPerspectiveTransform(pts1, pts2)
dst = cv2.warpPerspective(
frame, M, (tamanoFinal, tamanoFinal))
img = dst
gray2 = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh2 = cv2.threshold(
gray2, 155, 255, 1)
cv2.imshow("GRIS", thresh2)
cv2.imshow("gris2", gray2)
# Remove some small noise if any.
dilate = cv2.dilate(thresh2, None)
erode = cv2.erode(dilate, None)
# Find contours with cv2.RETR_CCOMP
contours, hierarchy = cv2.findContours(
erode, cv2.RETR_CCOMP,
cv2.CHAIN_APPROX_SIMPLE)
#cv2.drawContours(img,contours,-1,(255,0,255),3)
e1x = 300
e2x = 0
e1y = 300
e2y = 0
for p, cnt in enumerate(contours):
# Check if it is an external contour and its area is more than 100
if hierarchy[0, p,
3] == -1 and cv2.contourArea(
cnt) > 300:
x, y, w, h = cv2.boundingRect(cnt)
#cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255),2)
#cv2.circle(img,(x,y), 2, (0,0,255), -1)
#print(format(x)+'--'+format(y))
#cv2.imshow('Segundo',img)
if x <= e1x + 8 and y <= e1y:
e1x = x
e1y = y
#print(format(x)+'-2-'+format(y))
if x + w > e2x and y + h > e2y:
e2x = x + w
e2y = y + h
cv2.circle(img, (x + w, y + h), 2,
(0, 255, 0), -1)
#cv2.rectangle(img,(e1x,e1y),(e2x,e2y),(0,255,0),2)
#cv2.imshow('Segundo',img)
pts1 = np.float32([[e1x, e1y], [e2x, e1y],
[e1x, e2y], [e2x, e2y]])
pts2 = np.float32([[0, 0], [0, tamanoFinal],
[tamanoFinal, 0],
[tamanoFinal, tamanoFinal]])
M = cv2.getPerspectiveTransform(pts1, pts2)
dst2 = cv2.warpPerspective(
img, M, (tamanoFinal, tamanoFinal))
rows, cols = dst2.shape[:2]
r = rotarImagen(dst2,
int(self.tamanoMatriz.get()),
int(self.numeroColores.get()))
dst2 = r.rotar()
nlineas = int(self.tamanoMatriz.get()) + 3
tcuadrado = round(tamanoFinal / (nlineas + 1))
for x in range(nlineas):
#lineas verticales
pt1 = ((x + 1) * tcuadrado, 0)
pt2 = ((x + 1) * tcuadrado, tamanoFinal)
#cv2.line(dst2,pt1,pt2,(0,255,0),1)
#lineas horizontales
pt1 = (0, (x + 1) * tcuadrado)
pt2 = (tamanoFinal, (x + 1) * tcuadrado)
#cv2.line(dst2,pt1,pt2,(0,255,0),1)
cv2.imshow("Transformacion2", dst2)
coloresR = coloresReferencia(
dst2, self.tamanoMatriz.get(),
self.numeroColores.get())
#print(coloresR.obtenerColoresReferencia())
sincronizacion = CeldaSincronizacion(
self.tamanoMatriz.get())
celdaSincronizacion = sincronizacion.LeerCeldas(
dst2)
FuncionSincronizacion = Sincronizacion(
celdaSincronizacion,
self.sincronizacionAnterior)
comparar = FuncionSincronizacion.CompararCeldas(
)
self.sincronizacionAnterior = celdaSincronizacion
#if comparar==1:
#print('sincronización igual')
if self.cont == 1:
hilo2 = threading.Thread(
target=self.leerFrames)
hilo2.start()
cv2.imwrite(
"Nuevo16-" + str(self.cont) + ".png", dst2)
nombreImagen = 'Nuevo16-' + str(
self.cont) + '.png'
imgScreen = PhotoImage(file=nombreImagen)
self.l1.configure(image=imgScreen)
self.Interfaz.update()
#print("Nuevo16-"+str(self.cont))
#hilo2 = threading.Thread(name='%s' %self.cont,
# target=self.leerFrames)
#hilo2.start()
self.cont = self.cont + 1
#else:
# print('sincronización nuevo')
# if self.cont == 1:
# hilo2 = threading.Thread(target=self.leerFrames)
# hilo2.start()
# cv2.imwrite("Nuevo16-"+str(self.cont)+".png", dst2)
# print("Nuevo16-"+str(self.cont))
# self.cont = self.cont+1
except KeyboardInterrupt:
#pass
exit()
def PararCaptura(self):
#self.hilo.join()
#self.hilo2.join()
#cv2.destroyAllWindows()
self.cap.stop()
exit()
