1 files changed, 76 insertions, 0 deletions
diff --git a/OMChem/Cooler.py b/OMChem/Cooler.py
new file mode 100644
index 0000000..9a44033
--- /dev/null
+++ b/OMChem/Cooler.py
@@ -0,0 +1,76 @@
+from OMChem.EngStm import EngStm
+class Cooler():
+    counter = 1
+    def __init__(self,name='Cooler',PressureDrop = None, eff = None):
+        self.PressureDrop = PressureDrop
+        self.eff = eff
+        # self.name = name
+        self.OM_data_eqn = ''
+        self.OM_data_init = ''
+        self.InputStms = None
+        self.OutputStms = None
+        #self.heatRem = heatRem
+        self.EngStms = EngStm(name='EngStm')
+        self.type = 'Cooler'
+        self.mode = None
+        self.modeVal = None
+
+        self.Prop = {
+                'pressDrop':None,
+                'eff':None,
+                'outT':None,
+                'tempDrop':None,
+                'heatRem':None,
+            }
+        # new 
+        self.name = name + str(Cooler.counter) 
+        self.no_of_input = 1 
+        self.no_of_output = 1  
+        Cooler.counter += 1  
+
+    def getname(self):
+        return self.name
+
+    def connect(self,InputStms = None,OutputStms = None):
+        self.InputStms = InputStms
+        self.OutputStms = OutputStms
+
+    def modesList(self):
+        return ["heatRem","outT","outVapPhasMolFrac","tempDrop","enFlo"]
+
+    def paramgetter(self,mode="heatRem"):
+        self.mode = mode
+        dict = {"PressureDrop":None,"eff":None,self.mode:None}
+        return dict
+
+    def paramsetter(self,dict):
+        
+        self.PressureDrop = dict['PressureDrop']
+        self.eff = dict['eff']
+        self.modeVal = dict[self.mode]
+        
+    def OM_Flowsheet_Init(self, addedcomp):
+        self.OM_data_init = ''
+        comp_count = len(addedcomp)
+        self.OM_data_init = self.OM_data_init + 'Simulator.Streams.Energy_Stream '+self.EngStms.name+';\n'
+        self.OM_data_init = self.OM_data_init + (
+        "Simulator.Unit_Operations.Cooler " + 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 + ("},")
+        self.OM_data_init = self.OM_data_init + 'pressDrop = ' + str(self.PressureDrop) + ','
+        self.OM_data_init = self.OM_data_init + 'eff = ' + str(self.eff) + ');\n'
+        return self.OM_data_init
+
+    def OM_Flowsheet_Eqn(self, addedcomp):
+        self.OM_data_eqn = ''
+        # self.OM_data_eqn = self.name + '.pressDrop = ' + str(self.PressDrop) + ';\n'
+        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 + ('connect(' + self.EngStms.name + '.outlet,'+ self.name + '.energy);\n')
+        if(self.mode =="enFlo"):
+            self.OM_data_eqn = self.OM_data_eqn + (self.EngStms.name+'.'+self.mode+'='+ self.modeVal + ';\n')
+        else:    
+            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