store id in DB

CMakeLists.txt

@@ -33,6 +33,7 @@ add_library(shared STATIC
   shared/src/values/Similarity.cpp
   shared/src/TcpSocket.cpp
   shared/src/sql.cpp
+  shared/src/values/DnaDB.cpp
 )
 
 add_executable(app

app/inc/App.hpp

@@ -1,6 +1,7 @@
 #include <array>
 #include "canvas/Canvas.hpp"
 #include "DnaStore.hpp"
+
 class App
 {
 

app/inc/DnaStore.hpp

@@ -2,6 +2,8 @@
 
 namespace DnaStore
 {
+  void init();
+  void deinit();
   void load(DnaManagerData *data);
   void saveData(DnaManagerData *data);
   void saveVec(DnaManagerData *data);

app/src/App.cpp

@@ -64,7 +64,7 @@ void App::init(int screenWidth, int screenHeight)
   this->screenWidth = screenWidth;
   this->screenHeight = screenHeight;
   this->canvas.init(screenWidth);
-
+  DnaStore::init();
   // int s = MeasureText("GEN 9999: 999/999", 20);
   // TraceLog(LOG_INFO, "%d", s);
 
@@ -239,4 +239,5 @@ void App::deinit()
     UnloadRenderTexture(canvasTexure[i]);
   }
   canvas.deinit();
+  DnaStore::deinit();
 }

app/src/DnaStore.cpp

@@ -9,6 +9,7 @@
 #include "sys.hpp"
 #include "DnaStore.hpp"
 #include "TcpSocket.hpp"
+#include "values/DnaDB.hpp"
 
 #include <raylib.h>
 
@@ -17,19 +18,30 @@
 #define VECTOR_FILE_NAME "VECTOR.bin"
 #define GEN_FILE_PATTRN "gen/%04d.bin"
 #define HOST_NAME "localhost"
+#define DB_NAME "data.db"
+
+DnaDB dnaDB;
+
+void DnaStore::init(){
+  const char* filename = sys::transformFilePath(DB_NAME);
+  dnaDB.init(filename);
+}
+
+void DnaStore::deinit(){
+  dnaDB.deinit();
+}
 
 void DnaStore::load(DnaManagerData *data)
 {
   const char *dir = sys::transformFilePath("gen");
   std::filesystem::create_directory(dir);
-  if (sys::fileExists(ID_FILE_NAME))
-  {
-    sys::loadDataFromFile(ID_FILE_NAME, &data->id, sizeof(int64_t));
-  }
-  else
-  {
+
+  std::vector<int64_t> ids = dnaDB.getUserIds();
+  if(!ids.empty()){
+    data->id = ids[0];
+  }else{
     data->id = time(nullptr);
-    sys::saveDataToFile(ID_FILE_NAME, &data->id, sizeof(int64_t));
+    dnaDB.insertUser(data->id, 0);
   }
 
   if (sys::fileExists(DATA_FILE_NAME))

shared/inc/sql.hpp

@@ -33,7 +33,7 @@ constexpr char DB_NAME[] = "data.db";
 struct sqlite3;
 struct sqlite3_stmt;
 
-#define SQLITE_ROW 100  /* sqlite3_step() has another row ready */
+#define SQLITE_ROW 100 /* sqlite3_step() has another row ready */
 #define SQL_DONE 101
 #define SQL_NULL 5
 #define SQLITE_OK 0
@@ -44,6 +44,7 @@ namespace sql
   void init();
   void shutdown();
   int open(const char *filename, sqlite3 **ppDb);
+  int exec(sqlite3 *db, const char *sql, int (*callback)(void *, int, char **, char **), void *first_arg, char **errmsg);
   int prepare_v2(sqlite3 *db, const char *zSql, int nByte, sqlite3_stmt **pStmt, const char **pzTail);
   int bind_int64(sqlite3_stmt *pStmt, int column, int64_t value);
   int step(sqlite3_stmt *pStmt);

shared/inc/values/DnaDB.hpp

@@ -0,0 +1,20 @@
+#include <cinttypes>
+#include <vector>
+
+struct sqlite3;
+
+class DnaDB
+{
+  sqlite3 *db;
+
+public:
+  void init(const char *db_name = "data.db");
+  void deinit();
+  std::vector<int64_t> getUserIds();
+  int insertUser(int64_t id, int64_t max_gen);
+  int updateUser(int64_t id, int64_t max_gen);
+
+  void getGenerations();
+  void getLikesForGeneration();
+  void getVectorForGeneration();
+};

shared/inc/values/DnaManager.hpp

@@ -7,9 +7,9 @@
 
 enum Liked
 {
-  tbd,
-  yes,
-  no
+  tbd = 0,
+  yes = 1,
+  no = 2,
 };
 
 struct UiUnit

shared/src/sql.cpp

@@ -78,6 +78,10 @@ namespace sql
   {
     return sqlite3_open(filename, ppDb);
   }
+  int exec(sqlite3 *db, const char *sql, int (*callback)(void *, int, char **, char **), void *first, char **errmsg)
+  {
+    return sqlite3_exec(db, sql, callback, first, errmsg);
+  }
   int prepare_v2(sqlite3 *db, const char *zSql, int nByte, sqlite3_stmt **pStmt, const char **pzTail)
   {
     return sqlite3_prepare_v2(db, zSql, nByte, pStmt, pzTail);

shared/src/values/DnaDB.cpp

@@ -0,0 +1,69 @@
+#include "values/DnaDB.hpp"
+
+#include <sql.hpp>
+
+
+
+void DnaDB::init(const char * db_name)
+{
+  sql::init();
+  int res = sql::open(db_name, &db);
+
+  constexpr char create_user_table[] = "CREATE TABLE IF NOT EXISTS user_table ( ID INTEGER PRIMARY KEY, USER_ID INTEGER, MAX_GENERATION INTEGER);";
+  res = sql::exec(db, create_user_table, nullptr, nullptr, nullptr);
+}
+
+void DnaDB::deinit()
+{
+  sql::close(db);
+  sql::shutdown();
+}
+
+std::vector<int64_t> DnaDB::getUserIds()
+{
+  constexpr char sql[] = "SELECT USER_ID FROM user_table;";
+  sqlite3_stmt *stmt;
+  sql::prepare_v2(db, sql, -1, &stmt, nullptr);
+  std::vector<int64_t> ids;
+  while (sql::step(stmt) != SQL_DONE)
+  {
+    int64_t id = sql::column_int64(stmt, 0);
+    ids.push_back(id);
+  }
+  sql::finalize(stmt);
+  return ids;
+}
+
+int DnaDB::insertUser(int64_t id, int64_t max_gen){
+  constexpr char sql[] = "INSERT INTO user_table (USER_ID, MAX_GENERATION) VALUES (?,?);";
+  sqlite3_stmt *stmt;
+  sql::prepare_v2(db, sql, -1, &stmt, nullptr);
+  sql::bind_int64(stmt, 1, id);
+  sql::bind_int64(stmt, 2, max_gen);
+  int ret = sql::step(stmt);
+  sql::finalize(stmt);
+  return ret;
+}
+
+int DnaDB::updateUser(int64_t id, int64_t max_gen){
+  constexpr char sql[] = "UPDATE user_table SET MAX_GENERATION = ? WHERE USER_ID = ?;";
+  sqlite3_stmt *stmt;
+  sql::prepare_v2(db, sql, -1, &stmt, nullptr);
+  sql::bind_int64(stmt, 1, max_gen);
+  sql::bind_int64(stmt, 2, id);
+  int ret = sql::step(stmt);
+  sql::finalize(stmt);
+  return ret;
+}
+
+void DnaDB::getGenerations()
+{
+}
+
+void DnaDB::getLikesForGeneration()
+{
+}
+
+void DnaDB::getVectorForGeneration()
+{
+}

shared/src/values/Similarity.cpp

@@ -136,7 +136,11 @@ namespace Similarity
     }
     else if (f == &Similarity::hamming_distance)
     {
-      return "hamming";
+      return "hamming_distance";
+    }
+    else if (f == &Similarity::hamming_distance_without_seeds)
+    {
+      return "hamming_distance_without_seeds";
     }
     else if (f == &Similarity::levenshtein_distance)
     {
@@ -144,7 +148,7 @@ namespace Similarity
     }
     else
     {
-      return "unknown";
+      return "unknown nameofFunc";
     }
   }
 
@@ -167,7 +171,6 @@ namespace Similarity
     
     const auto int_ms = std::chrono::duration_cast<std::chrono::microseconds>(stop - start);
  
-    TraceLog(LOG_INFO, "%s, %d", nameofFunc(f), int_ms);
     return average_similarity * 100.0f;
   }
 

view/inc/Vapp.hpp

@@ -28,6 +28,7 @@ public:
 private:
   bool showSelection;
   bool showStats;
+  bool saveToFile;
   sqlite3 *db;
   sqlite3_stmt *get_gen_num;
   std::vector<int64_t> ids;

view/src/Vapp.cpp

@@ -110,7 +110,7 @@ void Vapp::update()
     stageOfDrawing = DrawingStage::save;
     break;
   case DrawingStage::save:
-    drawToFile();
+    if(saveToFile) drawToFile();
     stageOfDrawing = DrawingStage::done;
     break;
   case DrawingStage::done:
@@ -140,6 +140,10 @@ void Vapp::draw()
     {
       setUpTable();
     }
+    if(ImGui::MenuItem("Save to File", nullptr, saveToFile, true))
+    {
+      saveToFile = !saveToFile;
+    }
     ImGui::EndMainMenuBar();
   }
 
@@ -335,27 +339,28 @@ void Vapp::setUpTable()
     }
 
     sql::reset(get_gen_stmt);
-
-    int64_t id = ids[selected_id_index];
-    char buff[50];
-    sprintf(buff, "%ld.txt", id);
-    std::ofstream file(buff);
-
-  file << "| index | euclidean_distance | cosine_similarity | cosine_similarity_int | hamming_distance | levenshtein_distance | dot_minmax |\n";
-  file << "| ---   | ---                | ---               | ---                   | ---              | ---                  | ---        |\n";
-
-
-
-    for (size_t i = 0; i < similTable.size(); i++)
+    if(saveToFile)
     {
-      file << "|" << i << "|";
-      for (size_t j = 0; j < similTable[i].size(); j++)
+      int64_t id = ids[selected_id_index];
+      const char* buff = TextFormat("%ld.txt", id);
+      std::ofstream file(buff);
+
+      file << "| index | euclidean_distance | cosine_similarity | cosine_similarity_int | hamming_distance | levenshtein_distance | dot_minmax |\n";
+      file << "| ---   | ---                | ---               | ---                   | ---              | ---                  | ---        |\n";
+
+
+
+      for (size_t i = 0; i < similTable.size(); i++)
       {
-        file << similTable[i][j] << "|";
+        file << "|" << i << "|";
+        for (size_t j = 0; j < similTable[i].size(); j++)
+        {
+          file << similTable[i][j] << "|";
+        }
+        file << "\n";
       }
-      file << "\n";
     }
-  }
+ }
 
   sql::finalize(get_gen_stmt);
 }