Extending locations and simulations...

src/libs/models/locations.js

+/* These classes are defined to allow checking of location type with the
+instanceof operator. */
+
+class Location
+
 // Location types
-class City {
+class City extends Location{
     constructor(id) {
         this.id = id;
     }
 }
 
-class State {
+class State extends Location{
     constructor(id) {
         this.id = id;
     }
 }
 
-class Region {
+class Region extends Location{
     constructor(id) {
         this.id = id;
     }

src/libs/models/simulation.js

@@ -3,12 +3,16 @@ const log = require(`${libs}/log`)(module);
 
 const locations = require(`${libs}/models/locations`);
 
+// Should define this somewhere else
+const MAX_SIMULATION_TIME = 10;
+
+
 class Simulation {
     /* Simulation object to manage parameters and storing */
     constructor(in_object = null) {
         this.name = null;
         this.location = null;
-        this.sim_time = null;
+        this.time = null;
         this.goals = {
             care: null,
             inclusion: null,
@@ -20,6 +24,26 @@ class Simulation {
             }
         }
     }
+    setTime(t) {
+        t = parseInt(t, 10);
+        if(t>MAX_SIMULATION_TIME){
+            // Throw an error?
+            return;
+        }
+        this.time = t;
+    }
+    setLocation(l) {
+        // Should sanitize
+        this.location = l;
+    }
+    setCareGoals(g) {
+        // Should sanitize
+        this.goals.care = g;
+    }
+    setInclusionGoals(g) {
+        // Should sanitize
+        this.goals.inclusion = g;
+    }
     run() {
         /* Runs the Simulation with given parameters */
         // if (!this.name || !this.location || !this.time) {