AliReducedEmcalCluster.cxx

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
#include <TArrayD.h>
#include <TLorentzVector.h>
#include <TObjArray.h>

#include "AliReducedEmcalCluster.h"

ClassImp(HighPtTracks::AliReducedClusterParticle)
ClassImp(HighPtTracks::AliReducedEmcalCluster)

namespace HighPtTracks {

AliReducedEmcalCluster::AliReducedEmcalCluster() :
  fClusterID(-1),
  fEnergy(-1),
  fEta(-100),
  fPhi(-100),
  fM02(-1),
  fM20(-1),
  fContributors(NULL)
{
  memset(fCellEnergies, 0, sizeof(Float_t) * 3);
}

AliReducedEmcalCluster::AliReducedEmcalCluster(Int_t id, Float_t energy,Float_t eta, Float_t phi, Float_t m02, Float_t m20):
  TObject(),
  fClusterID(id),
  fEnergy(energy),
  fEta(eta),
  fPhi(phi),
  fM02(m02),
  fM20(m20),
  fContributors(NULL)
{
  memset(fCellEnergies, 0, sizeof(Float_t)*3);
}

AliReducedEmcalCluster::AliReducedEmcalCluster(const AliReducedEmcalCluster& ref) :
  TObject(ref),
  fClusterID(ref.fClusterID),
  fEnergy(ref.fEnergy),
  fEta(ref.fEta),
  fPhi(ref.fPhi),
  fM02(ref.fM02),
  fM20(ref.fM20),
  fContributors(NULL)
{
  ref.Copy(*this);
}

AliReducedEmcalCluster& AliReducedEmcalCluster::operator=(const AliReducedEmcalCluster& ref) {
  if(&ref != this){
    this->~AliReducedEmcalCluster();
    TObject::operator=(ref);
    ref.Copy(*this);
  }
  return *this;
}

AliReducedEmcalCluster::~AliReducedEmcalCluster() {
  if(fContributors) delete fContributors;
}

void AliReducedEmcalCluster::Copy(TObject &target) const {
  AliReducedEmcalCluster *targetcluster = dynamic_cast<AliReducedEmcalCluster *>(&target);
  if(!targetcluster) return;
  targetcluster->fClusterID = fClusterID;
  targetcluster->fEnergy = fEnergy;
  targetcluster->fEta = fEta;
  targetcluster->fPhi = fPhi;
  targetcluster->fM02 = fM02;
  targetcluster->fM20 = fM20;
  targetcluster->fContributors = NULL;
  memcpy(targetcluster->fCellEnergies, fCellEnergies, sizeof(Float_t)*3);
  if(fContributors){
    targetcluster->fContributors = new TObjArray;
    targetcluster->fContributors->SetOwner(kTRUE);
    for(TIter contiter = TIter(fContributors).Begin(); contiter != TIter::End(); ++contiter){
      targetcluster->fContributors->Add(new AliReducedClusterParticle(*(static_cast<AliReducedClusterParticle *>(*contiter))));
    }
  }
}

void AliReducedEmcalCluster::FillCellEnergies(TArrayD& target) {
  target.Set(3);
  for(Int_t icell = 0; icell < 3; icell++) target[icell] = fCellEnergies[icell];
}

void AliReducedEmcalCluster::AddTrueContributor(Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t energy) {
  if(!fContributors){
    fContributors = new TObjArray;
    fContributors->SetOwner(kTRUE);
  }
  fContributors->Add(new AliReducedClusterParticle(pdg, px, py, pz, energy));
}

HighPtTracks::AliReducedClusterParticle::AliReducedClusterParticle():
  fPdg(0),
  fEnergy(0)
{
  memset(fPvec, 0, sizeof(Float_t) * 3);
}

HighPtTracks::AliReducedClusterParticle::AliReducedClusterParticle(Int_t pdg,
    Double_t px, Double_t py, Double_t pz, Double_t energy) :
  fPdg(pdg),
  fEnergy(energy)
{
  fPvec[0] = px;
  fPvec[1] = py;
  fPvec[2] = pz;
}

HighPtTracks::AliReducedClusterParticle::~AliReducedClusterParticle() { }

void HighPtTracks::AliReducedClusterParticle::FillLorentzVector(TLorentzVector& target) const {
  target.SetPxPyPzE(fPvec[0],fPvec[1],fPvec[2],fEnergy);
}

} /* namespace HighPtTracks */