--- a/PLCGenerator.py	Wed Mar 21 16:38:28 2007 +0100
+++ b/PLCGenerator.py	Fri Apr 06 18:06:00 2007 +0200
@@ -30,6 +30,8 @@
                 "outputVars" : "VAR_OUTPUT", "inOutVars" : "VAR_IN_OUT", "externalVars" : "VAR_EXTERNAL",
                 "globalVars" : "VAR_GLOBAL", "accessVars" : "VAR_ACCESS"}
 
+pouTypeNames = {"function" : "FUNCTION", "functionBlock" : "FUNCTION_BLOCK", "program" : "PROGRAM"}
+
 def ReIndentText(text, nb_spaces):
     compute = ""
     lines = text.splitlines()
@@ -455,12 +457,8 @@
     program = ""
     for pou in project.getPous():
         pou_type = pou.getPouType().getValue()
-        if pou_type == "function":
-            pou_program = PouProgram(pou.getName(), "FUNCTION")
-        elif pou_type == "functionBlock":
-            pou_program = PouProgram(pou.getName(), "FUNCTION_BLOCK")
-        elif pou_type == "program":
-            pou_program = PouProgram(pou.getName(), "PROGRAM")
+        if pou_type in pouTypeNames:
+            pou_program = PouProgram(pou.getName(), pouTypeNames[pou_type])
         else:
             raise ValueError, "Undefined pou type"
         pou_program.GenerateInterface(pou.getInterface())