datasetIndexReceiver.py

# %% import libs
import socket
import time
import keyboard
import threading


class DatasetIndexReceiver():
    def __init__(self, IP_rcv='0.0.0.0', PORT_rcv=5556,
                       IP_snd='127.0.0.1', PORT_snd=5555):
        self.latest = 0

        # receiver socket
        self.sock = socket.socket(socket.AF_INET,  # Internet
                                  socket.SOCK_DGRAM)  # UDP
        self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.sock.bind((IP_rcv, PORT_rcv))

        # sender socket
        self.sock_send = socket.socket(socket.AF_INET,  # Internet
                                       socket.SOCK_DGRAM)  # UDP
        self.ADDR_snd = (IP_snd, PORT_snd)

        # start threads
        self.t1 = threading.Thread(target=self.reader, daemon=True).start()
        self.t2 = threading.Thread(target=self.controlHotKeys, daemon=True).start()

    # Receive messages from external app
    def reader(self):
        print('HRTF manager initialized')
        while True:
            data, addr = self.sock.recvfrom(1024)
            if len(data) > 0:
                tmp = data.decode()
                print
                if 'captur' in tmp.lower():  # if received cmd to capture position, request the pos receiver
                    self.sock_send.sendto(data, self.ADDR_snd)
                else:
                    self.latest = int(list(filter(str.isdigit, tmp))[0])
                    print(f'Dataset: {self.latest}')
            time.sleep(1)


    # Hotkey control
    def controlHotKeys(self): 
        savepos = False  
        changeDatasetIdx = None
 
        while True:
            time.sleep(0.1)  

            # capture position (use 'right' key)
            if keyboard.is_pressed('right'):
                savepos = True

            if (not keyboard.is_pressed('right') and savepos): 
                savepos = False
                # print('space') 
                self.sock_send.sendto('captur'.encode(), self.ADDR_snd)

            # change dataset index
            for number_key in range(10):
                if keyboard.is_pressed(f'alt gr+{number_key}'):
                    changeDatasetIdx = number_key

                if (not keyboard.is_pressed(f'alt gr+{number_key}') and isinstance(changeDatasetIdx, int)):  # dataset index
                    self.latest = changeDatasetIdx
                    changeDatasetIdx = None
                    print(f'Dataset: {self.latest}') 
                    
           
if __name__ == '__main__':
    a = DatasetIndexReceiver()

# %%