Ported python code to netCDF4 package (support most recent netcdf4 on-disk fornat)

This commit is contained in:
Guilhem Lavaux 2012-10-31 15:00:43 -04:00
parent e380538e66
commit d18fbdf10b
2 changed files with 18 additions and 14 deletions

View file

@ -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()

View file

@ -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()