from OMChem.EngStm import EngStm
class Heater():
    counter = 1
    def __init__(self,name='Heater',PressureDrop = None, eff = None):
        self.PressureDrop = PressureDrop
        self.eff = eff
        self.OM_data_eqn = ''
        self.OM_data_init = ''
        self.InputStms = None
        self.OutputStms = None
        self.EngStms = EngStm(name='EngStm'+name)
        self.type = 'Heater'
        self.mode = None
        self.modeVal = None

        self.Prop = {
                'pressDrop':None,
                'eff':None,
                'outT':None,
                'tempInc':None,
                'heatAdd':None,
            }

        # new 
        self.name = name + str(Heater.counter) 
        self.no_of_input = 1 
        self.no_of_output = 1  
        Heater.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 ["heatAdd","outT","outVapPhasMolFrac","tempInc","enFlo"]

    def paramgetter(self,mode="heatAdd"):
        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.Heater " + 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.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