nugrid_set.py 文件源码

def set_find_first_TP(self,dirs=[]):

        '''
            Find first TP of all runs and return array
        '''


                historydata=[]
                if (len(dirs)) == 0:
                        dirs=self.run_LOGS
                        historydata=self.run_historydata
                else:
                        for i in range(len(dirs)):
                                historydata.append(history_data(dirs[i]+"/LOGS"))
        t0_models=[]
        for j in range(len(dirs)):
            h1_boundary_mass  = historydata[j].get('h1_boundary_mass')
                    he4_boundary_mass = historydata[j].get('he4_boundary_mass')
                    star_mass         = historydata[j].get('star_mass')
                    #mx1_bot           = historyda.get('mx1_bot')*star_mass
                    #mx1_top           = historydata.get('mx1_top')*star_mass
                    mx2_bot           = historydata[j].get('mx2_bot')*star_mass
                    #mx2_top           = historydata.get('mx2_top')*star_mass
            he_lumi       = historydata[j].get('log_LHe')
            h_lumi        = historydata[j].get('log_LH')
            #model_number            = historydata[j].get('model_number')
            lum_array=[]
            activate=False
            models=[]
            for i in range(len(h1_boundary_mass)):
                if (h1_boundary_mass[i]-he4_boundary_mass[i] <0.1) and (he4_boundary_mass[i]>0.2):
                    if (mx2_bot[i]>he4_boundary_mass[i]) and (he_lumi[i]>h_lumi[i]):
                        activate=True
                        lum_array.append(he_lumi[i])
                        models.append(i)
                    if (activate == True) and (he_lumi[i]<h_lumi[i]):
                        break   
            t0_models.append(models[np.argmax(lum_array)]  )
        return t0_models