opt

[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliEmcalClusTrackMatcherTask.cxx

// $Id$
//
// Track/cluster matcher
// 
// Author: C.Loizides, S.Aiola

#include <TClonesArray.h>

#include "AliVCluster.h"
#include "AliVTrack.h"
#include "AliPicoTrack.h"
#include "AliEmcalParticle.h"
#include "AliLog.h"

#include "AliEmcalClusTrackMatcherTask.h"

ClassImp(AliEmcalClusTrackMatcherTask)

//________________________________________________________________________
AliEmcalClusTrackMatcherTask::AliEmcalClusTrackMatcherTask() : 
  AliAnalysisTaskEmcal("AliEmcalClusTrackMatcherTask"),
  fDoClusTrack(1),
  fDoTrackClus(0),
  fMaxDistance(0.1)
{
  // Constructor.
}

//________________________________________________________________________
AliEmcalClusTrackMatcherTask::AliEmcalClusTrackMatcherTask(const char *name) : 
  AliAnalysisTaskEmcal(name),
  fDoClusTrack(1),
  fDoTrackClus(0),
  fMaxDistance(0.1)
{
  // Standard constructor.
}

//________________________________________________________________________
AliEmcalClusTrackMatcherTask::~AliEmcalClusTrackMatcherTask()
{
  // Destructor.
}

//________________________________________________________________________
Bool_t AliEmcalClusTrackMatcherTask::Run() 
{
  // Run the matching for the selected options.
  
  if (fDoClusTrack) 
    DoMatching(fCaloClusters, fTracks);

  if (fDoTrackClus) 
    DoMatching(fTracks, fCaloClusters);
  
  return kTRUE;
}

//________________________________________________________________________
void AliEmcalClusTrackMatcherTask::DoMatching(TClonesArray *array1, TClonesArray *array2) 
{
  // Do the actual matching.

  if (!array1 || !array2)
    return;

  const Double_t maxd2 = fMaxDistance*fMaxDistance;

  const Int_t n1 = array1->GetEntries();
  const Int_t n2 = array2->GetEntries();
  for (Int_t i = 0; i < n1; ++i) {
    AliEmcalParticle *part1 = static_cast<AliEmcalParticle*>(array1->At(i));
    if (!part1)
      continue;
    if (!AcceptEmcalPart(part1))
      continue;

    AliVCluster *cluster1 = part1->GetCluster();
    AliVTrack   *track1   = part1->GetTrack()  ;

    part1->ResetMatchedObjects();

    for (Int_t j = 0; j < n2; ++j) {

      AliEmcalParticle *part2 = static_cast<AliEmcalParticle*>(array2->At(j));
      if (!part2)
	continue;
      if (!AcceptEmcalPart(part2))
        continue;

      AliVCluster *cluster2 = part2->GetCluster();
      AliVTrack   *track2   = part2->GetTrack()  ;

      Double_t deta = 999;
      Double_t dphi = 999;
      if (track1 && cluster2)
	AliPicoTrack::GetEtaPhiDiff(track1, cluster2, dphi, deta);
      else if (track2 && cluster1)
	AliPicoTrack::GetEtaPhiDiff(track2, cluster1, dphi, deta);
      else
	continue;

      Double_t d2 = deta * deta + dphi * dphi;
      if (d2 < maxd2) 
	part1->AddMatchedObj(j, TMath::Sqrt(d2));
    }
    
    if (part1->GetNumberOfMatchedObj() > 0) {
      if (track1) {
	track1->SetEMCALcluster(part1->GetMatchedObjId());
      } else if (cluster1) {
	const UInt_t matchedId = part1->GetMatchedObjId();
	Double_t deta = 999;
	Double_t dphi = 999;
	AliEmcalParticle *part2 = static_cast<AliEmcalParticle*>(array2->At(matchedId));
	AliVTrack *track2 = 0;
        if (part2)
          track2 = part2->GetTrack();
	if (track2) {
	  AliPicoTrack::GetEtaPhiDiff(track2, cluster1, dphi, deta);
	  cluster1->SetEmcCpvDistance(matchedId);
	  cluster1->SetTrackDistance(deta, dphi);
	}
      }
    }
  }
}