1 files changed, 58 insertions, 0 deletions
diff --git a/src/main/python/OMChem/Valve.py b/src/main/python/OMChem/Valve.py
new file mode 100644
index 0000000..05fa55f
--- /dev/null
+++ b/src/main/python/OMChem/Valve.py
@@ -0,0 +1,58 @@
+class Valve():
+    counter = 1
+    def __init__(self,name='Valve'):
+        self.mode = None
+        self.modeVal = None
+        self.OM_data_eqn = ''
+        self.OM_data_init = ''
+        self.InputStms = None
+        self.OutputStms = None
+        self.type = 'Valve'
+
+        self.Prop = {
+                'pressDrop':None,
+                'outP':None
+            }
+        # new 
+        self.name = name + str(Valve.counter) 
+        self.no_of_input = 1 
+        self.no_of_output = 1  
+        Valve.counter += 1  
+
+    def getname(self):
+        return self.name
+
+    def modesList(self):
+        return ["pressDrop","outP"]
+
+    def paramgetter(self,mode="pressDrop"):
+        self.mode = mode
+        dict = {self.mode:None}
+        return dict
+
+    def paramsetter(self,dict):
+
+        self.modeVal = dict[self.mode]      
+
+    def connect(self,InputStms = None,OutputStms = None):
+        self.InputStms = InputStms
+        self.OutputStms = OutputStms
+
+    def OM_Flowsheet_Init(self, addedcomp):
+        self.OM_data_init = ''
+        comp_count = len(addedcomp)
+        self.OM_data_init = self.OM_data_init + (
+        "Simulator.Unit_Operations.Valve " + self.name + "(Nc = " + str(comp_count))
+        self.OM_data_init = self.OM_data_init + (",comp = {")
+        comp = str(addedcomp).strip('[').strip(']')
+        comp = comp.replace("'", "")
+        self.OM_data_init = self.OM_data_init + comp + ("});\n")
+        return self.OM_data_init
+
+    def OM_Flowsheet_Eqn(self, addedcomp):
+        self.OM_data_eqn = ''
+        
+        self.OM_data_eqn = self.OM_data_eqn + ('connect(' + self.InputStms[0].name + '.outlet,' +  self.name + '.inlet' + ');\n')
+        self.OM_data_eqn = self.OM_data_eqn + ('connect(' + self.name + '.outlet,' + self.OutputStms[0].name + '.inlet);\n')
+        self.OM_data_eqn = self.OM_data_eqn + (self.name+'.'+self.mode+'='+ self.modeVal + ';\n')
+        return self.OM_data_eqn
+\ No newline at end of file