def readW7Data(dataFile=None): if dataFile==None: dataFile=os.path.join(config.getMainConfigDir(),'w7.combined.dat') data=np.loadtxt(dataFile) colHeaders=file(dataFile).readlines()[1].split() selProperties=[5,3] selElem=[9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42] colDescriptor=[] # Manually setting property colDescriptors colDescriptor.append('vph') colDescriptor.append('dens') #Automatically setting the element colDescriptors colDescriptor+=[string.upper(colHeaders[i][0])+colHeaders[i][1:] for i in selElem] dataDict=dict([(key,np.array(value)) for key,value in zip(colDescriptor,zip(*data[:,selProperties+selElem]))]) dataDict['vph']/=1e5 return dataDict
def initLumVphGrid(step=1,gridSize=10,doPickle=True): t=config.getTimeFromExplosion() vph=initialize.time2vph(t) mainConfigDir=config.getMainConfigDir() modelW7=initialize.readW7Data(os.path.join(mainConfigDir,'w7.combined.dat')) initDica=config.getVanillaDica() initDica['t']=t #Preparing the normalization initComp=config.getVanillaComp() initComp.update(initialize.getW7Comp(modelW7,t)) initComp=abund.setNiDecay(initComp,t) initComp=abund.setCONe(initComp) initComp=abund.normAbundances(initComp) initRunDir='init_lumvph_run/' initialVph=[0.6*vph,1.4*vph] return lumVphGrid(initialLum,initialVph,1,initRunDir,initComp=initComp,initDica=initDica,gridSize=gridSize,doPickle=doPickle)