mirror of
https://bitbucket.org/cosmicvoids/vide_public.git
synced 2025-07-04 23:31:12 +00:00
Ported python code to netCDF4 package (support most recent netcdf4 on-disk fornat)
This commit is contained in:
parent
e380538e66
commit
d18fbdf10b
2 changed files with 18 additions and 14 deletions
|
@ -13,8 +13,10 @@ import glob
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
from pylab import figure
|
from pylab import figure
|
||||||
from Scientific.IO.NetCDF import NetCDFFile
|
from netCDF4 import Dataset
|
||||||
import Scientific.N as Num
|
|
||||||
|
NetCDFFile = Dataset
|
||||||
|
ncFloat='f8' # Double precision
|
||||||
|
|
||||||
class Stack:
|
class Stack:
|
||||||
zMin = 0.0
|
zMin = 0.0
|
||||||
|
@ -760,7 +762,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posx.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posx.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -773,7 +775,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posy.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posy.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -786,7 +788,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posz.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posz.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -799,7 +801,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/redshifts.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/redshifts.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -812,7 +814,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/indexes.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/indexes.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
|
|
@ -14,11 +14,13 @@ import glob
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
from pylab import figure
|
from pylab import figure
|
||||||
from Scientific.IO.NetCDF import NetCDFFile
|
from netCDF4 import Dataset
|
||||||
import Scientific.N as Num
|
|
||||||
from void_python_tools.backend.classes import *
|
from void_python_tools.backend.classes import *
|
||||||
import void_python_tools.apTools as vp
|
import void_python_tools.apTools as vp
|
||||||
|
|
||||||
|
NetCDFFile = Dataset
|
||||||
|
ncFloat = 'f8' # Double precision
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
def launchGenerate(sample, binPath, workDir=None, inputDataDir=None,
|
def launchGenerate(sample, binPath, workDir=None, inputDataDir=None,
|
||||||
zobovDir=None, figDir=None, logFile=None, useLCDM=False,
|
zobovDir=None, figDir=None, logFile=None, useLCDM=False,
|
||||||
|
@ -626,7 +628,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posx.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posx.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -639,7 +641,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posy.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posy.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -652,7 +654,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/posz.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/posz.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -665,7 +667,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/redshifts.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/redshifts.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
@ -678,7 +680,7 @@ def launchCombine(sample, stack, voidDir=None, logFile=None,
|
||||||
dataTemp = np.append(dataTemp, dataTemp2)
|
dataTemp = np.append(dataTemp, dataTemp2)
|
||||||
outFile = NetCDFFile(voidDir+"/indexes.nc", mode='w')
|
outFile = NetCDFFile(voidDir+"/indexes.nc", mode='w')
|
||||||
outFile.createDimension("dim", len(dataTemp))
|
outFile.createDimension("dim", len(dataTemp))
|
||||||
v = outFile.createVariable("array", Num.Float, ("dim",))
|
v = outFile.createVariable("array", ncFloat, ("dim",))
|
||||||
v[:] = dataTemp
|
v[:] = dataTemp
|
||||||
outFile.close()
|
outFile.close()
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue