Started working on tech heaters.

1 files changed, 23 insertions, 0 deletions

diff --git a/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs b/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs
index e7a1cead1..36f78391e 100644
--- a/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs
+++ b/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs
@@ -47,6 +47,7 @@ namespace Tango.PMRGenerator.CLI
                 GenerateWindingMethods(db, pmrFolder);
                 GenerateSpoolTypes(db, pmrFolder);
                 GenerateProcessParameters(db, pmrFolder);
+                GenerateHeaterTypes(db, pmrFolder);
 
                 GenerateColorLabColorSpaces(db, pmrFolder);
                 GenerateColorLabLiquidTypes(db, pmrFolder);
@@ -415,6 +416,28 @@ namespace Tango.PMRGenerator.CLI
             File.WriteAllText(Path.Combine(pmrFolder, "Printing", enumFile.Name + ".proto"), enumString);
         }
 
+        private static void GenerateHeaterTypes(ObservablesContext db, String pmrFolder)
+        {
+            Console.WriteLine("Generating Heater Types...");
+
+            ProtoEnumFile enumFile = new ProtoEnumFile();
+            enumFile.Name = "HeaterType";
+            enumFile.Package = "Tango.PMR.Diagnostics";
+
+            foreach (var field in db.TechHeaters.ToList().OrderBy(x => x.Code))
+            {
+                enumFile.Fields.Add(new EnumerationField()
+                {
+                    Name = field.Name.Replace(" ", ""),
+                    Value = field.Code,
+                });
+            }
+
+            String enumString = enumFile.GenerateCode();
+
+            File.WriteAllText(Path.Combine(pmrFolder, "Diagnostics", enumFile.Name + ".proto"), enumString);
+        }
+
         private static void GenerateSpoolTypes(ObservablesContext db, String pmrFolder)
         {
             Console.WriteLine("Generating Spool Types...");