PndUnassignedHitsTask.h

Go to the documentation of this file.

/*
 * PndUnassignedHitsTask.h
 *
 *  Created on: Nov 22, 2018
 *      Author: Tobias Stockmanns <t.stockmanns@fz-juelich.de>
 */

#ifndef PndUnassignedHitsTask_H_
#define PndUnassignedHitsTask_H_

#include "PndPersistencyTask.h"

#include "FairMultiLinkedData.h"

#include <map>

class TClonesArray;
class FairMultiLinkedData;

class PndUnassignedHitsTask : public PndPersistencyTask {
 public:
  PndUnassignedHitsTask() : PndPersistencyTask("Unassigned Hits Task"){};
  virtual ~PndUnassignedHitsTask(){};

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

  virtual void Exec(Option_t *opt);

  virtual void AddHitBranch(TString branchName);
  virtual void AddTrackCands(TString trackCandName);

  void SetVerbose(Int_t verbose) { fVerbose = verbose; };

 protected:
  void RegisterBranches();
  void RegisterTrackCands();
  void FillTrackCands();
  void FillUnassignedHits(TString branchName);
  void FillOutputBranch(TString branchName, std::vector<int> unusedHits);

 private:
  std::map<TString, TClonesArray *> fHitBranches;           
  std::map<TString, TClonesArray *> fUnassignedHitBranches; 
  std::map<TString, TClonesArray *> fTrackCands;            
  FairMultiLinkedData fHitsInTracks;                        

  ClassDef(PndUnassignedHitsTask, 1);
};

#endif /* PndUnassignedHitsTask_H_ */