guilhem_lavaux/vide_public
0
0
Fork 0
mirror of https://bitbucket.org/cosmicvoids/vide_public.git synced 2025-07-04 23:31:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make particle loading more flexible

Browse source
This commit is contained in:
P.M. Sutter 2014-01-31 10:50:32 -06:00
parent 303650fa4b
commit eb7446e4f5
1 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
python_tools/void_python_tools/partUtil/partUtil.py
View file

@ -25,16 +25,20 @@ from netCDF4 import Dataset
import sys import sys
from void_python_tools.backend import * from void_python_tools.backend import *
import void_python_tools.apTools as vp import void_python_tools.apTools as vp
import pickle
NetCDFFile = Dataset NetCDFFile = Dataset
ncFloat = 'f8' ncFloat = 'f8'
# ----------------------------------------------------------------------------- # -----------------------------------------------------------------------------
def loadPart(workDir, sampleDir, sample): def loadPart(sampleDir):
#print " Loading particle data..." #print " Loading particle data..."
sys.stdout.flush() sys.stdout.flush()
infoFile = workDir+"/"+sampleDir+"/zobov_slice_"+sample.fullName+".par" with open(sampleDir+"/sample_info.dat", 'rb') as input:
sample = pickle.load(input)
infoFile = sampleDir+"/zobov_slice_"+sample.fullName+".par"
File = NetCDFFile(infoFile, 'r') File = NetCDFFile(infoFile, 'r')
ranges = np.zeros((3,2)) ranges = np.zeros((3,2))
ranges[0][0] = getattr(File, 'range_x_min') ranges[0][0] = getattr(File, 'range_x_min')
@ -49,7 +53,7 @@ def loadPart(workDir, sampleDir, sample):
mul = np.zeros((3)) mul = np.zeros((3))
mul[:] = ranges[:,1] - ranges[:,0] mul[:] = ranges[:,1] - ranges[:,0]
partFile = workDir+"/"+sampleDir+"/zobov_slice_"+sample.fullName partFile = sampleDir+"/zobov_slice_"+sample.fullName
iLine = 0 iLine = 0
partData = [] partData = []
part = np.zeros((3)) part = np.zeros((3))
@ -106,11 +110,14 @@ def loadPart(workDir, sampleDir, sample):
return partData, boxLen, volNorm, isObservationData return partData, boxLen, volNorm, isObservationData
# ----------------------------------------------------------------------------- # -----------------------------------------------------------------------------
def loadPartVel(workDir, sampleDir, sample): def loadPartVel(sampleDir):
#print " Loading particle velocities..." #print " Loading particle velocities..."
sys.stdout.flush() sys.stdout.flush()
infoFile = workDir+"/"+sampleDir+"/zobov_slice_"+sample.fullName+".par" with open(sampleDir+"/sample_info.dat", 'rb') as input:
sample = pickle.load(input)
infoFile = sampleDir+"/zobov_slice_"+sample.fullName+".par"
File = NetCDFFile(infoFile, 'r') File = NetCDFFile(infoFile, 'r')
isObservation = getattr(File, 'is_observation') isObservation = getattr(File, 'is_observation')