Automatic commit

[universe.git] / main.cc

diff --git a/main.cc b/main.cc
index b677526..405354c 100644 (file)
--- a/main.cc
+++ b/main.cc
@@ -20,6 +20,7 @@ using namespace std;
 #include "retina.h"
 #include "manipulator.h"
 #include "intelligence.h"
+#include "xfig_tracer.h"
 
 // To train
 // ./main --task hit_shape.task 0 --action-mode=random --nb-ticks=5000 --proportion-for-training=0.5 --save-file=dump.mem --no-window
@@ -394,8 +395,8 @@ int main(int argc, char **argv) {
                 retina.save_as_ppm("/tmp/retina.ppm");
                 cout << "Retina screen shot saved in /tmp/retina.ppm" << endl;
                 {
-                  ofstream out("/tmp/universe.fig");
-                  universe.print_fig(out);
+                  XFigTracer tracer("/tmp/universe.fig");
+                  universe.print_fig(&tracer);
                 }
               }
               else if(strcmp(se.key, "Shift_L") == 0 || strcmp(se.key, "Shift_R") == 0) press_shift = true;