ClusterIso.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ClusterIso.h

// This file is really -*- C++ -*-.
#ifndef CLUSTERISO_CLUSTERISO_H
#define CLUSTERISO_CLUSTERISO_H

#include <fun4all/SubsysReco.h>

#include <CLHEP/Vector/ThreeVector.h>

#include <cmath>
#include <string>

class PHCompositeNode;
class RawTowerGeom;

class ClusterIso : public SubsysReco
{
 public:
  ClusterIso(const std::string&, float eTCut, int coneSize, bool do_subtracted, bool do_unsubtracted);

  int Init(PHCompositeNode*) override;
  int process_event(PHCompositeNode*) override;
  int End(PHCompositeNode*) override;

  void seteTCut(float x);
  void setConeSize(int x);
  /*const*/ float geteTCut();
  /*const*/ int getConeSize();
  /*const*/ CLHEP::Hep3Vector getVertex();

 private:
  double getTowerEta(RawTowerGeom* tower_geom, double vx, double vy, double vz);
  float m_eTCut;     
  float m_coneSize;  
  float m_vx;        
  float m_vy;        
  float m_vz;        
  bool m_do_subtracted;
  bool m_do_unsubtracted;
};

inline /*const*/ float deltaR(float eta1, float eta2, float phi1, float phi2)
{
  float deta = eta1 - eta2;
  float dphi = phi1 - phi2;
  if (dphi > M_PI) dphi -= 2 * M_PI;       //corrects to keep range -pi to pi
  if (dphi < -1 * M_PI) dphi += 2 * M_PI;  //corrects to keep range -pi to pi
  return sqrt(deta * deta + dphi * dphi);
}

#endif