Example #1
0
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
Example #2
0
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)