catalog loading routine now also includes edge flag and particle adjancy info

This commit is contained in:
Paul M. Sutter 2024-06-07 22:02:33 +02:00
parent 1a678eda5d
commit ad752f21e8

View file

@ -435,12 +435,24 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True,
y = partData[i][1], y = partData[i][1],
z = partData[i][2], z = partData[i][2],
volume = 0, volume = 0,
adjs = 0,
ra = extraData[i][0], ra = extraData[i][0],
dec = extraData[i][1], dec = extraData[i][1],
redshift = extraData[i][2], redshift = extraData[i][2],
uniqueID = extraData[i][3])) uniqueID = extraData[i][3],
edgeFlag = 0))
if isObservationData:
print(" Loading edge flags...")
edgeFlagFile = sampleDir+"/galaxy_edge_flags.txt"
if os.path.isfile(edgeFlagFile):
edgeFlags = np.genfromtxt(edgeFlagFile, dtype=np.int32, skip_header=1)
for iEdge in range(len(edgeFlags)):
catalog.part[iEdge].edgeFlag = edgeFlags[iEdge]
else:
print(" Edge file not found!")
print(" Loading volumes...") print(" Loading volumes...")
if sample.hasWeightedVolumes: if sample.hasWeightedVolumes:
volFile = sampleDir+"/vol_weighted_"+sample.fullName+".dat" volFile = sampleDir+"/vol_weighted_"+sample.fullName+".dat"
@ -452,6 +464,14 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True,
for ivol in range(len(vols)): for ivol in range(len(vols)):
catalog.part[ivol].volume = vols[ivol] / volNorm catalog.part[ivol].volume = vols[ivol] / volNorm
print(" Loading adjacencies...")
adjFile = sampleDir+"adj_"+sample.fullName+".dat"
with open(adjFile, mode="rb") as File:
numPart = np.fromfile(adjFile, dtype=np.int32,count=1)[0]
for p in range(numPart):
numAdjs = np.fromfile(File, dtype=np.int32,count=1)[0]
catalog.part[p].adjs = np.fromfile(File, dtype=np.int32, count=numAdjs)
print(" Loading zone-void membership info...") print(" Loading zone-void membership info...")
zoneFile = sampleDir+"/voidZone_"+sample.fullName+".dat" zoneFile = sampleDir+"/voidZone_"+sample.fullName+".dat"
catalog.void2Zones = [] catalog.void2Zones = []