fix arrow styles not being reset

Author: hmsklee

index.html

@@ -173,9 +173,14 @@
   function changeDataset() {
     if (this.selectedIndex > 0) {
       parser.readJson(`./data/data${this.selectedIndex}.json`, function() {
-        console.log(parser);
         var words = parser.tokens.map((token, i) => new Word(token, i));
+        [].concat(parser.data.entities, parser.data.triggers).forEach(el => {
+          if (words[el.tokenIndex]) {
+            words[el.tokenIndex].tag = el.type;
+          }
+        });
         resetDrawing(words, []);
+        console.log(parser);
       });
     }
   };
@@ -184,11 +189,12 @@
     draw.clear();
     rowGroup = draw.group();
     groupAllElements = linkG = textG = arrowG = undefined;
-    dragElem = null;
     isDragging = false;
     rowOffsetX = 0;
     rowOffsetWord = null;
     prevX = -1;
+    styles = setupStyles(draw); //creates an associative array of whatever     
+    dragElem = null;
 
     rows = [];
     wordObjs = words || createTestWords(5, 6, 10);

js/parser.js

@@ -39,18 +39,23 @@ class Parser {
   parseData(data) {
     /* first get string tokens from the syntaxData */
     this.text = data.syntaxData.text;
+
+    const offsets = {};
     this.tokens = data.syntaxData.entities
       .map(x => x[2][0])
       .sort((a, b) => a[0] - b[0])
-      .map(interval => this.text.substring(interval[0], interval[1]));
+      .map((interval, i) => {
+        offsets[interval[0]] = i;
+        return this.text.substring(interval[0], interval[1]);
+      });
 
     /* parse the event data: entities, triggers, events, and relations */
     const e = data.eventData;
     const entities = e.entities.map(arr => {
           return {
               id: arr[0],
               type: arr[1],
-              interval: arr[2][0],
+              tokenIndex: offsets[arr[2][0][0]],
               string: e.text.substring(arr[2][0][0], arr[2][0][1])
           };
       });
@@ -59,7 +64,7 @@ class Parser {
           return {
               id: arr[0],
               type: arr[1],
-              interval: arr[2][0],
+              tokenIndex: offsets[arr[2][0][0]],
               string: e.text.substring(arr[2][0][0], arr[2][0][1])
           };
       });