CbmMCObject.h

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

/*
 * CbmMCStage.h
 *
 *  Created on: Dec 1, 2009
 *      Author: stockman
 */

#ifndef CBMMCOBJECT_H_
#define CBMMCOBJECT_H_

#include "CbmDetectorList.h"
#include "CbmMCEntry.h"

#include "TObject.h"

#include <vector>
#include <iostream>

class FairMultiLinkedData;

class CbmMCObject: public TObject {
 public:
  CbmMCObject();
 CbmMCObject(DataType type):TObject(), fStage(), fStageId(type){};
 CbmMCObject(const CbmMCObject& obj)
   : TObject(), fStage(obj.GetEntryVector()), fStageId(obj.GetStageId())
    {};
 CbmMCObject(DataType type, std::vector<CbmMCEntry> stage)
   : TObject(),
    fStage(stage),
    fStageId(type)
  {
  };
  virtual ~CbmMCObject();

  void SetStageId(DataType type){fStageId = type;}

  void SetEntry(std::vector<Int_t> type, std::vector<Int_t> link, int index);
  void SetEntry(FairMultiLinkedData* data, int index);
  void SetEntry(CbmMCEntry link);

  void SetStage(std::vector<CbmMCEntry> stage) {fStage = stage;}
  void SetLink(FairLink link, int index);
  void AddLink(FairLink link, int index);

  CbmMCEntry GetEntry(int index) const {return fStage[index];}
  FairLink  GetSingleLink(int entryIndex, int linkIndex) const {
    return fStage.at(entryIndex).GetLink(linkIndex);
  }

  DataType GetStageId(void) const {return fStageId;}

  CbmMCEntry GetMCLink(Int_t index){return fStage.at(index);}

  int GetNEntries() const {return fStage.size();}
  int GetNLinks(int entryIndex)const {return fStage.at(entryIndex).GetNLinks();}
  std::vector<CbmMCEntry> GetEntryVector() const{return fStage;}

//  FairMultiLinkedData PosInList(std::pair<int, int> link);
  FairMultiLinkedData PosInList(FairLink link);

  FairMultiLinkedData GetLinksWithType(DataType type){
    FairMultiLinkedData result;
    for (int i = 0; i < GetNEntries(); i++){
      result.AddLinks(GetMCLink(i).GetLinksWithType(type), false);
    }
    return result;
  }

  virtual void ClearEntries(){fStage.clear();}

  virtual void Print(std::ostream& out = std::cout){out << *this;}

  CbmMCObject& operator=(const CbmMCObject& obj){
    TObject::operator=(obj);
    fStageId = obj.GetStageId();
    fStage = obj.GetEntryVector();
          return *this;
  }

  friend std::ostream& operator<< (std::ostream& out, const CbmMCObject& obj){
    std::vector<CbmMCEntry> stages = obj.GetEntryVector();
    for (int i = 0; i < stages.size(); i++){
      out << i << ": ";
      stages[i].Print(out);
    }
    return out;
  }

private:
  void AdoptSize(int index);
  std::vector<CbmMCEntry> fStage;
  DataType fStageId;



  ClassDef(CbmMCObject, 0);
};

#endif /* PNDMCOBJECT_H_ */