# and set up the two color disentangler
simsAB = sim_utils.sim_twocolors2(
cryst_model,
detector,
iset.get_beam(0), [5000, None],
[parameters.ENERGY_LOW, parameters.ENERGY_HIGH], [1e14, 1e14],
pids=None,
Gauss=False,
oversample=4,
Ncells_abc=(20, 20, 20),
mos_dom=20,
mos_spread=0.0) # returns a dict of {0: 64panelsim, 1: 64panelsim }
if n_idx % save_sim == 0:
sim_fname = os.path.join(out_dir, "sim64_%d.h5" % idx)
sim_utils.save_twocolor(simsAB, iset, sim_fname, force=0)
simsAB_2 = sim_utils.sim_twocolors2(
crystalAB,
detector,
iset.get_beam(0), [5000, None],
[parameters.ENERGY_LOW, parameters.ENERGY_HIGH], [1e14, 1e14],
pids=None,
Gauss=False,
oversample=4,
Ncells_abc=(20, 20, 20),
mos_dom=20,
mos_spread=0.0) # returns a dict of {0: 64panelsim, 1: 64panelsim }
# Now, few things to do here:
# The above simulation object has 128 images
beamB = D['beamB']
det = D['detector']
cryst = D["optCrystal"]
simsAB_old = sim_utils.sim_twocolors2(
cryst,
det,
beamA,
FF, [parameters.ENERGY_LOW, parameters.ENERGY_HIGH],
FLUX,
pids=None,
Gauss=False,
oversample=2,
Ncells_abc=(7, 7, 7),
mos_dom=20,
mos_spread=0.0)
sim_utils.save_twocolor(simsAB_old, iset,
"run%d_shot%d.hdf5" % (run, shot_idx))
raws = iset.get_raw_data(0)
img = raws[62].as_numpy_array()
get_ipython().magic(u'pylab')
imshow(img, vmax=20)
imshow(img, vmax=200)
x = 145.537 - .5
y = 119.91 - .5
plot(x, y, 'o', ms=16, mec='r', mfc='none')
fig = figure()
ax = fig.gca(projection='3d')
from mpl_toolkits.mplot3d import Axes3D
ax = fig.gca(projection='3d')
Y, X = np.indices(img.shape)
Z = img
surf = ax.plot_surface(X,