avoid corrupt dumps by creating new files instead of appending, allow setting dumper and serum options via config file

Author: mkalkbrenner

include/DMDUtil/Config.h

@@ -55,6 +55,8 @@ class DMDUTILAPI Config
   void SetShowNotColorizedFrames(bool showNotColorizedFrames) { m_showNotColorizedFrames = showNotColorizedFrames; }
   bool IsDumpNotColorizedFrames() const { return m_dumpNotColorizedFrames; }
   void SetDumpNotColorizedFrames(bool dumpNotColorizedFrames) { m_dumpNotColorizedFrames = dumpNotColorizedFrames; }
+  bool IsDumpFrames() const { return m_dumpFrames; }
+  void SetDumpFrames(bool dumpFrames) { m_dumpFrames = dumpFrames; }
   bool IsFilterTransitionalFrames() const { return m_filterTransitionalFrames; }
   void SetFilterTransitionalFrames(bool filterTransitionalFrames)
   {
@@ -107,6 +109,7 @@ class DMDUTILAPI Config
   int m_framesToSkip;
   bool m_showNotColorizedFrames;
   bool m_dumpNotColorizedFrames;
+  bool m_dumpFrames;
   bool m_filterTransitionalFrames;
   bool m_zedmd;
   std::string m_zedmdDevice;

include/DMDUtil/DMD.h

@@ -183,6 +183,7 @@ class DMDUTILAPI DMD
   void AdjustRGB24Depth(uint8_t* pData, uint8_t* pDstData, int length, uint8_t* palette, uint8_t depth);
   void HandleTrigger(uint16_t id);
   void QueueSerumFrames(Update* dmdUpdate, bool render32 = true, bool render64 = true);
+  void GenerateRandomSuffix(char* buffer, size_t length);
 
   void DmdFrameThread();
   void LevelDMDThread();

src/Config.cpp

@@ -25,6 +25,7 @@ Config::Config()
   m_framesToSkip = 0;
   m_showNotColorizedFrames = false;
   m_dumpNotColorizedFrames = false;
+  m_dumpFrames = false;
   m_filterTransitionalFrames = false;
   m_zedmd = true;
   m_zedmdDevice.clear();
@@ -44,6 +45,7 @@ void Config::parseConfigFile(const char* path)
 {
   inih::INIReader r{path};
 
+  // DMDServer
   SetDMDServerAddr(r.Get<std::string>("DMDServer", "Addr", "localhost").c_str());
   SetDMDServerPort(r.Get<int>("DMDServer", "Port", 6789));
   SetAltColor(r.Get<bool>("DMDServer", "AltColor", true));
@@ -62,6 +64,14 @@ void Config::parseConfigFile(const char* path)
   // Pixelcade
   SetPixelcade(r.Get<bool>("Pixelcade", "Enabled", true));
   SetPixelcadeDevice(r.Get<std::string>("Pixelcade", "Device", "").c_str());
+  // Serum
+  SetIgnoreUnknownFramesTimeout(r.Get<int>("Serum", "IgnoreUnknownFramesTimeout", 0));
+  SetMaximumUnknownFramesToSkip(r.Get<int>("Serum", "MaximumUnknownFramesToSkip", 0));
+  SetShowNotColorizedFrames(r.Get<bool>("Serum", "ShowNotColorizedFrames", false));
+  // Dumper
+  SetDumpNotColorizedFrames(r.Get<bool>("Dumper", "DumpNotColorizedFrames", false));
+  SetDumpFrames(r.Get<bool>("Dumper", "DumpFrames", false));
+  SetFilterTransitionalFrames(r.Get<bool>("Dumper", "FilterTransitionalFrames", false));
 }
 
 }  // namespace DMDUtil

src/DMD.cpp

@@ -1463,6 +1463,18 @@ void DMD::AdjustRGB24Depth(uint8_t* pData, uint8_t* pDstData, int length, uint8_
   }
 }
 
+void DMD::GenerateRandomSuffix(char* buffer, size_t length)
+{
+  const char charset[] = "abcdefghijklmnopqrstuvwxyz0123456789";
+  size_t charsetSize = sizeof(charset) - 1;  // exclude null terminator
+
+  for (size_t i = 0; i < length; ++i)
+  {
+    buffer[i] = charset[rand() % charsetSize];
+  }
+  buffer[length] = '\0';
+}
+
 void DMD::DumpDMDTxtThread()
 {
   char name[DMDUTIL_MAX_NAME_SIZE] = {0};
@@ -1523,8 +1535,10 @@ void DMD::DumpDMDTxtThread()
           if (name[0] != '\0')
           {
             char filename[128];
-            snprintf(filename, DMDUTIL_MAX_NAME_SIZE + 5, "%s.txt", name);
-            f = fopen(filename, "a");
+            char suffix[9];  // 8 chars + null terminator
+            GenerateRandomSuffix(suffix, 8);
+            snprintf(filename, DMDUTIL_MAX_NAME_SIZE + 5, "%s-%s.txt", name, suffix);
+            f = fopen(filename, "w");
             update = true;
             memset(renderBuffer, 0, 2 * 256 * 64);
             passed[0] = passed[1] = 0;