1 files changed, 58 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 a86454e26..b2ddc9c7d 100644
--- a/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs
+++ b/Software/Visual_Studio/Utilities/Tango.PMRGenerator.CLI/Program.cs
@@ -59,6 +59,8 @@ namespace Tango.PMRGenerator.CLI
                 GenerateInsightsMonitors(db, pmrFolder);
                 GenerateEventTypes(db, pmrFolder);
                 GenerateInterfaceIOs(db, pmrFolder);
+                GenerateBtsrApplicationTypes(db, pmrFolder);
+                GenerateBtsrYarnTypes(db, pmrFolder);
             }
 
             Console.WriteLine("Done");
@@ -793,5 +795,61 @@ namespace Tango.PMRGenerator.CLI
 
             return null;
         }
+
+        private static void GenerateBtsrApplicationTypes(ObservablesContext db, String pmrFolder)
+        {
+            Console.WriteLine("Generating BTSR Application Types...");
+
+            ProtoEnumFile enumFile = new ProtoEnumFile();
+            enumFile.Name = "BtsrApplicationType";
+            enumFile.Package = "Tango.PMR.BTSR";
+
+            enumFile.Fields.Add(new EnumerationField()
+            {
+                Name = "UNDEFINED_APPLICATION",
+                Value = 0,
+            });
+
+            foreach (var field in db.BtsrApplicationTypes.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, "BTSR", enumFile.Name + ".proto"), enumString);
+        }
+
+        private static void GenerateBtsrYarnTypes(ObservablesContext db, String pmrFolder)
+        {
+            Console.WriteLine("Generating BTSR Yarn Types...");
+
+            ProtoEnumFile enumFile = new ProtoEnumFile();
+            enumFile.Name = "BtsrYarnType";
+            enumFile.Package = "Tango.PMR.BTSR";
+
+            enumFile.Fields.Add(new EnumerationField()
+            {
+                Name = "UNDEFINED_YARN",
+                Value = 0,
+            });
+
+            foreach (var field in db.BtsrYarnTypes.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, "BTSR", enumFile.Name + ".proto"), enumString);
+        }
     }
 }