--- UserCode/MitPhysics/Utils/interface/JetTools.h	2010/07/18 21:14:46	1.1
+++ UserCode/MitPhysics/Utils/interface/JetTools.h	2011/01/18 16:48:23	1.9
@@ -1,11 +1,3 @@
-//-------------------
-//
-// Jet Tools
-//
-// S Markson
-//
-//-------------------
-
 #ifndef MITPHYSICS_UTILS_JETTOOLS_H
 #define MITPHYSICS_UTILS_JETTOOLS_H
 
@@ -13,33 +5,42 @@
 #include "MitAna/DataUtil/interface/Debug.h"
 #include "MitAna/DataTree/interface/Names.h"
 #include "MitAna/DataTree/interface/ParticleCol.h"
+#include "MitAna/DataTree/interface/PFCandidateCol.h"
 #include "MitAna/DataTree/interface/JetCol.h"
-#include "MitAna/DataTree/interface/Jet.h"
+#include "MitAna/DataTree/interface/PFJetCol.h"
 #include "MitPhysics/Init/interface/ModNames.h"
 #include "MitAna/DataCont/interface/ObjArray.h"
 #include "MitAna/DataTree/interface/TrackCol.h"
+#include "MitAna/DataTree/interface/CaloTowerCol.h"
 #include "MitAna/DataTree/interface/MetCol.h"
+#include "MitAna/DataTree/interface/VertexCol.h"
 #include <TVector3.h>
 #include <TLorentzVector.h>
 
 namespace mithep {
   class JetTools {
-    public:
-      JetTools();
-      virtual ~JetTools();
-
-      static Double_t NJettiness(const ParticleOArr *particles, const JetOArr *jets, double Y = 0.0);
-      static Double_t NJettiness(const TrackOArr *tracks, const JetOArr *jets, double Y = 0.0);
-      static Double_t NJettiness(const JetOArr *jetsS, const JetOArr *jets, double Y = 0.0);
-      static Double_t M_r(const ParticleOArr *particles);
-      static Double_t Beta_r(const ParticleOArr *particles);
-      static Double_t M_r_t(const ParticleOArr *particles, const Met *met);
-      static Double_t Razor(const ParticleOArr *particles, const Met *met);
-      static Double_t CosineOmega(const ParticleOArr *particles);
-      static Double_t MtHiggs(const CompositeParticle *dilepton, const Met *met, int nsel);
-
-      ClassDef(JetTools, 0)
+  public:
+    JetTools();
+    virtual ~JetTools();
+    
+    static Double_t NJettiness(const ParticleOArr *particles, const JetOArr *jets, double Q = 1, double Y = 0.0);
+    static Double_t NJettiness(const PFCandidateOArr *particles, const JetOArr *jets, double Q = 1, double Y = 0.0);
+    static Double_t NJettiness(const TrackOArr *tracks, const JetOArr *jets, double Q = 1, double Y = 0.0);
+    static Double_t NJettiness(const JetOArr *jetsS, const JetOArr *jets, double Q = 1, double Y = 0.0);
+    static Double_t NJettiness(const CaloTowerOArr *calos, const JetOArr *jets, double Q = 1, double Y = 0.0);
+    
+    static Double_t M_r(const ParticleOArr *particles);
+    static Double_t Beta_r(const ParticleOArr *particles);
+    static Double_t M_r_t(const ParticleOArr *particles, const Met *met);
+    static Double_t Razor(const ParticleOArr *particles, const Met *met);
+    static Double_t CosineOmega(const Particle *particles0, const Particle *particles1);
+    static Double_t MtHiggs(const ParticleOArr *leptons, const Met *met, double metFraction[2], int nsel);
+    static void     Alpha(Double_t AlphaVar[2], const TrackCol *tracks, Jet *jet, const VertexCol *vertices, Double_t  delta_z, Double_t delta_cone);
+    static void     Alpha(Double_t AlphaVar[2], const PFJet *jet, const VertexCol *vertices, Double_t  delta_z);
+    
+    ClassDef(JetTools, 0)
   };
+
 }
 
 #endif