from pmts import build_pmt, build_light_collector, build_light_collector_from_file, build_pmt_shell
from chroma.optics import *
from os.path import dirname

def build_12inch_pmt(outer_material=water, nsteps=16):
    return build_pmt(dirname(__file__) + '/hamamatsu_12inch.txt', 0.003,
                     outer_material, nsteps)

def build_12inch_pmt_shell(outer_material=water, nsteps=16):
    return build_pmt_shell(dirname(__file__) + '/hamamatsu_12inch.txt')

# from Jelena Maricic
lc_12inch_a = 0.16597
lc_12inch_b = 0.584525
lc_12inch_d = 0.09548
lc_12inch_rmin = 0.1524
lc_12inch_rmax = 0.235072

def build_12inch_pmt_with_lc(outer_material=water, nsteps=16):
    pmt = build_12inch_pmt(outer_material, nsteps)
    return pmt + build_light_collector(pmt, a=lc_12inch_a, b=lc_12inch_b,
                                       d=lc_12inch_d, rmin=lc_12inch_rmin,
                                       rmax=lc_12inch_rmax)

def build_12inch_pmt_with_lc_hd(outer_material=water, nsteps=128):
    pmt = build_12inch_pmt(outer_material, nsteps)
    return pmt + build_light_collector(pmt, a=lc_12inch_a, b=lc_12inch_b,
                                       d=lc_12inch_d, rmin=lc_12inch_rmin,
                                       rmax=lc_12inch_rmax, npoints=100)

def build_8inch_pmt(outer_material=water, nsteps=24):
    return build_pmt(dirname(__file__) + '/sno_pmt_reduced.txt', 0.003,
                     outer_material, nsteps)

def build_8inch_pmt_with_lc(outer_material=water, nsteps=24):
    pmt = build_8inch_pmt(outer_material, nsteps)
    lc = build_light_collector_from_file(dirname(__file__) + '/sno_cone.txt',
                                         outer_material, nsteps)
    return pmt + lc