Added first variant of util generator.

astand · astand · commit ad787bc79ca9 · 2021-02-10T12:14:09.000+03:00
diff --git a/src/codegen/c-util-generator.cpp b/src/codegen/c-util-generator.cpp
@@ -0,0 +1,74 @@
+#include "c-util-generator.h"
+#include <algorithm>
+
+CiUtilGenerator::CiUtilGenerator()
+{
+  Clear();
+}
+
+void CiUtilGenerator::Clear()
+{
+  // clear all groups of messages
+  tx.clear();
+  rx.clear();
+  both.clear();
+}
+
+
+void CiUtilGenerator::Generate(std::vector<MessageDescriptor_t*>& msgs, const FsDescriptor_t& fsd,
+  const MsgsClassification& groups)
+{
+  Clear();
+
+  // 1 step is to prepare vectors of message's groups
+  for (auto m : msgs)
+  {
+    auto v = std::find_if(groups.Both.begin(), groups.Both.end(), [&](const uint32_t& msgid)
+    {
+      return msgid == m->MsgID;
+    });
+
+    if (v != std::end(groups.Both))
+    {
+      // message is in Both group, so put it to rx and tx containers
+      tx.push_back(m);
+      rx.push_back(m);
+      continue;
+    }
+
+    v = std::find_if(groups.Rx.begin(), groups.Rx.end(), [&](const uint32_t& msgid)
+    {
+      return msgid == m->MsgID;
+    });
+
+    if (v != std::end(groups.Rx))
+    {
+      rx.push_back(m);
+      continue;
+    }
+
+    v = std::find_if(groups.Tx.begin(), groups.Tx.end(), [&](const uint32_t& msgid)
+    {
+      return msgid == m->MsgID;
+    });
+
+    if (v != std::end(groups.Tx))
+    {
+      tx.push_back(m);
+      continue;
+    }
+  }
+
+  std::sort(rx.begin(), rx.end(), [&](const MessageDescriptor_t* a, const MessageDescriptor_t* b)
+  {
+    return a->MsgID < b->MsgID;
+  });
+  std::sort(tx.begin(), tx.end(), [&](const MessageDescriptor_t* a, const MessageDescriptor_t* b)
+  {
+    return a->MsgID < b->MsgID;
+  });
+
+  fdesc = &fsd;
+
+}
+
diff --git a/src/codegen/c-util-generator.h b/src/codegen/c-util-generator.h
@@ -0,0 +1,34 @@
+#pragma once
+
+#include "types/message.h"
+#include "fs-creator.h"
+
+class CiUtilGenerator {
+ public:
+  CiUtilGenerator();
+
+  void Clear();
+
+  // msgs - the whole list of messages in CAN matrix
+  // fsd - file and names descriptor
+  // groups - collection of msg IDs assigned to available groups (rx, tx, both)
+
+  // the output of this function source files which contain:
+  // - global TX and RX typedefs message struct
+  // - function to Unpack incoming frame to dedicated RX message struct field
+  // - optional (through define in global "dbccodeconf.h") variable allocation in source files
+  //
+  void Generate(std::vector<MessageDescriptor_t*>& msgs, const FsDescriptor_t& fsd, const MsgsClassification& groups);
+
+ private:
+  void PrintHeader();
+  void PrintSource();
+
+ private:
+
+  std::vector<MessageDescriptor_t*> tx;
+  std::vector<MessageDescriptor_t*> rx;
+  std::vector<MessageDescriptor_t*> both;
+
+  const FsDescriptor_t* fdesc;
+};
diff --git a/src/types/message.h b/src/types/message.h
@@ -131,3 +131,14 @@ typedef struct
   std::string CommentText;
 
 } MessageDescriptor_t;
+
+typedef struct
+{
+  std::vector<uint32_t> Rx;
+
+  std::vector<uint32_t> Tx;
+
+  std::vector<uint32_t> Both;
+
+} MsgsClassification;
+
