Fan control: Python script framework

The basics of a python script that will generate the
fan zone data structures for the fan control code
from zone and fan yaml files using a mako template.

Change-Id: I1d92dc55c2b6dd986e5aaae0d3440ef85cbe79c0
Signed-off-by: Matt Spinler <spinler@us.ibm.com>

1 files changed, 78 insertions, 0 deletions

diff --git a/control/gen-fan-zone-defs.py b/control/gen-fan-zone-defs.py
new file mode 100755
index 0000000..806b484
--- /dev/null
+++ b/control/gen-fan-zone-defs.py
@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+
+"""
+This script reads in fan definition and zone definition YAML
+files and generates a set of structures for use by the fan control code.
+"""
+
+import os
+import sys
+import yaml
+from argparse import ArgumentParser
+from mako.template import Template
+
+#Note: Condition is a TODO
+tmpl = '''/* This is a generated file. */
+#include "manager.hpp"
+
+using namespace phosphor::fan::control;
+
+const std::vector<ZoneGroup> Manager::_zoneLayouts
+{
+%for zone_group in zones:
+    std::make_tuple(
+            std::vector<Condition>{},
+            std::vector<ZoneDefinition>{
+            %for zone in zone_group['zones']:
+                std::make_tuple(${zone['num']},
+                                ${zone['initial_speed']},
+                                std::vector<FanDefinition>{
+                                    %for fan in zone['fans']:
+                                    std::make_tuple("${fan['name']}",
+                                        std::vector<std::string>{
+                                        %for sensor in fan['sensors']:
+                                           "${sensor}",
+                                        %endfor
+                                        }
+                                    ),
+                                    %endfor
+                                }
+                ),
+            %endfor
+            }
+    ),
+%endfor
+};
+'''
+
+if __name__ == '__main__':
+    parser = ArgumentParser(
+        description="Phosphor fan zone definition parser")
+
+    parser.add_argument('-z', '--zone_yaml', dest='zone_yaml',
+                        default="example/zones.yaml",
+                        help='fan zone definitional yaml')
+    parser.add_argument('-f', '--fan_yaml', dest='fan_yaml',
+                        default="example/fans.yaml",
+                        help='fan definitional yaml')
+    parser.add_argument('-o', '--output_dir', dest='output_dir',
+                        default=".",
+                        help='output directory')
+    args = parser.parse_args()
+
+    if not args.zone_yaml or not args.fan_yaml:
+        parser.print_usage()
+        sys.exit(-1)
+
+    with open(args.zone_yaml, 'r') as zone_input:
+        zone_data = yaml.safe_load(zone_input) or {}
+
+    with open(args.fan_yaml, 'r') as fan_input:
+        fan_data = yaml.safe_load(fan_input) or {}
+
+    #Fill in with using zone_data and fan_data with next commit
+    zone_data = []
+
+    output_file = os.path.join(args.output_dir, "fan_zone_defs.cpp")
+    with open(output_file, 'w') as output:
+        output.write(Template(tmpl).render(zones=zone_data))