diff --git a/companion_software/images/get_endpoint.py b/companion_software/images/get_endpoint.py deleted file mode 100644 index 713a0f3..0000000 --- a/companion_software/images/get_endpoint.py +++ /dev/null @@ -1,6 +0,0 @@ -import json -import robot_util - -url = '%s/v1/get_endpoint/jsmpeg_video_capture/%s' % ('https://api.robotstreamer.com', '4618') -response = robot_util.getWithRetry(url) -print(json.loads(response)) diff --git a/companion_software/images/robot_util.py b/companion_software/images/robot_util.py deleted file mode 100644 index a558584..0000000 --- a/companion_software/images/robot_util.py +++ /dev/null @@ -1,126 +0,0 @@ -import os -import time -import traceback -import ssl -import urllib.request -import getpass -import json -import _thread - - -ConfigFilename = "/home/pi/config_" + getpass.getuser() + ".json" #this is never used - -KeepAlivePeriod = 6 - - -def times(lst, number): - return [x*number for x in lst] - - -def getWithRetry(url, secure=True): - - for retryNumber in range(2000): - try: - print("GET", url) - if secure: - object = urllib.request.urlopen(url) - - else: - ctx = ssl.create_default_context() - ctx.check_hostname = False - ctx.verify_mode = ssl.CERT_NONE - object = urllib.request.urlopen(url, context=ctx) - - - - break - except: - print("could not open url", url) - traceback.print_exc() - time.sleep(2) - - data = object.read() - encoding = object.info().get_content_charset('utf-8') - return data.decode(encoding) - - -def getNoRetry(url, secure=True): - #socket test has retry - try: - print("GET", url) - if secure: - object = urllib.request.urlopen(url) - - else: - ctx = ssl.create_default_context() - ctx.check_hostname = False - ctx.verify_mode = ssl.CERT_NONE - object = urllib.request.urlopen(url, context=ctx) - - except: - print("could not open url", url) - return - - data = object.read() - encoding = object.info().get_content_charset('utf-8') - return data.decode(encoding) - - -def sendSerialCommand(ser, command): - - - print(ser.name) # check which port was really used - ser.nonblocking() - - # loop to collect input - #s = "f" - #print "string:", s - print(str(command.lower())) - ser.write(command.lower() + "\r\n") # write a string - #ser.write(s) - ser.flush() - - #while ser.in_waiting > 0: - # print "read:", ser.read() - - #ser.close() - - - -def makePOST(url, data): - - params = json.dumps(data).encode('utf8') - req = urllib.request.Request(url, - data=params, - headers={'content-type': 'application/json'}) - response = urllib.request.urlopen(req) - return response - - -def sendCameraAliveMessage(apiServer, cameraID, streamKey): - - print("sending camera alive message") - url = '%s/v1/set_camera_status' % (apiServer) - print("url", url) - try: - response = makePOST(url, {'camera_id': cameraID, - 'camera_status': 'online', - 'stream_key': streamKey, - 'type': 'robot_git'}) - #todo:send unified stream key here - except: - print("could not make post to", url) - - - -def aplayFile(filename): - for hardwareNumber in (2, 0, 1): - _thread.start_new_thread(os.system, ('aplay -D plughw:%d,0 %s' % (hardwareNumber, filename),)) - - -def handleSoundCommand(command, keyPosition): - print("command:", command, "key position:", keyPosition) - if len(command) >= 6: - if command[0:5] == "SOUND" and keyPosition == "down": - number = int(command[5:]) - aplayFile('/home/pi/sound/SOUND%d.WAV' % number) diff --git a/companion_software/main.py b/companion_software/main.py new file mode 100644 index 0000000..d2635bf --- /dev/null +++ b/companion_software/main.py @@ -0,0 +1,21 @@ +from PIL import Image, ImageDraw + +def testScreen(x=1280, y=720, image=None): + if image == None: + image = Image.new("RGB", (x, y)) + + draw = ImageDraw.Draw(image) + + + stripeList = ["red", "green", "blue", "orange", "purple"] + for i, stripe in enumerate(stripeList): + draw.rectangle(( + (x/len(stripeList)) * i, 0, + (x/len(stripeList)) * (i + 1), x), + fill=stripe) + + return image + +if __name__ == '__main__': + print('running') + testScreen().save('test_screen.png')