Prep_revan.py

# Author: Donggeun Tak (takdg123@gmail.com)
# Date: April 3rd, 2020

# Before using this code, check geofile, Log_E, and angles parameters.
# This code will generate runRevan.sh file.

import numpy as np
import math
import argparse
import os

Log_E=[2.2,2.5,2.7,3,3.2,3.5,3.7,4,4.2,4.5,4.7,5,5.2,5.5,5.7,6,6.2,6.5,6.7]
#angles  =[0,25.8,36.9,45.6,53.1,60]
angles = [0, 36.9]
OneBeam= 'FarFieldPointSource'
geometry_file='AmegoBase.geo.setup'


#gives the cosTheta array
def ang2cos(allAng):
    ang =[]
    for i in allAng:
        a= round(np.cos(math.radians(i)),1)
        ang.append(a) 
    return ang

#in keV [316,501,1000,1585, ... ]
def logE2ene(allEne):
    ene =[]
    for ee in allEne:
        a=int(10**ee)
        ene.append(a) 
    return ene

energies=logE2ene(Log_E)
cos_ang =ang2cos(angles)


def create_file(source_file):
    with open('./{}.revan.cfg'.format(source_file), mode='w') as cfg:
        with open('./{}'.format(base_file)) as base:
            for line in base.readlines():
                if line.find('<GeometryFileName>') != -1:
                    cfg.write('<GeometryFileName>{}/{}</GeometryFileName>\n'.format(geometry_dir,geometry_file))
                elif line.find('<DataFileName>') != -1:
                    cfg.write('<DataFileName>{}/{}.inc1.id1.sim</DataFileName>\n'.format(source_dir, source_file))
                else:
                    cfg.write(line)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("g",  help="Geometry file dir")
    parser.add_argument("-f", default=os.getcwd(), help="Source file dir (Default: current directory)")
    parser.add_argument("-b", default='revan_AMEGO_X.cfg', help="Base file")
    args = parser.parse_args()

    source_dir= args.f
    geometry_dir= args.g
    base_file= args.b

    args = parser.parse_args()
    with open("./runRevan.sh", mode='w') as f:
        for myene in energies:
            for cosTh,ang in zip(cos_ang,angles):
                source_file='%s_%.3fMeV_Cos%.1f'%(OneBeam,myene/1000.,cosTh)
                create_file(source_file)
                f.write("revan -a -n -f {}.inc1.id1.sim -g {}/{} -c {}.revan.cfg\n".format(source_file, geometry_dir,geometry_file, source_file))