Module `sipmarray.scripts.script_sipmunit`

Expand source code

#!/usr/bin/env python3

import argparse

import matplotlib.pyplot as plt

from sipmarray import SiPMunit


def main():
    model, plot, output, verbose = parse_args()

    if verbose: print('Building SiPM array.')

    sipm_unit = SiPMunit(model = model)
    
    if verbose: sipm_unit.print_properties()
    if plot: make_plot(model, sipm_unit)
    if output: raise NotImplementedError(
            'Output to file not implemented yet for single model.')
    if verbose: print('Done!')

def make_plot(model, sipm_unit):
    fig, ax = plt.subplots(1,1)
    fig, ax = sipm_unit.plot_model(figax = (fig, ax))
    fig.savefig(f'sipmunit_{model}.png')

def parse_args():
    parser = argparse.ArgumentParser(
        description=('Script to quickly get a SiPM unit plot or geometry.'))

    parser.add_argument('-m', '--model',
                        help='Define the SiPM model.',
                        default= '6x6',
                        type = str,
                        required=True)
    parser.add_argument('-p', '--plot',
                        help='Plot the SiPM unit.',
                        type= bool,
                        default= False,
                        required=False)
    parser.add_argument('-o', '--output',
                        help ='Output the SiPM unit geometry to a file.',
                        type = bool,
                        default = True,
                        required=False)
    parser.add_argument('-v', '--verbose',
                        help ='Print the SiPM unit properties in the terminal.',
                        type = bool,
                        default = False,
                        required=False)
    args = parser.parse_args()

    model = args.model
    plot = args.plot
    output = args.output
    verbose = args.verbose
    return model,plot,output,verbose

if __name__ == '__main__':
    main()

Functions

def main()

Expand source code

def main():
    model, plot, output, verbose = parse_args()

    if verbose: print('Building SiPM array.')

    sipm_unit = SiPMunit(model = model)
    
    if verbose: sipm_unit.print_properties()
    if plot: make_plot(model, sipm_unit)
    if output: raise NotImplementedError(
            'Output to file not implemented yet for single model.')
    if verbose: print('Done!')

def make_plot(model, sipm_unit)

Expand source code

def make_plot(model, sipm_unit):
    fig, ax = plt.subplots(1,1)
    fig, ax = sipm_unit.plot_model(figax = (fig, ax))
    fig.savefig(f'sipmunit_{model}.png')

def parse_args()

Expand source code

def parse_args():
    parser = argparse.ArgumentParser(
        description=('Script to quickly get a SiPM unit plot or geometry.'))

    parser.add_argument('-m', '--model',
                        help='Define the SiPM model.',
                        default= '6x6',
                        type = str,
                        required=True)
    parser.add_argument('-p', '--plot',
                        help='Plot the SiPM unit.',
                        type= bool,
                        default= False,
                        required=False)
    parser.add_argument('-o', '--output',
                        help ='Output the SiPM unit geometry to a file.',
                        type = bool,
                        default = True,
                        required=False)
    parser.add_argument('-v', '--verbose',
                        help ='Print the SiPM unit properties in the terminal.',
                        type = bool,
                        default = False,
                        required=False)
    args = parser.parse_args()

    model = args.model
    plot = args.plot
    output = args.output
    verbose = args.verbose
    return model,plot,output,verbose