Add savaToFile option
This commit is contained in:
@@ -28,6 +28,7 @@ public:
|
|||||||
private:
|
private:
|
||||||
bool showSelection;
|
bool showSelection;
|
||||||
bool showStats;
|
bool showStats;
|
||||||
|
bool saveToFile;
|
||||||
sqlite3 *db;
|
sqlite3 *db;
|
||||||
sqlite3_stmt *get_gen_num;
|
sqlite3_stmt *get_gen_num;
|
||||||
std::vector<int64_t> ids;
|
std::vector<int64_t> ids;
|
||||||
|
|||||||
@@ -110,7 +110,7 @@ void Vapp::update()
|
|||||||
stageOfDrawing = DrawingStage::save;
|
stageOfDrawing = DrawingStage::save;
|
||||||
break;
|
break;
|
||||||
case DrawingStage::save:
|
case DrawingStage::save:
|
||||||
drawToFile();
|
if(saveToFile) drawToFile();
|
||||||
stageOfDrawing = DrawingStage::done;
|
stageOfDrawing = DrawingStage::done;
|
||||||
break;
|
break;
|
||||||
case DrawingStage::done:
|
case DrawingStage::done:
|
||||||
@@ -140,6 +140,10 @@ void Vapp::draw()
|
|||||||
{
|
{
|
||||||
setUpTable();
|
setUpTable();
|
||||||
}
|
}
|
||||||
|
if(ImGui::MenuItem("Save to File", nullptr, saveToFile, true))
|
||||||
|
{
|
||||||
|
saveToFile = !saveToFile;
|
||||||
|
}
|
||||||
ImGui::EndMainMenuBar();
|
ImGui::EndMainMenuBar();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -335,27 +339,28 @@ void Vapp::setUpTable()
|
|||||||
}
|
}
|
||||||
|
|
||||||
sql::reset(get_gen_stmt);
|
sql::reset(get_gen_stmt);
|
||||||
|
if(saveToFile)
|
||||||
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++)
|
|
||||||
{
|
{
|
||||||
file << "|" << i << "|";
|
int64_t id = ids[selected_id_index];
|
||||||
for (size_t j = 0; j < similTable[i].size(); j++)
|
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);
|
sql::finalize(get_gen_stmt);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user