Merge pull request #84 from pkendall64/hdzero-osd

Add OSD support for HDZero goggles

Author: pkendall64

python/osd_test.py

@@ -0,0 +1,72 @@
+import time
+import serial
+import argparse
+import serials_find
+
+def crc8_dvb_s2(crc, a):
+  crc = crc ^ a
+  for ii in range(8):
+    if crc & 0x80:
+      crc = (crc << 1) ^ 0xD5
+    else:
+      crc = crc << 1
+  return crc & 0xFF
+
+def send_msp(port, baud, body):
+  s = serial.Serial(port=port, baudrate=baud,
+      bytesize=8, parity='N', stopbits=1,
+      timeout=1, xonxoff=0, rtscts=0)
+  crc = 0
+  for x in body:
+    crc = crc8_dvb_s2(crc, x)
+  msp = [ord('$'),ord('X'),ord('<')]
+  msp = msp + body
+  msp.append(crc)
+  s.write(msp)
+  print(msp)
+  time.sleep(2.0)
+  print(s.read_all())
+
+def send_clear(port, baud):
+  msp = [0,0xb6,0x00,1,0,0x02]
+  send_msp(port, baud, msp)
+
+def send_display(port, baud):
+  msp = [0,0xb6,0x00,1,0,0x04]
+  send_msp(port, baud, msp)
+
+def send_msg(port, baud, row, col, str):
+  l = 4+len(str)
+  msp = [0,0xb6,0x00,l%256,int(l/256),0x03,row,col,0]
+  for x in [*str]:
+    msp.append(ord(x))
+  send_msp(port, baud, msp)
+
+if __name__ == '__main__':
+  parser = argparse.ArgumentParser(
+    description="Initialize EdgeTX passthrough to internal module")
+  parser.add_argument("-b", "--baud", type=int, default=460800,
+    help="Baud rate for passthrough communication")
+  parser.add_argument("-p", "--port", type=str,
+    help="Override serial port autodetection and use PORT")
+  parser.add_argument("-c", "--clear", action='store_true',
+    help="Clear the OSD")
+  parser.add_argument("-d", "--display", action='store_true',
+    help="Draw the currently set OSD")
+  parser.add_argument("row", type=int, nargs='?', default=0,
+    help="Row where message is to be displayed")
+  parser.add_argument("col", type=int, nargs='?', default=0,
+    help="Column where message is to be displayed")
+  parser.add_argument("message", type=str, nargs='?', default='',
+    help="Message to be displayed")
+  args = parser.parse_args()
+
+  if (args.port == None):
+    args.port = serials_find.get_serial_port()
+
+  if args.clear:
+    send_clear(args.port, args.baud)
+  elif args.display:
+    send_display(args.port, args.baud)
+  else:
+    send_msg(args.port, args.baud, args.row, args.col, args.message)

src/hdzero.cpp

@@ -110,6 +110,13 @@ HDZero::SendHeadTrackingEnableCmd(bool enable)
     msp.sendPacket(&packet, m_port);
 }
 
+void
+HDZero::SetOSD(mspPacket_t *packet)
+{
+    MSP msp;
+    msp.sendPacket(packet, m_port);
+}
+
 void
 HDZero::SetRTC()
 {
@@ -131,4 +138,4 @@ HDZero::SetRTC()
     packet.addByte(timeData.tm_sec);
 
     msp.sendPacket(&packet, m_port);
-}
+}

src/hdzero.h

@@ -24,5 +24,6 @@ class HDZero : public MSPModuleBase
     uint8_t GetRecordingState();
     void SetRecordingState(uint8_t recordingState, uint16_t delay);
     void SendHeadTrackingEnableCmd(bool enable);
+    void SetOSD(mspPacket_t *packet);
     void SetRTC();
 };