Add meshes from CVCS_TG collective simulations
This commit is contained in:
27
plot_mayavi.py
Normal file
27
plot_mayavi.py
Normal file
@@ -0,0 +1,27 @@
|
||||
#!/usr/bin/env python
|
||||
#-*- coding: utf-8 -*-
|
||||
"""
|
||||
Script to generate mayavi model. Run in IPython console (not qtconsole).
|
||||
"""
|
||||
|
||||
from mayavi.tools import pipeline
|
||||
from plot_tally import *
|
||||
import numpy as np
|
||||
|
||||
# build geometry from CAD files (removed from repository)
|
||||
#mod=pipeline.open('cad_models/target_hall.stl')
|
||||
#pipeline.surface(mod, color=(0.8,0.8,0.8))
|
||||
#mod=pipeline.open('cad_models/shielding_wall.stl')
|
||||
#pipeline.surface(mod, color=(0.4,0.0,0.4))
|
||||
#mod=pipeline.open('cad_models/shielding_outside.stl')
|
||||
#pipeline.surface(mod, color=(0.0,0.6,0.0))
|
||||
|
||||
def load_data(fname, tally=None):
|
||||
x,y,z,I,dI=read_tally(fname, use_tally=tally)
|
||||
Y,X,Z=meshgrid((y[1:]+y[:-1])/0.2-21., (x[1:]+x[:-1])/0.2+151., (z[1:]+z[:-1])/0.2)
|
||||
Xr=X*cos(36.*pi/180.)+Y*sin(36.*pi/180.)
|
||||
Yr=-X*sin(36.*pi/180.)+Y*cos(36.*pi/180.)
|
||||
scal=pipeline.scalar_scatter(Xr, Yr, Z, log10(I+1e-9))
|
||||
scal=pipeline.delaunay3d(scal)
|
||||
pipeline.iso_surface(scal, colormap='Vega20c', vmin=-8, vmax=2,
|
||||
contours=[-6.01])
|
||||
Reference in New Issue
Block a user