SValidStore.cc

Go to the documentation of this file.

#include <sstream>
#include <iostream>
#include <fstream>
#include <algorithm>

#include "SValidStore.h"
#include "SIOValidationObj.h"
#include "FIOValidationObj.h"
#include "TValidationObj.h"
#include "TValidationPtr.h"

#include <utl/ErrorLogger.h>

#include <det/Detector.h>

#include <utl/TimeStamp.h>

#include <evt/Event.h>
#include <evt/ShowerRecData.h>
#include <evt/ShowerSRecData.h>

#include <sevt/SEvent.h>
#include <sevt/Header.h>
#include <sevt/Station.h>
#include <sevt/PMT.h>
#include <sevt/PMTRecData.h>
#include <sevt/StationRecData.h>
#include <sevt/SortCriteria.h>
#include <fwk/LocalCoordinateSystem.h>

#include <utl/Trace.h>
#include <utl/AugerUnits.h>
#include <utl/MathConstants.h>
#include <utl/PhysicalConstants.h>
#include <utl/TabulatedFunctionErrors.h>
#include <utl/UTMPoint.h>
#include <fwk/CentralConfig.h>

#include <TFile.h>
#include <TObjArray.h>
#include <TGraph.h>
#include <TGraphErrors.h>
#include <TMultiGraph.h>
#include <TF1.h>
#include <TH2D.h>
#include <TVector3.h>
#include <TMarker.h>
#include <TCanvas.h>
#include <TTree.h>
#include <TFolder.h>
#include <TString.h>
#include <TPaveText.h>

using namespace det;
using namespace evt;
using namespace sevt;
using namespace std;
using namespace fwk;


SValidStore::SValidStore() :
  vOutputFile(0),
  vReferenceTree(0),
  _TvObjPtr(0)
{
}


SValidStore::~SValidStore()
{
  delete vOutputFile;
}


fwk::VModule::ResultFlag
SValidStore::Init()
{
  // Initialize your module here. This method
  // is called once at the beginning of the run.
  // The eSuccess flag indicates the method ended
  // successfully.  For other possible return types,
  // see the VModule documentation.

  fwk::CentralConfig* theCC = fwk::CentralConfig::GetInstance();
  _xmlSValidStore = theCC->GetTopBranch("SValidStore");
  string outFileName;
  _xmlSValidStore.GetChild("Filename").GetData(outFileName);
  _xmlSValidStore.GetChild("Mode").GetData(_Mode);

  if (_Mode != "check") {
    vOutputFile = new TFile(outFileName.c_str(), _Mode.c_str());
    _TvObjPtr = new TValidationPtr();
  } else {
    vOutputFile = new TFile(outFileName.c_str());
    _TvObjPtr = new TValidationPtr();
  }

  list<string> module;
  _xmlSValidStore.GetChild("StoreObj").GetData(module);

  for (list<string>::const_iterator m = module.begin();
       m != module.end(); ++m)
    if (_ObjList.find(*m) == _ObjList.end()) {
      ValidationObj* vObj = Factory(*m);
      if (vObj)
        _ObjList[*m] = vObj;
    }

  return eSuccess;
}


fwk::VModule::ResultFlag
SValidStore::Run(Event& event)
{
  INFO("SValidStore::Run()");

  if (_Mode != "check")
    for (map<string, ValidationObj*>::const_iterator vo =_ObjList.begin();
         vo != _ObjList.end(); ++vo) {
      ValidationObj* vObj = vo->second;
      TValidationObj* TvObj = vObj->GetTObject();
      _TvObjPtr->SetTVObj(TvObj, vObj->GetModuleName());

      if (vObj->Fill(event)) {
        vOutputFile->cd();
        _TvObjPtr->Write((vObj->GetModuleName()).c_str(), TObject::kWriteDelete);
      }
    }
  else {
    vOutputFile->cd();

    for (map<string, ValidationObj*>::const_iterator vo = _ObjList.begin();
         vo != _ObjList.end(); ++vo) {
      ValidationObj* const vObj = vo->second;

      cout << "Retrieving Module Name = " << vObj->GetModuleName() << " ... " << endl;

      _TvObjPtr = (TValidationPtr*)vOutputFile->Get((vObj->GetModuleName()).c_str());
      cout << "Module name retrieved!" << endl;

      TValidationObj* const TvObj = _TvObjPtr->GetTVObj();
      cout << "INFO::before fill!" << endl;

      if (vObj->Fill(event)) {
        if (vObj->Validate(TvObj)) {

          cout << "\n\n\n Obj " << vObj->GetModuleName() << " validated!\n\n\n";
          vObj->SetValidationOk();
          return eSuccess;

        } else {

          cout << "\n+++++++++++++++++++++++++++++++++++++++++++++++++++++++\n"
                  "\n\n\n Obj " << vObj->GetModuleName() << " NOT Validated!\n\n\n"
                  "\nIn Reference file:" << endl;
          TvObj->Dump();
          cout << "\nIn last checkout:" << endl;
          TValidationObj* CurTvObj = vObj->GetTObject();
          CurTvObj->Dump();
          cout << "\n+++++++++++++++++++++++++++++++++++++++++++++++++++++++\n" << endl;
          return eFailure;
        }
      }
    }
  }
  return eSuccess;
}


fwk::VModule::ResultFlag
SValidStore::Finish()
{
  INFO("SValidStore::Finish()");
  if (_Mode == "check")
    for (map<string, ValidationObj*>::const_iterator vo = _ObjList.begin();
         vo != _ObjList.end(); ++vo) {
      ValidationObj* const vObj = vo->second;
      if (!vObj->IsValidate()) {
          cout << "\n+++++++++++++++++++++++++++++++++++++++++++++++++++++++\n"
                  "\n\n\n Obj " << vObj->GetModuleName() << " NOT Validated!\n\n\n"<<endl;
          cout<<  "\n\n\n Event rejected before the Validation Module in this Chain \n\n\n" <<endl;
          cout << "\n+++++++++++++++++++++++++++++++++++++++++++++++++++++++\n" << endl;
          return eFailure;
      }
    }
  vOutputFile->cd();
  vOutputFile->Close();

  return eSuccess;
}


void
SValidStore::ExamineEvent(const Event& event)
{
  // Plotting of stations is switched off for now; causes memeory leak
  //  PlotStations(event.GetSEvent());

  if (event.HasRecShower()) {
    RecInfo(event);
    //PlotLDF(event);
  }
}


//OK, let's dump out some reconstruction information

void
SValidStore::RecInfo(const evt::Event& /*event*/)
{ }


ValidationObj*
SValidStore::Factory(const string& objName)
{
  if (objName == "SIOValidation" || objName == "SIOValidationObj")
    return new SIOValidationObj();
  else if (objName == "FIOValidation" || objName == "FIOValidationObj")
    return new FIOValidationObj();
  else return 0;
}