first commit

helix/presenters/panel_presenter.py

@@ -0,0 +1,99 @@
+import math
+
+from helix.models.coordinate import Coordinate
+import sys
+
+class ProjectPresenter(object):
+    def __init__(self, system_type, module_type):
+        self.offset = None
+        self.system_type = system_type
+        self.module_type = module_type
+
+    '''"This function expects coordinates to be in unit values, processed in coordinates_calculator.py" '''
+    def get_panel_data(self, panels, subarrays, max_y = None):
+        if self.offset is not None:
+            raise RuntimeError("ProjectPresenter panels must be computed before buildings")
+
+        table_data = []
+
+        system_constants = self.system_type.system_constants()
+        module_constants = self.system_type.module_constants(self.module_type)
+        spacing_x, spacing_y = module_constants.presenter_spacing
+        wind_zones = system_constants.wind_zones
+
+
+
+        for panel in panels:
+            subarray = [x for x in subarrays if x.subarray_number == panel.subarray][0]
+            origin = subarray.origin
+
+            table_data.append({
+                'x': (panel.coordinate.x + origin.x) * spacing_x,
+                'y': (panel.coordinate.y + origin.y) * spacing_y,
+                'width': spacing_x,
+                'height': spacing_y,
+                'data': {
+                    'panel_id': panel.id,
+                    'panel_type': panel.panel_type.number(),
+                    'ballast': panel.ballast,
+                    'wind_anchors': panel.wind_anchors,
+                    'seismic_anchors': panel.seismic_anchors,
+                    'wind_zones': wind_zones[panel.wind_zone],
+                    'link_trays': panel.presented_link_tray,
+                    'cross_trays': panel.cross_tray,
+                    'psf': round(panel.pressure, 2),
+                    'subarray': panel.subarray
+                }
+            })
+
+        # Move coordinates to reflect origin being at top-left
+        # (as per canvas) instead of bottom-left
+        # if max_y is not passed - use the panels instead of buildings
+        if max_y is None:
+            height = max(map(lambda row: row['y'], table_data))
+        else:
+            height = max_y
+        self.offset = height
+        for panel in table_data:
+            panel['y'] = height - panel['y']
+        return table_data
+
+    def get_buildings(self, buildings):
+        if self.offset is None:
+            self.offset = 0
+
+        module_constants = self.system_type.module_constants(self.module_type)
+        spacing_x, spacing_y = module_constants.presenter_spacing
+        # max_y = -sys.maxsize - 1 # for flipping the y coordinate
+
+        result = []
+        for building in buildings:
+            presentable_building = []
+            result.append(presentable_building)
+            # origin = self.find_origin(building)
+            for point in building:
+                point.x = point.x * spacing_x
+                point.y = point.y * spacing_y
+                presentable_building.append(point.__dict__)
+
+        for presentable_building in result:
+            for point in presentable_building:
+                point['y'] = self.offset - point['y']
+
+        return result
+
+    def get_max_y(self,buildings, panels):
+
+        module_constants = self.system_type.module_constants(self.module_type)
+        _, spacing_y = module_constants.presenter_spacing
+        all_y = []
+
+        if buildings is None or buildings == []: # no buildings in the file, probably CSV coordinates loaded
+            for panel in panels:
+                all_y.append(panel.coordinate.y)
+
+        for building in buildings:
+            for point in building:
+                all_y.append(point.y)
+        return max(all_y) * spacing_y
+