Changeset 30 in tmcsimulator-scriptbuilder for trunk/src/scriptbuilder/structures/SimulationScript.java

Timestamp:

08/02/2017 12:48:06 PM (9 years ago)

Author:

bmcguffin

Message:

Made TimeSlice?, ScriptIncident?, and SimulationScript? all implement I_XML_Writable. Simulationscript also now implements a saveScriptToFile() method.

File:

• trunk/src/scriptbuilder/structures/SimulationScript.java (modified) (7 diffs)

trunk/src/scriptbuilder/structures/SimulationScript.java

@@ -2 +2 @@
 
 import java.awt.Color;
+import java.io.BufferedWriter;
 import java.io.File;
+import java.io.FileWriter;
+import java.io.OutputStream;
+import java.io.OutputStreamWriter;
 import java.util.ArrayList;
 import java.util.List;
@@ -25 +29 @@
  * @version 2017/06/22
  */
-public class SimulationScript extends Observable
+public class SimulationScript extends Observable implements I_XML_Writable
 {
 
@@ -45 +49 @@
     };
 
+    public String title = "";
+
     /**
      * The incidents displayed by the GUI.
@@ -93 +99 @@
         }
         /*
-         // Add some demo events
+         Add some demo events
          incidents.add(new ScriptIncident(incidentColors[1], 174, "Blueberry Truck",
          "Blueberry truck crashed on the freeway.", 8800, this));
@@ -109 +115 @@
          "Truck stalled on the freeway.", 2800, this));
          numberOfIncidents++;
-         //incidents.add(new ScriptIncident(incidentColors[6], 179, "Tomato Truck Spill",
-         //        "Tomato trucked spilt tomatos all over the freeway.", 3200, this));
-         //incidents.add(new ScriptIncident(incidentColors[7], 180, "Crash at Intersection",
-         //        "Crash at the intersection of Tree Road and Red Road.", 4300, this));
-         //incidents.add(new ScriptIncident(incidentColors[8], 181, "Bomb Threat",
-         //        "Bomb threat and Road X.", 6000, this));
+         incidents.add(new ScriptIncident(incidentColors[6], 179, "Tomato Truck Spill",
+         "Tomato trucked spilt tomatos all over the freeway.", 3200, this));
+         incidents.add(new ScriptIncident(incidentColors[7], 180, "Crash at Intersection",
+         "Crash at the intersection of Tree Road and Red Road.", 4300, this));
+         incidents.add(new ScriptIncident(incidentColors[8], 181, "Bomb Threat",
+         "Bomb threat and Road X.", 6000, this));
          incidents.add(null);
          incidents.add(null);
@@ -146 +152 @@
          incidents.get(4).setOffset(4400);
          incidents.get(5).setOffset(1400);
-         //incidents.get(6).setOffset(7600);
-         //incidents.get(7).setOffset(2400);
-         //incidents.get(8).setOffset(4800); 
+         incidents.get(6).setOffset(7600);
+         incidents.get(7).setOffset(2400);
+         incidents.get(8).setOffset(4800); 
          }
          */
@@ -217 +223 @@
         this.update();
     }
+
+    public void saveScriptToFile(File f)
+    {
+        try
+        {
+            f.createNewFile();
+
+            BufferedWriter bw = new BufferedWriter(new FileWriter(f));
+            bw.write(this.toXML());
+            bw.flush();
+            bw.close();
+
+        }
+        catch (Exception ex)
+        {
+            System.out.println("ERROR SAVING SCRIPT");
+            ex.printStackTrace();
+        }
+    }
+
+    @Override
+    public String toXML()
+    {
+        String output = "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>\n";
+        output += "<!DOCTYPE TMC_SCRIPT SYSTEM \"script.dtd\">";
+        ArrayList<TimeSlice> slices = allSlices();
+        output += openTag(ELEMENT.TMC_SCRIPT.tag + " title=\"" + this.title + "\"");
+        for (TimeSlice slice : slices)
+        {
+            output += slice.toXML();
+        }
+        output += closeTag(ELEMENT.TMC_SCRIPT.tag);
+        return output;
+    }
+
+    @Override
+    public String openTag(String s)
+    {
+        return "<" + s + ">\n";
+    }
+
+    @Override
+    public String closeTag(String s)
+    {
+        return "</" + s + ">\n";
+    }
+
+    @Override
+    public String emptyTag(String s)
+    {
+        return "<" + s + "/>\n";
+    }
+
+    /**
+     * Arranges all timeslices in this script in chronological order, then by
+     * incident number.
+     *
+     * @return a list of all timeslices in the simulation script
+     */
+    public ArrayList<TimeSlice> allSlices()
+    {
+        ArrayList<TimeSlice> list = new ArrayList<TimeSlice>();
+        int length = absoluteLength();
+
+        for (int i = 0; i < length; i++)
+        {
+            for (ScriptIncident inc : incidents)
+            {
+                
+                if (inc != null && inc.slices.get(i) != null)
+                {
+                    list.add(inc.slices.get(i));
+                }
+            }
+        }
+        return list;
+    }
+
+    /**
+     * Gets the total length of the simulation in seconds.
+     *
+     * @return
+     */
+    public int absoluteLength()
+    {
+        int length = 0;
+        for (ScriptIncident inc : incidents)
+        {
+            if (inc != null)
+            {
+                int currentLength = inc.length + inc.offset;
+                if (currentLength > length)
+                {
+                    length = currentLength;
+                }
+            }
+        }
+        return length;
+    }
+
 }