GPU Workflor: add possibility to run TRD tracking within GPU workflow

davidrohr · davidrohr · commit bce63c1a6362 · 2022-01-30T19:16:48.000+01:00
diff --git a/GPU/Workflow/include/GPUWorkflow/GPUWorkflowSpec.h b/GPU/Workflow/include/GPUWorkflow/GPUWorkflowSpec.h
@@ -47,6 +47,7 @@ struct Config {
   bool processMC = false;
   bool sendClustersPerSector = false;
   bool askDISTSTF = true;
+  bool runTRDTracking = false;
   bool readTRDtracklets = false;
 };
 using CompletionPolicyData = std::vector<framework::InputSpec>;
diff --git a/GPU/Workflow/src/GPUWorkflowSpec.cxx b/GPU/Workflow/src/GPUWorkflowSpec.cxx
@@ -231,6 +231,10 @@ DataProcessorSpec getGPURecoWorkflowSpec(gpuworkflow::CompletionPolicyData* poli
           throw std::invalid_argument("Cannot run TPC decompression with a sector mask");
         }
       }
+      if (specconfig.runTRDTracking) {
+        config.configWorkflow.inputs.setBits(GPUDataTypes::InOutType::TRDTracklets, true);
+        config.configWorkflow.steps.setBits(GPUDataTypes::RecoStep::TRDTracking, true);
+      }
       if (specconfig.outputSharedClusterMap) {
         config.configProcessing.outputSharedClusterMap = true;
       }
diff --git a/GPU/Workflow/src/gpu-reco-workflow.cxx b/GPU/Workflow/src/gpu-reco-workflow.cxx
@@ -45,7 +45,7 @@ void customize(std::vector<ConfigParamSpec>& workflowOptions)
 
   std::vector<ConfigParamSpec> options{
     {"input-type", VariantType::String, "digits", {"digitizer, digits, zsraw, zsonthefly, clustersnative, compressed-clusters-root, compressed-clusters-ctf, trd-tracklets"}},
-    {"output-type", VariantType::String, "tracks", {"clustersnative, tracks, compressed-clusters-ctf, qa, no-shared-cluster-map, send-clusters-per-sector"}},
+    {"output-type", VariantType::String, "tracks", {"clustersnative, tracks, compressed-clusters-ctf, qa, no-shared-cluster-map, send-clusters-per-sector, trd-tracks"}},
     {"disable-root-input", VariantType::Bool, true, {"disable root-files input reader"}},
     {"disable-mc", VariantType::Bool, false, {"disable sending of MC information"}},
     {"ignore-dist-stf", VariantType::Bool, false, {"do not subscribe to FLP/DISTSUBTIMEFRAME/0 message (no lost TF recovery)"}},
@@ -80,6 +80,7 @@ enum struct ioType { Digits,
                      Tracks,
                      QA,
                      TRDTracklets,
+                     TRDTracks,
                      NoSharedMap,
                      SendClustersPerSector };
 
@@ -98,7 +99,8 @@ static const std::unordered_map<std::string, ioType> OutputMap{
   {"compressed-clusters-ctf", ioType::CompClustCTF},
   {"qa", ioType::QA},
   {"no-shared-cluster-map", ioType::NoSharedMap},
-  {"send-clusters-per-sector", ioType::SendClustersPerSector}};
+  {"send-clusters-per-sector", ioType::SendClustersPerSector},
+  {"trd-tracks", ioType::TRDTracks}};
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
 {
@@ -141,6 +143,7 @@ WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
   cfg.sendClustersPerSector = isEnabled(outputTypes, ioType::SendClustersPerSector);
   cfg.askDISTSTF = !cfgc.options().get<bool>("ignore-dist-stf");
   cfg.readTRDtracklets = isEnabled(inputTypes, ioType::TRDTracklets);
+  cfg.runTRDTracking = isEnabled(outputTypes, ioType::TRDTracks);
   specs.emplace_back(o2::gpu::getGPURecoWorkflowSpec(&gPolicyData, cfg, tpcSectors, gTpcSectorMask, "gpu-reconstruction"));
 
   if (!cfgc.options().get<bool>("ignore-dist-stf")) {

Original file line number	Diff line number	Diff line change
`@@ -231,6 +231,10 @@ DataProcessorSpec getGPURecoWorkflowSpec(gpuworkflow::CompletionPolicyData* poli`
`231`	`231`	`throw std::invalid_argument("Cannot run TPC decompression with a sector mask");`
`232`	`232`	`}`
`233`	`233`	`}`
	`234`	`+ if (specconfig.runTRDTracking) {`
	`235`	`+ config.configWorkflow.inputs.setBits(GPUDataTypes::InOutType::TRDTracklets, true);`
	`236`	`+ config.configWorkflow.steps.setBits(GPUDataTypes::RecoStep::TRDTracking, true);`
	`237`	`+ }`
`234`	`238`	`if (specconfig.outputSharedClusterMap) {`
`235`	`239`	`config.configProcessing.outputSharedClusterMap = true;`
`236`	`240`	`}`