AtTrack.cxx

Go to the documentation of this file.

#include "AtTrack.h"
 
#include "AtPattern.h"
 
#include <Math/Vector3D.h> // for DisplacementVector3D
#include <Rtypes.h>
 
#include <iterator>
#include <numeric>
 
constexpr auto cRED = "\033[1;31m";
constexpr auto cYELLOW = "\033[1;33m";
constexpr auto cNORMAL = "\033[0m";
constexpr auto cGREEN = "\033[1;32m";
 
ClassImp(AtTrack);
 
AtTrack &AtTrack::operator=(AtTrack obj)
{
   swap(*this, obj);
   return *this;
}
 
AtTrack::AtTrack(const AtTrack &o)
   : fTrackID(o.fTrackID), fIsMerged(o.fIsMerged), fVertexToZDist(o.fVertexToZDist), fGeoThetaAngle(o.fGeoThetaAngle),
     fGeoPhiAngle(o.fGeoPhiAngle), fGeoRadius(o.fGeoRadius), fGeoCenter(o.fGeoCenter),
     fHitClusterArray(o.fHitClusterArray)
{
   fPattern = (o.fPattern != nullptr) ? o.fPattern->Clone() : nullptr;
   for (auto &hit : o.fHitArray)
      fHitArray.push_back(hit->Clone());
}
 
void AtTrack::AddClusterHit(std::shared_ptr<AtHitCluster> hitCluster)
{
   fHitClusterArray.push_back(std::move(*hitCluster));
}
 
AtTrack::XYZPoint AtTrack::GetLastPoint()
{
   Double_t maxR = 0.;
   XYZPoint maxPos;
   for (auto &nHit : fHitArray) {
      auto temp = nHit->GetPosition();
      if (temp.Rho() > maxR) {
         maxR = temp.Rho();
         maxPos = temp;
      }
   }
   return maxPos;
}
 
AtTrack::XYZPoint AtTrack::GetFirstPoint()
{
   Double_t minR = 999.;
   XYZPoint minPos;
   for (auto &nHit : fHitArray) {
      auto temp = nHit->GetPosition();
      if (temp.Rho() < minR) {
         minR = temp.Rho();
         minPos = temp;
      }
   }
   return minPos;
}
 
Double_t AtTrack::GetMeanTime()
{
   Double_t meanTime = 0.0;
 
   if (fHitArray.size() > 0) {
      Int_t sum = std::accumulate(begin(fHitArray), end(fHitArray), 0,
                                  [](int i, HitPtr &hit) { return hit->GetTimeStamp() + i; });
      return sum / (Double_t)fHitArray.size();
   } else
      return meanTime;
}
 
Double_t AtTrack::GetLinearRange()
{
   if (fHitArray.size() > 0) {
      AtHit fhit = *fHitArray.front(); // Last hit of the track (Low TB)
      AtHit lhit = *fHitArray.back();  // First hit of the track (High TB)
      return GetLinearRange(fhit.GetPosition(), lhit.GetPosition());
   } else
      return 0;
}
 
Double_t AtTrack::GetLinearRange(XYZPoint vertex)
{
   if (fHitArray.size() > 0) {
      AtHit fhit = *fHitArray.front();
      return GetLinearRange(fhit.GetPosition(), vertex);
   } else
      return 0;
}
 
Double_t AtTrack::GetLinearRange(const XYZPoint &vertex, const XYZPoint &maxPos)
{
   return (vertex - maxPos).R();
}
 
Double_t AtTrack::GetGeoQEnergy()
{
 
   Double_t charge = 0;
 
   if (fHitArray.size() > 0) {
      return std::accumulate(begin(fHitArray), end(fHitArray), 0.0,
                             [](double i, const HitPtr &hit) { return hit->GetCharge() + i; });
   } else
      return -10.0;
}
 
void AtTrack::SortHitArrayTime()
{
   std::sort(fHitArray.begin(), fHitArray.end(), AtHit::SortHitTimePtr);
}
 
void AtTrack::SortClusterHitArrayZ()
{
 
   std::sort(fHitClusterArray.begin(), fHitClusterArray.end(), SortClusterHitZ);
}