mirror of
https://bitbucket.org/cosmicvoids/vide_public.git
synced 2025-07-04 23:31:12 +00:00
added option to load trimmed or untrimmed catalog
This commit is contained in:
parent
c59b393ed3
commit
9b5961aca4
1 changed files with 14 additions and 5 deletions
|
@ -273,11 +273,15 @@ class Catalog:
|
||||||
sampleInfo = None
|
sampleInfo = None
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
|
def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True,
|
||||||
|
untrimmed=False):
|
||||||
|
|
||||||
# loads a void catalog
|
# loads a void catalog
|
||||||
|
# by default, loads parent-level voids with central densities greater than 0.2*mean
|
||||||
# sampleDir: path to VIDE output directory
|
# sampleDir: path to VIDE output directory
|
||||||
# dataPortion: "central" or "all"
|
# dataPortion: "central" or "all"
|
||||||
# loadParticles: if True, also load particle information
|
# loadParticles: if True, also load particle information
|
||||||
|
# untrimmed: if True, catalog contains all voids, regardless of density or hierarchy level
|
||||||
|
|
||||||
sys.stdout.flush()
|
sys.stdout.flush()
|
||||||
|
|
||||||
|
@ -304,8 +308,13 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
|
||||||
volNorm = getVolNorm(sampleDir)
|
volNorm = getVolNorm(sampleDir)
|
||||||
catalog.volNorm = volNorm
|
catalog.volNorm = volNorm
|
||||||
|
|
||||||
|
if unfiltered:
|
||||||
|
prefix = "untrimmed_"
|
||||||
|
else:
|
||||||
|
prefix = ""
|
||||||
|
|
||||||
print "Loading voids..."
|
print "Loading voids..."
|
||||||
fileName = sampleDir+"/untrimmed_voidDesc_"+dataPortion+"_"+sample.fullName+".out"
|
fileName = sampleDir+"/"+prefix+"voidDesc_"+dataPortion+"_"+sample.fullName+".out"
|
||||||
catData = np.loadtxt(fileName, comments="#", skiprows=2)
|
catData = np.loadtxt(fileName, comments="#", skiprows=2)
|
||||||
catalog.voids = []
|
catalog.voids = []
|
||||||
for line in catData:
|
for line in catData:
|
||||||
|
@ -334,7 +343,7 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
|
||||||
|
|
||||||
print "Loading barycenters..."
|
print "Loading barycenters..."
|
||||||
iLine = 0
|
iLine = 0
|
||||||
for line in open(sampleDir+"/untrimmed_barycenters_"+dataPortion+"_"+sample.fullName+".out"):
|
for line in open(sampleDir+"/"+prefix+"barycenters_"+dataPortion+"_"+sample.fullName+".out"):
|
||||||
line = line.split()
|
line = line.split()
|
||||||
catalog.voids[iLine].barycenter[0] = float(line[1])
|
catalog.voids[iLine].barycenter[0] = float(line[1])
|
||||||
catalog.voids[iLine].barycenter[1] = float(line[2])
|
catalog.voids[iLine].barycenter[1] = float(line[2])
|
||||||
|
@ -342,7 +351,7 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
|
||||||
iLine += 1
|
iLine += 1
|
||||||
|
|
||||||
print "Loading derived void information..."
|
print "Loading derived void information..."
|
||||||
fileName = sampleDir+"/untrimmed_centers_"+dataPortion+"_"+sample.fullName+".out"
|
fileName = sampleDir+"/"+prefix+"centers_"+dataPortion+"_"+sample.fullName+".out"
|
||||||
catData = np.loadtxt(fileName, comments="#")
|
catData = np.loadtxt(fileName, comments="#")
|
||||||
for (iLine,line) in enumerate(catData):
|
for (iLine,line) in enumerate(catData):
|
||||||
catalog.voids[iLine].volume = float(line[6])
|
catalog.voids[iLine].volume = float(line[6])
|
||||||
|
@ -353,7 +362,7 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
|
||||||
catalog.voids[iLine].centralDen = float(line[13])
|
catalog.voids[iLine].centralDen = float(line[13])
|
||||||
iLine += 1
|
iLine += 1
|
||||||
|
|
||||||
fileName = sampleDir+"/untrimmed_shapes_"+dataPortion+"_"+sample.fullName+".out"
|
fileName = sampleDir+"/"+prefix+"shapes_"+dataPortion+"_"+sample.fullName+".out"
|
||||||
catData = np.loadtxt(fileName, comments="#")
|
catData = np.loadtxt(fileName, comments="#")
|
||||||
for (iLine,line) in enumerate(catData):
|
for (iLine,line) in enumerate(catData):
|
||||||
catalog.voids[iLine].eigenVals[0] = float(line[1])
|
catalog.voids[iLine].eigenVals[0] = float(line[1])
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue