TrainingData.cxx

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// \file    TrainingData.h
/// \brief   The TrainingData objects contains a PixelMap and the
///          output class type, and any other bit that goes into the ANN
// \author   radovic -- a.radovic@gmail.com

//#include "art/Framework/Services/Registry/ServiceHandle.h"

#include "dunereco/CVN/func/TrainingData.h"
//#include "MCCheater/BackTracker.h"

namespace cvn
{

  TrainingData::TrainingData(const InteractionType& interaction,
                             float nuEnergy, float lepEnergy,
                             float nueEnergy, float numuEnergy,
                             float nutauEnergy, float weight,
                             const PixelMap& pMap):
  fInt(interaction),
  fNuEnergy(nuEnergy),
  fLepEnergy(lepEnergy),
  fRecoNueEnergy(nueEnergy),
  fRecoNumuEnergy(numuEnergy),
  fRecoNutauEnergy(nutauEnergy),
  fEventWeight(weight),
  fUseTopology(false),
  fNuPDG(0),
  fNProton(-1),
  fNPion(-1),
  fNPizero(-1),
  fNNeutron(-1),
  fTopologyType(-1),
  fTopologyTypeAlt(-1),
  fPMap(pMap)
  {  }


  void TrainingData::FillOutputVector(float* output) const
  {
    for(unsigned int i = 0; i < kNIntType; ++i)
      output[i] = 0;

    output[fInt] = 1;
  }

  void TrainingData::SetTopologyInformation(int pdg, int nproton,
    int npion, int npizero, int nneutron, int toptype,
    int toptypealt){

    fUseTopology = true;

    fNuPDG = pdg;
    fNProton = nproton;
    fNPion = npion;
    fNPizero = npizero;
    fNNeutron = nneutron;

    fTopologyType = toptype;
    fTopologyTypeAlt = toptypealt;

  }

} // end namespace cvn
////////////////////////////////////////////////////////////////////////