MotionStyleTransferInHand/HandLMServer.py at main · Louislar/MotionStyleTransferInHand · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
from http.server import BaseHTTPRequestHandler, HTTPServer
from httpServer import MyServer, hostName, serverPort
import threading
import time

stopThread = False

class HandLMServer():
    def __init__(self, hostIP=None, hostPort=None) -> None:
        self.hostName=hostIP
        self.serverPort=hostPort
        self.webServer=None
        self.httpServerThread=None
        self.curSentMsg=['', '']    # List for pass by reference(Not sure if it works? -> it works :-))
        self.getMsg=['']

    '''
    Reason of this is because we want to send a variable in the _requestHandler class,
    Python did not allow inner class use instances in outer class!!
    '''
    def _requestHandlerClassFunc(self):
        msgStringNeedToSend = self.curSentMsg
        msgStringReceived = self.getMsg
        class _requestHandler(BaseHTTPRequestHandler):
            def do_GET(self):
                self.send_response(200)
                self.send_header("Content-type", "text/html")
                self.end_headers()
                # TODO: 利用self.requestline區隔想要讀取的資料
                # TODO: _request[1]代表GET請求的路徑資料, 目前UnityDefault是傳送'/'而已
                #   之後可以傳送有意義的訊息
                # TODO: 之後還會傳送更多不同的URL, 請求不同動作種類估計的pose
                #       只需要在下一個frame回傳新的動作種類估計的pose就好, 現在還是回傳上一個frame估計的動作.
                #       也就是delay一個frame.
                # e.g. full body pose or wrist position
                _request = self.requestline.split(' ')
                print(_request)
                print(_request[1])
                if _request[1] == '/wrist':
                    self.wfile.write(bytes(msgStringNeedToSend[1], "utf-8"))
                else:
                    self.wfile.write(bytes(msgStringNeedToSend[0], "utf-8"))
                    msgStringReceived[0] = _request[1]
        return _requestHandler

    def httpServerServeForever(self):
        self.webServer.serve_forever()

    def startHTTPServerThread(self):
        self._requestHandler = self._requestHandlerClassFunc()
        self.webServer=HTTPServer((self.hostName, self.serverPort), self._requestHandler)
        print("Server started http://%s:%s" % (hostName, serverPort))
        httpServerThread = threading.Thread(target=self.httpServerServeForever,)
        httpServerThread.setDaemon(True)    # When Main thead ends, the sub thread will terminate
        httpServerThread.start()

    def stopHTTPServer(self):
        self.webServer.shutdown()
        self.webServer.server_close()
        print("Server stopped.")

if __name__=='__main__':
    handLMServer = HandLMServer(hostIP=hostName, hostPort=serverPort)
    handLMServer.startHTTPServerThread()
    curTime = time.time()
    while True:
        try:
            if time.time() - curTime > 3:
                handLMServer.curSentMsg[0] = str(curTime)
                handLMServer.curSentMsg[1] = str(curTime%10)
                curTime=time.time()
            pass
        except KeyboardInterrupt:
            print('keyboard interrupt')
            handLMServer.stopHTTPServer()
            break


# def httpServerServeForever(server: HTTPServer):
#     try:
#         server.serve_forever()
#     except KeyboardInterrupt:
#         pass

# if __name__=='__main__':
#     webServer = HTTPServer((hostName, serverPort), MyServer)
#     print("Server started http://%s:%s" % (hostName, serverPort))
#     httpServerThread = threading.Thread(target=httpServerServeForever, args=(webServer, ))
#     httpServerThread.setDaemon(True)    # When Main thead ends, the sub thread will terminate
#     httpServerThread.start()
#     # httpServerThread.join()    # Wait for the thread to terminate
#     while True:
#         try:
#             pass
#         except KeyboardInterrupt:
#             print('keyboard interrupt')
#             webServer.shutdown()
#             webServer.server_close()
#             print("Server stopped.")
#             break
#     # while not stopThread:
#     #     pass