// Author: David Lawrence  June 24, 2004
//
//
// DEventSourceHDDM
//
/// Implements JEventSource for HDDM files
//
// Changes:   Oct 3, 2012 Yi Qiang: add classes for Cerenkov detector
//         OCt 10, 2012 Yi Qiang: modifed Cerenkov classes with general Cere hits
//         Oct 8, 2013 Yi Qiang: added dedicated object for RICH Truth Hit
//

#ifndef _JEVENT_SOURCEHDDM_H_
#define _JEVENT_SOURCEHDDM_H_

#include <vector>
#include <string>
using namespace std;

#include <pthread.h>

#include <JANA/JEventSource.h>
#include <JANA/jerror.h>
#include <JANA/JCalibration.h>

#include <HDDM/hddm_s.hpp>

#include "TRACKING/DMCTrackHit.h"
#include "TRACKING/DMCThrown.h"
#include "TRACKING/DMCTrajectoryPoint.h"
#include "BCAL/DBCALSiPMHit.h"
#include "BCAL/DBCALSiPMSpectrum.h"
#include "BCAL/DBCALHit.h"
#include "BCAL/DBCALIncidentParticle.h"
#include "BCAL/DBCALTDCHit.h"
#include "BCAL/DBCALTruthShower.h"
#include "BCAL/DBCALTruthCell.h"
#include "CDC/DCDCHit.h"
#include "FDC/DFDCHit.h"
#include "FCAL/DFCALTruthShower.h"
#include "FCAL/DFCALHit.h"
#include "CCAL/DCCALTruthShower.h"
#include "CCAL/DCCALHit.h"
#include "TOF/DTOFTruth.h"
#include "TOF/DTOFHit.h"
#include "TOF/DTOFHitMC.h"
#include "START_COUNTER/DSCTruthHit.h"
#include "START_COUNTER/DSCHit.h"
#include <PID/DMCReaction.h>
#include <PID/DBeamPhoton.h>
#include <TRACKING/DTrackTimeBased.h>
#include <TAGGER/DTAGMHit.h>
#include <TAGGER/DTAGHHit.h>
// load CERE headers, yqiang Oct 3, 2012
// modified by yqiang, Oct 10 2012
// added RichTruthHit object, yqiang, Oct 7, 2013
#include <CERE/DCereHit.h>
#include <RICH/DRichHit.h>
#include <RICH/DRichTruthHit.h>
#include <PID/DRFTime.h>
#include <DANA/DApplication.h>
#include "PAIR_SPECTROMETER/DPSHit.h"
#include "PAIR_SPECTROMETER/DPSCHit.h"
#include "PAIR_SPECTROMETER/DPSTruthHit.h"
#include "PAIR_SPECTROMETER/DPSCTruthHit.h"

class DEventSourceHDDM:public JEventSource
{
   public:
      DEventSourceHDDM(const char* source_name);
      virtual ~DEventSourceHDDM();      
      virtual const char* className(void){return static_className();}
      static const char* static_className(void){return "DEventSourceHDDM";}

      jerror_t GetEvent(JEvent &event);
      void FreeEvent(JEvent &event);
      jerror_t GetObjects(JEvent &event, JFactory_base *factory);
      
      jerror_t Extract_DMCTrackHit(hddm_s::HDDM *record, JFactory<DMCTrackHit> *factory, string tag);
      jerror_t GetCDCTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetFDCTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetBCALTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetTOFTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetCherenkovTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetFCALTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetCCALTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t GetSCTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);

	   jerror_t Extract_DRFTime(hddm_s::HDDM *record, JFactory<DRFTime> *factory, JEventLoop* locEventLoop);
      jerror_t Extract_DBCALTruthShower(hddm_s::HDDM *record, JFactory<DBCALTruthShower> *factory, string tag);
      jerror_t Extract_DBCALTruthCell(hddm_s::HDDM *record, JFactory<DBCALTruthCell> *factory, string tag);
      jerror_t Extract_DBCALSiPMHit(hddm_s::HDDM *record, JFactory<DBCALSiPMHit> *factory, string tag);
      jerror_t Extract_DBCALSiPMSpectrum(hddm_s::HDDM *record, JFactory<DBCALSiPMSpectrum>* factory, string tag);
      jerror_t Extract_DBCALHit(hddm_s::HDDM *record, JFactory<DBCALHit> *factory, string tag);
      jerror_t Extract_DBCALIncidentParticle(hddm_s::HDDM *record, JFactory<DBCALIncidentParticle> *factory, string tag);
      jerror_t Extract_DBCALTDCHit(hddm_s::HDDM *record, JFactory<DBCALTDCHit> *factory, string tag);
      jerror_t Extract_DMCReaction(hddm_s::HDDM *record, JFactory<DMCReaction> *factory, string tag, JEventLoop *loop);
      jerror_t Extract_DBeamPhoton(hddm_s::HDDM *record, JFactory<DBeamPhoton> *factory, string tag, JEventLoop *loop);
      jerror_t Extract_DMCThrown(hddm_s::HDDM *record, JFactory<DMCThrown> *factory, string tag);
      jerror_t Extract_DCDCHit(JEventLoop* locEventLoop, hddm_s::HDDM *record, JFactory<DCDCHit> *factory, string tag);
      jerror_t Extract_DFDCHit(hddm_s::HDDM *record, JFactory<DFDCHit> *factory, string tag);
      jerror_t Extract_DFCALTruthShower(hddm_s::HDDM *record, JFactory<DFCALTruthShower> *factory, string tag);
      jerror_t Extract_DCCALHit(hddm_s::HDDM *record, JFactory<DCCALHit> *factory, string tag, JEventLoop* eventLoop);
      jerror_t Extract_DCCALTruthShower(hddm_s::HDDM *record, JFactory<DCCALTruthShower> *factory, string tag);
      jerror_t Extract_DFCALHit(hddm_s::HDDM *record, JFactory<DFCALHit> *factory, string tag, JEventLoop* eventLoop);
      jerror_t Extract_DMCTrajectoryPoint(hddm_s::HDDM *record, JFactory<DMCTrajectoryPoint> *factory, string tag);
      jerror_t Extract_DTOFTruth(hddm_s::HDDM *record,  JFactory<DTOFTruth> *factory, string tag);

      jerror_t Extract_DTOFHit( hddm_s::HDDM *record,  JFactory<DTOFHit>* factory, JFactory<DTOFHitMC>* factoryMC, string tag);
      jerror_t Extract_DTOFHitMC( hddm_s::HDDM *record,  JFactory<DTOFHitMC>* factoryMC, JFactory<DTOFHit>* factory, string tag);

      jerror_t Extract_DSCHit(hddm_s::HDDM *record,  JFactory<DSCHit> *factory, string tag);
      jerror_t Extract_DSCTruthHit(hddm_s::HDDM *record,  JFactory<DSCTruthHit> *factory, string tag);

      jerror_t Extract_DTrackTimeBased(hddm_s::HDDM *record,  JFactory<DTrackTimeBased> *factory, string tag, int runnumber);
      string StringToDMatrixDSym(string &str_vals, DMatrixDSym &mat, int Nrows, int Ncols);

      jerror_t Extract_DTAGMHit( hddm_s::HDDM *record,  JFactory<DTAGMHit>* factory, string tag);
      jerror_t Extract_DTAGHHit( hddm_s::HDDM *record,  JFactory<DTAGHHit>* factory, string tag);

      jerror_t Extract_DPSHit(hddm_s::HDDM *record,JFactory<DPSHit>* factory, 
			      string tag);
      jerror_t Extract_DPSCHit(hddm_s::HDDM *record,JFactory<DPSCHit>* factory, 
			       string tag);
      jerror_t Extract_DPSTruthHit(hddm_s::HDDM *record,JFactory<DPSTruthHit>* factory, 
			      string tag);
      jerror_t Extract_DPSCTruthHit(hddm_s::HDDM *record,JFactory<DPSCTruthHit>* factory, 
			       string tag);

      Particle_t IDTrack(float locCharge, float locMass) const;

      // add RICH hit and Truth, yqiang Oct 3, 2012
      // modifed by yqiang, Oct 10 2012 now include both truth hits in DMCThrown
      // Oct 8, 2013, added dedicated object for RICH truth hit
      jerror_t GetRichTruthHits(hddm_s::HDDM *record, vector<DMCTrackHit*>& data);
      jerror_t Extract_DCereHit(hddm_s::HDDM *record, JFactory<DCereHit> *factory, string tag);
      jerror_t Extract_DRichHit(hddm_s::HDDM *record, JFactory<DRichHit> *factory, string tag);
      jerror_t Extract_DRichTruthHit(hddm_s::HDDM *record, JFactory<DRichTruthHit> *factory, string tag);

      std::ifstream *ifs;
      hddm_s::istream *fin;
      DApplication *dapp;
      const DMagneticFieldMap *bfield;
      const DGeometry *geom;

   private:
      bool initialized;
   
      pthread_mutex_t rt_mutex;
      map<hddm_s::HDDM*, vector<DReferenceTrajectory*> > rt_by_event;
      list<DReferenceTrajectory*> rt_pool;

      map<unsigned int, double> dTargetCenterZMap; //unsigned int is run number
      map<unsigned int, double> dRFFrequencyMap; //unsigned int is run number

      JCalibration *jcalib;
      float uscale[192],vscale[192];

};

#endif //_JEVENT_SOURCEHDDM_H_