PndPidIdealAssociatorTask.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#ifndef PID_IDEALASSOCIATOR_H
#define PID_IDEALASSOCIATOR_H

// **************************************************************************
//  Author: Ralf Kliemt e-mail: r.kliemt@physik.tu-dresden.de
//
//  pid simple classifier
//  Warning - will use Monte-Carlo information
//
// Created: 01-09-09
// Modified:
//
// **************************************************************************

#include "TClonesArray.h"
#include "FairTask.h"

class PndPidCandidate;
class PndPidProbability;

class PndPidIdealAssociatorTask : public FairTask {

 protected:
  // TString fNeutralBranchName;
  TString fChargedBranchName;

  TString fTrackBranchNamePidHypo;

  TClonesArray *fPidChargedCand; 
  // TClonesArray* fPidNeutralCand;    //! PndPidCandidate TCA for neutral particles
  TClonesArray *fPidChargedProb; 
  // TClonesArray* fPidNeutralProb;    //! PndPidProbability TCA for neutral particles
  TClonesArray *fMCTrack; 

  void DoChargedPidMatch(PndPidCandidate *pidcand, PndPidProbability *prob); // cheating for each candidate
  // void DoNeutralPidMatch(PndPidCandidate* pidcand, PndPidProbability* prob); // cheating for each candidate

 public:
  virtual void Exec(Option_t *option);
  virtual InitStatus Init(); //

  void Register();
  void Reset();

  PndPidIdealAssociatorTask(const char *name, const char *title = "Pnd Task");
  PndPidIdealAssociatorTask();
  virtual ~PndPidIdealAssociatorTask();

  void SetToOnlyOne();
  void SetVerbose(Bool_t verb) { fVerbose = verb; };

  virtual void SetParContainers();
  virtual void Finish();

  void SetOutputBranch(TString branch) { fTrackBranchNamePidHypo = branch; };

  ClassDef(PndPidIdealAssociatorTask, 1) // PndPidIdealAssociatorTask
};

#endif