Added util source print.

astand · astand · commit 4fd51160d2ef · 2021-02-13T23:48:24.000+03:00
diff --git a/src/codegen/c-util-generator.cpp b/src/codegen/c-util-generator.cpp
@@ -14,6 +14,7 @@ CiUtilGenerator::CiUtilGenerator()
 {
   Clear();
   tof = new FileWriter;
+  condtree = new ConditionalTree;
 }
 
 void CiUtilGenerator::Clear()
@@ -84,10 +85,11 @@ void CiUtilGenerator::Generate(std::vector<MessageDescriptor_t*>& msgs, const Fs
 
   fdesc = &fsd;
 
+  // print header for util code
   PrintHeader();
 
+  // print main source for util code
   PrintSource();
-
 }
 
 void CiUtilGenerator::PrintHeader()
@@ -169,4 +171,105 @@ void CiUtilGenerator::PrintHeader()
 
 void CiUtilGenerator::PrintSource()
 {
+  tof->AppendLine(StrPrint("#include \"%s\"", fdesc->util_h.fname.c_str()), 2);
+
+  // optional RX and TX struct allocations
+  if (rx.size() > 0 || tx.size() > 0)
+  {
+    tof->AppendLine(StrPrint("#ifdef __DEF_%s__", fdesc->DRVNAME.c_str()), 2);
+
+    if (rx.size() > 0)
+    {
+      tof->AppendLine(StrPrint("%s_rx_t %s_rx;", fdesc->drvname.c_str(), fdesc->drvname.c_str()), 2);
+    }
+
+    if (tx.size() > 0)
+    {
+      tof->AppendLine(StrPrint("%s_tx_t %s_tx;", fdesc->drvname.c_str(), fdesc->drvname.c_str()), 2);
+    }
+
+    tof->AppendLine(StrPrint("#endif // __DEF_%s__", fdesc->DRVNAME.c_str()), 2);
+  }
+
+  if (rx.size() > 0)
+  {
+    // tree will be created inside (in dynamic memory) so this
+    // scope is responsible for deletion these resources
+    // tree is the struct tree-view which is used to execute
+    // binary search on FrameID for selecting unpacking function
+    auto tree = FillTreeLevel(rx, 0, rx.size());
+
+    tof->AppendLine(StrPrint("uint32_t %s_Receive(%s_rx_t* _m, const uint8_t* _d, uint32_t _id, uint8_t dlc_)",
+        fdesc->drvname.c_str(), fdesc->drvname.c_str()));
+
+    tof->AppendLine("{");
+    tof->AppendLine(" uint32_t recid = 0;");
+
+    // put tree-view struct on code (in treestr variable)
+    std::string treestr;
+    condtree->Clear();
+    tof->AppendLine(condtree->WriteCode(tree, treestr, 1));
+
+    tof->AppendLine(" return recid;");
+    tof->AppendLine("}", 2);
+
+    // clear tree after using
+    condtree->DeleteTree(tree);
+  }
+
+  tof->Flush(fdesc->util_c.fpath);
+}
+
+ConditionalTree_t* CiUtilGenerator::FillTreeLevel(std::vector<MessageDescriptor_t*>& list,
+  int32_t l,
+  int32_t h,
+  bool started)
+{
+  int32_t span = h - l;
+  int32_t lowhalf = span / 2;
+  int32_t highhalf = span - lowhalf;
+
+  treestarted = started;
+
+  if (h < 1)
+  {
+    return nullptr;
+  }
+
+  ConditionalTree_t* ret = new ConditionalTree_t;
+
+  if (!treestarted && h == 1)
+  {
+    ret->Type = ConditionalType::Cond;
+    auto msg = list[l];
+    ret->ConditionExpresion = StrPrint("_id == 0x%XU", msg->MsgID);
+    ret->High = new ConditionalTree_t;
+    ret->Type = ConditionalType::Express;
+    ret->UtilCodeBody = StrPrint("recid = Unpack_%s_%s(&(_m->%s), _d, dlc_);",
+        msg->Name.c_str(), code_drvname.c_str(), msg->Name.c_str());
+    return ret;
+  }
+
+  if (span > 1)
+  {
+    ret->Type = ConditionalType::Cond;
+
+    if (lowhalf > 1)
+      ret->ConditionExpresion = StrPrint("(_id >= 0x%XU) && (_id < 0x%XU)", list[l]->MsgID, list[(l + lowhalf)]->MsgID);
+    else
+      ret->ConditionExpresion = StrPrint("_id == 0x%XU", list[l]->MsgID);
+
+    ret->High = FillTreeLevel(list, l, l + lowhalf, true);
+    ret->Low = FillTreeLevel(list, l + lowhalf, h, true);
+  }
+  else
+  {
+    ret->Type = ConditionalType::Express;
+    auto msg = list[l];
+    ret->ConditionExpresion = StrPrint("_id == 0x%XU", msg->MsgID);
+    ret->UtilCodeBody = StrPrint("recid = Unpack_%s_%s(&(_m->%s), _d, dlc_);",
+        msg->Name.c_str(), code_drvname.c_str(), msg->Name.c_str());
+  }
+
+  return ret;
 }
diff --git a/src/codegen/c-util-generator.h b/src/codegen/c-util-generator.h
@@ -3,6 +3,7 @@
 #include "types/message.h"
 #include "fs-creator.h"
 #include "filewriter.h"
+#include "conditional-tree.h"
 
 class CiUtilGenerator {
  public:
@@ -25,6 +26,8 @@ class CiUtilGenerator {
  private:
   void PrintHeader();
   void PrintSource();
+  ConditionalTree_t* FillTreeLevel(std::vector<MessageDescriptor_t*>& msgs,
+    int32_t l, int32_t h, bool started = false);
 
  private:
 
@@ -39,4 +42,7 @@ class CiUtilGenerator {
   std::string file_drvname;
 
   const FsDescriptor_t* fdesc;
+  ConditionalTree* condtree;
+
+  bool treestarted;
 };
diff --git a/src/codegen/conditional-tree.cpp b/src/codegen/conditional-tree.cpp
@@ -0,0 +1,88 @@
+#include "conditional-tree.h"
+#include "helpers/formatter.h"
+
+ConditionalTree::ConditionalTree()
+{
+}
+
+// this method performs printing tree-viewed frames collection
+// to proper source code
+std::string ConditionalTree::WriteCode(const ConditionalTree_t* tree, std::string& outstr, uint8_t intend)
+{
+  if (tree != nullptr)
+  {
+    std::string temp;
+
+    if (tree->Type == ConditionalType::Cond)
+    {
+      temp = StrPrint("if (%s) {", tree->ConditionExpresion.c_str());
+      PrintCode(temp, intend);
+
+      WriteCode(tree->High, outstr, intend + 1);
+
+      if (tree->Low != nullptr)
+      {
+        if (tree->Low->Type == ConditionalType::Express)
+        {
+          temp = StrPrint("} else if (%s) {", tree->Low->ConditionExpresion.c_str());
+          PrintCode(temp, intend);
+        }
+        else
+        {
+          temp = "} else {";
+          PrintCode(temp, intend);
+        }
+
+        WriteCode(tree->Low, outstr, intend + 1);
+      }
+      else
+      {
+        temp = "} else {";
+        PrintCode(temp, intend);
+      }
+
+      if (tree->Low != nullptr)
+      {
+        temp = "}";
+        PrintCode(temp, intend);
+      }
+    }
+    else
+    {
+      temp = StrPrint("%s", tree->UtilCodeBody.c_str());
+      PrintCode(temp, intend);
+    }
+  }
+
+  return codestr;
+}
+
+void ConditionalTree::PrintCode(std::string& str, uint8_t indent)
+{
+  while (indent--)
+  {
+    codestr += " ";
+  }
+
+  codestr += str;
+  codestr += "\n";
+}
+
+
+void ConditionalTree::DeleteTree(ConditionalTree_t* tree)
+{
+  if (tree != nullptr)
+  {
+    if (tree->Low != nullptr)
+    {
+      DeleteTree(tree->Low);
+    }
+
+    if (tree->High != nullptr)
+    {
+      DeleteTree(tree->High);
+    }
+
+    delete tree;
+  }
+}
diff --git a/src/codegen/conditional-tree.h b/src/codegen/conditional-tree.h
@@ -0,0 +1,44 @@
+#pragma once
+
+#include <string>
+
+enum class ConditionalType { Cond, Express, Single };
+
+struct ConditionalTree_t
+{
+  ConditionalTree_t* High;
+
+  ConditionalTree_t* Low;
+
+  ConditionalType Type;
+
+  std::string ConditionExpresion;
+
+  std::string UtilCodeBody;
+
+  ConditionalTree_t() {
+    High = nullptr;
+    Low = nullptr;
+    Type = ConditionalType::Single;
+  }
+};
+
+class ConditionalTree {
+ public:
+  ConditionalTree();
+
+  void Clear() {
+    codestr.clear();
+  }
+
+  std::string WriteCode(const ConditionalTree_t* tree, std::string& outstr, uint8_t intend);
+
+  // deletes all memory allocated to tree members
+  void DeleteTree(ConditionalTree_t* tree);
+
+ private:
+
+  void PrintCode(std::string& str, uint8_t indent);
+
+  std::string codestr;
+};
