simpler version of JobShop that we'll merge later

dream/simulation/GUI/SimpleJobShop.py

+from copy import copy
+import json
+import time
+import random
+import operator
+from datetime import datetime
+
+from dream.simulation.GUI.Default import Simulation as DefaultSimulation
+from dream.simulation.GUI.Default import schema
+
+
+class Simulation(DefaultSimulation):
+  def getConfigurationDict(self):
+    conf = DefaultSimulation.getConfigurationDict(self)
+    conf["Dream-MachineJobShop"] = {
+        "property_list": [
+          schema["operationType"] # XXX always manual
+        ],
+        "_class": 'Dream.MachineJobShop',
+        "name": 'Machine', # XXX or simply processing step ?
+        "short_id": "M",
+    }
+    conf["Dream-QueueJobShop"] = {
+        "property_list": [
+          schema["capacity"],
+          schema["schedulingRule"]
+        ],
+        "_class": 'Dream.QueueJobShop',
+        "name": 'Queue',
+        "short_id": "Q",
+    }
+    conf["Dream-ExitJobShop"] = {
+        "_class": 'Dream.ExitJobShop',
+        "name": 'Exit',
+        "short_id": "E",
+    }
+    conf["Dream-Configuration"]["gui"]["wip_part_spreadsheet"] = 1
+    conf["Dream-Configuration"]["gui"]["job_schedule_spreadsheet"] = 1
+    conf["Dream-Configuration"]["gui"]["job_gantt"] = 1
+    conf["Dream-Configuration"]["gui"]["queue_stat"] = 1
+
+    conf["Dream-Configuration"]["gui"]["debug_json"] = 1
+
+    # remove tools that does not make sense here
+    conf.pop('Dream-Machine')
+    conf.pop('Dream-Queue')
+    conf.pop('Dream-Exit')
+    conf.pop('Dream-Repairman')
+    conf.pop('Dream-Source')
+    conf.pop('Dream-EventGenerator')
+    return conf
+
+  def getRouteList(self, sequence_list, processing_time_list, prerequisite_list):
+    # use to record which predecessor has been already done, used to avoid doing
+    # two times Decomposition
+    predecessor_set = set()
+    route_list = []
+    for j, sequence_step in enumerate(sequence_list):
+      route = {"stationIdsList": sequence_step.split("-"),
+                "processingTime": {"distributionType": "Fixed",
+                                  "mean": float(processing_time_list[j])},
+                "setupTime": {"distributionType": "Fixed",
+                              "mean": .5}, # XXX hardcoded value
+              }
+      if prerequisite_list:
+        route["prerequisites"] = prerequisite_list
+      route_list.append(route)
+    return route_list
+
+  def getListFromString(self, my_string):
+    my_list = []
+    if not my_string in (None, ''):
+      my_list = my_string.split('-')
+    return my_list
+
+  def _preprocess(self, in_data):
+    """ Set the WIP in queue from spreadsheet data.
+    """
+    data = copy(DefaultSimulation._preprocess(self, in_data))
+    self.data = data
+
+    now = datetime.now()
+    if data['general']['currentDate']:
+      now = datetime.strptime(data['general']['currentDate'], '%Y/%m/%d')
+
+    if 'wip_part_spreadsheet' in data:
+      wip_list = []
+      i = 0
+      wip_part_spreadsheet_length = len(data['wip_part_spreadsheet'])
+      while i < wip_part_spreadsheet_length:
+        value_list = data['wip_part_spreadsheet'][i]
+        if value_list[0] == 'Order ID' or not value_list[4]:
+          i += 1
+          continue
+        order_dict = {}
+        wip_list.append(order_dict)
+        order_id, due_date, priority, project_manager, part, part_type,\
+          sequence_list, processing_time_list, prerequisite_string = value_list
+        due_date = (datetime.strptime(due_date, '%Y/%m/%d') - now).days * 24
+        prerequisite_list = self.getListFromString(prerequisite_string)
+        sequence_list = sequence_list.split('-')
+        processing_time_list = processing_time_list.split('-')
+
+        order_dict["_class"] = "Dream.Order"
+        order_dict["id"] = "%i" % i # XXX hack, we use it in UI to retrieve spreadsheet line
+        order_dict["manager"] = project_manager
+        order_dict["name"] = order_id
+        order_dict["dueDate"] = due_date
+        # XXX make it dynamic by writing a function that will reuse the
+        # code available a bit after
+        order_dict["route"] = self.getRouteList(sequence_list, processing_time_list,
+                                                prerequisite_list)
+        i += 1
+        component_list = []
+        if i < wip_part_spreadsheet_length:
+          while data['wip_part_spreadsheet'][i][0] in (None, ''):
+            value_list = data['wip_part_spreadsheet'][i]
+            if value_list[4] in (None, ''):
+              break
+            order_id, due_date, priority, project_manager, part, part_type,\
+              sequence_list, processing_time_list, prerequisite_string = value_list
+            sequence_list = sequence_list.split('-')
+            prerequisite_list = self.getListFromString(prerequisite_string)
+            processing_time_list = processing_time_list.split('-')
+            component_dict = {}
+            component_dict["_class"] = "Dream.OrderComponent"
+            if part_type == "Mould":
+              component_dict["_class"] = "Dream.Mould"
+            component_dict["componentType"] = part_type
+            component_dict["id"] = "%i" % i # XXX hack, we use it in UI to retrieve spreadsheet line
+            component_dict["name"] = part
+            component_list.append(component_dict)
+            route_list = self.getRouteList(sequence_list, processing_time_list,
+                                           prerequisite_list)
+            if part_type == "Mould":
+              route_list = route_list[1:]
+            component_dict["route"] = route_list
+            i+=1
+          order_dict["componentsList"] = component_list
+      data["nodes"]["QStart"]["wip"] = wip_list # XXX
+      del(data['wip_part_spreadsheet'])
+    return data