nugrid_set.py 文件源码

def set_get_abu_distr_decay(self,cycles=20*[-1],mass_range=20*[[0,0]],ylim=20*[[0,0]],isotopes=['all']):
                import nugridse as mp
                import utils as u
                print self.runs_H5_restart
                massfrac_all=[]
                iso_all=[]
                for i in range(len(self.runs_H5_restart)):      
                        sefiles=mp.se(self.runs_H5_restart[i])
                        cycle=cycles[i]
                        if cycle==-1:
                                cycle=int(sefiles.se.cycles[-1])
                        if mass_range[i][0] ==0 and mass_range[i][1]==0:
                                mass_range[i][1]=sefiles.get(cycle,'mass')[-1]
            print 'use cycle',cycle
            sefiles.average_iso_abund_marco(mass_range=mass_range[i],cycle=cycle,stable=True,i_decay=2)         
            massfrac1=mp.average_mass_frac_decay
            other_name_scheme=u.back_ind.keys()
            isos=[]
            massfrac=[]
            print len(massfrac),len(other_name_scheme)
            for kk in range(len(other_name_scheme)):
                list1=re.split('(\d+)',other_name_scheme[kk])
                newname=list1[0].capitalize().strip()+'-'+list1[1].strip()
                #print other_name_scheme[kk],newname,massfrac[kk]
                isos.append(newname)
                massfrac.append(massfrac1[u.back_ind[other_name_scheme[kk]]])
            massfrac_all.append(massfrac)
            iso_all.append(isos)
        return iso_all,massfrac_all