// -------------------------------------------------------------------------
// -----                       CbmVisMuch source file                  -----
// -----                  Created 14/03/07  by E. Kryshen              -----
// -------------------------------------------------------------------------


#include "CbmVisMuch.h"

#include "TApplication.h"
#include "TClonesArray.h"
#include "TGClient.h"
#include "TGButton.h"
#include "TGFrame.h"
#include "TGStatusBar.h"

#include "FairRootManager.h"
#include "FairRunAna.h"
#include "FairRuntimeDb.h"
#include "CbmGeoMuchPar.h"
#include "CbmMCTrack.h"

#include "CbmMuchStation.h"
#include "CbmMuchSector.h"
#include "CbmMuchPoint.h"
#include "CbmMuchDigi.h"
#include "CbmMuchDigiMatch.h"
#include "CbmMuchCluster.h"
#include "CbmMuchPixelHit.h"
#include "CbmMuchStrawHit.h"
#include "CbmMuchGeoScheme.h"

#include "CbmVisMuchStationFrame.h"
#include "CbmVisMuchModuleFrame.h"
#include "CbmVisPoint.h"
#include "CbmVisHit.h"
#include "CbmVisPixelHit.h"
#include "CbmVisStripHit.h"
#include "CbmVisMuchCluster.h"
#include "CbmMuchLayer.h"
#include "CbmMuchLayerSide.h"
#include "CbmMuchModule.h"
#include "CbmMuchModuleGem.h"
#include "CbmMuchPad.h"
#include "TColor.h"
#include "TFile.h"

// -----   Default constructor   ------------------------------------------
CbmVisMuch::CbmVisMuch():FairTask("Task",0){
  fDigiFileName = "";
  fEvent  = 0;
  fMCTracks    = NULL;
  fPoints      = NULL;
  fDigis       = NULL;
  fDigiMatches = NULL;
  fPixelHits   = NULL;
  fStripHits   = NULL;
  fNstations   = 0;
  fOpenModules = new TObjArray();
  fSumEvents   = kFALSE;
  fVisClusters = new TClonesArray("CbmVisMuchCluster",10);
}
// -------------------------------------------------------------------------


// -----   Destructor   ----------------------------------------------------
CbmVisMuch::~CbmVisMuch() {
  if ( fOpenModules ) fOpenModules->Delete(); delete fOpenModules;
}
// -------------------------------------------------------------------------


// -----   Private method SetParContainers   -------------------------------
void CbmVisMuch::SetParContainers() {
//  FairRunAna* fRun = FairRunAna::Instance();
//  FairRuntimeDb* db = fRun->GetRuntimeDb();
//  fGeoPar  = (CbmGeoMuchPar*)  db->getContainer("CbmGeoMuchPar");
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
InitStatus CbmVisMuch::Init() {
  TObjArray* stations = (TObjArray*) fDigiFile->Get("stations");
  if (!stations) Fatal("Init","No array of CbmMuchStations");

  fGeoScheme=CbmMuchGeoScheme::Instance();
  fGeoScheme->Init(stations);
  fGeoScheme->CreatePointArrays();
  fGeoScheme->CreateHitArrays();
  fGeoScheme->CreateClusterArrays();

  fRootManager = FairRootManager::Instance();
  fMCTracks    = (TClonesArray*) fRootManager->GetObject("MCTrack");
  fPoints      = (TClonesArray*) fRootManager->GetObject("MuchPoint");
  fPixelHits   = (TClonesArray*) fRootManager->GetObject("MuchPixelHit");
  fStripHits   = (TClonesArray*) fRootManager->GetObject("MuchStrawHit");
  fDigis       = (TClonesArray*) fRootManager->GetObject("MuchDigi");
  fDigiMatches = (TClonesArray*) fRootManager->GetObject("MuchDigiMatch");
  fClusters    = (TClonesArray*) fRootManager->GetObject("MuchCluster");

  CreateDisplay();
  return kSUCCESS;
}
// -------------------------------------------------------------------------

void CbmVisMuch::CloseWindow(){
    gApplication->Terminate(0);
}


// -------------------------------------------------------------------------
void CbmVisMuch::CreateDisplay(){
  fDisplay = new TGMainFrame(gClient->GetRoot(),800,800,kVerticalFrame);
  fDisplay->SetWindowName("MUCH Event Display");
  fDisplay->Connect("CloseWindow()","TApplication",gApplication,"Terminate()");

  fMain = new TGVerticalFrame(fDisplay);

  Int_t parts[] = {20,65,15};
  fStatusBar = new TGStatusBar(fDisplay,50,10,kHorizontalFrame);
  fStatusBar->SetParts(parts,3);

  fStationFrame = new CbmVisMuchStationFrame(fMain,this);

  //Button Frame
  fButtonFrame = new TGHorizontalFrame(fMain,500,100);

  fPrev = new TGTextButton(fButtonFrame,"&Previous");
  fNext = new TGTextButton(fButtonFrame,"&Next");
  fExit = new TGTextButton(fButtonFrame,"&Exit","gApplication->Terminate(0)");

  fNext->Connect("Clicked()","CbmVisMuch",this,"ReadNextEvent()");
  fPrev->Connect("Clicked()","CbmVisMuch",this,"ReadPrevEvent()");

  TGLayoutHints* lhButton = new TGLayoutHints(kLHintsCenterX,5,5,2,2);
  fButtonFrame->AddFrame(fPrev,lhButton);
  fButtonFrame->AddFrame(fNext,lhButton);
  fButtonFrame->AddFrame(fExit,lhButton);

  fMain->AddFrame(fStationFrame, new TGLayoutHints(kLHintsExpandX|kLHintsExpandY,2,2,2,2));
  fMain->AddFrame(fButtonFrame , new TGLayoutHints(kLHintsBottom|kLHintsCenterX,2,2,2,2));

  fDisplay->AddFrame(fMain     ,new TGLayoutHints(kLHintsTop|kLHintsLeft|kLHintsExpandX|kLHintsExpandY,2,2,2,2));
  fDisplay->AddFrame(fStatusBar,new TGLayoutHints(kLHintsBottom|kLHintsLeft |kLHintsExpandX,0,0,2,0));


  fDisplay->MapSubwindows();
  fDisplay->Resize(fDisplay->GetDefaultSize());
  fDisplay->MapWindow();
  ReadEvent(fEvent);
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
void CbmVisMuch::ReadEvent(Int_t event){
  int verbose = 0;
  fEvent=event;
  printf("Event:%i\n",fEvent);
  fRootManager->ReadEvent(fEvent);

  //Clear point and hit arrays
  fGeoScheme->ClearPointArrays();
  fGeoScheme->ClearHitArrays();
  fGeoScheme->ClearClusterArrays();

  printf("Reset pads\n");
  vector<CbmMuchModule*> modules = fGeoScheme->GetModules();
  for (Int_t im=0;im<modules.size();im++){
    CbmMuchModule* module = modules[im];
    if (module->GetDetectorType()!=1) continue;
    CbmMuchModuleGem* gem_module = (CbmMuchModuleGem*) module;
    vector<CbmMuchPad*> pads = gem_module->GetPads();
    for (Int_t ip=0;ip<pads.size();ip++) pads[ip]->Reset();
  }

  fVisClusters->Clear();


  printf("Fill arrays of CbmVisPoints... nPoints = %i\n",fPoints->GetEntriesFast());
  //Create arrays of CbmVisPoints
  for (Int_t i=0;i<fPoints->GetEntriesFast();i++){
    CbmMuchPoint*  point  = (CbmMuchPoint*) fPoints->At(i);
    if ( ! point->IsUsable() ) continue;
    CbmMuchModule* module = fGeoScheme->GetModuleByDetId(point->GetDetectorID());
    TClonesArray &points = *(module->GetPoints());
    Int_t index = points.GetEntriesFast();
    new (points[index]) CbmVisPoint(point);
    SetPointInfo((CbmVisPoint*) points[index]);
  }

  printf("Fill arrays of CbmVisPixelHits... nHits = %i\n",fPixelHits->GetEntriesFast());
  //Create arrays of CbmVisHits
  for (Int_t i=0;i<fPixelHits->GetEntriesFast();i++){
    CbmMuchPixelHit* hit = (CbmMuchPixelHit*) fPixelHits->At(i);
    CbmMuchModule* module = fGeoScheme->GetModuleByDetId(hit->GetDetectorId());
    TClonesArray &hits = *(module->GetHits());
    Int_t index = hits.GetEntriesFast();
    new (hits[index]) CbmVisPixelHit(hit);
    //SetHitInfo((CbmVisHit*) hits[index]);
  }

  if (fStripHits) {
    printf("Fill arrays of CbmVisStripHits... nHits = %i\n",fStripHits->GetEntriesFast());
    //Create arrays of CbmVisHits
    for (Int_t i=0;i<fStripHits->GetEntriesFast();i++){
      CbmMuchStrawHit* hit = (CbmMuchStrawHit*) fStripHits->At(i);
      CbmMuchModule* module = fGeoScheme->GetModuleByDetId(hit->GetDetectorId());
      TClonesArray &hits = *(module->GetHits());
      Int_t index = hits.GetEntriesFast();
      new (hits[index]) CbmVisStripHit(hit);
      //SetHitInfo((CbmVisHit*) hits[index]);
    }
  }

  printf("Mark fired pads ... nPads = %i\n",fDigis->GetEntriesFast());
  for (Int_t i=0;i<fDigis->GetEntriesFast();i++){
    CbmMuchDigi* digi = (CbmMuchDigi*) fDigis->At(i);
    CbmMuchModule* module = (CbmMuchModule*) fGeoScheme->GetModuleByDetId(digi->GetDetectorId());
    if (module->GetDetectorType()!=1) continue;
    CbmMuchModuleGem* gem_module = (CbmMuchModuleGem*) module;
    CbmMuchPad* pad = gem_module->GetPad(digi->GetChannelId());
    pad->SetFired(i,digi->GetADCCharge());
  }

  printf("Fill arrays of CbmVisMuchClusters... nClusters = %i\n",fClusters->GetEntriesFast());
  for (Int_t i=0;i<fClusters->GetEntriesFast();i++){
    new ((*fVisClusters)[i]) CbmVisMuchCluster(i);
  }

//  for (Int_t i=0;i<fClusters->GetEntriesFast();i++){
//    CbmMuchCluster* cluster = (CbmMuchCluster*) fClusters->At(i);
//    if (!cluster->GetNDigis()) {
//      Error("ReadEvent","Cluster with empty array of digi indices");
//      continue;
//    }
//    CbmMuchDigi* digi      = (CbmMuchDigi*) fDigis->At(cluster->GetDigiIndex(0));
//    CbmMuchModule* module  = fGeoScheme->GetModuleByDetId(digi->GetDetectorId());
//    TClonesArray &clusters = *(module->GetClusters());
//    Int_t index = clusters.GetEntriesFast();
//    new (clusters[index]) CbmVisMuchCluster(i);
//  }



  // Update current station view
  fStationFrame->UpdateStation();

  // Update open modules
  for (Int_t i=0;i<fOpenModules->GetEntriesFast();i++){
    CbmVisMuchModuleFrame* modFrame = (CbmVisMuchModuleFrame*) fOpenModules->At(i);
    if (!modFrame) continue;
    modFrame->UpdateModule();
  }
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
void CbmVisMuch::ReadNextEvent(){
  fEvent++;
  ReadEvent(fEvent);
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
void CbmVisMuch::ReadPrevEvent(){
  fEvent--;
  ReadEvent(fEvent);
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
Int_t CbmVisMuch::SetPointInfo(CbmVisPoint* vpoint){
  Int_t trackId = vpoint->GetTrackId();
  if (trackId<0) return 2;
  CbmMCTrack* track = (CbmMCTrack*) fMCTracks->At(trackId);
  vpoint->SetTrack(track);
  if (!track) return 3;
  vpoint->SetMotherPdg(GetMotherPdg(track));
  return 0;
}
// -------------------------------------------------------------------------


// -------------------------------------------------------------------------
Int_t CbmVisMuch::SetHitInfo(CbmVisHit* vhit){
//  CbmMuchPixelHit* hit  = (CbmMuchPixelHit*) vhit->GetHit();
//  Int_t digiId = hit->GetDigi();
//  CbmMuchDigiMatch* match = (CbmMuchDigiMatch*) fDigiMatches->At(digiId);
//
//  Int_t pointId = match->GetRefIndex();
//  vhit->SetPointId(pointId);
//
//  if (pointId<0) return 1;
//  FairMCPoint*  point  = (FairMCPoint*) fPoints->At(pointId);
//  Int_t trackId = point->GetTrackID();
//  vhit->SetTrackId(trackId);
//
//  if (trackId<0) return 2;
//  CbmMCTrack* track = (CbmMCTrack*) fMCTracks->At(trackId);
//  if (!track) return 3;
//  vhit->SetTrack(track);
//  vhit->SetMotherPdg(GetMotherPdg(track));
  return 0;
}
// -------------------------------------------------------------------------

Int_t CbmVisMuch::GetMotherPdg(CbmMCTrack* track){
  Int_t motherId = track->GetMotherId();
  if (motherId<0) return 0;
  CbmMCTrack* mother = (CbmMCTrack*) fMCTracks->At(motherId);
  if (!mother) return 0;
  return mother->GetPdgCode();
}

ClassImp(CbmVisMuch)