guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for multiple attributes in CTPV

Browse Source
This commit is contained in:
Guilhem Lavaux 2020-03-06 08:23:55 +01:00
parent cf90393c36
commit bc2e126075
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/cosmotool/ctpv.py
View File

@ -87,7 +87,7 @@ class ProgrammableParticleHistoryLoad(object):
class ProgrammableDensityLoad(object): class ProgrammableDensityLoad(object):
@staticmethod @staticmethod
def main_script(source, density, extents=None, aname="default"): def main_script(source, density, extents=None, aname="default", extra_arrays={}):
import vtk import vtk
from vtk.util import numpy_support from vtk.util import numpy_support
@ -117,6 +117,11 @@ class ProgrammableDensityLoad(object):
arr.SetName(aname) arr.SetName(aname)
ido.GetPointData().AddArray(arr) ido.GetPointData().AddArray(arr)
for k in extra_arrays.keys():
arr = numpy_support.numpy_to_vtk(extra_arrays[k].transpose().astype(np.float64).ravel(), deep=1)
arr.SetName(k)
ido.GetPointData().AddArray(arr)
@staticmethod @staticmethod
def request_information(source, density=None, dims=None): def request_information(source, density=None, dims=None):
import vtk import vtk