PndMissingPzCleanerTask.h

Go to the documentation of this file.

/*
 * PndMissingPzCleanerTask.h
 *
 * \brief Task to flag or to remove tracks with missing Pz information
 *
 * This task takes as an input any PndTrack branch. It goes through all the tracks. For those tracks only consisting of STTHits
 * and with a ParamFirst::pz value between 0.015 and 0.025 it either sets the PndTrack::fFlag to -30 or deletes the track.
 * What action is taken is selected by SetRemoveTrack(). If SetRemoveTrack is set to kTRUE the track is deleted otherwise flaged.
 * The filtered PndTracks are written in an output branch either with a given name by SetOutputTrackBranch() of, if no branch name is
 * given by InputTrackBranchName + "_filtered".
 *
 *  Created on: 22.3.2017
 *      Author: t.stockmanns
 */

#ifndef PndMissingPzCleanerTask_H_
#define PndMissingPzCleanerTask_H_

#include "PndPersistencyTask.h"
#include "TString.h"

class TClonesArray;

class PndMissingPzCleanerTask : public PndPersistencyTask {
 public:
  PndMissingPzCleanerTask()
    : PndPersistencyTask("Missing Pz Track Cleaner Task"), fInputTrackBranch(""), fOutputTrackBranch(""), fTracks(nullptr), fFilteredTracks(nullptr), fRemoveTrack(kFALSE)
  {
    SetPersistency(kTRUE);
  };
  PndMissingPzCleanerTask(const TString &branchName)
    : PndPersistencyTask("Missing Pz Track Cleaner Task"), fInputTrackBranch(branchName), fOutputTrackBranch(""), fTracks(nullptr), fFilteredTracks(nullptr), fRemoveTrack(kFALSE)
  {
    SetPersistency(kTRUE);
  };
  virtual ~PndMissingPzCleanerTask(){};

  virtual void SetParContainers();
  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);
  virtual void FinishEvent();
  virtual void Finish();

  void SetInputTrackBranch(const TString &name) { fInputTrackBranch = name; }
  void SetOutputTrackBranch(const TString &name) { fOutputTrackBranch = name; }
  void SetRemoveTrack(const Bool_t &remove) { fRemoveTrack = remove; }
  void SetPersistence(const Bool_t &val) { SetPersistency(val); }

 private:
  TString fInputTrackBranch;
  TString fOutputTrackBranch;

  TClonesArray *fTracks;
  TClonesArray *fFilteredTracks;

  Bool_t fRemoveTrack; 

  ClassDef(PndMissingPzCleanerTask, 1);
};

#endif /* PndMissingPzCleanerTask_H_ */