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(0), fFilteredTracks(0), fRemoveTrack(kFALSE){SetPersistency(kTRUE); };
        PndMissingPzCleanerTask(const TString& branchName)
                : PndPersistencyTask("Missing Pz Track Cleaner Task"), fInputTrackBranch(branchName), fOutputTrackBranch(""),
                  fTracks(0), fFilteredTracks(0), 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_ */