CbmMCMatchSelectorTask.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmMCMatchSelectorTask.h

// -------------------------------------------------------------------------
// -----                 PNDMCMATCHSELECTORTASK header file             -----
// -----                  Created 18/01/10  by T.Stockmanns             -----
// -------------------------------------------------------------------------


#ifndef CBMMCMATCHSELECTORTASK_H
#define CBMMCMATCHSELECTORTASK_H


// framework includes
#include "FairTask.h"
//#include "CbmMCMatch.h"
#include "CbmDetectorList.h"


#include <vector>
#include <list>

class CbmMCMatch;
class TClonesArray;

class CbmMCMatchSelectorTask : public FairTask
{
 public:

  CbmMCMatchSelectorTask();

  CbmMCMatchSelectorTask(DataType start, DataType stop);

  virtual ~CbmMCMatchSelectorTask();


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


  virtual void Exec(Option_t* opt);

  virtual void Finish();

  virtual void SetStart(DataType type){fStart = type;}
  virtual void SetStop(DataType type){fStop = type;}

  virtual void SetAllWeights(Float_t weight) {fCommonWeight = weight;}
  virtual void SetWeightStage(Int_t type, Float_t weight){
    fStageWeights.push_back(std::pair<DataType, Float_t>(static_cast<DataType>(type), weight));
  }

  virtual void SetWeights();



 private:
  CbmMCMatch* fMCMatch;
  DataType fStart;
  DataType fStop;

  std::vector<std::pair<DataType, Float_t> > fStageWeights;
  Float_t fCommonWeight;

  void Register();

  void Reset();

  void ProduceHits();

  CbmMCMatchSelectorTask(const CbmMCMatchSelectorTask&);
  CbmMCMatchSelectorTask& operator=(const CbmMCMatchSelectorTask&);

  ClassDef(CbmMCMatchSelectorTask,1);

};

#endif