MCTruthToDk2NuHackItr.h

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
/// \file  MCTruthToDk2NuHackItr.cxx
/// \brief allow easy iteration over MCTruth and associated GTruth/MCFlux
///
/// \version $Id: $
/// \author  rhatcher@fnal.gov
////////////////////////////////////////////////////////////////////////
#ifndef EVG_MCTRUTHTODK2NUHACKITR_H
#define EVG_MCTRUTHTODK2NUHACKITR_H

#include <vector>
#include <utility>   // for pair<>
#include <set>

#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Principal/Run.h"
#include "art/Framework/Principal/SubRun.h"

#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include "nusimdata/SimulationBase/MCTruth.h"
#include "nusimdata/SimulationBase/MCFlux.h"
#include "nusimdata/SimulationBase/GTruth.h"

#include "dk2nu/tree/dk2nu.h"
#include "dk2nu/tree/NuChoice.h"

namespace evg {

  class MCTruthToDk2NuHackItr {

  public:
    MCTruthToDk2NuHackItr(art::Event const & evtIn,
                          std::vector<std::string> const & labels,
                          int verbose=0);

    virtual ~MCTruthToDk2NuHackItr() { ; }

    bool Next();   // move to next
    const simb::MCTruth*   GetMCTruth()  const { return thisMCTruth;  }
    const simb::GTruth*    GetGTruth()   const { return thisGTruth;   }
    const simb::MCFlux*    GetMCFlux()   const { return thisMCFlux;   }
    const bsim::Dk2Nu*     GetDk2Nu()    const { return thisDk2Nu;    }
    const bsim::NuChoice*  GetNuChoice() const { return thisNuChoice; }

    std::string            GetLabel()    const { return thisLabel;    } 
    // return associated label???

    art::Ptr<simb::MCTruth> GetMCTruthPtr() const { return thisMCTruthPtr; }

  private:

    bool pretty_darn_close(double a, double b, double eps_precision=10.0);

    bool match_mctruth_nuchoice(const simb::MCTruth* pmctruth,
                                const bsim::NuChoice* pnchoice);

    art::Event const &                evt;
    std::vector<std::string> const &  fInputModuleLabels;

    std::vector< art::Handle< std::vector<simb::MCTruth> > > mclists;

    std::set<std::pair<int,int> >                 indices;
    std::set<std::pair<int,int> >::const_iterator indx_itr;
    std::vector<std::string>                      outlabels;

    int                               nmctruth;
    int                               imctruth;

    const simb::MCTruth*              thisMCTruth;
    const simb::GTruth*               thisGTruth;
    const simb::MCFlux*               thisMCFlux;
    const bsim::Dk2Nu*                thisDk2Nu;
    const bsim::NuChoice*             thisNuChoice;
    std::string                       thisLabel;

    art::Ptr<simb::MCTruth>           thisMCTruthPtr;

    int                               verbose;

  }; // end-of-class MCTruthToDk2NuHackItr

} // end-of-namespace evg

#endif // EVG_MCTRUTHTODK2NUHACKITR_H