adding upc selections

Author: nzardosh

PWGJE/Core/JetDerivedDataUtilities.h

@@ -17,6 +17,8 @@
 #ifndef PWGJE_CORE_JETDERIVEDDATAUTILITIES_H_
 #define PWGJE_CORE_JETDERIVEDDATAUTILITIES_H_
 
+#include "PWGUD/Core/SGSelector.h"
+
 #include "Common/CCDB/EventSelectionParams.h"
 #include "Common/CCDB/RCTSelectionFlags.h"
 #include "Common/CCDB/TriggerAliases.h"
@@ -45,7 +47,10 @@ enum JCollisionSel {
   selNoSameBunchPileup = 6,
   selIsGoodZvtxFT0vsPV = 7,
   selNoCollInTimeRangeStandard = 8,
-  selNoCollInRofStandard = 9
+  selNoCollInRofStandard = 9,
+  selUpcSingleGapA = 10,
+  selUpcSingleGapC = 11,
+  selUpcDoubleGap = 12,
 };
 
 enum JCollisionSubGeneratorId {
@@ -169,11 +174,21 @@ std::vector<int> initialiseEventSelectionBits(const std::string& eventSelectionM
     eventSelectionMaskBits.push_back(JCollisionSel::sel7);
     eventSelectionMaskBits.push_back(JCollisionSel::selKINT7);
   }
+  if (eventSelectionMasksContainSelection(eventSelectionMasks, "selUPCSingleGapA")) {
+    eventSelectionMaskBits.push_back(JCollisionSel::selUpcSingleGapA);
+  }
+  if (eventSelectionMasksContainSelection(eventSelectionMasks, "selUPCSingleGapC")) {
+    eventSelectionMaskBits.push_back(JCollisionSel::selUpcSingleGapC);
+  }
+  if (eventSelectionMasksContainSelection(eventSelectionMasks, "selUPCDoubleGap")) {
+    eventSelectionMaskBits.push_back(JCollisionSel::selUpcDoubleGap);
+  }
+
   return eventSelectionMaskBits;
 }
 
 template <typename T>
-uint16_t setEventSelectionBit(T const& collision)
+uint16_t setEventSelectionBit(T const& collision, int upcSelectionResult = o2::aod::sgselector::TrueGap::NoGap)
 {
   uint16_t bit = 0;
   if (collision.sel8()) {
@@ -206,6 +221,16 @@ uint16_t setEventSelectionBit(T const& collision)
   if (collision.selection_bit(o2::aod::evsel::kNoCollInRofStandard)) {
     SETBIT(bit, JCollisionSel::selNoCollInRofStandard);
   }
+  if (upcSelectionResult == o2::aod::sgselector::SingleGapA) {
+    SETBIT(bit, JCollisionSel::selUpcSingleGapA);
+  }
+  if (upcSelectionResult == o2::aod::sgselector::SingleGapC) {
+    SETBIT(bit, JCollisionSel::selUpcSingleGapC);
+  }
+  if (upcSelectionResult == o2::aod::sgselector::DoubleGap) {
+    SETBIT(bit, JCollisionSel::selUpcDoubleGap);
+  }
+
   return bit;
 }
 

PWGJE/DataModel/JetReducedData.h

@@ -88,6 +88,11 @@ DECLARE_SOA_COLUMN(CentFT0C, centFT0C, float);
 DECLARE_SOA_COLUMN(CentFT0M, centFT0M, float);
 DECLARE_SOA_COLUMN(CentFT0CVariant1, centFT0CVariant1, float);
 DECLARE_SOA_COLUMN(CentralityVariant1, centralityVariant1, float);
+DECLARE_SOA_COLUMN(AmplitudesFV0, amplitudesFV0, std::vector<float>);
+DECLARE_SOA_COLUMN(AmplitudesFT0A, amplitudesFT0A, std::vector<float>);
+DECLARE_SOA_COLUMN(AmplitudesFT0C, amplitudesFT0C, std::vector<float>);
+DECLARE_SOA_COLUMN(AmplitudesFDDA, amplitudesFDDA, std::vector<float>);
+DECLARE_SOA_COLUMN(AmplitudesFDDC, amplitudesFDDC, std::vector<float>);
 DECLARE_SOA_COLUMN(HadronicRate, hadronicRate, float);
 DECLARE_SOA_COLUMN(Weight, weight, float);
 DECLARE_SOA_COLUMN(GetSubGeneratorId, getSubGeneratorId, int);
@@ -134,6 +139,11 @@ DECLARE_SOA_TABLE_STAGED(JCollisions, "JCOLLISION",
                          jcollision::CentFT0C,
                          jcollision::CentFT0M,
                          jcollision::CentFT0CVariant1,
+                         jcollision::AmplitudesFV0,
+                         jcollision::AmplitudesFT0A,
+                         jcollision::AmplitudesFT0C,
+                         jcollision::AmplitudesFDDA,
+                         jcollision::AmplitudesFDDC,
                          jcollision::HadronicRate,
                          jcollision::TrackOccupancyInTimeRange,
                          jcollision::Alias,

PWGJE/TableProducer/CMakeLists.txt

@@ -15,7 +15,7 @@ if(FastJet_FOUND)
 
 o2physics_add_dpl_workflow(jet-deriveddata-producer
                     SOURCES derivedDataProducer.cxx
-                    PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::PWGJECore O2Physics::AnalysisCore O2Physics::AnalysisCCDB O2Physics::EventFilteringUtils
+                    PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::PWGJECore O2Physics::AnalysisCore O2Physics::AnalysisCCDB O2Physics::EventFilteringUtils O2Physics::SGCutParHolder
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(jet-deriveddata-trigger-producer