LArTrainingData.h

Go to the documentation of this file.

// \file    TrainingData.h
// \author   radovic -- a.radovic@gmail.com
#ifndef LCVN_TRAININGDATA_H
#define LCVN_TRAININGDATA_H

#include <iostream>
#include <ostream>

#include "larrecodnn/CVN/func/AssignLabels.h"
#include "larrecodnn/CVN/func/InteractionType.h"
#include "larrecodnn/CVN/func/PixelMap.h"

namespace lcvn {

  class TDNuInfo {
  public:
    TDNuInfo();

    void SetTruthInfo(float nuEnergy, float lepEnergy, float lepAngle, float weight);
    void SetRecoInfo(float nueEnergy, float numuEnergy, float nutauEnergy);

    // Set topology information separately to save having a large number of
    // arguments in the constructor.
    void SetTopologyInformation(int pdg,
                                int nproton,
                                int npion,
                                int npizero,
                                int nneutron,
                                int toptype,
                                int toptypealt);

    friend std::ostream& operator<<(std::ostream& os, const TDNuInfo& td);

  private:
    float fNuEnergy;    
    float fLepEnergy;   
    float fLepAngle;    
    float fEventWeight; 

    float fRecoNueEnergy;   
    float fRecoNumuEnergy;  
    float fRecoNutauEnergy; 

    // If we are using topology information, store it here
    bool fUseTopology;
    int fNuPDG;
    int fNProton;
    int fNPion;
    int fNPizero;
    int fNNeutron;
    int fTopologyType;
    int fTopologyTypeAlt;
  };


  template <class T>
  class LArTrainingData {

  public:
    LArTrainingData(){};
    LArTrainingData(const InteractionType& interaction, const PixelMap& pMap, const T info);

    unsigned int NOutput() const { return (unsigned int)kNIntType; };

    void FillOutputVector(float* output) const;

    InteractionType fInt; 
    PixelMap fPMap;       
    T fInfo;
  };

  typedef LArTrainingData<lcvn::TDNuInfo> LArTrainingNuData;
} // end namespace

#endif // CVN_TRAININGDATA_H