HepEntityI.h

#ifndef  D0RunIconeJets_HepEntity_class
#define  D0RunIconeJets_HepEntity_class
 
#include "inline_maths.h"
#include <fastjet/internal/base.hh>
 
FASTJET_BEGIN_NAMESPACE
 
namespace d0runi{
 
//Author: Lars Sonnenschein 15/Sep/2009
//This is an example class fulfilling the minimal requirements needed by the
//D0 RunI cone jet algorithm implementation, which is an inlined template class
 
// This file is distributed with FastJet under the terms of the GNU
// General Public License (v2). Permission to do so has been granted
// by Lars Sonnenschein and the D0 collaboration (see COPYING for
// details)
//
// History of changes in FastJet compared tothe original version of
// HepEntity.h
//
// 2011-12-13  Gregory Soyez  <soyez@fastjet.fr>
// 
//        * added license information
//
// 2011-11-14  Gregory Soyez  <soyez@fastjet.fr>
//
//         * removed some harmless warnings coming with the -Wshadow gcc option
// 
// 2011-10-06  Gregory Soyez  <soyez@fastjet.fr>
//
//        * put the code in the fastjet::d0runi namespace
 
class HepEntityI {
 
 public:
 
  HepEntityI() {
    Et=0.;
    eta=0.;
    phi=0.;
    index = -1;
    return;
  }
 
 
  HepEntityI(double E_in, double px_in, double py_in, double pz_in,
             int index_in = -1) : index(index_in) {
    //Snowmass Et scheme    
    double pt = sqrt(px_in*px_in+py_in*py_in);
    double p = sqrt(pt*pt+pz_in*pz_in);
    phi = inline_maths::phi(px_in,py_in);
    double theta = asin(pt/p);
    eta = inline_maths::eta(theta);
 
    Et = E_in*sin(theta);
    
    return;
  }
 
 
 
   HepEntityI(const HepEntityI& in) : Et(in.Et), eta(in.eta), phi(in.phi), index(in.index) {
    return;
  }
 
 
 
  
  inline double pT() const {
    return Et;
  }
 
  inline double px() const {
    return Et*cos(phi);
  }
 
  inline double py() const {
    return Et*sin(phi);
  }
 
  inline double pz() const {
    return Et*sinh(eta);
  }
  
  inline double E() const {
    return Et*cosh(eta);
  }
 
  
  inline void p4vec(float* p) const {
    p[0] = Et*cos(phi);
    p[1] = Et*sin(phi);
    p[2] = Et*sinh(eta);
    p[3] = Et*cosh(eta); //E
    return;
  }
  
 
  inline void Add(const HepEntityI el) {
    //assumes Et, eta and phi stored accurately
    double w2 = el.Et;
    Et += el.Et;
    w2 /= Et;
    
    eta += w2*(el.eta - eta);
    phi += w2*inline_maths::delta_phi(el.phi, phi); 
 
    return; 
  }
 
 
  inline void Fill(double E_in, double px_in, double py_in, double pz_in, int index_in) {
    double pt = sqrt(px_in*px_in+py_in*py_in);
    double p = sqrt(pt*pt+pz_in*pz_in);
    phi = inline_maths::phi(px_in,py_in);
    double theta = asin(pt/p);
    eta = inline_maths::eta(theta);
    
    Et = E_in*sin(theta);
 
    index = index_in;
    
    return;
  }
 
 
  double Et;
  double eta;
  double phi;
  int index;
 
 private:
 
 
 
};
//end of class HepEntityI;
 
} // end of namespace d0runi
 
FASTJET_END_NAMESPACE
 
#endif