PndEventCheck.h

Go to the documentation of this file.

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

#pragma once

#include "PndParticleRequirements.h"
#include "TVector3.h"
#include "TVector2.h"
#include "PndEventRequirements.h"
#include "TClonesArray.h"
#include <iostream>
#include <vector>

class PndEventCheck {
 public:
  PndEventCheck();
  PndEventCheck(PndEventRequirements &requirements);

  virtual ~PndEventCheck();

  void Print(std::ostream &out = std::cout);

  void InitRequirements() { fEventRequirements.InitRequirements(); }

  friend std::ostream &operator<<(std::ostream &out, PndEventCheck &event)
  {
    event.Print(out);
    return out;
  }
  void SetRequirements(PndEventRequirements &req) { fEventRequirements = req; }
  PndEventRequirements GetEventRequirements() const { return fEventRequirements; }

  std::vector<std::vector<int>> CheckEvent(TClonesArray *mcTrackInfo);
  std::vector<std::vector<int>> GetEventResult() const { return fEventResult; }

 protected:
  int CheckElectronPair(PndMCTrackInfo *trackInfo, TClonesArray *mcTrackInfos);

 private:
  PndEventRequirements fEventRequirements;
  std::vector<std::vector<int>> fEventResult;

  ClassDef(PndEventCheck, 1);
};