s

view/inc/Vapp.hpp

@@ -11,6 +11,7 @@ enum DrawingStage
   drawTree,
   drawBig,
   calSim,
+  save,
   done,
 };
 

view/src/Vapp.cpp

@@ -107,9 +107,12 @@ void Vapp::update()
     simil[3] = Similarity::calc_similarity(manager.vector, Similarity::hamming_distance);
     simil[4] = Similarity::calc_similarity(manager.vector, Similarity::hamming_distance_without_seeds);
     simil[5] = Similarity::calc_similarity(manager.vector, Similarity::levenshtein_distance);
+    stageOfDrawing = DrawingStage::save;
+    break;
+  case DrawingStage::save:
+    drawToFile();
     stageOfDrawing = DrawingStage::done;
     break;
-
   case DrawingStage::done:
     enableAll = true;
     break;
@@ -137,10 +140,6 @@ void Vapp::draw()
     {
       setUpTable();
     }
-    if (ImGui::MenuItem("DrawToFile", nullptr, false, enableAll))
-    {
-      drawToFile();
-    }
     ImGui::EndMainMenuBar();
   }
 
@@ -363,12 +362,10 @@ void Vapp::setUpTable()
 
 void Vapp::drawToFile()
 {
-  enableAll = false;
   int64_t id = ids[selected_id_index];
   char buff[50];
   sprintf(buff, "%ld_%d.png", id, selected_gen);
   Image image = LoadImageFromTexture(bigTexture.texture);
   ExportImage(image, buff);
   UnloadImage(image);
-  enableAll = true;
 }