Changeset 264 in svn for trunk/Utilities/ExRootAnalysis/interface/BlockClasses.h

Timestamp:

Feb 11, 2009, 10:22:30 AM (16 years ago)

Author:

Xavier Rouby

Message:

first test 2.0

File:

• trunk/Utilities/ExRootAnalysis/interface/BlockClasses.h (modified) (10 diffs)

trunk/Utilities/ExRootAnalysis/interface/BlockClasses.h

@@ -13 +13 @@
  *  present in the data members of the Block TRoot class.
  *
- *  $Date: 2009-02-02 11:32:12 $
- *  $Revision: 1.11 $
+ *  $Date: 2009-02-11 09:19:14 $
+ *  $Revision: 1.12 $
  *
  *  
@@ -24 +24 @@
 #include "TObject.h"
 #include "BlockCompare.h"
+#include "interface/D_Constants.h"
+#include "interface/CaloUtil.h"
 
 class TSortableObject: public TObject
@@ -147 +149 @@
   float Pz; // particle momentum vector (z component) in GeV
 
-  float PT; // particle transverse momentum in GeV
   float Eta; // particle pseudorapidity  
   float Phi; // particle azimuthal angle in rad 
 
+  float EtaCalo; // particle pseudorapidity when entering the calo, 
+  float PhiCalo; // particle azimuthal angle in rad when entering the calo
+
   void Set(const TLorentzVector& momentum);
   void Set(const float px, const float py, const float pz, const float e);
-  static TCompare *fgCompare; //!
-  const TCompare *GetCompare() const { return fgCompare; }
+  void SetEtaPhi(const float eta, const float phi) {Eta=eta; Phi=phi;};
+  void SetEtaPhiCalo(const float eta, const float phi) {EtaCalo=eta; PhiCalo=phi;};
+  void SetEtaPhiEET(const float eta, const float phi, const float e, const float et);
+  static TCompare *fgCompare; //!
+  const TCompare *GetCompare() const { return fgCompare; }
+  //void SetEem_Ehad(const float sE_em, const float sE_had) {} ; // sets E_em and E_had, computes E and ET
+  float PT; // particle transverse momentum in GeV
+  //const float getEtaGen() const {return _EtaGen;}
+  //const float getPhiGen() const {return _PhiGen;}
+
+
+//protected:
+  //float EGen, _EtaGen, _PhiGen; // from the generator: energy, eta, phi
 
   ClassDef(TRootParticle, 1)
@@ -165 +180 @@
 
 public:
-  TRootGenParticle() {};
+  TRootGenParticle() {_initialised=false;}
   int PID; // particle HEP ID number [RawHepEventParticle::pid()]
   int Status; // particle status [RawHepEventParticle::status()]
@@ -172 +187 @@
   int D1; // particle 1st daughter [RawHepEventParticle::daughter1() - 1]
   int D2; // particle 2nd daughter [RawHepEventParticle::daughter2() - 1]
-  float Charge; // electrical charge
 
   float T; // particle vertex position (t component) [RawHepEventParticle::t()]
@@ -179 +193 @@
   float Z; // particle vertex position (z component) [RawHepEventParticle::z()]
   float M;
-  static TCompare *fgCompare; //!
-
+  void setFractions();
+  const float getCharge() const {return Charge;};
+  const float getFem()  {if(!_initialised) setFractions(); return _Fem;}
+  const float getFhad() {if(!_initialised) setFractions(); return _Fhad;}
+
+  float Charge;      // electrical charge
+
+  static TCompare *fgCompare; //!
+ protected:
+  float _Fem, _Fhad; // fractions of energy deposit
+  bool _initialised;
   ClassDef(TRootGenParticle, 1)
 };
+
 
 //------------------------------------------------------------------------------
@@ -225 +249 @@
 //---------------------------------------------------------------------------
 
-class TRootTracks: public TRootParticle
-{
-public:
-  TRootTracks() : Etaout(0), Phiout(0) {};
-  static TCompare *fgCompare; //!
- 
+class D_Track : public TSortableObject {
+//class D_Track : public TRootParticle {
+ public:
+    D_Track(); // needed for storage in ExRootAnalysis
+    D_Track(const D_Track& track);
+    D_Track(const float inEta, const float inPhi, const float outEta, const float outPhi, const float pt);
+    D_Track& operator=(const D_Track& track);
+    void Set(const float inEta, const float inPhi, const float outEta, const float outPhi, const float pt);
+    const TLorentzVector GetFourVector() const;
+    const float getEta() const {return Eta;}
+    const float getPhi() const {return Phi;}
+    const float getEtaOuter() const {return EtaOuter;}
+    const float getPhiOuter() const {return PhiOuter;}
+    const float getE() const {return E;}
+    const float getPx() const {return Px;}
+    const float getPy() const {return Py;}
+    const float getPz() const {return Pz;}
+    const float getPT() const {return PT;}
+
+  static TCompare *fgCompare; //!
+  const TCompare *GetCompare() const { return fgCompare; }
+ protected:
+    float Eta, Phi; // (eta,phi) at the beginning of the track
+    float EtaOuter, PhiOuter; // (eta,phi) at the end of the track
+    float PT, E, Px, Py, Pz;  // transverse momentum
+ ClassDef(D_Track, 1)
+};
+
+
+class TRootTracks: public TRootParticle  {
+public:
+  TRootTracks() {} // : Etaout(0), Phiout(0) {};
+  static TCompare *fgCompare; //!
+  TRootTracks(const D_Track& track);
+  float E;  // particle energy in GeV
+  float Px; // particle momentum vector (x component) in GeV
+  float Py; // particle momentum vector (y component) in GeV
+  float Pz; // particle momentum vector (z component) in GeV
+
+  float Eta; // particle pseudorapidity
+  float Phi; // particle azimuthal angle in rad
+
+  float EtaCalo; // particle pseudorapidity when entering the calo,
+  float PhiCalo; // particle azimuthal angle in rad when entering the calo 
 //  float X, Y, Z; // coordinates of the beginning of the track
 //  float Xout, Yout, Zout; // coordinates of the end of the track
-  float Etaout, Phiout; // coordinates of the end of the track
+  //float Etaout, Phiout; // coordinates of the end of the track
 //  void SetPosition(const float x, const float y, const float z) {X=x; Y=y; Z=z;}
 //  void SetPositionOut(const float x, const float y, const float z) {Xout=x; Yout=y; Zout=z;}
@@ -241 +303 @@
 //---------------------------------------------------------------------------
 
-class TRootCalo: public TRootParticle
-{
+/*class TRootCalo: public TRootParticle
 public:
   TRootCalo() {};
@@ -248 +309 @@
 
   ClassDef(TRootCalo, 1)
+};
+*/
+
+//class TRootCalo: public TSortableObject {
+class TRootCalo: public TRootParticle {
+ public:
+   TRootCalo() ;
+   TRootCalo(const TRootCalo& cal);
+   TRootCalo& operator=(const TRootCalo& cal);
+   void set(const D_CaloTower&  cal);
+   static TCompare *fgCompare; //!
+ //  const TCompare *GetCompare() const { return fgCompare; }
+   const float getET() const {return ET;}
+ 
+ protected:
+    //float Eta, Phi;    // segmented eta, phi
+    float E_em, E_had; // electromagnetic and hadronic components of the tower energy
+    //float E;
+    float ET;       // total energy and transverse energy   
+   ClassDef(TRootCalo, 1)
 };
 
@@ -340 +421 @@
 };
 
-//---------------------------------------------------------------------------
-
 #endif // BLOCKCLASSES_H