PndEventRequirements.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      PndEventRequirements header file                   -----
// -----                  Created 16/09/19  by T.Stockmann             -----
// -------------------------------------------------------------------------

#pragma once

#include "PndParticleRequirements.h"
#include "TVector3.h"
#include "TVector2.h"
#include <iostream>

class PndEventRequirements {

 public:
  PndEventRequirements();

  virtual ~PndEventRequirements();

  // needs to be called before first use once the FairRootManager is available (in the init of the task using it)
  void InitRequirements();

  friend std::ostream &operator<<(std::ostream &out, PndEventRequirements &track)
  {
    int i = 0;
    for (auto req : track.fTrackRequirements) {
      out << i++ << " : " << req << std::endl;
    }
    return out;
  }

  void AddTrackRequirement(PndParticleRequirements val)
  {
    if (val.GetCheckTracking() == true)
      fChargedTracks.push_back(fTrackRequirements.size());
    if (val.GetCheckNeutral() == true)
      fNeutralTracks.push_back(fTrackRequirements.size());
    fTrackRequirements.push_back(val);
  }

  std::vector<PndParticleRequirements> GetRequirements() { return fTrackRequirements; }
  int GetNRequirements() const { return fTrackRequirements.size(); }
  int GetNParticles() { return fTrackRequirements.size(); }
  std::vector<int> GetChargedRequirements() const { return fChargedTracks; }
  std::vector<int> GetNeutralRequirements() const { return fNeutralTracks; }
  void ClearFound()
  {
    for (auto &track : fTrackRequirements)
      track.ClearFound();
  }

 protected:

 private:
  std::vector<PndParticleRequirements> fTrackRequirements;
  std::vector<int> fChargedTracks;  // index to requirement of charged particle
  std::vector<int> fNeutralTracks;  // index to requirement of neutral particle

  ClassDef(PndEventRequirements, 1);
};