ENCODING!! and lots of organization

2026-08-10 01:43:46 +00:00 · 2023-09-13 14:50:27 -05:00
parent 2ab90d0146
commit fa0160fc54
37 changed files with 651 additions and 222 deletions
@@ -0,0 +1,142 @@
+const PacketParser = require('./decoders/packet');
+const InfoPacket = require('./models/packets/info-packet');
+const DataPacket = require('./models/packets/data-packet');
+
+class Decoder {
+  lastInfoPacketHeader;
+
+  lastDataPacketHeader;
+
+  infoPacketFrames = {};
+
+  dataPacketFrames = {};
+
+  constructor() {
+    this.info = {
+      trackers: {},
+    };
+
+    this.data = {
+      trackers: {},
+    };
+  }
+
+  updateInfo(framePackets) {
+    // console.log('info update');
+    framePackets.forEach((packet) => {
+      packet.subChunks?.forEach((subChunk) => {
+        if (subChunk.id === 0x0001) {
+          // NOTE(jwetzell): system name subChunk
+          this.info.system_name = subChunk.system_name;
+        } else if (subChunk.id === 0x0002) {
+          //   console.log(subChunk);
+          subChunk.trackers?.forEach((tracker) => {
+            if (this.info.trackers[tracker.id] === undefined) {
+              this.info.trackers[tracker.id] = {};
+            }
+            this.info.trackers[tracker.id].name = tracker.tracker_name.tracker_name;
+          });
+        }
+      });
+    });
+  }
+
+  updateData(framePackets) {
+    // console.log('data update');
+    framePackets.forEach((packet) => {
+      packet.subChunks?.forEach((subChunk) => {
+        if (subChunk.id === 0x0001) {
+          subChunk.trackers?.forEach((tracker) => {
+            if (this.data.trackers[tracker.id] === undefined) {
+              this.data.trackers[tracker.id] = {};
+            }
+            tracker.fields?.forEach((field) => {
+              Object.entries(field).forEach(([key, value]) => {
+                this.data.trackers[tracker.id][key] = value;
+              });
+            });
+          });
+        }
+      });
+    });
+  }
+
+  // TODO(jwetzell): support multiple frame packets this might require some rework of how chunks are assumed
+  decode(packetBuf) {
+    const packet = PacketParser.parse(packetBuf);
+    // console.log(packet);
+
+    if (packet.id === 0x6756) {
+      const infoPacket = new InfoPacket(packet);
+      if (infoPacket) {
+        if (infoPacket.subChunks?.length > 0) {
+          const currentInfoPacketHeader = infoPacket.getHeaderPacket();
+          // console.log(currentInfoPacketHeader);
+          if (!currentInfoPacketHeader) {
+            // NOTE(jwetzell): not sure that info packets without a header subchunk are valid?
+            return;
+          }
+
+          const systemSubChunk = infoPacket.getSystemPacket();
+          // console.log(systemSubChunk);
+          if (!systemSubChunk) {
+            // NOTE(jwetzell): not sure that info packets without a system subchunk are valid?
+            return;
+          }
+
+          if (this.infoPacketFrames[currentInfoPacketHeader.frame_id] === undefined) {
+            this.infoPacketFrames[currentInfoPacketHeader.frame_id] = [];
+          }
+          this.infoPacketFrames[currentInfoPacketHeader.frame_id].push(infoPacket);
+
+          if (
+            this.infoPacketFrames[currentInfoPacketHeader.frame_id].length ===
+            currentInfoPacketHeader.frame_packet_count
+          ) {
+            // console.log('found complete info frame');
+            this.updateInfo(this.infoPacketFrames[currentInfoPacketHeader.frame_id]);
+            delete this.infoPacketFrames[currentInfoPacketHeader.frame_id];
+          } else {
+            // TODO(jwetzell): need to compute whether a frame is complete
+            // console.log('compute frame completion');
+          }
+
+          this.lastInfoPacketHeader = currentInfoPacketHeader;
+        }
+      }
+    } else if (packet.id === 0x6755) {
+      const dataPacket = new DataPacket(packet);
+      //   console.log(dataPacket);
+      if (dataPacket) {
+        if (dataPacket.subChunks?.length > 0) {
+          const currentDataPacketHeader = dataPacket.getHeaderPacket();
+          if (!currentDataPacketHeader) {
+            // NOTE(jwetzell): not sure that info packets without a header subchunk are valid?
+            console.error('data packet with no header');
+            return;
+          }
+
+          if (this.dataPacketFrames[currentDataPacketHeader.frame_id] === undefined) {
+            this.dataPacketFrames[currentDataPacketHeader.frame_id] = [];
+          }
+          this.dataPacketFrames[currentDataPacketHeader.frame_id].push(dataPacket);
+
+          if (
+            this.dataPacketFrames[currentDataPacketHeader.frame_id].length ===
+            currentDataPacketHeader.frame_packet_count
+          ) {
+            this.updateData(this.dataPacketFrames[currentDataPacketHeader.frame_id]);
+            delete this.dataPacketFrames[currentDataPacketHeader.frame_id];
+          } else {
+            // TODO(jwetzell): need to compute whether a frame is complete
+            // console.log('compute data frame completion');
+          }
+
+          this.lastDataPacketHeader = currentDataPacketHeader;
+        }
+      }
+    }
+  }
+}
+
+module.exports = Decoder;