//*-- AUTHOR : Vladimir Pechenov
//*-- Modified : 31.05.2001 by Vladimir Pechenov

///////////////////////////////////////////////////////////////////////////////
//  HMdcSegSim
//
// This class is derived from HMdcSim for simulated events.
// Additional data elements are following:
//   nTracks - the number of tracks in the list of tracks numbers
//   listTracks[5] - the list of tracks numbers
//   nTimes[5] - num. of drift times from each track in this segment hit
//     (not cells because one cell can have 2 drift times from 2 tracks)
// This information can be accessed via the functions:
//    Int_t getNTracks(void)
//    Int_t getTrack(const Int_t n)      0 <= n < getNTracks()
//    UChar_t getNTimes(const Int_t n)
//
// The list of tracks is sorted. listTracks[0] is a track which have
// most number of drift times and etc.
// 
// The list of tracks is calculated after filling of HMdcSeg container
// via function  Int_t calcNTracks(void) which return nTracks.
// 
///////////////////////////////////////////////////////////////////////////////

#include "hmdcsegsim.h"
#include "hades.h"                                                                  
#include "hmdcdef.h"                                                          
#include "hmdccal1sim.h"                                                            
#include "hevent.h"                                                                 
#include "hcategory.h" 
#include "hmessagemgr.h"

ClassImp(HMdcSegSim)

void HMdcSegSim::setNTracks(const Int_t nTr, const Int_t* listTr, 
    const UChar_t* nTm) {
  if(nTr<=0) nTracks=-1;
  else {
    nTracks=(nTr<6) ? nTr:5;
    for(Int_t n=0; n<nTracks; n++) {
      listTracks[n]=listTr[n];
      nTimes[n]=nTm[n];
    }
  }
}

Int_t HMdcSegSim::calcNTracks(void) {
  HCategory *fCalCat=gHades->getCurrentEvent()->getCategory(catMdcCal1);
  if( !fCalCat ) {
    nTracks=0;
    return -1;
  }
  HLocation loc;
  loc.set(4,getSec(),0,0,0);
  Int_t seg=getIOSeg();
  Int_t lsTr[96];
  Int_t nTimesTot=0;
  for(Int_t layer=0; layer<12; layer++) {
    loc[1]=seg*2+layer/6;
    loc[2]=layer%6;
    for(Int_t n=0; n<4; n++) {
      UChar_t nCTimes=getSignId(layer,n);
      if(!nCTimes) continue;
      loc[3]=getCell(layer,n);
      HMdcCal1Sim *fCal1Sim=(HMdcCal1Sim*)fCalCat->getObject(loc);
      if(nCTimes&1) lsTr[nTimesTot++]=fCal1Sim->getNTrack1();
      if(nCTimes&2) lsTr[nTimesTot++]=fCal1Sim->getNTrack2();
    }
  }
  Int_t ind[96];
  UChar_t nTm[96];
  Int_t nTr=0;
  for(Int_t n=0; n<nTimesTot; n++) {
    if(lsTr[nTr]<=0) continue;
    lsTr[nTr]=lsTr[n];
    nTm[nTr]=1;
    ind[nTr]=nTr;
    for(Int_t m=n+1; m<nTimesTot; m++) {
      if(lsTr[m] != lsTr[nTr]) continue;
      nTm[nTr]++;
      lsTr[m]=-1;
    }
    nTr++;
  }
  //Sorting:
  if(nTr>1) {
    Bool_t sort=kTRUE;
    while(sort) {
      sort=kFALSE;
      for(Int_t n=0; n<nTr-1; n++) {
        if(nTm[ind[n]]<nTm[ind[n+1]]) {
          sort=kTRUE;
          Int_t i=ind[n];
          ind[n]=ind[n+1];
          ind[n+1]=i;
        }
      }
    }
  }
  nTracks=(nTr>5) ? 5:nTr;
  for(Int_t n=0; n<nTracks; n++) {
    listTracks[n]=lsTr[ind[n]];
    nTimes[n]=nTm[ind[n]];
  }
  return nTracks;
}

void HMdcSegSim::print(void) {
  gHades->getMsg()->info(10,DET_MDC,GetName(),"Sec.%i Seg.%i, %i tracks: ",
			 getSec(),getIOSeg(),nTracks);
  for(Int_t n=0; n<nTracks; n++) gHades->getMsg()->info(10,DET_MDC,GetName(),
						"%i(%i) ",listTracks[n],nTimes[n]);	
  INFO_msg(10,DET_MDC,"[trNum(NOfTimes)]");
}