#include <cometEventLoop.hxx>
#include <IValidationUtils.hxx>
#include <IRecPackManager.hxx>
#include <IReconGlobalUtils.hxx>
#include <ICOMETOutput.hxx>
#include <IReconBase.hxx>
#include <IReconPID.hxx>


class IReconGlobalValidTree: public COMET::ICOMETEventLoopFunction {
  public:
    IReconGlobalValidTree() {}
    virtual ~IReconGlobalValidTree() {}


    void Initialize();
    void Finalize(COMET::ICOMETOutput* output);

    bool operator () (COMET::ICOMETEvent& event);

    IReconGlobalUtils fUtils;
};


//**************************************************************  
void IReconGlobalValidTree::Initialize(){
//**************************************************************
    COMET::vman().AddTreeValid("COMETValidTree");

}


//**************************************************************
bool IReconGlobalValidTree::operator () (COMET::ICOMETEvent& event) {
//**************************************************************
  COMET::IHandle<COMET::IAlgorithmResult> GlobRec = event.GetFit("ReconGlobal");
  if ( GlobRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = GlobRec->GetResultsContainer("final");
    if ( allTracks ) {

//    std::cout << std::endl;
//    std::cout << "********** SUMMARY OF GLOBAL RECON *****************" << std::endl;    
//    COMET::tman().DumpContainerWithHistory(*allTracks);   
//    std::cout << "****************************************************" << std::endl;
//    std::cout << std::endl;

      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconGlobal");
        // Get the inner object (P0D+TRACKER)
        if (!((*it)->UsesDetector(COMET::IReconBase::kTPC)) || (*it)->CheckStatus(COMET::IReconBase::kLikelihoodFit)) continue;
        COMET::IHandle<COMET::IReconBase> innerObject = *it;
        if (innerObject->UsesDetector(COMET::IReconBase::kSMRD))
          innerObject = COMET::tman().GetConstituentNotUsingDetector(innerObject,COMET::IReconBase::kSMRD);
        if (innerObject->UsesDetector(COMET::IReconBase::kECal))
          innerObject = COMET::tman().GetConstituentNotUsingDetector(innerObject,COMET::IReconBase::kECal);
        if (innerObject->UsesDetector(COMET::IReconBase::kDSECal))
          innerObject = COMET::tman().GetConstituentNotUsingDetector(innerObject,COMET::IReconBase::kDSECal);
        COMET::IHandle<COMET::IReconPID> ThePID = innerObject;
        if (ThePID){
          const COMET::IReconObjectContainer& alter = ThePID->GetAlternates();
          for ( std::vector< COMET::IHandle<COMET::IReconBase> >::const_iterator ot=alter.begin(); ot!=alter.end(); ot++ ){
            COMET::vman().FillTreeValid("COMETValidTree", *ot, "ReconGlobalAltPID");
          }
        }
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> P0dRec = event.GetFit("ReconP0D");
  if ( P0dRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allVertices = P0dRec->GetResultsContainer("final");
    COMET::IHandle<COMET::IReconObjectContainer> allTracks( new COMET::IReconObjectContainer("p0d"));
      fUtils.GetObjectsFromVertices(allVertices, allTracks);
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconP0D");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> TrackerRec = event.GetFit("ReconTracker");
  if ( TrackerRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = TrackerRec->GetResultsContainer("final");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconTracker");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> TpcRec = event.GetFit("ReconTPC");
  if ( TpcRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = TpcRec->GetResultsContainer("final");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconTPC");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> FgdRec = event.GetFit("ReconFGD");
  if ( FgdRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allIsoTracks = FgdRec->GetResultsContainer("FGDPids");
    if ( allIsoTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allIsoTracks->begin(); it!=allIsoTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconFGD");
      }
    }
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = FgdRec->GetResultsContainer("fittedFgdTpcReconObjects");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconFGD");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> EcalRec = event.GetFit("ReconECal");
  if ( EcalRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = EcalRec->GetResultsContainer("final");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconECal");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> P0decalRec = event.GetFit("ReconP0DECal");
  if ( P0decalRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = P0decalRec->GetResultsContainer("final");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconP0DECal");
      }
    }
  }

  COMET::IHandle<COMET::IAlgorithmResult> SmrdRec = event.GetFit("ReconSMRD");
  if ( SmrdRec ) {
    COMET::IHandle<COMET::IReconObjectContainer> allTracks = SmrdRec->GetResultsContainer("final");
    if ( allTracks ) {
      for ( COMET::IReconObjectContainer::iterator it=allTracks->begin(); it!=allTracks->end(); it++ ){
        COMET::vman().FillTreeValid("COMETValidTree", *it, "ReconSMRD");
      }
    }
  }


  return false;
}




//*****************************************************************************
void IReconGlobalValidTree::Finalize (COMET::ICOMETOutput* output){
//*****************************************************************************
  //Remove COMET Structure from outputted root file.
  output->Delete("COMETEvents;*");
  output->Delete("COMETGeometry*;*");

}



//*****************************************************************************
int main(int argc, char **argv) {
//*****************************************************************************
  IReconGlobalValidTree userCode;
  cometEventLoop(argc,argv,userCode);
}