[PWGJE] First update tagging QA (#4589)

* First update tagging QA

* add cmakelist for tagging qa

* remove num

* add line eof

* delete line

* add line

* remove line

* remove comma

* remove species number

Author: hanseopark

PWGJE/Core/JetTaggingUtilities.h

@@ -38,6 +38,8 @@ enum JetTaggingSpecies {
 
 namespace jettaggingutilities
 {
+const int cmTomum = 10000; // using cm -> #mum for impact parameter (dca)
+
 /**
  * returns the globalIndex of the earliest mother of a particle in the shower. returns -1 if a suitable mother is not found
  *
@@ -275,6 +277,36 @@ int jetOrigin(T const& jet, U const& particles, float dRMax = 0.25)
   return 0;
 }
 
+/**
+ * return geometric sign which is calculated scalar product between jet axis with DCA (track propagated to PV )
+ * positive and negative value are expected from primary vertex
+ * positive value is expected from secondary vertex
+ *
+ * @param collision which is needed external table of collision due to postion X and Y
+ * @param jet
+ * @param track which is needed each DCA_X and Y which is measured in jettaggerhfExtension.cxx
+ */
+template <typename T, typename U, typename V>
+int getGeoSign(T const& collision, U const& jet, V const& track)
+{
+  auto trackPar = getTrackPar(track);
+  auto xyz = trackPar.getXYZGlo();
+  auto dcaX = xyz.X() - collision.posX();
+  auto dcaY = xyz.Y() - collision.posY();
+  auto sign = TMath::Sign(1, dcaX * jet.px() + dcaY * jet.py() + track.dcaZ() * jet.pz());
+  if (sign < -1 || sign > 1)
+    LOGF(info, Form("Sign is %d", sign));
+  return sign;
+}
+
+void calculateDcaXYZ(float& dcaXYZ, float& sigmaDcaXYZ2, float dcaXY, float dcaZ, float cYY, float cZY, float cZZ, float sigmaDcaXY2, float sigmaDcaZ2)
+{
+  dcaXYZ = std::sqrt(dcaXY * dcaXY + dcaZ * dcaZ);
+  float dFdxy = 2 * dcaXY / dcaXYZ;
+  float dFdz = 2 * dcaZ / dcaXYZ;
+  sigmaDcaXYZ2 = std::abs(cYY * dFdxy * dFdxy + cZZ * dFdz * dFdz + 2 * cZY * dFdxy * dFdz);
+}
+
 }; // namespace jettaggingutilities
 
 #endif // PWGJE_CORE_JETTAGGINGUTILITIES_H_

PWGJE/TableProducer/jettaggerhf.cxx

@@ -12,6 +12,7 @@
 // Task to produce a table joinable to the jet tables for hf jet tagging
 //
 /// \author Nima Zardoshti <nima.zardoshti@cern.ch>
+/// \author Hanseo Park <hanseo.park@cern.ch>
 
 #include "Framework/AnalysisTask.h"
 #include "Framework/AnalysisDataModel.h"
@@ -34,9 +35,10 @@ struct JetTaggerHFTask {
   Produces<JetTaggingTableData> taggingTableData;
   Produces<JetTaggingTableMCD> taggingTableMCD;
 
-  Configurable<bool> doAlgorithm1{"doAlgorithm1", false, "fill table for algoithm 1"};
-  Configurable<bool> doAlgorithm2{"doAlgorithm2", false, "fill table for algoithm 2"};
-  Configurable<bool> doAlgorithm3{"doAlgorithm3", false, "fill table for algoithm 3"};
+  Configurable<bool> doWShower{"doWShower", false, "find jet origin included gluon spliting"}; // true:: remove gluon spliting
+  Configurable<bool> doTC{"doTC", false, "fill table for track counting algorithm"};
+  Configurable<bool> doSV{"doSV", false, "fill table for secondary vertex algorithm"};
+  Configurable<bool> doML{"doML", false, "fill table for machine learning"};
   Configurable<float> maxDeltaR{"maxDeltaR", 0.25, "maximum distance of jet axis from flavour initiating parton"};
 
   void processDummy(JetCollisions const& collision)
@@ -51,9 +53,9 @@ struct JetTaggerHFTask {
       int algorithm1 = jet.globalIndex(); // This needs to be changed. It is only done because O2Physics compilation breaks if jet is unused
       int algorithm2 = 0;
       int algorithm3 = 0;
-      // if (doAlgorithm1) algorithm1 = jettaggingutilities::Algorithm1((mcdjet, tracks);
-      // if (doAlgorithm2) algorithm2 = jettaggingutilities::Algorithm2((mcdjet, tracks);
-      // if (doAlgorithm3) algorithm3 = jettaggingutilities::Algorithm3((mcdjet, tracks);
+      // if (doTC) algorithm1 = jettaggingutilities::Algorithm1((mcdjet, tracks);
+      // if (doSV) algorithm2 = jettaggingutilities::Algorithm2((mcdjet, tracks);
+      // if (doML) algorithm3 = jettaggingutilities::Algorithm3((mcdjet, tracks);
       taggingTableData(0, algorithm1, algorithm2, algorithm3);
     }
   }
@@ -62,14 +64,18 @@ struct JetTaggerHFTask {
   void processMCD(JetCollision const& collision, JetTableMCD const& mcdjets, JetTracksMCD const& tracks, JetParticles const& particles)
   {
     for (auto& mcdjet : mcdjets) {
-
-      int origin = jettaggingutilities::mcdJetFromHFShower(mcdjet, tracks, particles, maxDeltaR);
+      typename JetTracksMCD::iterator hftrack;
+      int origin = 0;
+      if (!doWShower)
+        origin = jettaggingutilities::mcdJetFromHFShower(mcdjet, tracks, particles, maxDeltaR);
+      else
+        origin = jettaggingutilities::jetTrackFromHFShower(mcdjet, tracks, particles, hftrack);
       int algorithm1 = 0;
       int algorithm2 = 0;
       int algorithm3 = 0;
-      // if (doAlgorithm1) algorithm1 = jettaggingutilities::Algorithm1((mcdjet, tracks);
-      // if (doAlgorithm2) algorithm2 = jettaggingutilities::Algorithm2((mcdjet, tracks);
-      // if (doAlgorithm3) algorithm3 = jettaggingutilities::Algorithm3((mcdjet, tracks);
+      // if (doTC) algorithm1 = jettaggingutilities::Algorithm1((mcdjet, tracks);
+      // if (doSV) algorithm2 = jettaggingutilities::Algorithm2((mcdjet, tracks);
+      // if (doML) algorithm3 = jettaggingutilities::Algorithm3((mcdjet, tracks);
       taggingTableMCD(origin, algorithm1, algorithm2, algorithm3);
     }
   }

PWGJE/Tasks/CMakeLists.txt

@@ -112,6 +112,10 @@ if(FastJet_FOUND)
                         SOURCES phiInJets.cxx
                         PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::PWGJECore O2Physics::AnalysisCore
                         COMPONENT_NAME Analysis)
+    o2physics_add_dpl_workflow(jet-taggerhf-qa
+                        SOURCES jettaggerhfQA.cxx
+                        PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::PWGJECore O2Physics::AnalysisCore
+                        COMPONENT_NAME Analysis)
     o2physics_add_dpl_workflow(jet-lund-reclustering
                         SOURCES jetLundReclustering.cxx
                         PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::PWGJECore O2Physics::AnalysisCore FastJet::FastJet FastJet::Contrib