ParticleID.h

Go to the documentation of this file.

// \version
//
// \brief Definition of data product to hold ParticleID information
//
// \author brebel@fnal.gov, tjyang@fnal.gov
//
#ifndef ANAB_PARTICLEID_H
#define ANAB_PARTICLEID_H

#include "larcoreobj/SimpleTypesAndConstants/geo_types.h"
#include "lardataobj/AnalysisBase/ParticleID_VariableTypeEnums.h"
#include <bitset>
#include <iomanip>
#include <iosfwd>
#include <iostream>
#include <vector>

namespace anab {

  struct sParticleIDAlgScores { 
    std::string
      fAlgName; 
    kVariableType
      fVariableType; 
    kTrackDir
      fTrackDir; 
    int
      fNdf; 
    int
      fAssumedPdg; 
    float fValue; 
    std::bitset<8>
      fPlaneMask; 

    sParticleIDAlgScores()
    {
      fAlgName = "AlgNameNotSet";
      fVariableType = kNotSet;
      fTrackDir = kNoDirection;
      fAssumedPdg = 0;
      fNdf = -9999;
      fValue = -9999.;
      // fPlaneMask will use default constructor: sets all values to 0
    }
  };

  class ParticleID {
  private:
    std::vector<sParticleIDAlgScores>
      fParticleIDAlgScores; 
    geo::PlaneID fPlaneID;  

  public:
    ParticleID();
    ParticleID(const std::vector<anab::sParticleIDAlgScores>& ParticleIDAlgScores,
               const geo::PlaneID& planeID);

    friend std::ostream& operator<<(std::ostream& o, ParticleID const& a);

    const std::vector<anab::sParticleIDAlgScores>& ParticleIDAlgScores() const;

    const geo::PlaneID& PlaneID() const;
  };

}

inline const std::vector<anab::sParticleIDAlgScores>& anab::ParticleID::ParticleIDAlgScores() const
{
  return fParticleIDAlgScores;
}

inline const geo::PlaneID& anab::ParticleID::PlaneID() const
{
  return fPlaneID;
}

#endif //ANAB_PARTICLEID_H