GPU Standalone: Add recreateTrivialCalibObjects option

Author: davidrohr

Detectors/TPC/reconstruction/include/TPCReconstruction/TPCFastTransformHelperO2.h

@@ -69,7 +69,6 @@ class TPCFastTransformHelperO2
   {
     return updateCalibrationImpl(fastTransform, TimeStamp, vDriftFactor, vDriftRef, driftTimeOffset);
   }
-
   /// _______________  Utilities   ________________________
 
   const TPCFastTransformGeo& getGeometry() { return mGeo; }

GPU/GPUTracking/Definitions/GPUSettingsList.h

@@ -638,6 +638,7 @@ AddOption(runCompression, int32_t, 1, "", 0, "Enable TPC Compression")
 AddOption(runTransformation, int32_t, 1, "", 0, "Enable TPC Transformation")
 AddOption(runRefit, bool, false, "", 0, "Enable final track refit")
 AddOption(setO2Settings, bool, false, "", 0, "Set O2 defaults for output of shared cluster map, referenceX")
+AddOption(recreateTrivialCalibObjects, bool, false, "", 0, "Recreate trivial calibration objects (TPCTransform, dEdxCorrection) from scratch")
 AddHelp("help", 'h')
 AddHelpAll("helpall", 'H')
 AddSubConfig(GPUSettingsRec, rec)

GPU/GPUTracking/Standalone/Benchmark/CMakeLists.txt

@@ -20,14 +20,15 @@ if(ALIGPU_BUILD_TYPE STREQUAL "O2")
     o2_add_executable(standalone-benchmark
                       COMPONENT_NAME gpu
                       TARGETVARNAME targetName
-                      PUBLIC_LINK_LIBRARIES O2::GPUO2Interface O2::GPUTracking
+                      PUBLIC_LINK_LIBRARIES O2::GPUO2Interface O2::GPUTracking O2::TPCReconstruction
                       SOURCES ${SRCS})
 endif()
 
 if(ALIGPU_BUILD_TYPE STREQUAL "Standalone")
     add_executable(ca ${SRCS})
     set(targetName ca)
-    target_link_libraries(${targetName} PUBLIC GPUTracking)
+    target_link_libraries(${targetName} PUBLIC GPUTracking standalone_support2)
+    install(TARGETS ca)
 endif()
 
 install(DIRECTORY ../tools DESTINATION ${CMAKE_INSTALL_DATAROOTDIR}/GPU)

GPU/GPUTracking/Standalone/Benchmark/standalone.cxx

@@ -24,14 +24,20 @@
 #include "display/GPUDisplayInterface.h"
 #include "genEvents.h"
 
-#include "TPCFastTransformPOD.h"
 #include "GPUTPCGMMergedTrack.h"
 #include "GPUSettings.h"
 #include "GPUConstantMem.h"
 
 #include "GPUO2DataTypes.h"
 #include "GPUChainITS.h"
 
+// For creating default objects
+#include "TPCReconstruction/TPCFastTransformHelperO2.h"
+#include "CalibdEdxContainer.h"
+#include "TPCFastTransformPOD.h"
+#include "GPUTRDRecoParam.h"
+#include "TPCZSLinkMapping.h"
+
 #include "DataFormatsTPC/CompressedClusters.h"
 
 #include <iostream>
@@ -286,6 +292,31 @@ int32_t ReadConfiguration(int argc, char** argv)
   return (0);
 }
 
+void CreateTrivialCalibObjects()
+{
+  GPUCalibObjectsConst calib;
+
+  aligned_unique_buffer_ptr<TPCFastTransformPOD> tmpFastTransformBuffer;
+  TPCFastTransformPOD::create(tmpFastTransformBuffer, *o2::tpc::TPCFastTransformHelperO2::instance()->create(0));
+  calib.fastTransform = tmpFastTransformBuffer.get();
+  auto tmpTRDGeometry = std::make_unique<o2::trd::GeometryFlat>();
+  calib.trdGeometry = tmpTRDGeometry.get();
+  auto tmpTRDRecoParam = std::make_unique<GPUTRDRecoParam>();
+  calib.trdRecoParam = tmpTRDRecoParam.get();
+  auto tmpdEdxCalibContainer = std::make_unique<o2::tpc::CalibdEdxContainer>();
+  tmpdEdxCalibContainer->setDefaultZeroSupresssionThreshold();
+  tmpdEdxCalibContainer->setDefaultPolTopologyCorrection();
+  calib.dEdxCalibContainer = tmpdEdxCalibContainer.get();
+  auto tmpTPCPadGainCalib = std::make_unique<TPCPadGainCalib>();
+  calib.tpcPadGain = tmpTPCPadGainCalib.get();
+  auto tmpTPCZSLinkMapping = std::make_unique<TPCZSLinkMapping>();
+  calib.tpcZSLinkMapping = tmpTPCZSLinkMapping.get();
+
+  chainTracking->SetCalibObjects(calib);
+  rec->DumpSettings("./");
+  printf("Wrote trivial calibration objects to current folder\n");
+}
+
 int32_t SetupReconstruction()
 {
   if (!configStandalone.eventGenerator) {
@@ -770,6 +801,11 @@ int32_t main(int argc, char** argv)
     }
   }
 
+  if (configStandalone.recreateTrivialCalibObjects) {
+    CreateTrivialCalibObjects();
+    return 0;
+  }
+
   if (SetupReconstruction()) {
     return 1;
   }

GPU/GPUTracking/Standalone/CMakeLists.txt

@@ -159,6 +159,9 @@ add_library(standalone_support SHARED
             ${O2_DIR}/Detectors/Base/src/Ray.cxx
             ${O2_DIR}/Detectors/Base/src/Propagator.cxx)
 
+add_library(standalone_support2 SHARED
+            ${O2_DIR}/Detectors/TPC/reconstruction/src/TPCFastTransformHelperO2.cxx)
+
 target_compile_definitions(standalone_support PUBLIC $<TARGET_PROPERTY:O2::GPUTracking,COMPILE_DEFINITIONS>)
 target_include_directories(standalone_support PUBLIC $<TARGET_PROPERTY:O2::GPUTracking,INCLUDE_DIRECTORIES>)
 
@@ -196,15 +199,20 @@ target_include_directories(standalone_support PUBLIC
                            ${O2_DIR}/Detectors/TRD/base/include
                            ${O2_DIR}/Detectors/TRD/base/src
                            ${O2_DIR}/Framework/Foundation/3rdparty/include)
+target_include_directories(standalone_support2 PUBLIC
+                           ${O2_DIR}/Detectors/TPC/reconstruction/include)
+
 
-target_link_libraries(standalone_support PUBLIC#
+target_link_libraries(standalone_support PUBLIC
                       dl
                       pthread
                       Microsoft.GSL::GSL)
 
 target_link_libraries(GPUTracking PUBLIC standalone_support)
 target_link_libraries(TPCFastTransformation PUBLIC standalone_support)
 
+target_link_libraries(standalone_support2 PUBLIC GPUTracking TPCFastTransformation standalone_support)
+
 if(NOT GPUCA_CONFIG_O2)
   target_compile_definitions(GPUTracking PRIVATE GPUCA_RUN2)
 endif()
@@ -243,7 +251,7 @@ if (GPUCA_BUILD_DEBUG_SANITIZE AND CMAKE_CXX_COMPILER MATCHES "clang\\+\\+")
 endif()
 
 # Installation
-install(TARGETS ca TPCFastTransformation standalone_support)
+install(TARGETS standalone_support standalone_support2)
 install(FILES "cmake/makefile" DESTINATION "${CMAKE_INSTALL_PREFIX}")
 install(CODE "execute_process(COMMAND ${CMAKE_COMMAND} -E create_symlink ${O2_DIR} ${CMAKE_INSTALL_PREFIX}/src)")
 install(CODE "execute_process(COMMAND ${CMAKE_COMMAND} -E create_symlink ${CMAKE_BINARY_DIR}/config.cmake ${CMAKE_INSTALL_PREFIX}/config.cmake)")

GPU/TPCFastTransformation/CMakeLists.txt

@@ -123,6 +123,7 @@ if(ALIGPU_BUILD_TYPE STREQUAL "Standalone")
   add_library(${MODULE} SHARED ${SRCS})
   set(targetName ${MODULE})
   target_include_directories(${targetName} PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})
+  install(TARGETS TPCFastTransformation)
 endif()
 
 install(FILES ${HDRS_CINT_O2} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/GPU)