align to pirmin's gbb

BBAnalysis/BBHeppy/BuildFile.xml

@@ -0,0 +1,28 @@
+
+<use name="CommonTools/Utils"/>
+<use name="TrackingTools/IPTools"/>
+<use name="SimDataFormats/GeneratorProducts"/>
+
+<use name="root"/>
+<use name="rootcore"/>
+<use name="rootinteractive"/>
+
+<use name="rootpy"/>
+<use name="clhepheader"/>
+<use name="DataFormats/Candidate"/>
+<use name="DataFormats/PatCandidates"/>
+<use name="DataFormats/HepMCCandidate"/>
+<use name="fastjet"/>
+<use name="ktjet"/>
+<use name="EgammaAnalysis/ElectronTools"/>
+<use name="MagneticField/UniformEngine"/>
+<use name="MagneticField/ParametrizedEngine"/>
+<use name="RecoVertex/KalmanVertexFit"/>
+<use name="RecoVertex/VertexTools"/>
+<use name="boost"/>
+<use name="lhapdf"/>
+<export>
+  <lib name="1"/>
+</export>
+
+

BBAnalysis/BBHeppy/interface/BBClusterizer.h

@@ -0,0 +1,62 @@
+#ifndef PhysicsTools_Heppy_BBClusterizer_h
+#define PhysicsTools_Heppy_BBClusterizer_h
+
+#include <vector>
+#include <iostream>
+#include <cmath>
+#include <TLorentzVector.h>
+#include <TMath.h>
+#include "DataFormats/Math/interface/LorentzVector.h"
+
+#include <boost/shared_ptr.hpp>
+#include <fastjet/internal/base.hh>
+#include "fastjet/PseudoJet.hh"
+#include "fastjet/JetDefinition.hh"
+#include "fastjet/ClusterSequence.hh"
+#include "fastjet/Selector.hh"
+#include "fastjet/PseudoJet.hh"
+
+namespace heppy{
+class BBClusterizer {
+
+ public:
+  typedef math::XYZTLorentzVector LorentzVector;
+
+  BBClusterizer(const std::vector<LorentzVector> & ,
+		   const LorentzVector &,const LorentzVector &,
+                   double ktpower, double rparam);
+
+  BBClusterizer(const std::vector<LorentzVector> & ,
+		   const LorentzVector &,const LorentzVector &, const LorentzVector &,const LorentzVector &,
+                   double ktpower, double rparam);
+
+  /// get grouping (inclusive jets)
+  std::vector<LorentzVector> getBJets();
+  std::vector<LorentzVector> getBJets4B();
+  std::vector<LorentzVector> GetLeadingJets();
+
+ private:
+  // pack the returns in a fwlite-friendly way
+  std::vector<LorentzVector> makeP4s(const std::vector<fastjet::PseudoJet> &jets) ;
+
+  // look if there is the SV in the jet
+  int SVIsInTheJet(const fastjet::PseudoJet & pj);
+
+  // divide the jet in 2
+  std::vector<fastjet::PseudoJet>  DivideTheJet(const fastjet::PseudoJet & pj);
+
+  // used to handle the inputs
+  std::vector<fastjet::PseudoJet> fjInputs;        // fastjet inputs
+
+  LorentzVector b1_;
+  LorentzVector b2_;
+  double ktpower_;
+  double rparam_;
+
+  /// fastjet outputs
+  typedef boost::shared_ptr<fastjet::ClusterSequence>  ClusterSequencePtr;
+  ClusterSequencePtr fjClusterSeq_;    
+};
+}
+#endif   
+

BBAnalysis/BBHeppy/interface/SVUtils.h

@@ -0,0 +1,37 @@
+#ifndef PhysicsTools_Heppy_SVUtils_h
+#define PhysicsTools_Heppy_SVUtils_h
+
+#include <vector>
+#include <iostream>
+#include <cmath>
+#include <TLorentzVector.h>
+#include <TMath.h>
+#include "DataFormats/Math/interface/LorentzVector.h"
+
+#include "RecoVertex/VertexPrimitives/interface/ConvertToFromReco.h"
+#include "RecoVertex/VertexPrimitives/interface/VertexState.h"
+#include "RecoVertex/VertexTools/interface/VertexDistance3D.h"
+#include "RecoVertex/VertexTools/interface/VertexDistanceXY.h"
+#include "RecoVertex/VertexTools/interface/VertexDistance.h"
+#include "RecoVertex/VertexTools/interface/SharedTracks.h"
+#include <RecoVertex/VertexPrimitives/interface/BasicVertexState.h>
+
+
+namespace heppy{
+class SVUtils {
+
+ public:
+  SVUtils(const reco::Vertex & pv, const  reco::VertexCompositePtrCandidate & sv): 
+  pv_(RecoVertex::convertPos(pv.position()),RecoVertex::convertError(pv.error())),
+  sv_(RecoVertex::convertPos(sv.position()),RecoVertex::convertError(sv.error()))
+  {}
+  Measurement1D dist3D();
+  Measurement1D dist2D();
+
+ private:
+    VertexState pv_;
+    VertexState sv_;
+};
+}
+#endif   
+

BBAnalysis/BBHeppy/interface/SignedImpactParameter.h

@@ -0,0 +1,60 @@
+#ifndef SignedImpactParmeter_h
+#define SignedImpactParmeter_h
+
+#include "DataFormats/Candidate/interface/VertexCompositePtrCandidate.h"
+#include "DataFormats/VertexReco/interface/Vertex.h"
+#include "DataFormats/TrackReco/interface/Track.h"
+#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1D.h"
+
+#if defined( __GXX_EXPERIMENTAL_CXX0X__)
+#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
+#endif
+
+struct MagneticField;
+
+namespace heppy {
+class SignedImpactParameter {
+    public:
+        SignedImpactParameter() ;
+        ~SignedImpactParameter() ;
+
+        bool qualitytrk(const reco::Track trk, const reco::Vertex &vtx);
+        Measurement1D signedIP3D(const reco::Track &tk, const reco::Vertex &vtx, const reco::Track::Vector jetdir) const ;
+        Measurement1D signedIP3D(const reco::Track &tk, const reco::VertexCompositePtrCandidate &vtx, const reco::Track::Vector jetdir) const ;
+        Measurement1D signedIP2D(const reco::Track &tk, const reco::Vertex &vtx, const reco::Track::Vector jetdir) const ;
+        Measurement1D signedIP2D(const reco::Track &tk, const reco::VertexCompositePtrCandidate &vtx, const reco::Track::Vector jetdir) const ;
+        Measurement1D IP3D(const reco::Track &tk, const reco::Vertex &vtx) const ;
+        Measurement1D IP3D(const reco::Track &tk, const reco::VertexCompositePtrCandidate &vtx) const ;
+        Measurement1D IP2D(const reco::Track &tk, const reco::Vertex &vtx) const ;
+        Measurement1D IP2D(const reco::Track &tk, const reco::VertexCompositePtrCandidate &vtx) const ;
+
+        std::pair<double,double> twoTrackChi2(const reco::Track &tk1, const reco::Track &tk2) const ;
+
+        //For the vertex related variables
+        //A = selectedLeptons[0], B = selectedLeptons[1], C = selectedLeptons[2], D = selectedLeptons[3]
+
+        #if defined( __GXX_EXPERIMENTAL_CXX0X__)
+        //Helping functions
+        std::vector<reco::TransientTrack> ttrksf(const reco::Track &trkA, const reco::Track &trkB, const reco::Track &trkC, const reco::Track &trkD, int nlep) const;  
+        std::vector<reco::TransientTrack> ttrksbuthef(const reco::Track &trkA, const reco::Track &trkB, const reco::Track &trkC, const reco::Track &trkD, int nlep, int iptrk) const; 
+        reco::TransientTrack thettrkf(const reco::Track &trkA, const reco::Track &trkB, const reco::Track &trkC, const reco::Track &trkD, int nlep, int iptrk) const;
+        #endif
+
+        //Variables related to IP
+        //Of one lepton w.r.t. the PV of the event
+        std::pair<double,double> absIP3D(const reco::Track &trk, const reco::Vertex &pv) const;
+        //Of one lepton w.r.t. the PV of the PV of the other leptons only
+        std::pair<double,double> absIP3Dtrkpvtrks(const reco::Track &trkA, const reco::Track &trkB, const reco::Track &trkC, const reco::Track &trkD, int nlep, int iptrk) const;
+       //Variables related to chi2
+       std::pair<double,double> chi2pvtrks(const reco::Track &trkA, const reco::Track &trkB, const reco::Track &trkC, const reco::Track &trkD, int nlep) const;
+
+        Measurement1D vertexD3d(const reco::VertexCompositePtrCandidate &sv, const reco::Vertex &pv) const ;
+        Measurement1D vertexDxy(const reco::VertexCompositePtrCandidate &sv, const reco::Vertex &pv) const ;
+        float vertexDdotP(const reco::VertexCompositePtrCandidate &sv, const reco::Vertex &pv) const ;
+    private:
+        //MagneticField *bfield_;
+        static MagneticField *paramField_;
+};
+
+}
+#endif

BBAnalysis/BBHeppy/plugins/BuildFile.xml

@@ -0,0 +1,21 @@
+<use name="FWCore/Framework"/>
+<use name="FWCore/PluginManager"/>
+<use name="FWCore/ParameterSet"/>
+<use name="DataFormats/PatCandidates"/>
+<use name="DataFormats/Candidate"/>
+<use name="DataFormats/HepMCCandidate"/>
+<use name="RecoVertex/VertexPrimitives"/>
+<use name="RecoVertex/VertexTools"/>
+<use name="RecoBTag/SecondaryVertex"/>
+<use name="DataFormats/BTauReco"/>
+<use name="FWCore/ServiceRegistry"/>
+<use name="CommonTools/UtilAlgos"/>
+<flags EDM_PLUGIN="1"/>
+
+
+<!--<use name="DataFormats/TrackReco"/>-->
+<!--<use name="CommonTools/UtilAlgos"/>-->
+<!--<use name="PhysicsTools/UtilAlgos"/>-->
+<!--<use name="root"/>-->
+<!--<use name="rootcore"/>-->
+