comparison yetilab/plot/plot.yeti @ 137:b28512329efc

Offscreen rendering & save-to-file (save-to-file works, offscreen currently doesn't)

 import org.jzy3d.colors.colormaps: ColorMapRainbow;
 import org.jzy3d.colors: ColorMapper, Color;
 import org.jzy3d.plot3d.rendering.canvas: Quality;
 import org.jzy3d.plot3d.rendering.view.modes: ViewPositionMode;
 
+import javax.imageio: ImageIO;
+
+import java.io: File;
+
 chartColours = array [
     { r = 82,  g = 126, b = 154 }, // dark steel blue
     { r = 161, g = 54,  b = 2   }, // red
     { r = 207, g = 228, b = 148 }, // grey-green
     { r = 21,  g = 183, b = 197 }, // light blue

         var keys = array (sort defaultKeys),
         var labels = [:],
         var animated = false,
         var normalised = false,
         var unit = "",
-        var xkeys = array (sort defaultXKeys)
+        var xkeys = array (sort defaultXKeys),
+        var saveTo = "",
+        var display = true,
     };
     for options
        \case of
         Keys kk: parsed.keys := array kk;
         XKeys xk: parsed.xkeys := array xk;
         Animated a: parsed.animated := a;
         Normalised n: parsed.normalised := n;
         Unit u: parsed.unit := u;
         Labels ll: parsed.labels := ll;
+        SaveTo file: parsed.saveTo := file;
+        Display d: parsed.display := d;
         esac;
     if empty? parsed.labels then
         parsed.labels := mapIntoHash id id parsed.keys
     fi;
     parsed);
 
+newChart opts =
+   (quality = Quality#Fastest;
+    quality#setAnimated(opts.animated);
+//    if opts.display then
+        new Chart(quality);
+//    else
+//        new Chart(quality, "offscreen,640,640");
+//    fi);
+    );
+
+showChart opts chart is 'a -> ~Chart -> () =
+   (//if opts.display then
+        \() ChartLauncher#openChart(chart);
+//    fi;
+    if opts.saveTo != "" then
+        \() chart#screenshot(opts.saveTo);
+    fi);
+    
 plotMatrix matrix =
    (mapper = newMatrixMapper matrix;
     size = matrix.size;
     xrange = new Range(0, size.columns - 1);
     yrange = new Range(0, size.rows - 1);

     ChartLauncher#openChart(chart);
     ());
 
 plotBarChart options values =
    (opts = parseOptions options (keys values) [];
-    quality = Quality#Fastest;
-    quality#setAnimated(opts.animated);
-    chart = new Chart(quality);
+    chart = newChart opts;
     var n = length opts.keys;
     scene = chart#getScene();
-    ticks = new float[n+1];
+    ticks = new double[n+1];
     tickLabels = new TickLabelMap();
     var i = 0;
     var x = n - i - 1;
     total = sum (map do k: if k in values then values[k] else 0 fi done opts.keys);
     for opts.keys do k:

         axes#setZAxeLabel(opts.unit);
     fi;
     axes#setXTickProvider(new StaticTickProvider(ticks));
     axes#setXTickRenderer(tickLabels);
     axes#setYTickLabelDisplayed(false);
-    ChartLauncher#openChart(chart);
-    ());
+    showChart opts chart);
 
 plotLines options values =
    (opts = parseOptions options (keys values) (keys values[head (keys values)]);
-    quality = Quality#Fastest;
-    quality#setAnimated(opts.animated);
-    chart = new Chart(quality);
+    chart = newChart opts;
     scene = chart#getScene();
     n = length opts.xkeys;
     var z = 0;
     for opts.keys do k:
         v = values[k];

     axes#setXAxeLabelDisplayed(false);
     axes#setYAxeLabelDisplayed(false);
     axes#setZAxeLabelDisplayed(true);
     axes#setZAxeLabel(opts.unit);
     axes#setYTickLabelDisplayed(false);
-    ChartLauncher#openChart(chart);
-    ());
+    showChart opts chart);
 
 stack keys xkeys values normalised =
    (stacked = mapIntoHash id \(mapIntoHash id \{ y0 = 0, y1 = 0 } xkeys) keys;
     prev = mapIntoHash id \0 xkeys;
     valueOf k xk = if xk in values[k] then values[k][xk] else 0 fi;

     poly#setColor(colour);
     poly);
 
 plotStacked options values =
    (opts = parseOptions options (keys values) (keys values[head (keys values)]);
-    quality = Quality#Fastest;
-    quality#setAnimated(opts.animated);
-    chart = new Chart(quality);
+    chart = newChart opts;
     scene = chart#getScene();
     stacked = stack opts.keys opts.xkeys values opts.normalised;
     var z = 0;
     var ty = 0;
     xtickLabels = new TickLabelMap();

         axes#setZTickRenderer(newPercentTickRenderer ());
     else
         axes#setZAxeLabel(opts.unit);
     fi;
     axes#setYTickLabelDisplayed(false);
-    ChartLauncher#openChart(chart);
-    ());
+    showChart opts chart);
 
 plotStructure structure =
     case structure of
     Grid matrix:
         plotMatrix matrix;