106X - Adding the DV-like inputs version 1

DeepNtuplizer/BuildFile.xml

@@ -4,7 +4,7 @@
 <use name="DataFormats/PatCandidates"/>
 <use name="TrackingTools/IPTools"/>
 <use name="DeepNTuples/JetAnalysis"/>
-<use name="RecoBTag/TensorFlow"/>
+<use name="RecoBTag/ONNXRuntime"/>
 <use name="RecoBTag/TrackProbability"/>
 <use name="CondFormats/DataRecord"/>
 <use name="CondFormats/BTauObjects"/>

DeepNtuplizer/bin/mergeSamples.cc

@@ -40,6 +40,8 @@
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_JetInfo.h"
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_pfCands.h"
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_SV.h"
+#include "DeepNTuples/DeepNtuplizer/interface/ntuple_LT.h"
+#include "DeepNTuples/DeepNtuplizer/interface/ntuple_pairwise.h"
 
 #include <dirent.h>
 #include <stdlib.h>
@@ -97,9 +99,10 @@ int main(int argc, char *argv[]){
     std::vector<ntuple_content*> branchinfos;
     branchinfos.push_back(new ntuple_JetInfo());
     branchinfos.push_back(new ntuple_SV());
+    branchinfos.push_back(new ntuple_LT());
     branchinfos.push_back(new ntuple_bTagVars());
     branchinfos.push_back(new ntuple_pfCands());
-
+    branchinfos.push_back(new ntuple_pairwise());
 
 
     //simple opt parsing

DeepNtuplizer/bin/mergeSamples_parallel.cc

@@ -40,6 +40,8 @@
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_JetInfo.h"
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_pfCands.h"
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_SV.h"
+#include "DeepNTuples/DeepNtuplizer/interface/ntuple_LT.h"
+#include "DeepNTuples/DeepNtuplizer/interface/ntuple_pairwise.h"
 #include "DeepNTuples/DeepNtuplizer/interface/ntuple_FatJetInfo.h"
 
 #include <dirent.h>
@@ -121,8 +123,10 @@ std::vector<TChain* > createChains(const std::vector<std::vector<TString> >& inf
 
     branchinfos.push_back(new ntuple_JetInfo());
     branchinfos.push_back(new ntuple_SV());
+    branchinfos.push_back(new ntuple_LT());
     branchinfos.push_back(new ntuple_bTagVars());
     branchinfos.push_back(new ntuple_pfCands());
+    branchinfos.push_back(new ntuple_pairwise());
     branchinfos.push_back(new ntuple_FatJetInfo());
 
     std::vector<TChain* > chains;

DeepNtuplizer/bin/printBranches.cc

@@ -2,10 +2,12 @@
 
 #include "../interface/ntuple_JetInfo.h"
 #include "../interface/ntuple_SV.h"
+#include "../interface/ntuple_LT.h"
 #include "../interface/ntuple_bTagVars.h"
 #include "../interface/ntuple_pfCands.h"
-#include "../interface/ntuple_FatJetInfo.h"
-#include "../interface/ntuple_DeepVertex.h"
+//#include "../interface/ntuple_FatJetInfo.h"
+//#include "../interface/ntuple_DeepVertex.h"
+#include "../interface/ntuple_pairwise.h"
 #include "TFile.h"
 #include <vector>
 #include "TH1F.h"
@@ -21,10 +23,12 @@ int main(int argc, char *argv[]) {
     std::vector<ntuple_content*> branchinfos;
     branchinfos.push_back(new ntuple_JetInfo());
     branchinfos.push_back(new ntuple_SV());
+    branchinfos.push_back(new ntuple_LT());
     branchinfos.push_back(new ntuple_bTagVars());
     branchinfos.push_back(new ntuple_pfCands());
-  //  branchinfos.push_back(new ntuple_FatJetInfo());
+    branchinfos.push_back(new ntuple_pairwise());
   //  branchinfos.push_back(new ntuple_DeepVertex());
+    //    branchinfos.push_back(new ntuple_GraphB());
 
     if (argc < 3)
         return -1;

DeepNtuplizer/interface/TrackPairInfoBuilder.h

@@ -0,0 +1,102 @@
+#ifndef BTAGHELPERS_INTERFACE_TRACKPARINFOBUILDER_H_
+#define BTAGHELPERS_INTERFACE_TRACKPAIRINFOBUILDER_H_
+
+#include "DataFormats/GeometrySurface/interface/Line.h"
+#include "TrackingTools/Records/interface/TransientTrackRecord.h"
+#include "TrackingTools/IPTools/interface/IPTools.h"
+#include "TrackingTools/PatternTools/interface/TwoTrackMinimumDistance.h"
+#include "DataFormats/PatCandidates/interface/Jet.h"
+
+namespace deepntuples {
+
+  class TrackPairInfoBuilder {
+  public:
+    TrackPairInfoBuilder();
+
+    void buildTrackPairInfo(const reco::TransientTrack it,
+                            const reco::TransientTrack tt,
+                            const reco::Vertex& pv,
+                            const pat::Jet &jet
+                            );
+
+    const float track_i_pt() const { return track_i_pt_; }
+    const float track_t_pt() const { return track_t_pt_; }
+    const float track_eta() const { return track_eta_; }
+    const float track_phi() const { return track_phi_; }
+    const float track_dz() const { return track_dz_; }
+    const float track_dxy() const { return track_dxy_; }
+    const float pca_distance() const { return pca_distance_; }
+    const float pca_significance() const { return pca_significance_; }
+    const float pcaSeed_x() const { return pcaSeed_x_; }
+    const float pcaSeed_y() const { return pcaSeed_y_; }
+    const float pcaSeed_z() const { return pcaSeed_z_; }
+    const float pcaSeed_xerr() const { return pcaSeed_xerr_; }
+    const float pcaSeed_yerr() const { return pcaSeed_yerr_; }
+    const float pcaSeed_zerr() const { return pcaSeed_zerr_; }
+    const float pcaTrack_x() const { return pcaTrack_x_; }
+    const float pcaTrack_y() const { return pcaTrack_y_; }
+    const float pcaTrack_z() const { return pcaTrack_z_; }
+    const float pcaTrack_xerr() const { return pcaTrack_xerr_; }
+    const float pcaTrack_yerr() const { return pcaTrack_yerr_; }
+    const float pcaTrack_zerr() const { return pcaTrack_zerr_; }
+    const float dotprodTrack() const { return dotprodTrack_; }
+    const float dotprodSeed() const { return dotprodSeed_; }
+    const float pcaSeed_dist() const { return pcaSeed_dist_; }
+    const float pcaTrack_dist() const { return pcaTrack_dist_; }
+    const float track_candMass() const { return track_candMass_; }
+    const float track_ip2d() const { return track_ip2d_; }
+    const float track_ip2dSig() const { return track_ip2dSig_; }
+    const float track_ip3d() const { return track_ip3d_; }
+    const float track_ip3dSig() const { return track_ip3dSig_; }
+    const float dotprodTrackSeed2D() const { return dotprodTrackSeed2D_; }
+    const float dotprodTrackSeed2DV() const { return dotprodTrackSeed2DV_; }
+    const float dotprodTrackSeed3D() const { return dotprodTrackSeed3D_; }
+    const float dotprodTrackSeed3DV() const { return dotprodTrackSeed3DV_; }
+    const float pca_jetAxis_dist() const { return pca_jetAxis_dist_; }
+    const float pca_jetAxis_dotprod() const { return pca_jetAxis_dotprod_; }
+    const float pca_jetAxis_dEta() const { return pca_jetAxis_dEta_; }
+    const float pca_jetAxis_dPhi() const { return pca_jetAxis_dPhi_; }
+
+  private:
+    float track_i_pt_;
+    float track_t_pt_;
+    float track_eta_;
+    float track_phi_;
+    float track_dz_;
+    float track_dxy_;
+    float pca_distance_;
+    float pca_significance_;
+    float pcaSeed_x_;
+    float pcaSeed_y_;
+    float pcaSeed_z_;
+    float pcaSeed_xerr_;
+    float pcaSeed_yerr_;
+    float pcaSeed_zerr_;
+    float pcaTrack_x_;
+    float pcaTrack_y_;
+    float pcaTrack_z_;
+    float pcaTrack_xerr_;
+    float pcaTrack_yerr_;
+    float pcaTrack_zerr_;
+    float dotprodTrack_;
+    float dotprodSeed_;
+    float pcaSeed_dist_;
+    float pcaTrack_dist_;
+    float track_candMass_;
+    float track_ip2d_;
+    float track_ip2dSig_;
+    float track_ip3d_;
+    float track_ip3dSig_;
+    float dotprodTrackSeed2D_;
+    float dotprodTrackSeed2DV_;
+    float dotprodTrackSeed3D_;
+    float dotprodTrackSeed3DV_;
+    float pca_jetAxis_dist_;
+    float pca_jetAxis_dotprod_;
+    float pca_jetAxis_dEta_;
+    float pca_jetAxis_dPhi_;
+  };
+
+}  // namespace btagbtvdeep
+
+#endif  //RecoBTag_FeatureTools_TrackPairInfoBuilder_h