MadNsID.h

Go to the documentation of this file.

//
// MadNsID
//
// A class to calculate Niki Saoulidou's ANN based CC PID
//
// Created:  N. Saoulidou -- Analysis first presented early  2004, Code implementation in May-2005
//
// Converted to this format: M. Kordosky, 12 August 2005
//
// $Author: niki $ 
//
// $Revision: 1.10 $
// 
// $Name:  $
//
// $Id: MadNsID.h,v 1.10 2007/03/19 23:51:00 niki Exp $
//
// $Log: MadNsID.h,v $
// Revision 1.10  2007/03/19 23:51:00  niki
// New 7-Input Variable ANN
//
// Revision 1.9  2007/03/15 19:45:16  niki
// Small changes in order to be able to call methods from NCUtils (If I have commited this twice I
// apologize but I am having some cvs problems)
//
// Revision 1.8  2007/01/30 03:17:21  rhatcher
// No longer user BeamType "class" found in MCReweight, but rather the
// BeamType "namespace" found in Conventions package.
// Also covert to using "Detector::" rather than "DetectorType::", though
// for now they are synonyms.
//
// Revision 1.7  2006/05/08 21:27:44  boehm
// Primary change is the addition of functions to calculate the PID's from NtpSt objects.  Both PIDs may now be generated without relying on the rest of the Mad infrastructure.  Mad based code though should not be changed.
//
// Also removed the if(!ntpTrack.fit.pass) return;  from MadDpID.cxx as it does not reflect the current state of the CC analysis. This was removed in the branched version but remained in the development version.
//
// Revision 1.6  2005/12/12 15:36:36  niki
// Corresponding .h file
//
// Revision 1.5  2005/10/06 15:03:26  kordosky
// various improvements, pids finally integrated into pan construction routine, still not fully debugged though
//
// Revision 1.4  2005/09/14 13:32:45  kordosky
// DpID and NsID classes now fully working and validated with MadTestAnalysis by comparing nannpid vs. annpid and npid0 vs. pid0.  pdf files and weights stored in data subdirectory. FD pdfs needed for DpID.
//
//


#ifndef madnsid_h
#define madnsid_h

#include <string>
#include <cmath>

//#include "Mad/MadQuantities.h"
//#include "TH1.h"
#include "Rtypes.h"
#include "Conventions/Detector.h"
#include "Conventions/BeamType.h"
#include "Mad/AnnInputBlock.h"

class MadQuantities;
//class TH1;
//class TFile;

class NtpSREvent;
class NtpSRTrack;
class NtpSRShower;
class NtpStRecord;


class MadNsID {

  //  AnnInputBlock AnnVar(Int_t event=0);
  //  Double_t GetAnnPid(AnnInputBlock anninput,Int_t det, Int_t tar, Int_t fid, Int_t prior);  
 public:
  MadNsID();
  bool GetPID(MadQuantities* mq, Int_t event, 
              const Detector::Detector_t& det, Double_t& pid);
  bool GetPID(NtpSREvent* ntpEvent, NtpSRTrack *ntpTrack,
              NtpSRShower *ntpShower, NtpStRecord* st,
              const Detector::Detector_t& det, Double_t& pid);

  bool ChooseWeightFile(const Detector::Detector_t& det,
                     const BeamType::BeamType_t& beam, 
                     Int_t fid=1, Int_t prior=1);
  void SetWeightFile(const char* file);

  bool CompareAnnBlocks(MadQuantities* mq, 
                        Int_t event, const AnnInputBlock& ext);

  bool CalcVars(NtpSREvent* ntpEvent, NtpSRTrack *ntpTrack,
                       NtpSRShower *ntpShower, NtpStRecord* st,
                        AnnInputBlock& anninput);


 std::string GetWeightFileName(const Detector::Detector_t& det,
                                const BeamType::BeamType_t& beam,
                                Int_t fid, Int_t prior);

 bool ReadWeights(const char* file);

 Double_t Sigmoid(Double_t x);
 Double_t CalcPID(const AnnInputBlock& b,
                   const Detector::Detector_t& det);

 private:

  bool CalcVars(MadQuantities* mq, Int_t event, AnnInputBlock& anninput);

 // Double_t CalcPID(const AnnInputBlock& b, 
 //                const Detector::Detector_t& det);
 // Double_t Sigmoid(Double_t x);


//  static const Int_t    inneuron  = 13;
  static const Int_t    inneuron  = 7;
  static const Int_t    hidneuron = 15;
  
  //  Double_t out[hidneuron]; // input neurons
  Double_t rin[hidneuron]; // first hidden layer
  
  Double_t weight1[inneuron][hidneuron];// weights first layer
  Double_t constant1[hidneuron];   // constant first layer
  
  Double_t weighto[hidneuron];// weights second  (output) layer
  Double_t constanto[1];   // constant second (output) layer  
  bool weights_read;
  std::string weight_fname;

  Detector::Detector_t cache_det;
  BeamType::BeamType_t cache_beam; 
  Int_t cache_fid;
  Int_t cache_prior;

};

inline Double_t MadNsID::Sigmoid(Double_t x)
{
  double sig;
      if(x>37.0){
         sig = 1.0;
      }
      else if(x<-37.0){
         sig = 0.0;
      }
      else {
         sig = 1./(1.+std::exp(-x));
      }
      return sig;
}

#endif

---