AlgFitShowerEM.cxx

Go to the documentation of this file.

// $Id: AlgFitShowerEM.cxx,v 1.3 2007/11/11 08:30:04 rhatcher Exp $
//
// AlgFitShowerEM.cxx
//
// AlgFitShowerEM is a concrete FitShowerEM Algorithm class.
//

#include <assert.h>

#include "Algorithm/AlgConfig.h"
#include "Candidate/CandContext.h"
#include "CandData/CandHeader.h"
#include "CandFitShowerEM/AlgFitShowerEM.h"
#include "CandFitShowerEM/CandFitShowerEMHandle.h"
#include "CandShowerEM/CandShowerEMHandle.h"
#include "Conventions/Mphysical.h"
#include "Conventions/Munits.h"
#include "Conventions/PlaneView.h"
#include "MessageService/MsgService.h"
#include "MessageService/MsgFormat.h"
#include "MinosObjectMap/MomNavigator.h"
#include "Navigation/NavKey.h"
#include "Navigation/NavSet.h"
#include "RecoBase/CandSliceHandle.h"
#include "RecoBase/CandStripHandle.h"
#include "RecoBase/CandTrackHandle.h"
#include "RecoBase/ArrivalTime.h"
#include "RecoBase/PropagationVelocity.h"
#include "RecoBase/LinearFit.h"
#include "UgliGeometry/UgliGeomHandle.h"
#include "CandFitShowerEM/FitterEM.h"

ClassImp(AlgFitShowerEM)

//______________________________________________________________________
CVSID("$Id: AlgFitShowerEM.cxx,v 1.3 2007/11/11 08:30:04 rhatcher Exp $");

//______________________________________________________________________
AlgFitShowerEM::AlgFitShowerEM()
{
}

//______________________________________________________________________
AlgFitShowerEM::~AlgFitShowerEM()
{
  MSG("FitShowerEM",Msg::kDebug) << "Deleting AlgFitShowerEM" << endl;
}

//______________________________________________________________________
void AlgFitShowerEM::RunAlg(AlgConfig &ac, CandHandle &ch,
                            CandContext &cx)
{
  MSG("Alg", Msg::kDebug) << "Starting AlgFitShowerEM::RunAlg()" << endl;

 
  assert(ch.InheritsFrom("CandFitShowerEMHandle"));
  CandFitShowerEMHandle &cfsemh = dynamic_cast<CandFitShowerEMHandle &>(ch);

  cfsemh.SetPass(0);
  
  assert(cx.GetDataIn());
  assert(cx.GetDataIn()->InheritsFrom("TObjArray"));
  
  const CandShowerEMHandle *shower0 = 0;
  const TObjArray *tclist = 0;
  const TObjArray *cxin = dynamic_cast<const TObjArray *>(cx.GetDataIn());
  for (Int_t i=0; i<=cxin->GetLast(); i++) {
    TObject *tobj = cxin->At(i);
    if (tobj->InheritsFrom("CandShowerEMHandle")) {
      shower0 = dynamic_cast<CandShowerEMHandle*>(tobj);
    }
    if (tobj->InheritsFrom("TObjArray")) {
      tclist = dynamic_cast<TObjArray*>(tobj);
    }
  }

  if(!shower0) return;

  fVldContext = const_cast<VldContext*>(shower0->GetVldContext());
  assert(fVldContext);
  fDetector = fVldContext->GetDetector();
  UgliGeomHandle ugh(*fVldContext);

  MSG("FitShowerEM", Msg::kDebug) 
    << "event number " << shower0->GetCandSlice()->GetCandRecord()
    ->GetCandHeader()->GetSnarl() << endl;

  Double_t theEnergy = shower0->GetEnergy()/2150.;
  Double_t vtx[3] = {0};
  vtx[0] = shower0->GetVtxU();
  vtx[1] = shower0->GetVtxV();
  vtx[2] = shower0->GetVtxZ();
  Double_t duds = shower0->GetDirCosU();
  Double_t dvds = shower0->GetDirCosV();
  Double_t dzds = shower0->GetDirCosZ();

  Double_t stepSize[6];
  stepSize[0] = ac.GetDouble("EnergyStep");
  stepSize[1] = ac.GetDouble("UVtxStep");
  stepSize[2] = ac.GetDouble("VVtxStep");
  stepSize[3] = ac.GetDouble("ZVtxStep");
  stepSize[4] = ac.GetDouble("UAngStep");
  stepSize[5] = ac.GetDouble("VAngStep");

  Int_t nSteps[6];  
  nSteps[0]   = ac.GetInt("EnergyNSteps");
  nSteps[1]   = ac.GetInt("UVtxNSteps");
  nSteps[2]   = ac.GetInt("VVtxNSteps");
  nSteps[3]   = ac.GetInt("ZVtxNSteps");
  nSteps[4]   = ac.GetInt("UAngNSteps");
  nSteps[5]   = ac.GetInt("VAngNSteps");

  CopyFromShowerEM(shower0,cfsemh);
  
  int nstp = shower0->GetNStrip();
  CandStripHandleItr cshi(shower0->GetDaughterIterator());

  FitterEM *ffitterEM = new FitterEM(nstp,cshi);
  ffitterEM->SetInputParams(theEnergy,vtx,duds/dzds,dvds/dzds);
  ffitterEM->SetStepSize(stepSize);
  ffitterEM->SetNSteps(nSteps);
  ffitterEM->DoFit();

  if(ffitterEM->Pass()){
    MSG("FitShowerEM",Msg::kDebug) << "Passed" << endl;
    cfsemh.SetPass(true);
    cfsemh.SetChi2(ffitterEM->GetChi2());
    cfsemh.SetNDOF(ffitterEM->GetNDF());    
    Double_t *fitparms = ffitterEM->GetFittedPars();
    cfsemh.SetMomentum(fitparms[0]);
    cfsemh.SetEnergy(fitparms[0]);
    cfsemh.SetVtxU(fitparms[1]);
    cfsemh.SetVtxV(fitparms[2]);
    cfsemh.SetVtxZ(fitparms[3]);
    cfsemh.SetDirCosU(fitparms[4]);
    cfsemh.SetDirCosV(fitparms[5]);
    cfsemh.SetNIterate(ffitterEM->GetNIterations());
    MSG("FitShowerEM",Msg::kDebug) << "with Chi2/NDOF = " 
                                   << ffitterEM->GetChi2()
                                   << "/" << ffitterEM->GetNDF() << endl;
    MSG("FitShowerEM",Msg::kDebug) << "and after " 
                                   << ffitterEM->GetNIterations()
                                   << " iterations" << endl;
  }
  else{
    MSG("FitShowerEM",Msg::kDebug) << "Failed" << endl;
    cfsemh.SetPass(false);
    cfsemh.SetChi2(-1);
    cfsemh.SetNDOF(-1);
    cfsemh.SetNIterate(ffitterEM->GetNIterations());
  }
  delete ffitterEM;
  
  return;
}

void AlgFitShowerEM::CopyFromShowerEM(const CandShowerEMHandle *shower0,
                                      CandFitShowerEMHandle &cfsemh){
  
  cfsemh.SetVtxPlane(shower0->GetVtxPlane());
  cfsemh.SetVtxZ(shower0->GetVtxZ());
  cfsemh.SetVtxU(shower0->GetVtxU());
  cfsemh.SetVtxV(shower0->GetVtxV());
  cfsemh.SetVtxT(shower0->GetVtxT());
  cfsemh.SetDirCosU(shower0->GetDirCosU());
  cfsemh.SetDirCosV(shower0->GetDirCosV());
  cfsemh.SetDirCosZ(shower0->GetDirCosZ());
  
  cfsemh.SetEigenVectors(shower0->GetEigenVectors());
  cfsemh.SetEigenValues(shower0->GetEigenValues());
  cfsemh.SetAvgDev(shower0->GetAvgDev());
  cfsemh.SetOutPH(shower0->GetOutPH());
  cfsemh.SetShwStatus(shower0->GetShwStatus());
  cfsemh.SetEnergy(shower0->GetEnergy());

  CandStripHandleItr stripItr(shower0->GetDaughterIterator());
  while (CandStripHandle *strip = stripItr()) {
    cfsemh.AddDaughterLink(*strip);
  }

  //Calibrate(&cfsemh);

}

---