NCExtractionANN.cxx

Go to the documentation of this file.

//
//
//NCExtractionANN.cxx
//
//Module for analyzing beam data for nc analysis
//
//R. Pittam 04/2007

#include <cassert>

#include "NCUtils/NCType.h"

#include "NCUtils/Extraction/NCExtractionANN.h"
#include "NCUtils/Extraction/RPTrackANN.h"
#include "NCUtils/Extraction/RPnearTrackANN.h"
#include "NCUtils/Extraction/RPnearTrackLowEANN.h"
#include "NCUtils/Extraction/RPfarTrackANN.h"
#include "NCUtils/Extraction/RPfarTrackLowEANN.h"

#include "NCUtils/Cuts/NCAnalysisCuts.h"
#include "NCUtils/Cuts/NCAnalysisCutsNC.h"

#include "MessageService/MsgService.h"
#include "Conventions/Detector.h"
#include "Conventions/SimFlag.h"

ClassImp(NCExtractionANN)
ClassImp(NCExtractionANNNear)
ClassImp(NCExtractionANNFar)

CVSID("$Id: NCExtractionANN.cxx,v 1.8 2009/09/23 14:03:49 bckhouse Exp $");

#include "NCUtils/Extraction/MicroDSTMaker.h"
REGISTER_NCEXTRACTION(NCExtractionANN, ANN)
REGISTER_NCEXTRACTION(NCExtractionANNNear, ANNNear)
REGISTER_NCEXTRACTION(NCExtractionANNFar, ANNFar)


//......................................................................
NCExtractionANN::NCExtractionANN(NCAnalysisCuts* cuts,
                                 const Registry& r) :
  NCExtraction(cuts, r),
  fDetector(Detector::kUnknown),
  fRPannUseLowETrain(false)
{
  int         tmpb;
  if (r.Get("RPAnnUseLowETrain",tmpb)) fRPannUseLowETrain = tmpb;
}

//-----------------------------------------------------------------------
double NCExtractionANN::GetIdProbability(NCEventInfo& evtInfo, int /*beamType*/)
{
  MAXMSG("NCExtractionANN", Msg::kDebug,1) << "GetIdProbabilty from ANN"
                                             << endl;

  // set up ANN input parameters from info objects
  // all ANN inputs must be doubles for some reason

  Double_t evtPlanes = evtInfo.event->planes;
  Double_t evtPHperPlane = evtInfo.event->phPerPlane;
  Double_t evtPHperStrip = evtInfo.event->phPerStrip;
  Double_t nshowers = evtInfo.event->showers;

  Double_t shwPlanes = 0.;
  if (nshowers>0) {
    shwPlanes = (Double_t) evtInfo.shower->planes;
  }

  // if longer than 40 planes, assign 1.5 (very CC like)
  if (evtPlanes>40) return 1.5;

  // now get the ANN values
  if (evtInfo.event->tracks>0) {
    Double_t trkPlanes = (Double_t) evtInfo.track->planes;
    Double_t trackExtension = (Double_t) trkPlanes - shwPlanes;
    Double_t trkdCosZvtx = evtInfo.track->dcosZVtx;
    Double_t trkredChi2 = evtInfo.track->reducedChi2;
    Double_t trkStrips = evtInfo.track->totalStrips;
    Double_t trkPHFraction = evtInfo.track->trackSigCorFraction;

    Detector::Detector_t det=(Detector::Detector_t)evtInfo.header->detector;
    assert(det==Detector::kNear || det==Detector::kFar);

    RPTrackANN* ann=0;

    // This is which detector we should actually use. If fDetector is
    // unset, we use the detector of the files we're running on,
    // otherwise we use the value of fDetector
    Detector::Detector_t detToUse=(fDetector==Detector::kUnknown) ? det : fDetector;

    if(fRPannUseLowETrain){
      MAXMSG("NCExtractionANN", Msg::kInfo, 1) 
        << "Using Low Energy Trained ANNs" << endl;
      // Boo, the ternary operator won't implicitly upcast here

      if(detToUse==Detector::kNear) ann=new RPnearTrackLowEANN;
      else                          ann=new RPfarTrackLowEANN;
      
    } 
    else{
      MAXMSG("NCExtractionANN", Msg::kInfo, 1) 
        << "Using All Energy Trained ANNs" << endl;
      if(detToUse==Detector::kNear) ann=new RPnearTrackANN;
      else                          ann=new RPfarTrackANN;
    }
    
    double ret = ann->value(0,
                            evtPHperPlane,
                            evtPHperStrip,
                            trkPHFraction,
                            trkStrips,
                            trkdCosZvtx,
                            trackExtension,
                            trkredChi2);
    // Make sure not to leak the ann
    delete ann;
    return ret;
    
  }//if tracks > 0
  else return 0; //no tracks

}

---