splitcalCluster Class Reference

#include <splitcalCluster.h>

Inheritance diagram for splitcalCluster:

Collaboration diagram for splitcalCluster:

Public Member Functions
	splitcalCluster ()
	splitcalCluster (splitcalHit *h)
virtual	~splitcalCluster ()
virtual void	Print () const
void	SetEtaPhiE (double &eta, double &phi, double &e)
void	SetEta (double &eta)
void	SetPhi (double &phi)
void	SetEnergy (double &e)
void	SetIndex (int i)
void	SetStartPoint (const double &x, const double &y, const double &z)
void	SetStartPoint (splitcalHit *&h)
void	SetEndPoint (const double &x, const double &y, const double &z)
void	SetEndPoint (splitcalHit *&h)
void	SetVectorOfHits (std::vector< splitcalHit * > &v)
void	AddHit (splitcalHit *h)
int	GetIndex ()
double	GetEta ()
double	GetPhi ()
double	GetEnergy ()
double	GetPx ()
double	GetPy ()
double	GetPz ()
double	GetEx ()
double	GetEy ()
double	GetEz ()
TVector3	GetStartPoint ()
TVector3	GetEndPoint ()
std::vector< splitcalHit * > &	GetVectorOfHits ()
regression	LinearRegression (std::vector< double > &x, std::vector< double > &y)
void	ComputeEtaPhiE ()
double	GetSlopeZX ()
double	GetInterceptZX ()
double	GetSlopeZY ()
double	GetInterceptZY ()

Private Member Functions
	splitcalCluster (const splitcalCluster &cluster)
splitcalCluster	operator= (const splitcalCluster &cluster)
	ClassDef (splitcalCluster, 1)

Private Attributes
int	_index
double	_eta
double	_phi
double	_energy
TVector3	_start
TVector3	_end
std::vector< splitcalHit * >	_vectorOfHits
double	_mZX
double	_qZX
double	_mZY
double	_qZY

Detailed Description

Definition at line 25 of file splitcalCluster.h.

Constructor & Destructor Documentation

splitcalCluster() [1/3]

splitcalCluster::splitcalCluster

(

)

Constructors

Definition at line 22 of file splitcalCluster.cxx.

{
}

splitcalCluster() [2/3]

splitcalCluster::splitcalCluster

(

splitcalHit *

h

)

Definition at line 26 of file splitcalCluster.cxx.

{
  _vectorOfHits.push_back(h);
 
  if ( _vectorOfHits.size() == 1){ //first element added to the cluster
    SetStartPoint(h);
    SetEndPoint(h);
  } else {
    if (_start.Z() > h->GetZ()) SetStartPoint(h); // start point is the hit with the smalllest z
    if (_end.Z() < h->GetZ()) SetEndPoint(h); // end point is the hit with the largest z
  }
 
}

~splitcalCluster()

splitcalCluster::~splitcalCluster ( )

virtual

Destructor

Definition at line 221 of file splitcalCluster.cxx.

{ }

splitcalCluster() [3/3]

splitcalCluster::splitcalCluster ( const splitcalCluster &  cluster )

private

Copy constructor

Member Function Documentation

AddHit()

void splitcalCluster::AddHit ( splitcalHit *  h )

inline

Definition at line 50 of file splitcalCluster.h.

{_vectorOfHits.push_back(h);}

ClassDef()

splitcalCluster::ClassDef ( splitcalCluster  , 1    )

private

ComputeEtaPhiE()

void splitcalCluster::ComputeEtaPhiE

(

)

Definition at line 41 of file splitcalCluster.cxx.

{
  
  
  // Compute energy weighted average for hits in the same layer 
  // This is in preparation of the linear fit to get the cluster eta and phi 
 
  // maps to copute the weighted average of all the hits in the same layer
  std::map<int, double> mapLayerWeigthedX;
  std::map<int, double> mapLayerWeigthedY;
  std::map<int, double> mapLayerZ1;
  std::map<int, double> mapLayerZ2;
  std::map<int, double> mapLayerSumWeigthsX;
  std::map<int, double> mapLayerSumWeigthsY;
 
  // loop over hits to compute cluster energy sum and to compute the coordinates weighted average per layer
  double energy = 0.;
  for (auto hit : _vectorOfHits){
    energy += hit->GetEnergyForCluster(_index);
    int layer = hit->GetLayerNumber();
    // hits from high precision layers give both x and y coordinates --> use if-if instead of if-else
    if (hit->IsX()){
      if (mapLayerWeigthedX.count(layer)==0) { //if key is not yet in map, initialise element to 0
    mapLayerWeigthedX[layer] = 0.; 
    mapLayerSumWeigthsX[layer] = 0.;
      }
      mapLayerWeigthedX[layer] += hit->GetX()*hit->GetEnergyForCluster(_index);
      mapLayerSumWeigthsX[layer] += hit->GetEnergyForCluster(_index);
      mapLayerZ1[layer] = hit->GetZ();
    }
    if (hit->IsY()){ 
      if (mapLayerWeigthedY.count(layer)==0) { //if key is not yet in map, initialise element to 0
    mapLayerWeigthedY[layer] = 0.; 
    mapLayerSumWeigthsY[layer] = 0.;
      }
      mapLayerWeigthedY[layer] += hit->GetY()*hit->GetEnergyForCluster(_index);
      mapLayerSumWeigthsY[layer] += hit->GetEnergyForCluster(_index);
      mapLayerZ2[layer] = hit->GetZ();
    }
  }//end loop on hit
 
 
  // FIXME: regression fit seems instable --> for the moment commented it out in favour of simple direction from initial and end point 
 
  auto const& firstElementX = mapLayerWeigthedX.begin();
  int minLayerX = firstElementX->first;
  double minX = firstElementX->second/mapLayerSumWeigthsX[minLayerX];
  double minZ1 = mapLayerZ1[minLayerX];
 
  auto const& firstElementY = mapLayerWeigthedY.begin();
  int minLayerY = firstElementY->first;
  double minY = firstElementY->second/mapLayerSumWeigthsY[minLayerY];
  double minZ2 = mapLayerZ1[minLayerY];
  
  double minZ = (minZ1+minZ2)/2.;
  
  SetStartPoint(minX, minY, minZ);
 
  auto const& lastElementX = mapLayerWeigthedX.rbegin();
  int maxLayerX = lastElementX->first;
  double maxX = lastElementX->second/mapLayerSumWeigthsX[maxLayerX];
  double maxZ1 = mapLayerZ1[maxLayerX];
 
  auto const& lastElementY = mapLayerWeigthedY.rbegin();
  int maxLayerY = lastElementY->first;
  double maxY = lastElementY->second/mapLayerSumWeigthsY[maxLayerY];
  double maxZ2 = mapLayerZ1[maxLayerY];
  
  double maxZ = (maxZ1+maxZ2)/2.;
  
  SetEndPoint(maxX, maxY, maxZ);
 
  // get direction vector from end-strat vector difference
  TVector3 direction;
  direction = _end - _start;
  double eta = direction.Eta();
  double phi = direction.Phi();
  SetEtaPhiE(eta, phi, energy);
 
 
  // // vectors holding the weighted coordinates per layer: x and z for the eta fit, and y and z for the phi fit
  // std::vector<double > x;
  // std::vector<double > z1;
  // std::vector<double > y;
  // std::vector<double > z2;
 
  // for (auto const& element : mapLayerWeigthedX){
  //   int key = element.first;
  //   x.push_back(element.second/mapLayerSumWeigthsX[key]);
  //   z1.push_back(mapLayerZ1[key]);
  // }
 
  // regression resultZX = LinearRegression(z1,x);
  // double alpha = acos(resultZX.slope);
 
  // for (auto const& element : mapLayerWeigthedY){
  //   int key = element.first;
  //   y.push_back(element.second/mapLayerSumWeigthsY[key]);
  //   z2.push_back(mapLayerZ2[key]);
  // }
 
  // regression resultZY = LinearRegression(z2,y);
  // double eta_check = acos(resultZY.slope);
 
  // // regression resultXY = LinearRegression(x,y);
  // // double phi = acos(resultXY.slope);
 
  // std::cout<<" ---- before fit " << std::endl;
  // _start.Print();
  // _end.Print();
 
  // // replace the x and y of start and end points with the re-evaluated values from the corresponding fit
  // _start.SetX( resultZX.slope * _start.Z() + resultZX.intercept ); 
  // _start.SetY( resultZY.slope * _start.Z() + resultZY.intercept ); 
 
  // _end.SetX( resultZX.slope * _end.Z() + resultZX.intercept ); 
  // _end.SetY( resultZY.slope * _end.Z() + resultZY.intercept ); 
  
  // // get direction vector from end-strat vector difference
  // TVector3 direction;
  // direction = _end - _start;
  // double eta = direction.Eta();
  // double phi = direction.Phi();
 
  // std::cout<<" ---- after fit " << std::endl;
  // _start.Print();
  // _end.Print();
 
  // std::cout<<" -- eta = "<< eta << std::endl;
  // std::cout<<" -- eta_check = "<< eta_check << std::endl;
  // std::cout<<" -- phi = "<< phi << std::endl;
  // std::cout<<" -- energy = "<< energy << std::endl;
  
  // SetEtaPhiE(eta, phi, energy);
 
  // // temporary for test
  // _mZX = resultZX.slope;
  // _qZX = resultZX.intercept;
  // _mZY = resultZY.slope;
  // _qZY = resultZY.intercept;
  
  return;
}

GetEndPoint()

TVector3 splitcalCluster::GetEndPoint ( )

inline

Definition at line 63 of file splitcalCluster.h.

{return _end; }

GetEnergy()

double splitcalCluster::GetEnergy ( )

inline

Definition at line 55 of file splitcalCluster.h.

{return _energy;}

GetEta()

double splitcalCluster::GetEta ( )

inline

Definition at line 53 of file splitcalCluster.h.

{return _eta;}

GetEx()

double splitcalCluster::GetEx ( )

inline

Definition at line 59 of file splitcalCluster.h.

{return GetPx();}

GetEy()

double splitcalCluster::GetEy ( )

inline

Definition at line 60 of file splitcalCluster.h.

{return GetPy();}

GetEz()

double splitcalCluster::GetEz ( )

inline

Definition at line 61 of file splitcalCluster.h.

{return GetPz();}

GetIndex()

int splitcalCluster::GetIndex ( )

inline

Definition at line 52 of file splitcalCluster.h.

{return _index;}

GetInterceptZX()

double splitcalCluster::GetInterceptZX ( )

inline

Definition at line 71 of file splitcalCluster.h.

{return _qZX;}

GetInterceptZY()

double splitcalCluster::GetInterceptZY ( )

inline

Definition at line 73 of file splitcalCluster.h.

{return _qZY;}

GetPhi()

double splitcalCluster::GetPhi ( )

inline

Definition at line 54 of file splitcalCluster.h.

{return _phi;}

GetPx()

double splitcalCluster::GetPx ( )

inline

Definition at line 56 of file splitcalCluster.h.

{return _energy*sin(_eta)*cos(_phi);}

GetPy()

double splitcalCluster::GetPy ( )

inline

Definition at line 57 of file splitcalCluster.h.

{return _energy*sin(_eta)*sin(_phi);}

GetPz()

double splitcalCluster::GetPz ( )

inline

Definition at line 58 of file splitcalCluster.h.

{return _energy*cos(_eta);}

GetSlopeZX()

double splitcalCluster::GetSlopeZX ( )

inline

Definition at line 70 of file splitcalCluster.h.

{return _mZX;}

GetSlopeZY()

double splitcalCluster::GetSlopeZY ( )

inline

Definition at line 72 of file splitcalCluster.h.

{return _mZY;}

GetStartPoint()

TVector3 splitcalCluster::GetStartPoint ( )

inline

Definition at line 62 of file splitcalCluster.h.

{return _start; }

GetVectorOfHits()

std::vector< splitcalHit * > & splitcalCluster::GetVectorOfHits ( )

inline

Definition at line 64 of file splitcalCluster.h.

{return _vectorOfHits;}

LinearRegression()

regression splitcalCluster::LinearRegression	(	std::vector< double > &	x,
		std::vector< double > &	y
	)

Definition at line 187 of file splitcalCluster.cxx.

                                                                                         {
 
  const auto n    = x.size();
  const auto s_x  = std::accumulate(x.begin(), x.end(), 0.);
  const auto s_y  = std::accumulate(y.begin(), y.end(), 0.);
  const auto s_xx = std::inner_product(x.begin(), x.end(), x.begin(), 0.);
  const auto s_xy = std::inner_product(x.begin(), x.end(), y.begin(), 0.);
 
  regression result;
 
  result.slope = (n * s_xy - s_x * s_y) / (n * s_xx - s_x * s_x);
  result.intercept = (s_x * s_x * s_y - s_xy * s_x) / (n * s_xx - s_x * s_x);
 
  std::cout<< "--- LinearRegression ---" <<std::endl;
  std::cout<< "--------- slope = " <<  result.slope <<std::endl;
  std::cout<< "--------- intercept = " <<  result.intercept <<std::endl;
 
  return result;
 
}

operator=()

splitcalCluster splitcalCluster::operator= ( const splitcalCluster &  cluster )

private

Print()

void splitcalCluster::Print ( ) const

virtual

Methods

Definition at line 225 of file splitcalCluster.cxx.

{
 
  std::cout<< "-I- splitcalCluster: " <<std::endl;
  std::cout<< "    (eta,phi,energy) = " 
       << _eta << " ,  "
       << _phi << " ,  "
       << _energy << std::endl;  
  std::cout<< "    start(x,y,z) = " 
       << _start.X() << " ,  "
       << _start.Y() << " ,  "
       << _start.Z() << std::endl;  
  std::cout<< "    end(x,y,z) = " 
       << _end.X() << " ,  "
       << _end.Y() << " ,  "
       << _end.Z() << std::endl;  
   std::cout<< "------- " <<std::endl;
}

SetEndPoint() [1/2]

void splitcalCluster::SetEndPoint ( const double &  x, const double &  y, const double &  z  )

inline

Definition at line 47 of file splitcalCluster.h.

{_end.SetXYZ(x,y,z);}

SetEndPoint() [2/2]

void splitcalCluster::SetEndPoint

(

splitcalHit *&

h

)

Definition at line 213 of file splitcalCluster.cxx.

                                                  {
  SetEndPoint(h->GetX(),h->GetY(),h->GetZ());
} 

SetEnergy()

void splitcalCluster::SetEnergy ( double &  e )

inline

Definition at line 43 of file splitcalCluster.h.

{_energy = e;}

SetEta()

void splitcalCluster::SetEta ( double &  eta )

inline

Definition at line 41 of file splitcalCluster.h.

{_eta = eta;}

SetEtaPhiE()

void splitcalCluster::SetEtaPhiE ( double &  eta, double &  phi, double &  e  )

inline

Definition at line 40 of file splitcalCluster.h.

{_eta = eta; _phi = phi; _energy = e;}

SetIndex()

void splitcalCluster::SetIndex ( int  i )

inline

Definition at line 44 of file splitcalCluster.h.

{_index = i;}

SetPhi()

void splitcalCluster::SetPhi ( double &  phi )

inline

Definition at line 42 of file splitcalCluster.h.

{_phi = phi;}

SetStartPoint() [1/2]

void splitcalCluster::SetStartPoint ( const double &  x, const double &  y, const double &  z  )

inline

Definition at line 45 of file splitcalCluster.h.

{_start.SetXYZ(x,y,z); }

SetStartPoint() [2/2]

void splitcalCluster::SetStartPoint

(

splitcalHit *&

h

)

Definition at line 209 of file splitcalCluster.cxx.

                                                   {
  SetStartPoint(h->GetX(),h->GetY(),h->GetZ());
} 

SetVectorOfHits()

void splitcalCluster::SetVectorOfHits ( std::vector< splitcalHit * > &  v )

inline

Definition at line 49 of file splitcalCluster.h.

{_vectorOfHits = v;}

Member Data Documentation

_end

TVector3 splitcalCluster::_end

private

Definition at line 83 of file splitcalCluster.h.

_energy

double splitcalCluster::_energy

private

Definition at line 81 of file splitcalCluster.h.

_eta

double splitcalCluster::_eta

private

Definition at line 81 of file splitcalCluster.h.

_index

int splitcalCluster::_index

private

Definition at line 80 of file splitcalCluster.h.

_mZX

double splitcalCluster::_mZX

private

Definition at line 87 of file splitcalCluster.h.

_mZY

double splitcalCluster::_mZY

private

Definition at line 88 of file splitcalCluster.h.

_phi

double splitcalCluster::_phi

private

Definition at line 81 of file splitcalCluster.h.

_qZX

double splitcalCluster::_qZX

private

Definition at line 87 of file splitcalCluster.h.

_qZY

double splitcalCluster::_qZY

private

Definition at line 88 of file splitcalCluster.h.

_start

TVector3 splitcalCluster::_start

private

Definition at line 82 of file splitcalCluster.h.

_vectorOfHits

std::vector<splitcalHit* > splitcalCluster::_vectorOfHits

private

Definition at line 84 of file splitcalCluster.h.

---

The documentation for this class was generated from the following files:

• splitcal/splitcalCluster.h
• splitcal/splitcalCluster.cxx