ARICHMaterialScan.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
# --------------------------------------------------------------------
# This script generate root which contains the map of ARICH material budget
# Initial script : simulation/examples/MaterialScan.py
#
# Author: Leonid Burmistrov (May 2018)
# --------------------------------------------------------------------
 
import sys
import os
 
from basf2 import logging, LogLevel, create_path, process
from optparse import OptionParser
 
parser = OptionParser()
parser.add_option('-f', '--file', dest='filename', default='ARICHMaterialScan.root')
(options, args) = parser.parse_args()
 
home = os.environ['BELLE2_LOCAL_DIR']
 
print("output file :", end=" ")
print(home, end="")
print("/arich/examples/", end="")
print(options.filename)
 
# Don't show all the info messages
logging.log_level = LogLevel.ERROR
 
# create path
main = create_path()
 
# We need to process one event so we need the EventInfoSetter
main.add_module("EventInfoSetter", evtNumList=[1])
 
# We need the geometry parameters
main.add_module("Gearbox")
 
# Build only ARICH detector
geometry = main.add_module("Geometry", logLevel=LogLevel.INFO, assignRegions=True,
                           components=['ARICH'])
 
# MaterialScan uses the Geant4 setup which is created by the FullSim module so
# we need this as well
main.add_module('FullSim')
 
# And finally the MaterialScan module
materialscan = main.add_module("MaterialScan", logLevel=LogLevel.INFO)
 
# Create a detailed Materialbudget for the beampipe
# 1) 100x100 raster starting from IP looking at the acceptance
# 2) 200x200 raster of the ZX plane between Z=-50 to 60 cm and X=-16 to
#    16 cm, starting at Y=-16cm and scanning the next 32cm along Y
 
materialscan.param({
    # Filename for output File
    'Filename': options.filename,
    # Do Spherical scan?
    'spherical': True,
    # Origin of the spherical scan
    'spherical.origin': [0, 0, 0],
    # Number of steps in theta
    'spherical.nTheta': 100,
    # Minimal theta value
    'spherical.minTheta': 17,
    # Maximal theta value
    'spherical.maxTheta': 150,
    # If set to true, the scan will be done uniform in cos(theta)
    'spherical.cosTheta': True,
    # Number of steps in theta
    'spherical.nPhi': 100,
    # Minimal theta value
    'spherical.minPhi': 0,
    # Maximal theta value
    'spherical.maxPhi': 360,
    # Depth of the scan: 0 for scan the whole defined geometry. Any value >0
    # will stop the ray after reaching the given distance from the start point
    'spherical.maxDepth': 0,
    # Split output by Material names instead of by Region
    'spherical.splitByMaterials': False,
    # Specify the names of Materials to ignore in the scan.
    # Default is Air and Vacuum
    'spherical.ignored': ['Air', 'Vacuum', 'G4_AIR', 'ColdAir'],
    # Do Planar scan?
    'planar': True,
    # Name of the plane for scanning.  One of 'XY', 'XZ', 'YX', 'YZ', 'ZX',
    # 'ZY' or 'custom' to define the plane directly
    'planar.plane': 'custom',
    # Define a custom plane for the scan, only used if planar.plane='custom'.
    # This is a list of 9 values where the first three values specify the
    # origin, the second 3 values specify the first axis and the last 3 values
    # specify the second axis. In this case we scan the ZX plane but the origin
    # is not at the IP at y=-1 to create a symmetric cut view in |y|< 1 cm
    'planar.custom': [
        0, 0, 0,  # Origin
        1, 0, 0,  # First axis
        0, 1, 0,  # Second axis
    ],
    # Depth of the scan: 0 for scan the whole defined geometry. Any value >0
    # will stop the ray after reaching the given distance from the start point
    'planar.maxDepth': 0,
    # Number of steps along the first axis
    'planar.nU': 120,
    # 'planar.nU': 5,
    # Minimum value for the first axis
    'planar.minU': -120.0,
    # Maximum value for the first axis
    'planar.maxU': 120,
    # Number of steps along the second axis
    'planar.nV': 120,
    # 'planar.nV': 5,
    # Minimum value for the second axis
    'planar.minV': -120,
    # Maximum value for the second axis
    'planar.maxV': 120,
    # Split output by Material names insted of by Region
    'planar.splitByMaterials': True,
    # Specify the names of Materials to ignore in the scan.
    # Default is Air and Vacuum
    'planar.ignored': ['Air', 'G4_AIR', 'ColdAir'],
})
 
# Do the MaterialScan, can take some time depending on the number of steps
process(main)
 
# Make basic performance plots
com = 'root -l ' + options.filename + ' ' + home + '/arich/utility/scripts/plotARICHmaterialbudget.C'
os.system(com)