Skip to content
Snippets Groups Projects
Commit 1d9c082d authored by Sergei Zharko's avatar Sergei Zharko
Browse files

mcbm2024: a script for data reconstruction steps execution

parent 704c4a7f
No related branches found
No related tags found
1 merge request!1865mcbm2024 data recostruction chain:
Pipeline #30089 failed
Stage: checkRepository
    Stage: checkFormat
      Stage: build
        Stage: package
          Stage: verify
            Stage: finalise
              Hide whitespace changes
              Inline Side-by-side
              algo/ca/core/CMakeLists.txt
              + 21
              − 21
              View file @ 1d9c082d
              ...@@ -82,8 +82,8 @@ install( ...@@ -82,8 +82,8 @@ install(
              #data/CaGridEntry.h #data/CaGridEntry.h
              #data/CaHit.h #data/CaHit.h
              #data/CaInputData.h #data/CaInputData.h
              #data/CaTrackParam.h data/CaTrackParam.h
              #data/CaTrack.h data/CaTrack.h
              #data/CaMeasurementU.h #data/CaMeasurementU.h
              #data/CaMeasurementXy.h #data/CaMeasurementXy.h
              #data/CaMeasurementTime.h #data/CaMeasurementTime.h
              ...@@ -94,38 +94,38 @@ install( ...@@ -94,38 +94,38 @@ install(
              #data/CaGridArea.h #data/CaGridArea.h
              #data/CaTriplet.h #data/CaTriplet.h
              #data/CaBranch.h #data/CaBranch.h
              #data/CaWindowData.h data/CaWindowData.h
              #data/CaTimesliceHeader.h #data/CaTimesliceHeader.h
              #pars/CaConstants.h pars/CaConstants.h
              #pars/CaField.h pars/CaField.h
              #pars/CaInitManager.h pars/CaInitManager.h
              #pars/CaIteration.h pars/CaIteration.h
              #pars/CaMaterialMap.h pars/CaMaterialMap.h
              #pars/CaMaterialMonitor.h pars/CaMaterialMonitor.h
              #pars/CaParameters.h pars/CaParameters.h
              #pars/CaSearchWindow.h pars/CaSearchWindow.h
              #pars/CaStation.h pars/CaStation.h
              #pars/CaStationInitializer.h #pars/CaStationInitializer.h
              #utils/CaTrackingMonitor.h utils/CaTrackingMonitor.h
              #utils/CaEnumArray.h utils/CaEnumArray.h
              #utils/CaMonitor.h utils/CaMonitor.h
              #utils/CaMonitorData.h utils/CaMonitorData.h
              #utils/CaObjectInitController.h #utils/CaObjectInitController.h
              #utils/CaSimd.h #utils/CaSimd.h
              #utils/CaSimdVc.h #utils/CaSimdVc.h
              #utils/CaSimdPseudo.h #utils/CaSimdPseudo.h
              #utils/CaTimer.h #utils/CaTimer.h
              #utils/CaVector.h utils/CaVector.h
              #utils/CaUtils.h utils/CaUtils.h
              #utils/CaDefines.h utils/CaDefines.h
              #tracking/CaCloneMerger.h #tracking/CaCloneMerger.h
              #tracking/CaFramework.h #tracking/CaFramework.h
              #tracking/CaTrackExtender.h #tracking/CaTrackExtender.h
              #tracking/CaTrackFinder.h #tracking/CaTrackFinder.h
              #tracking/CaTrackFinderWindow.h #tracking/CaTrackFinderWindow.h
              #tracking/CaTrackFit.h tracking/CaTrackFit.h
              #tracking/CaTrackFitter.h tracking/CaTrackFitter.h
              #tracking/CaTripletConstructor.h #tracking/CaTripletConstructor.h
              DESTINATION DESTINATION
              include/ include/
              ......
              macro/beamtime/mcbm2024/reco_mcbm.sh 0 → 100755
              + 274
              − 0
              View file @ 1d9c082d
              #!/bin/bash
              # Copyright (C) 2024 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
              # SPDX-License-Identifier: GPL-3.0-only
              # Authors: Sergei Zharko [committer]
              #
              # @file reco_mcbm.sh
              # @brief Script to run the TSA file reconstruction in mCBM
              # @since 21.05.2024
              # @author Sergei Zharko <s.zharko@gsi.de>
              #
              # **********************************************************************************************************************
              # *** User Manual (v. 0.0.1, 19.06.2024) ***
              # **********************************************************************************************************************
              #
              # 1. Definitions
              # <setup>:
              # An official setup tag, e.g. "mcbm_beam_2024_05_08_nickel".
              # <label>:
              # A common part of name bases of the data-files, e.g. "3105_node8_05_0002"
              # NOTE: <label> is strictly determined at least for a particular setup.
              # <top_dir>:
              # A top "data" directory.
              #
              # 2. Introduction
              #
              # This script aims to provide a generic data reconstruction scenario in mCBM, which includes:
              # - setup files generation: [--setup]
              # - TSA file unpacking [--unpack]
              # - reconstruction [--reco]
              # - main QA [--qa]
              # - reconstruciton QA (Alex) [--qa-module]
              #
              # Required options:
              # --tsa <path> Path to the input TSA file
              # NOTE: must contain a run index as a first integer in its base name.
              # --param-online <path> Path to the online parameters
              # NOTE: must be ${VMCWORKDIR}/parameters/online, but the current parameter repositories
              # a diverged for the detector and online binary parameters.
              # Auxiliary options:
              # -n [--nts] <N_TS> Number of timeslices to procede
              # --setup, --unpack, --reco, --qa, --qa-module
              #
              # 3. File names involved:
              # <geo>: Input geometry file: <top_dir>/<setup>.geo.root
              # <par>: Input parameter file: <top_dir>/<setup>.par.root --copy--> <top_dir>/<label>.par.root
              # <tra>: Dummy transport file: <top_dir>/<setup>.tra.root
              # <digi>: Digitization output: <top_dir>/<label>.digi.root --ln--> <top_dir>/<label>.raw.root
              # <reco>: Reconstruction output: <top_dir>/<label>.reco.root --ln--> <top_dir>/<label>.rec.root
              # <qa>: Main QA output: <top_dir>/<label>.qa.root
              # <qa-module>: Reconstruction module QA: <top_dir>/<label>.rqa.root
              #
              #
              # 4. Parameters situation (for 20.06.2024)
              # The cbmroot_parameter repository is diverged: the valid online binary parameters are stored in the main CbmSoft/
              # cbmroot_parameter repository, and the detector parameters are stored on the mCBM fork of this repository. This
              # will be changed with the merger request !1864 to the cbmroot master. For now one can checkout the main parameter
              # repository with:
              # cd /path/to/dir
              # git clone https://git.cbm.gsi.de/CbmSoft/cbmroot_parameter.git
              # cd cbmroot_parameter
              # git remote add upstream https://git.cbm.gsi.de/CbmSoft/cbmroot_parameter.git
              # git fetch --al
              # Provide a path --param-online /path/to/dir/cbmroot_parameter/online
              #
              # To configure CbmRoot with correct mcbm parameters, one must provide a parameter -DCBM_MCBM_PARS=1 in the
              # cmake configuration command.
              #
              #
              # *********************
              # ** USER PARAMETERS **
              # *********************
              # Data directory
              DATA_TOP_DIR='./data'
              # Subsystem flags
              RECO_MVD=0
              RECO_STS=1
              RECO_TRD=1
              RECO_TRD2d=1
              RECO_RICH=0
              RECO_MUCH=0
              RECO_TOF=1
              RECO_FSD=0
              RECO_PSD=0
              # Algorithm flags
              RECO_TOFtr=0
              RECO_ALI=0
              RECO_EvB=1
              RECO_CA=1
              RECO_QA=0
              # ----------------------------------------------------------------------------------------------------------------------
              # ----------------------------------------------------------------------------------------------------------------------
              # *************************
              # ** Parameter selection **
              # *************************
              # ----- Execution steps
              DO_CREATE_SETUP=0
              DO_UNPACK=0
              DO_RECO=0
              DO_QA=0 # Main QA Macro
              DO_QA_MODULE=0 # QA Macro by Alexandru
              ONLINE_PAR=
              # ----- Run information
              RUN=-1
              TSA=
              N_TS=-1
              while [[ $# > 0 ]]; do
              case ${1} in
              --setup )
              CREATE_SETUP=1
              ;;
              --unpack )
              DO_UNPACK=1
              ;;
              --reco )
              DO_RECO=1
              ;;
              --qa )
              DO_QA=1
              ;;
              --qa-module )
              DO_QA_MODULE=1
              ;;
              --tsa )
              TSA=${2}
              ;;
              --param-online )
              ONLINE_PAR=${2}
              ;;
              -n | --nts )
              N_TS=${2}
              ;;
              esac
              shift
              done
              # ----- Check the environment
              #
              if [[ -z "${VMCWORKDIR}" ]]; then
              printf "E- CBM environment is not defined (VMCWORKDIR is not found). Please, configure your CbmRoot\n"
              exit 1
              fi
              # ----- Select the reconstruction binary
              ONLINE_BINARY="${VMCWORKDIR}/../../bin/cbmreco"
              if [[ ! -x "${ONLINE_BINARY}" ]]; then
              ONLINE_BINARY=$(which cbmreco)
              if [[ ! -x "${ONLINE_BINARY}" ]]; then
              printf "E- Online binary was not found. Please, install and configure your CbmRoot "
              printf "or provide the bin directory in the PATH environmental variable\n"
              exit 2
              fi
              fi
              printf "I- Online binary path: %s, with parameters: %s\n" ${ONLINE_BINARY} ${ONLINE_PAR}
              # ----- Check the TSA input and retrieve the run index
              if [[ -z ${TSA} ]]; then
              printf "E- TSA input file is not defined\n"
              exit 3
              fi
              TSA_INFO=($(basename $(echo ${TSA}) | grep -oE '[0-9]*'))
              RUN=${TSA_INFO[0]} # Implying, that TSA basename contains run index as the first integer
              if [[ ${RUN} -lt 0 ]]; then
              printf "E- Run number is undefined, please try again with option -r <RUN> or --run <RUN>\n"
              fi
              mkdir -p ${DATA_TOP_DIR}
              # ----- Filenames definition
              TSA_INP=${TSA}
              MACRO_SETUP="${VMCWORKDIR}/macro/run/create_mcbm_geo_setup.C"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2022/mcbm_event_reco_L1.C"
              MACRO_QA="${VMCWORKDIR}/macro/mcbm/mcbm_qa.C"
              MACRO_QA_MODULE="${VMCWORKDIR}/macro/qa/run_recoQa.C"
              SETUP_NAME=
              if [[ ${RUN} -ge 2350 && ${RUN} -le 2397 ]]; then
              SETUP_NAME="mcbm_beam_2022_05_23_nickel"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2022/mcbm_event_reco_L1.C"
              elif [[ ${RUN} -ge 2454 && ${RUN} -le 2497 ]]; then
              SETUP_NAME="mcbm_beam_2022_06_16_gold"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2022/mcbm_event_reco_L1.C"
              elif [[ ${RUN} -ge 2498 && ${RUN} -le 2610 ]]; then
              SETUP_NAME="mcbm_beam_2022_06_18_gold"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2022/mcbm_event_reco_L1.C"
              elif [[ ${RUN} -ge 2724 && ${RUN} -le 2917 ]]; then
              SETUP_NAME="mcbm_beam_2024_03_22_gold"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2024/mcbm_event_reco_L1.C"
              elif [[ ${RUN} -ge 2918 ]]; then
              SETUP_NAME="mcbm_beam_2024_05_08_nickel"
              MACRO_RECO="${VMCWORKDIR}/macro/beamtime/mcbm2024/mcbm_event_reco_L1.C"
              else
              printf "E- Run ${RUN} is undefined. Exiting"
              exit 4
              fi
              # -----
              # File
              FILE_LABEL=$(printf $(basename ${TSA_INP}) | cut -f 1 -d '.')
              INP_FILE="${FILE_LABEL}"
              OUT_DIGI=
              if [[ ${RUN} -ge 2724 ]]; then
              OUT_DIGI="${DATA_TOP_DIR}/${FILE_LABEL}.digi.out"
              INP_FILE="${DATA_TOP_DIR}/${FILE_LABEL}.digi.root"
              fi
              # *********************
              # ** Steps execution **
              # *********************
              # ----- Create setup files
              if [[ ${CREATE_SETUP} -eq 1 ]]; then
              root -b -l -q ${MACRO_SETUP}"(${RUN}, \"${DATA_TOP_DIR}\")"
              fi
              # ----- Run unpacker
              if [[ ${DO_UNPACK} -eq 1 ]]; then
              # TODO: Define unpack options for different setups
              ${ONLINE_BINARY} --steps Unpack -i ${TSA_INP} -r ${RUN} -p ${ONLINE_PAR} -O DigiTimeslice -o ${OUT_DIGI} -s STS BMON TOF TRD RICH
              root -l -b -q ${VMCWORKDIR}/macro/run/run_inspect_digi_timeslice.C"(\"${OUT_DIGI}\", \"${INP_FILE}\")"
              rm ${OUT_DIGI}
              fi
              # ----- Run reconstruction
              if [[ ${DO_RECO} -eq 1 ]]; then
              UNP_FILE_ID=-1
              cp "${DATA_TOP_DIR}/${SETUP_NAME}.par.root" "${DATA_TOP_DIR}/${FILE_LABEL}.par.root"
              DATA_PREF="${DATA_TOP_DIR}"
              PARS="${RUN},${N_TS},\"${DATA_PREF}\",\"${DATA_PREF}\",${UNP_FILE_ID},${RECO_MVD},${RECO_STS},${RECO_TRD}"
              PARS="${PARS},${RECO_TRD2d},${RECO_RICH},${RECO_MUCH},${RECO_TOF},${RECO_TOFtr},${RECO_PSD},${RECO_ALI},${RECO_EvB}"
              PARS="${PARS},${RECO_CA},${RECO_QA},${RECO_FSD},\"${INP_FILE}\""
              root -b -l -q ${MACRO_RECO}"(${PARS})"
              ln -s "${FILE_LABEL}.digi.root" "${DATA_TOP_DIR}/${FILE_LABEL}.raw.root" # TMP for QA
              ln -s "${FILE_LABEL}.reco.root" "${DATA_TOP_DIR}/${FILE_LABEL}.rec.root" # TMP for QA
              fi
              # ----- Run QA of reco modules
              if [[ ${DO_QA_MODULE} -eq 1 ]]; then
              QA_GEO="${DATA_TOP_DIR}/${SETUP_NAME}.geo.root"
              QA_REC="${DATA_TOP_DIR}/${FILE_LABEL}.rec.root"
              ln -s ${QA_GEO} "${SETUP_NAME}.geo.root"
              PARS="-1,\"${QA_REC}\",\"${SETUP_NAME}\",kFALSE"
              root -b -l -q ${MACRO_QA_ALEX}"(${PARS})"
              fi
              # ----- Run QA
              if [[ ${DO_QA} -eq 1 ]]; then
              USE_MC="kFALSE"
              CONFIG=""
              BENCHMARK=""
              #PARS="1,\"${DATA_TOP_DIR}/${FILE_LABEL}\",\"${SETUP_NAME}\",${USE_MC},\"${CONFIG}\",\"${BENCHMARK}\""
              QA_RAW="${DATA_TOP_DIR}/${FILE_LABEL}.raw.root"
              QA_REC="${DATA_TOP_DIR}/${FILE_LABEL}.rec.root"
              QA_PAR="${DATA_TOP_DIR}/${FILE_LABEL}.par.root"
              QA_GEO="${DATA_TOP_DIR}/${SETUP_NAME}.geo.root"
              QA_OUT="${DATA_TOP_DIR}/${FILE_LABEL}.qa.root"
              PARS="0,\"\",\"${QA_RAW}\",\"${QA_REC}\",\"${QA_PAR}\",\"${QA_GEO}\",\"${QA_OUT}\",\"${SETUP_NAME}\""
              PARS="${PARS},${USE_MC},\"${CONFIG}\",\"${BENCHMARK}\""
              root -b -l -q ${MACRO_QA}"(${PARS})"
              fi
              printf "Reconstruction of ${TSA_INP} succeeded\n"
              0% or .
              You are about to add 0 people to the discussion. Proceed with caution.
              Finish editing this message first!
              Please register or to comment