TrainingData.h
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////
2 // \file TrainingData.h
3 /// \brief The TrainingData objects contains a PixelMap and the
4 /// output class type, and any other bit that goes into the ANN
5 // \author rocco -- rocco@physics.umn.edu
6 ////////////////////////////////////////////////////////////////////////
7 #ifndef CVN_TRAININGDATA_H
8 #define CVN_TRAININGDATA_H
9 
15 #include "CVN/func/ParticlesType.h"
16 namespace cvn
17 {
18 
19 
20  /// \brief The TrainingData objects contains a PixelMap and the
21  /// output class type, and any other bit that goes into the ANN
22 
24  {
25 
26  public:
28  TrainingData(const InteractionType& interaction,
29  const ParentParticleType& parentparticle,
30  const FinalStateType& finalstate,
31  const FinalStateProngsType& finalstateprongs,
32  const ParticlesType& finalstatelong,
33  float nuEnergy, float lepEnergy,
34  double vtxx, double vtxy, double vtxz,
35  const PixelMap& pMap);
36 
37  TrainingData(const InteractionType& interaction,
38  const FinalStateType& finalstate,
39  const FinalStateProngsType& finalstateprongs,
40  float nuEnergy, float lepEnergy,
41  const PixelMap& pMap);
42 
43  unsigned int NOutput() const {return (unsigned int)kNIntType;};
44  unsigned int NPPtOutput() const {return (unsigned int)kNPPtType;};
45  unsigned int NFStOutput() const {return (unsigned int)kNFStType;};
46  unsigned int NFSpOutput() const {return (unsigned int)kNFSpType;};
47  unsigned int NSimpleOutput() const
48  {return (unsigned int) kNSimpleNeutrinoType;};
49 
50 
51 
52  void FillOutputVector(float* output) const;
53  void FillPPtOutputVector(float* output) const;
54  void FillFStOutputVector(float* output) const;
55  void FillFSpOutputVector(float* output) const;
56  void FillFSlongOutputVector(float* output) const;
57  void FillSimpleOutputVector(float* output) const;
58 
59  ParentParticleType fPPt; ///< Class of the event by parent particle
60  InteractionType fInt; ///< Class of the event
61  FinalStateType fFSt; ///< Class of the event by final state
62  FinalStateProngsType fFSp; ///< Class of the event by primary n prongs
63  ParticlesType fFSlong; ///< Class of the event by primary n prongs
64  float fNuEnergy; ///< True energy of neutrino event
65  float fLepEnergy; ///< True energy of outgoing lepton
66  double fVtxX; ///< True vertex of X position
67  double fVtxY; ///< True vertex of Y position
68  double fVtxZ; ///< True vertex of Z position
69  PixelMap fPMap; ///< PixelMap for the event
70  };
71 
72 } // end namespace
73 
74 #endif // CVN_TRAININGDATA_H
75 //////////////////////////////////////////////////////////////////////////////
ofstream output
Number of Parent Particle types, used like a vector size.
InteractionType fInt
Class of the event.
Definition: TrainingData.h:60
void FillFStOutputVector(float *output) const
unsigned int NSimpleOutput() const
Definition: TrainingData.h:47
unsigned int NOutput() const
Definition: TrainingData.h:43
void FillPPtOutputVector(float *output) const
Number of interaction types, used like a vector size.
ParticlesType fFSlong
Class of the event by primary n prongs.
Definition: TrainingData.h:63
float vtxx
enum cvn::Interaction InteractionType
Defines an enumeration for prong classification.
void FillOutputVector(float *output) const
PixelMap fPMap
PixelMap for the event.
Definition: TrainingData.h:69
double fVtxY
True vertex of Y position.
Definition: TrainingData.h:67
void FillFSpOutputVector(float *output) const
float fLepEnergy
True energy of outgoing lepton.
Definition: TrainingData.h:65
float fNuEnergy
True energy of neutrino event.
Definition: TrainingData.h:64
Number of interaction types, used like a vector size.
float vtxy
enum cvn::FinalState FinalStateType
unsigned int NFStOutput() const
Definition: TrainingData.h:45
enum cvn::Particles ParticlesType
Number of interaction types, used like a vector size.
double fVtxX
True vertex of X position.
Definition: TrainingData.h:66
Number of types, used like a vector size.
float vtxz
void FillSimpleOutputVector(float *output) const
FinalStateProngsType fFSp
Class of the event by primary n prongs.
Definition: TrainingData.h:62
enum cvn::ParentParticle ParentParticleType
The TrainingData objects contains a PixelMap and the output class type, and any other bit that goes i...
Definition: TrainingData.h:23
PixelMap, basic input to CVN neural net.
Definition: PixelMap.h:23
unsigned int NPPtOutput() const
Definition: TrainingData.h:44
enum cvn::FinalStateProngs FinalStateProngsType
double fVtxZ
True vertex of Z position.
Definition: TrainingData.h:68
ParentParticleType fPPt
Class of the event by parent particle.
Definition: TrainingData.h:59
void FillFSlongOutputVector(float *output) const
unsigned int NFSpOutput() const
Definition: TrainingData.h:46
FinalStateType fFSt
Class of the event by final state.
Definition: TrainingData.h:61