diff options
Diffstat (limited to 'solids/pmts.py')
-rw-r--r-- | solids/pmts.py | 87 |
1 files changed, 0 insertions, 87 deletions
diff --git a/solids/pmts.py b/solids/pmts.py deleted file mode 100644 index 43aed3f..0000000 --- a/solids/pmts.py +++ /dev/null @@ -1,87 +0,0 @@ -import numpy as np -from chroma.geometry import Solid -from chroma.make import rotate_extrude -from chroma.optics import * -from chroma.tools import read_csv, offset - -def get_lc_profile(radii, a, b, d, rmin, rmax): - c = -b*np.sqrt(1 - (rmin-d)**2/a**2) - return -c - b*np.sqrt(1-(radii-d)**2/a**2) - -def build_light_collector(pmt, a, b, d, rmin, rmax, npoints=10): - if not isinstance(pmt, Solid): - raise Exception('`pmt` must be an instance of %s' % Solid) - - lc_radii = np.linspace(rmin, rmax, npoints) - lc_profile = get_lc_profile(lc_radii, a, b, d, rmin, rmax) - - pmt_face_profile = pmt.profile[pmt.profile[:,1] > -1e-3] - - lc_offset = np.interp(lc_radii[0], list(reversed(pmt_face_profile[:,0])), list(reversed(pmt_face_profile[:,1]))) - - lc_mesh = rotate_extrude(lc_radii, lc_profile + lc_offset, pmt.nsteps) - - return Solid(lc_mesh, pmt.outer_material, pmt.outer_material, surface=shiny_surface) - -def build_pmt_shell(filename, outer_material=water, nsteps=16): - profile = read_csv(filename) - - # slice profile in half - profile = profile[profile[:,0] < 0] - profile[:,0] = -profile[:,0] - # order profile from base to face - profile = profile[np.argsort(profile[:,1])] - # set x coordinate to 0.0 for first and last profile along the profile - # so that the mesh is closed - profile[0,0] = 0.0 - profile[-1,0] = 0.0 - # convert mm -> m - profile /= 1000.0 - - return Solid(rotate_extrude(profile[:,0], profile[:,1], nsteps), glass, outer_material, color=0xeeffffff) - -def build_pmt(filename, glass_thickness, outer_material=water, nsteps=16): - profile = read_csv(filename) - - # slice profile in half - profile = profile[profile[:,0] < 0] - profile[:,0] = -profile[:,0] - # order profile from base to face - profile = profile[np.argsort(profile[:,1])] - # set x coordinate to 0.0 for first and last profile along the profile - # so that the mesh is closed - profile[0,0] = 0.0 - profile[-1,0] = 0.0 - # convert mm -> m - profile /= 1000.0 - - offset_profile = offset(profile, -glass_thickness) - - outer_envelope_mesh = rotate_extrude(profile[:,0], profile[:,1], nsteps) - inner_envelope_mesh = rotate_extrude(offset_profile[:,0], offset_profile[:,1], nsteps) - - outer_envelope = Solid(outer_envelope_mesh, glass, outer_material) - - photocathode = np.mean(inner_envelope_mesh.assemble(), axis=1)[:,1] > 0 - - inner_envelope = Solid(inner_envelope_mesh, vacuum, glass, surface=np.where(photocathode, r7081hqe_photocathode, shiny_surface), color=np.where(photocathode, 0xff00, 0xff0000)) - - pmt = outer_envelope + inner_envelope - - # profile points, outer_material, and theta are used to construct the - # light collector - pmt.profile = profile - pmt.outer_material = outer_material - pmt.nsteps = nsteps - - return pmt - -def build_light_collector_from_file(filename, outer_material, nsteps=48): - profile = read_csv(filename) - - # Convert mm to m - profile /= 1000.0 - - mesh = rotate_extrude(profile[:,0], profile[:,1], nsteps) - solid = Solid(mesh, outer_material, outer_material, surface=shiny_surface) - return solid |