From f05f3ba70879d0312446f930881f32192fae42d6 Mon Sep 17 00:00:00 2001
From: Chip Morningstar <chip@fudco.com>
Date: Tue, 19 May 2020 13:12:59 -0700
Subject: [PATCH] fix: deal more gracefully with data sources that have a
 common base file name

---
 packages/stat-logger/src/statGraph.js        | 16 ++++++++--------
 packages/swingset-runner/src/dataGraphApp.js |  9 +++++++++
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/packages/stat-logger/src/statGraph.js b/packages/stat-logger/src/statGraph.js
index 5c88564c9c6..d2ab13c3237 100644
--- a/packages/stat-logger/src/statGraph.js
+++ b/packages/stat-logger/src/statGraph.js
@@ -3,7 +3,6 @@ import path from 'path';
 import * as vega from 'vega';
 
 export function initGraphSpec(statsPath, xField, xLabel, yField, yLabel) {
-  const tag = path.basename(statsPath);
   const spec = {
     $schema: 'https://vega.github.io/schema/vega/v5.json',
     width: 1500,
@@ -24,7 +23,7 @@ export function initGraphSpec(statsPath, xField, xLabel, yField, yLabel) {
         type: 'linear',
         range: 'width',
         nice: true,
-        domain: { data: tag, field: xField },
+        domain: { data: statsPath, field: xField },
       },
       {
         name: 'y',
@@ -32,7 +31,7 @@ export function initGraphSpec(statsPath, xField, xLabel, yField, yLabel) {
         range: 'height',
         nice: true,
         zero: true,
-        domain: { data: tag, field: yField },
+        domain: { data: statsPath, field: yField },
       },
       {
         name: 'legend',
@@ -69,17 +68,18 @@ export function initGraphSpec(statsPath, xField, xLabel, yField, yLabel) {
 }
 
 export function addDataToGraphSpec(spec, statsPath) {
-  const tag = path.basename(statsPath);
   const dataElement = {
-    name: tag,
+    name: statsPath,
     format: { type: 'tsv' },
     url: `file://${statsPath}`,
   };
   spec.data.push(dataElement);
 }
 
-export function addGraphToGraphSpec(spec, statsPath, yField, color) {
-  const tag = path.basename(statsPath);
+export function addGraphToGraphSpec(spec, tag, statsPath, yField, color) {
+  if (!tag) {
+    tag = path.basename(statsPath);
+  }
   const legendElement = {
     text: `${tag} - ${yField}`,
     color,
@@ -87,7 +87,7 @@ export function addGraphToGraphSpec(spec, statsPath, yField, color) {
   spec.data[0].values.push(legendElement);
   const lineElement = {
     type: 'line',
-    from: { data: tag },
+    from: { data: statsPath },
     encode: {
       enter: {
         x: { scale: 'x', field: spec.scales[0].domain.field },
diff --git a/packages/swingset-runner/src/dataGraphApp.js b/packages/swingset-runner/src/dataGraphApp.js
index ad444ba5c31..d4f29837738 100644
--- a/packages/swingset-runner/src/dataGraphApp.js
+++ b/packages/swingset-runner/src/dataGraphApp.js
@@ -22,6 +22,7 @@ export async function dataGraphApp(xField, xLabel, yField, yLabel, lineFields) {
 
   let outfile = null;
   const datafiles = [];
+  const tags = [];
   let type = 'png';
 
   while (argv[0]) {
@@ -35,11 +36,18 @@ export async function dataGraphApp(xField, xLabel, yField, yLabel, lineFields) {
         case '--pdf':
           type = 'pdf';
           break;
+        case '--label':
+        case '-l':
+          tags.push(argv.shift());
+          break;
         default:
           throw new Error(`invalid flag ${arg}`);
       }
     } else {
       datafiles.push(arg);
+      if (datafiles.length < tags.length) {
+        tags.push(undefined);
+      }
     }
   }
   if (datafiles.length < 1) {
@@ -53,6 +61,7 @@ export async function dataGraphApp(xField, xLabel, yField, yLabel, lineFields) {
     for (let lineIdx = 0; lineIdx < lineFields.length; lineIdx += 1) {
       addGraphToGraphSpec(
         spec,
+        tags[dataIdx],
         datafiles[dataIdx],
         lineFields[lineIdx],
         groupColors[lineIdx % groupColors.length],