#include "hpidfl.h"
#include "hruntimedb.h"
#include "hades.h"
#include "hgeantkine.h"
#include "hgeantrich.h"
#include "hgeantmdc.h"
#include "hmdcseg.h"
#include "hmdcsegsim.h"
#include "hrichhitsim.h"
#include "hmdctrkcand.h"
#include "hdebug.h"
#include "hgeomvector2.h"
#include "hgeomvector.h"

#include "hparset.h"

#include "hgeomtransform.h"
#include "hmdcsizescells.h"
//#include "mptools.h"
#include "hlinearcategory.h"
#include "hmatrixcategory.h"
#include "hmatrixcatiter.h"
#include "htofhit.h"
#include "htofhitsim.h"
#include "htofcluster.h"
#include "htofclustersim.h"
#include "hiterator.h"
#include "hrichcal.h"
#include "richdef.h"
#include "hgeantdef.h"
#include "hpidgeanttrackset.h"
#include "hpidtrackcand.h"
#include "hpidtrackcandsim.h"
#include "hkicktrack.h"
#include "hkicktrackB.h"
#include "hkicktracksim.h"
#include "hsplinetrack.h"
#include "hrichgeometrypar.h"
#include "hrichpad.h"
#include "hshowerhittof.h"
#include "hshowerhittoftrack.h"
#include "kickdef.h"
#include "hmdctrackgdef.h"
#include "tofdef.h"
#include "hmdcdef.h"
#include "TH1.h"
#include "TVector3.h"
#include "piddef.h"
#include "hmetamatch.h"
#include "hdatasource.h"
#include "hrktrackB.h"

//_HADES_CLASS_DESCRIPTION
///////////////////////////////////////////////////////////
// HPidFL
//
// library of static functions
// look here for auxiliary functions for:
// physics analysis, simulation analysis, ...
///////////////////////////////////////////////////////////

ClassImp(HPidFL)

  static HPidFL* gHPidFL=NULL;

HPidFL::HPidFL()
{
  //getParameterContainers();
}

HPidFL* HPidFL::Instance()
{
  if(gHPidFL)
    return gHPidFL;
  return new HPidFL();
}

HPidFL::~HPidFL(void) 
{

}

TObject* HPidFL::getHitObject(Int_t Index, Cat_t category, Bool_t bWarning) //OK
{
  HCategory* hitCategory = getCategory(category,kTRUE);
  if(hitCategory)
    return hitCategory->getObject(Index);
  else
    {
      if(bWarning)
	::Error("HPidFL::getHitObject","Category %d not found",category);
      return NULL;
    }
}


HCategory* HPidFL::getCategory(Cat_t cat, Bool_t bWarning) //OK
{
// Find HCategory if gHades->getCurrentEvent() is accessible

HEvent    *pEvent;
HCategory *pCat;

    if((gHades == NULL) || ((pEvent = gHades->getCurrentEvent()) == NULL))
    {
        if(bWarning)
        {
            ::Error("HPidFL::getCategory",
                            "Cannot access current event");
        }

        return NULL;
    }

    if((pCat = pEvent->getCategory(cat)) == NULL)
    {
        if(bWarning)
            ::Error("HPidFL::getCategory", "No category: %d", cat);
    }

    return pCat;
}

Float_t HPidFL::getMdcPhi(Int_t iSector, Float_t fPhiMdc) //OK
{
// Convert MDC's phi angle to the coordinate system used in other detectors

Float_t fPhi = TMath::RadToDeg() * fPhiMdc;

    switch(iSector)
    {
        case 0:
            break;

        case 1:
        case 2:
        case 3:
        case 4:
            fPhi += 60.0f * iSector;
            break;

        default:
            fPhi -= 60.0f;
            break;
    }

    return fPhi;
}

// -----------------------------------------------------------------------------

Float_t HPidFL::getMdcTheta(Float_t fThetaMdc) //OK
{
// Convert MDC's theta angle to the coordinate system used in other detectors

    return  TMath::RadToDeg() * fThetaMdc;
}

// -----------------------------------------------------------------------------

HKickTrack* HPidFL::getKickTrack(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HKickTrack object corresponding to iId.
// pCat must point on the catKickTrack category or gHades->getCurrentEvent()
// must be accessible.
// In case catKickTrack is category of HKickTrackSim then the object of this
// class is returned.

    if(iId < 0)
      {
	::Error("HPidFL::getKickTrack","Index <0 not defined!");
	return NULL;
      }
    if((pCat == NULL) && ((pCat = getCategory(catKickTrack)) == NULL))
    {
        ::Error("HPidFL::getKickTrack", "No catKickTrack category");
        return NULL;
    }

    return (HKickTrack *)pCat->getObject(iId);
}
// -----------------------------------------------------------------------------

HKickTrackB* HPidFL::getKickTrackB(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HKickTrack object corresponding to iId.
// pCat must point on the catKickTrack category or gHades->getCurrentEvent()
// must be accessible.
// In case catKickTrack is category of HKickTrackSim then the object of this
// class is returned.

    if(iId < 0)
      {
	::Error("HPidFL::getKickTrackB","Index <0 not defined!");
	return NULL;
      }
    if((pCat == NULL) && ((pCat = getCategory(catKickTrackB)) == NULL))
    {
        ::Error("HPidFL::getKickTrack", "No catKickTrackB category");
        return NULL;
    }

    return (HKickTrackB *)pCat->getObject(iId);
}
// -----------------------------------------------------------------------------


/*
HKickTrackSim* HPidFL::getKickTrackSim(Int_t iId, HCategory *pCat)
{
// Static function which returns HKickTrackSim object corresponding to iId.
// Returns HKickTrackSim object corresponding to nKickTrackId (if exists)
// pCat must point on the catKickTrack category or gHades->getCurrentEvent()
// must be accessible.

    return (HKickTrackSim *)getKickTrack(iId, pCat);
}
*/
// -----------------------------------------------------------------------------

HSplineTrack* HPidFL::getSplineTrack(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HSplineTrack object corresponding to iId.
// pCat must point on the catSplineTrack category or gHades->getCurrentEvent()
// must be accessible.
// In case catKickTrack is category of HKickTrackSim then the object of this
// class is returned.

    if(iId < 0)
      {
	::Error("HPidFL::getSplineTrack", "Index <0 not defined!");
        return NULL;
      }

    if((pCat == NULL) && ((pCat = getCategory(catSplineTrack)) == NULL))
      {
        ::Error("HPidFL::getSplineTrack", "No catSplineTrack category");
        return NULL;
      }

    return (HSplineTrack *)pCat->getObject(iId);
}

// -----------------------------------------------------------------------------
HRKTrackB* HPidFL::getRKTrack(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HRKTrackB  object corresponding to iId.
// pCat must point on the catRKTrackB category or gHades->getCurrentEvent()
// must be accessible.
    if(iId < 0)
      {
	::Error("HPidFL::getRKTrack", "Index <0 not defined!");
        return NULL;
      }

    if((pCat == NULL) && ((pCat = getCategory(catRKTrackB)) == NULL))
      {
        ::Error("HPidFL::getRKTrack", "No catRKTrackB category");
        return NULL;
      }

    return (HRKTrackB *)pCat->getObject(iId);
}
// -----------------------------------------------------------------------------

HMdcTrkCand* HPidFL::getMdcTrkCand(Int_t iId, HCategory *pCat) //OK
{
  // Static function which returns HMdcTrkCand object corresponding to iId.
  // pCat must point on the catMdcTrkCand category or gHades->getCurrentEvent()
  // must be accessible.
  
  if(iId < 0)
    {
      ::Error("HPidFL::getMdcTrkCand","Index <0 not defined!");
      return NULL;
    }
  if((pCat == NULL) && ((pCat = getCategory(catMdcTrkCand)) == NULL))
    {
      ::Error("HPidFL::getMdcTrkCand", "No catMdcTrkCand category");
      return NULL;
    }
  
  return (HMdcTrkCand *)pCat->getObject(iId);
}
/*
HMdcSeg* HPidFL::getMdcSeg(Int_t iSector, Int_t iId, HCategory *pCat)
{
// Static function which returns HMdcSeg object corresponding to iId
// in sector iSector.
// pCat must point on the catMdcSeg category or gHades->getCurrentEvent()
// must be accessible.

HLocation  loc;

    if((iSector < 0) || (iSector > 5) || (iId < 0))
    {
      ::Error("HPidFL::getMdcSeg","invalid access-index!");
        return NULL;
    }

    if((pCat == NULL) && ((pCat = getCategory(catMdcSeg)) == NULL))
    {
        ::Error("HPidFL::getMdcSeg", "No catMdcSeg category");
        return NULL;
    }

    loc.setNIndex(3);
    loc[0] = iSector;
    loc[1] = 0;
    loc[2] = iId;

    return (HMdcSeg *)pCat->getObject(loc);
}
// -----------------------------------------------------------------------------

HMdcSegSim* HPidFL::getMdcSegSim(Int_t iSector, Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HMdcSegSim object corresponding to iId
// in sector iSector.
// pCat must point on the catMdcSeg category or gHades->getCurrentEvent()
// must be accessible.

    return (HMdcSegSim *) getMdcSeg(iSector, iId, pCat);
}
*/
// -----------------------------------------------------------------------------

HRichHit* HPidFL::getRichHit(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HRichHit object corresponding to iId index.
// pCat must point on the catRichHit category or gHades->getCurrentEvent()
// must be accessible.

    if(iId < 0)
      {
	::Error("HPidFL::getRichHit","invalid access-index!");
        return NULL;
      }
    
    if((pCat == NULL) && ((pCat = getCategory(catRichHit)) == NULL))
    {
        ::Error("HPidFL::getRichHit", "No catRichHit category");
        return NULL;
    }

    return (HRichHit *)pCat->getObject(iId);
}

// -----------------------------------------------------------------------------

HRichHitSim* HPidFL::getRichHitSim(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HRichHitSim object corresponding to iId index.
// pCat must point on the catRichHitSim category or gHades->getCurrentEvent()
// must be accessible.
// In the second case catRichHit is used instead of catRichHitSim because
// some inaccurate in RICH code.


    if(iId < 0)
      {
	::Error("HPidFL::getRichHitSim","invalid access-index:%d", iId);
        return NULL;
      }


    if((pCat == NULL) && ((pCat = getCategory(catRichHit)) == NULL))
    {
        ::Error("HPidFL::getRichHit", "No catRichHit category");
        return NULL;
    }

    return (HRichHitSim *)pCat->getObject(iId);
}
// -----------------------------------------------------------------------------

HTofHit* HPidFL::getTofHit(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HTofHit object corresponding to iId.
// pCat must point on the catTofHit category or gHades->getCurrentEvent()
// must be accessible.
// In case HPidFL isSim() then HTofHitSim is returned.


    if(iId < 0)
      {
	::Error("HPidFL::getTofHit","invalid access-index!");
        return NULL;
      }



    if((pCat == NULL) && ((pCat = getCategory(catTofHit)) == NULL))
    {
        ::Error("HPidFL::getTofHit", "No catTofHit category");
        return NULL;
    }
    //cout << "Searching for location!" << endl;
    //HLocation loc;

    //loc.setNIndex(3);
    //loc[0] = iId / 1000;
    //loc[1] = (iId % 1000) / 100;
    //loc[2] = iId % 100;
    //cout << loc[0] << " " <<loc[1] << " " <<loc[2] << endl;
    //return (HTofHit *)pCat->getObject(loc);
    return (HTofHit *)pCat->getObject(iId);
}

// -----------------------------------------------------------------------------

HTofHitSim* HPidFL::getTofHitSim(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HTofHitTrack object corresponding to iId index.
// pCat must point on the catTofHit category or gHades->getCurrentEvent()
// must be accessible.

    return (HTofHitSim *)getTofHit(iId, pCat);
}

// -----------------------------------------------------------------------------

HTofCluster *  HPidFL::getTofCluster(Int_t iId, HCategory *pCat){ //OK
//Static function which returns HTofCluster
  if(iId < 0)
    {
      ::Error("HPidFL::getTofCluster","invalid access-index!");
      return NULL;
    }
  
  
  
  if((pCat == NULL) && ((pCat = getCategory(catTofCluster)) == NULL))
    {
      ::Error("HPidFL::HTofCluster", "No catTofCluster category");
      return NULL;
    }
  //HLocation loc;
  
  //loc.setNIndex(3);
  //loc[0] = iId / 1000;
  //loc[1] = (iId % 1000) / 100;
    //loc[2] = iId % 100;
  return (HTofCluster *)pCat->getObject(iId);
  
}
// -----------------------------------------------------------------------------
HTofClusterSim*  HPidFL::getTofClusterSim(Int_t iId, HCategory *pCat){ //OK
	return (HTofClusterSim*) getTofCluster(iId, pCat);
}
// -----------------------------------------------------------------------------

HShowerHitTof* HPidFL::getShowerHitTof(Int_t iId, HCategory *pCat)
{
// Static function which returns HShowerHitTof object corresponding to iId.
// pCat must point on the catShowerHitTof category or gHades->getCurrentEvent()
// must be accessible.

  if(iId < 0)
    {
      ::Error("HPidFL::getShowerHitTof","invalid access-index!");
      return NULL;
    }


  if((pCat == NULL) && ((pCat = getCategory(catShowerHitTof)) == NULL))
    {
      ::Error("HPidFL::getShowerHitTof",
	      "No catShowerHitTof category");
      return NULL;
    }
  
  return (HShowerHitTof *)pCat->getObject(iId);
}

// -----------------------------------------------------------------------------

HShowerHitTofTrack* HPidFL::getShowerHitTofTrack(Int_t iId, HCategory *pCat) //OK
{
// Static function which returns HShowerHitTofTrack object corresponding to iId.
// pCat must point on the catShowerHitTof category or gHades->getCurrentEvent()
// must be accessible.

    if(iId < 0)
    {
      ::Error("HPidFL::getShowerHitTofTrack","invalid access-index!");
      return NULL;
    }


    if((pCat == NULL) && ((pCat = getCategory(catShowerHitTofTrack)) == NULL))
    {
        ::Error("HPidFL::getShowerHitTofTrack",
                "No catShowerHitTofTrack category");
        return NULL;
    }

    return (HShowerHitTofTrack *)pCat->getObject(iId);
}

// -----------------------------------------------------------------------------

HMdcSeg* HPidFL::getMdcSeg(Int_t iId, HCategory *pCat)//OK
{
  // Static function which returns HMdcSeg object corresponding to linear index iId
  // in category
  // pCat must point on the catMdcSeg category or gHades->getCurrentEvent()
  // must be accessible.
  if(iId < 0)
    {
      ::Error("HPidFL::getMdcSeg","invalid access-index!");
      return NULL;
    }
  
  if((pCat == NULL) && ((pCat = getCategory(catMdcSeg)) == NULL))
    {
      ::Error("HPidFL::getMdcSeg", "No catMdcSeg category");
      return NULL;
    }
  
  return (HMdcSeg *)pCat->getObject(iId);
  
}
// -----------------------------------------------------------------------------
HMdcSegSim* HPidFL::getMdcSegSim(Int_t iId, HCategory *pCat) //OK
{
  // Static function which returns HMdcSegSim object corresponding to iId
  // in category
  // pCat must point on the catMdcSeg category or gHades->getCurrentEvent()
  // must be accessible.
  
    return (HMdcSegSim*) getMdcSeg(iId,pCat);
}

// -----------------------------------------------------------------------------
HRichHitIPU* HPidFL::getRichHitIPU(Int_t iId, HCategory *pCat) //OK
{
  // Static function which returns HRichHitIPU object corresponding to iId
  // in category
  // pCat must point on the catRichHitIPU category or gHades->getCurrentEvent()
  // must be accessible.
#if 0 
 if(iId < 0)
    {
      ::Error("HPidFL::getRichHitIPU","invalid access-index!");
      return NULL;
    }

  
    if((pCat == NULL) && ((pCat = getCategory(catHardRichHit)) == NULL))
    {
        ::Error("HPidFL::getRichHitIPU", "No catRichHitIPU category");
        return NULL;
    }

    return (HRichHitIPU *)pCat->getObject(iId);
#endif
return 0;
}

// -----------------------------------------------------------------------------
HRichHitIPUSim*  HPidFL::getRichHitIPUSim(Int_t iId, HCategory *pCat) //OK
{
//  return (HRichHitIPUSim*) getRichHitIPU(iId);
return 0;
}
// -----------------------------------------------------------------------------

HPidCandidate* HPidFL::getPidCandidate(Int_t candIndex, HCategory *pCat) //OK
{
// Returns HPidCandidate object corresponding to candIndex (if exists)
// Works when pCat is set or gHades->getCurrentEvent() is accessible

  if(candIndex < 0)
    {
      ::Error("HPidFL::getPidCandidate","invalid access-index!");
      return NULL;
    }

    if((pCat == NULL) && ((pCat = getCategory(catPidCandidate)) == NULL))
    {
        ::Error("HPidFL::getCandidate",
                "No catPidCandidate category");
        return NULL;
    }
    
    return (HPidCandidate *)pCat->getObject(candIndex);
}

HPidTrackCand* HPidFL::getTrackCand(Int_t trackCandIndex ,HCategory *pCat) //OK
{
// Returns HPidTrackCand object corresponding to candIndex (if exists)
// Works when pCat is set or gHades->getCurrentEvent() is accessible

  if(trackCandIndex < 0)
    {
      ::Error("HPidFL::getTrackCand","invalid access-index!");
      return NULL;
    }


  if((pCat == NULL) && ((pCat = getCategory(catPidTrackCand)) == NULL))
    {
      ::Error("HPidFL::getTrackCand",
	      "No catPidTrackCand category");
      return NULL;
    }
  
  return (HPidTrackCand *)pCat->getObject(trackCandIndex);
}


UInt_t HPidFL::countLeptonsInInnerMdc(HPidGeantTrackSet* theTrackSet,Bool_t verbose) //OK
{
  Int_t leptonCounter=0;
  Int_t wait;
  //Loop over all track ids contributing to this set
  if(verbose) cout <<"This particle has : " << theTrackSet->getNCorrelatedTrackIds() << " track ids " << endl;
  for(UInt_t i=0;i<theTrackSet->getNCorrelatedTrackIds();i++)
    { 
      if(verbose){
	cout <<"The correlation is: "  <<theTrackSet->getCorrelationFlag(i) << "Ckecking now!" <<endl;
	cout << "kSeenInInnerMMDC is coded as: " << kSeenInInnerMDC << endl;
	cout << ((UInt_t)(theTrackSet->getCorrelationFlag(i)) & kSeenInInnerMDC) << endl;
	cout << theTrackSet->wasSeenInDetector(theTrackSet->getCorrelationFlag(i),kSeenInInnerMDC) << endl;
      }
      if(theTrackSet->wasSeenInDetector((UInt_t)(theTrackSet->getCorrelationFlag(i)),
					kSeenInInnerMDC,verbose)) 
	{//If this particle was seen in the inner MDC
	  if(verbose)cout << "This was in mdc" << endl;
	  if(theTrackSet->getGeantPID(i)==2 || theTrackSet->getGeantPID(i)==3) //This was a lepton!
	    {
	      if(verbose) cout << "This was a lepton in mdc" << endl;
	      leptonCounter++;
	    }
	}
    }
  if(verbose) cin >>wait;
  return leptonCounter;

}

Bool_t HPidFL::getGeantMatches(HPidTrackCandSim* giveCandidate,Int_t& nMatchesMdcMeta, 
			       Int_t& nMatchesFull,Int_t& nLeptonicMatchesMdcMeta, Int_t& GeantPid) //OK
{
  Int_t wait;
  nMatchesMdcMeta=0;
  nMatchesFull=0;
  nLeptonicMatchesMdcMeta=0;
  GeantPid=-99;
  HPidGeantTrackSet* theGeantTrackSet = giveCandidate->getGeantTrackSet();
  if(theGeantTrackSet->getCorrelationFlag(0)>=kTrackKick) //This tracks was seen at least in MDC and META
    GeantPid=theGeantTrackSet->getGeantPID(0);

  //Loop over all correlated track ids
  for(UInt_t i = 0;i<theGeantTrackSet->getNCorrelatedTrackIds() ; i++)
    {
      if(theGeantTrackSet->getCorrelationFlag(i)>=kTrackKick) //This particle was seen before and behind
	{                                                     //the magnetic field
	  nMatchesMdcMeta++;
	  if(theGeantTrackSet->getGeantPID(i)==2 || theGeantTrackSet->getGeantPID(i)==3)
	    nLeptonicMatchesMdcMeta++;
	}
      if(theGeantTrackSet->getCorrelationFlag(i)>=kTrackKickRICH)
	{
	  nMatchesFull++;
	}
    }

  if((nLeptonicMatchesMdcMeta>0 || nMatchesMdcMeta>0) && GeantPid==-99)
    {
      cout << "DIAGNOSIS REQUIRED!" << endl;
      for(UInt_t j=0;j<theGeantTrackSet->getNCorrelatedTrackIds();j++)
	{
	  cout << "Correlation: " << theGeantTrackSet->getCorrelationFlag(j) << endl;
	  cout << "Pid: " << theGeantTrackSet->getGeantPID(j) << endl;
	  cout << "Track ID: " << theGeantTrackSet->getGeantTrackID(j) << endl;
	}
      cin >> wait;
    }
  return kTRUE;
}

Bool_t HPidFL::isLepOnMirror(HRichHitSim* r,Int_t &nTrkNbOnMirror) //Copied from Laura ...
{

    HMatrixCategory * mirrcat = (HMatrixCategory*) getCategory(catRichGeantRaw+2);

    HRichGeometryPar* pGeo = (HRichGeometryPar*) getParamContainer("RichGeometryParameters");
    Float_t fYShift = pGeo->getSectorShift(); // to account for shifted volumes in HGEANT !
    fYShift = fYShift/TMath::Cos( 20. * TMath::DegToRad() );//20 degrees is padplane tilt
    HRichPadTab* pPadsTab = ((HRichGeometryPar*)pGeo)->getPadsPar();
    if (!pPadsTab) return kFALSE;
    HRichPad *pPad = 0;
    if(!r) cout << "No hitsim!" <<endl;

    if (!r) return kFALSE;
    if (r)
    {
 	Int_t xRing=r->getRingCenterX();
	Int_t yRing=r->getRingCenterY();
	Int_t secRing=r->getSector();
	// iterate over hits on mirror and compare hit ring x,y with
	// center of reflected photons, calculate pad hit by photon ctr
	HGeantRichMirror *mirr=0;
	HIterator *iter_mirror = (HIterator *)(mirrcat->MakeIterator("native"));
	iter_mirror->Reset();
	while((mirr=(HGeantRichMirror *)iter_mirror->Next())!=0)
	{
	    
	    Int_t aTrk,aID;
	    mirr->getTrack(aTrk,aID);
	    if(matchRingMirrTracks(r,aTrk,aID)>0) // ring has track of kine obj seen traversing the mirror
	      {

		// project back from mirror to padplane
		Float_t x=mirr->getYRing()/10.;//in mm of the padplane
		Float_t y=mirr->getXRing()/10. + fYShift;
		pPad=pPadsTab->getPad(x, // they are swapped, sic !
				      y );// divide by 10 to account for mm->cm
		if (!pPad) 
		  {cout << "No pad" << endl;
		  return kFALSE;
		  }
		Int_t sec=mirr->getSector();
		Int_t xMirrPhotCM=pPad->getPadX();
		Int_t yMirrPhotCM=pPad->getPadY();
		
		// check position of ring cntr and photon cntr
		if (
		    TMath::Abs(xMirrPhotCM-xRing) <  2 && 
		    TMath::Abs(yMirrPhotCM-yRing) <  2 && 
		    sec == secRing
		    )
		  {
		    nTrkNbOnMirror = aTrk;
		    return kTRUE; // ring stems from lepton that made photons with cm
		                  // in position found by pattern recognition in correct place.
		  } 
	      }
	}
    }
    return kFALSE;
}
HGeantKine* HPidFL::getGeantKineObjFromTrkNbFast(Int_t nTrkNb)//OK
{
  HLinearCategory * cat = (HLinearCategory*) getCategory(catGeantKine);
  if(nTrkNb <= 0)
    return NULL;
  
  if(cat == NULL)
    {
      ::Error("HPidFL::getGeantKineObjFromTrkNbFast", "No catGeantKine category");
      return NULL;
    }
  HGeantKine* pKine = (HGeantKine*) cat->getObject(nTrkNb - 1);
  if(pKine->getTrack() == nTrkNb)
    {
      return pKine;
    }
  else 
    {
      ::Error("HPidFL::getGeantKineObjFromTrkNbFast",
	      "Object at position %d has track number different from %d",nTrkNb-1,nTrkNb);
      return NULL;
    }
}

Int_t HPidFL::getGeantParID(Int_t nTrkNb) //OK
{
  HLinearCategory * cat = (HLinearCategory*) getCategory(catGeantKine);
  Int_t dummy,aID;
  aID=0;
  if (!cat) 
    { 
      ::Error("HPidFL::getGeantPiarID","No catgory HGeantKine");
      return -1;
    }
  HGeantKine *kine = getGeantKineObjFromTrkNbFast(nTrkNb);
  if (kine)
    {
      kine->getParticle(dummy,aID);
    }
  return aID;
}


Int_t HPidFL::matchRingMirrTracks(HRichHitSim *r,Int_t trackMirr,Int_t idMirr) //Copied from Laura ...
{
  
  // return weight of photon contribution of track on mirror and in ring
  Int_t t1,t2,t3;
  t1=t2=t3=-1;
  t1=r->track1;t2=r->track2;t3=r->track3;
  // get particle id of 
  Int_t p1,p2,p3;p1=p2=p3=-1;
  if (t1>-1) p1=getGeantParID(t1);
  if (t2>-1) p2=getGeantParID(t2);
  if (t3>-1) p3=getGeantParID(t3);
  
  
  if (t1==trackMirr && idMirr==p1) return r->weigTrack1;
  if (t2==trackMirr && idMirr==p2) return r->weigTrack2;
  if (t3==trackMirr && idMirr==p3) return r->weigTrack3;
  return 0; 
}

HParSet* HPidFL::getParamContainer(const Char_t * contname) //OK
{
  return (HParSet*)((gHades->getRuntimeDb())->getContainer(contname));
}

Bool_t HPidFL::isDoubleLepOnMirror(HRichHitSim* r,
				   Int_t &nTrkNbOnMirror1, Int_t &nTrkNbOnMirror2) //Copied from Laura...
{

    HMatrixCategory * mirrcat = (HMatrixCategory*) getCategory(catRichGeantRaw+2);

    HRichGeometryPar* pGeo = (HRichGeometryPar*) getParamContainer("RichGeometryParameters");
    Float_t fYShift = pGeo->getSectorShift(); // to account for shifted volumes in HGEANT !
    fYShift = fYShift/TMath::Cos( 20. * TMath::DegToRad() );//20 degrees is padplane tilt
    HRichPadTab* pPadsTab = ((HRichGeometryPar*)pGeo)->getPadsPar();
    if (!pPadsTab) return kFALSE;
    HRichPad *pPad = 0;

    Bool_t kFirstLepton,kSecondLepton;kFirstLepton = kSecondLepton = kFALSE;

    if (!r) {
      cout << "No richhit" << endl;
      return kFALSE;
    }
    if (r)
    {
	Int_t xRing=r->getRingCenterX();
	Int_t yRing=r->getRingCenterY();
	Int_t secRing=r->getSector();
	// iterate over hits on mirror and compare hit ring x,y with
	// center of reflected photons, calculate pad hit by photon ctr
	HGeantRichMirror *mirr=0;
	HIterator *iter_mirror = (HIterator *)(mirrcat->MakeIterator("native"));
	iter_mirror->Reset();
	while((mirr=(HGeantRichMirror *)iter_mirror->Next())!=0)
	{

	  Int_t aTrk,aID;
	  mirr->getTrack(aTrk,aID);
	  if(matchRingMirrTracks(r,aTrk,aID)>0) // ring has track of kine obj seen traversing the mirror
	    {
	      // project back from mirror to padplane
	      Float_t x=mirr->getYRing()/10.;//in mm of the padplane
	      Float_t y=mirr->getXRing()/10. + fYShift;
	      pPad=pPadsTab->getPad(x, // they are swapped, sic !
				    y );// divide by 10 to account for mm->cm
	      //cout << "pPad:" <<pPad << endl;
	      if (!pPad) {
		cout << "No pad" << endl;
		return kFALSE;
	      }
	      //cout << "pad reconstructed" << endl;
	      Int_t sec=mirr->getSector();
	      Int_t xMirrPhotCM=pPad->getPadX();
	      Int_t yMirrPhotCM=pPad->getPadY();
	      
	      if (
		  TMath::Abs(xMirrPhotCM-xRing) <  2 && 
		  TMath::Abs(yMirrPhotCM-yRing) <  2 && 
		  sec == secRing                     &&
		  !kFirstLepton
		  )
		{
		  nTrkNbOnMirror1 = aTrk; 
		  kFirstLepton = kTRUE;
		  //cout << "first lepton found" << endl;
		} 
	      else if (
		       TMath::Abs(xMirrPhotCM-xRing) <  4 && 
		       TMath::Abs(yMirrPhotCM-yRing) <  4 && 
		       sec == secRing                     &&
		       !kSecondLepton
		       ) 
		{
		  nTrkNbOnMirror2 = aTrk;
		  kSecondLepton = kTRUE;
		  //cout << "second lepton found" << endl;
		}
	    }
	}
	if ( kFirstLepton && kSecondLepton ) return kTRUE;
    }

    return kFALSE;
}

HPidParticle* HPidFL::getPidParticle(Int_t iIdx, HCategory* pCat) //OK
{
  if(iIdx < 0)
    {
      ::Error("HPidFL::getPidParticle","invalid access-index!");
      return NULL;
    }

    if((pCat == NULL) && ((pCat = getCategory(catPidPart, kFALSE)) == NULL))
    {
        ::Error("HPidParticle::getPidParticle", "No catPidPart category");

        return NULL;
    }

    return (HPidParticle *) pCat->getObject(iIdx);
}

//When a track has multiple rings matching to it we pick the best match according to the angular difference of the coordinates
Int_t HPidFL::pickBestRing(HMetaMatch* pMatch, Float_t fMdcPhi, Float_t fMdcTheta) //OK
{
  TVector3 MdcCoord;
  TVector3 RichCoord;
  MdcCoord.SetMagThetaPhi(1.0,fMdcTheta*TMath::DegToRad(),fMdcPhi*TMath::DegToRad());

  Int_t bestRingIndex=-1;

  Float_t dist[3];
  Float_t closestDistance=-1.0;

  HRichHit* pRing0=getRichHit(pMatch->getARichInd(0));
  if(!pRing0)
    {
      ::Error("HPidFL::pickBestRing","No first ring found!");
      return -1;
    }
  bestRingIndex=0;
  RichCoord.SetMagThetaPhi(1.0,pRing0->getTheta()*TMath::DegToRad(),
			   pRing0->getPhi()*TMath::DegToRad());
  dist[0]=TMath::RadToDeg()*MdcCoord.Angle(RichCoord);


  if(pMatch->getARichInd(1)>-1)
    {
      //cout << "multiple rings. Ind 1 is: " << pMatch->getARichInd(1)<< endl;
      HRichHit* pRing1=getRichHit(pMatch->getARichInd(1));
      RichCoord.SetMagThetaPhi(1.0,pRing1->getTheta()*TMath::DegToRad(),
			       pRing1->getPhi()*TMath::DegToRad());
      //compare 0 and 1
      dist[1]=TMath::RadToDeg()*MdcCoord.Angle(RichCoord);

      if(dist[1] < dist[0])
	{
	  //second ring matches better than first one
	  bestRingIndex=1;
	  closestDistance=dist[1];
	}
      else
	{
	  closestDistance=dist[0];
	}
    }
  
  
  if(pMatch->getARichInd(2)>-1)
    {
      HRichHit* pRing2=getRichHit(pMatch->getARichInd(2));
      RichCoord.SetMagThetaPhi(1.0,pRing2->getTheta()*TMath::DegToRad(),
			       pRing2->getPhi()*TMath::DegToRad());

      dist[2]=TMath::RadToDeg()*MdcCoord.Angle(RichCoord);

      if(dist[2]<closestDistance)
	{
	  //third ring matches best!
	  bestRingIndex=2;
	  closestDistance=dist[2];
	}
    }
  return bestRingIndex;
}