// -------------------------------------------------------------------------
// -----                        CbmTrdDigiDraw source file                  -----
// -----                  Created 03/01/08  by M. Al-Turany            -----
// -------------------------------------------------------------------------
#include "CbmTrdDigiDraw.h"

#include "CbmTrdAddress.h"      // for CbmTrdAddress
#include "CbmTrdDigi.h"         // for CbmTrdDigi
#include "CbmTrdParModDigi.h"   // for CbmTrdParModDigi
#include "CbmTrdParSetDigi.h"   // for CbmTrdParSetDigi

#include <FairLogger.h>         // for Logger, LOG
#include <FairEventManager.h>   // for FairEventManager
#include <FairRootManager.h>    // for FairRootManager
#include <FairRunAna.h>         // for FairRunAna
#include <FairRuntimeDb.h>      // for FairRuntimeDb

#include <TClonesArray.h>       // for TClonesArray
#include <TEveBoxSet.h>         // for TEveBoxSet, TEveBoxSet::kBT_AABox
#include <TEveManager.h>        // for TEveManager, gEve
#include <TVector3.h>           // for TVector3

#include <iomanip>              // for setprecision

using std::setprecision;

// -----   Default constructor   -------------------------------------------
CbmTrdDigiDraw::CbmTrdDigiDraw() 
  : FairTask(),
    fVerbose(0),
    fPointList(nullptr),
    fDigiPar(nullptr),
    fModuleInfo(nullptr),
    fEventManager(nullptr),
    fq(nullptr),
    fColor(),
    fStyle(),
    fActiveLayers()
{
}
// -------------------------------------------------------------------------

                                      

// -----   Standard constructor   ------------------------------------------
CbmTrdDigiDraw::CbmTrdDigiDraw(const char* name, Color_t color ,Style_t mstyle,Int_t iVerbose) 
  : FairTask(name, iVerbose),
    fVerbose(iVerbose),
    fPointList(nullptr),
    fDigiPar(nullptr),
    fModuleInfo(nullptr),
    fEventManager(nullptr),
    fq(nullptr),
    fColor(color),
    fStyle(mstyle),
    fActiveLayers()
{
  for(Int_t i=0; i<15; i++){
    fActiveLayers[i]=kTRUE;
  } 
}
// ----  Initialisation  ----------------------------------------------
void CbmTrdDigiDraw::SetParContainers()
{
    LOG(debug) <<" * CbmTrdDigiDraw * :: SetParContainers() ";


    // Get Base Container
    FairRunAna* ana = FairRunAna::Instance();
    FairRuntimeDb* rtdb=ana->GetRuntimeDb();

    fDigiPar = (CbmTrdParSetDigi*)(rtdb->getContainer("CbmTrdParSetDigi"));

}
// --------------------------------------------------------------------

// -------------------------------------------------------------------------
InitStatus CbmTrdDigiDraw::Init()
{
  LOG(debug) <<  "CbmTrdDigiDraw::Init()";
   FairRootManager* fManager = FairRootManager::Instance();
   fPointList = (TClonesArray *)fManager->GetObject(GetName());
   if(fPointList==0){
      LOG(warn) << "CbmTrdDigiDraw::Init()  branch " << GetName() << " Not found! Task will be deactivated "; 
      SetActive(kFALSE);
    }
   LOG(debug1) <<  "CbmTrdDigiDraw::Init() get track list" <<  fPointList;
   fEventManager =FairEventManager::Instance();
   LOG(debug1) <<  "CbmTrdDigiDraw::Init() get instance of FairEventManager ";
   fq =0;
      
  // gEve->AddElement(fq, fEventManager ); 

   return kSUCCESS;
}
// -------------------------------------------------------------------------  
void CbmTrdDigiDraw::Exec(Option_t* /*option*/)
{
  if (IsActive()){
    
    CbmTrdDigi *digi=0;
    
    Int_t npoints=fPointList->GetEntriesFast();
    Reset();

    TEveBoxSet *q = new TEveBoxSet(GetName(),"");
    q->Reset(TEveBoxSet::kBT_AABox, kTRUE, npoints);
    q->SetMainColor(kBlue);

    //   q->SetMainAlpha(1.);
    //   q->SetMarkerSize(1.5);
    //   q->SetMarkerStyle(fStyle);
    
    //   q->SetDefDepth();
    //   q->SetDefHeight();
    //   q->SetDefDepth();
    
   Int_t refCounter=0;

   for (Int_t i=0; i<npoints; ++i) {   
     
     digi =  (CbmTrdDigi*) fPointList->At(i);
     

     if(digi!=0) {
       
    
       //Double_t ELoss = digi-> GetCharge();

       Int_t address = digi->GetAddress();
       Int_t Col = CbmTrdAddress::GetColumnId(address);
       Int_t Row = CbmTrdAddress::GetRowId(address);
       Int_t Sector = CbmTrdAddress::GetSectorId(address);
       Int_t moduleId = CbmTrdAddress::GetModuleAddress(address);
       Int_t layerId = CbmTrdAddress::GetLayerId(address);

       fModuleInfo = (CbmTrdParModDigi*)fDigiPar->GetModulePar(moduleId);

       TVector3 posHit, padSize;
       fModuleInfo->GetPosition(/*moduleId, */Sector, Col, Row, posHit, padSize);

       // The given point is used as the edge of the box but it is
       // the middle point of the pad. So we have to do a transformation
     
      
       Double_t X = posHit.X()-(padSize.X()/2);
       Double_t Y = posHit.Y()-(padSize.Y()/2);


//       LOG(bebug1)<<"*** CbmTrdHitProducerDigi::CalculateHitPosition ***";
       LOG(debug1)<<"Col: "<< Col;
       LOG(debug1)<<"Row: "<< Row;
       LOG(debug1)<<setprecision(5)<<"fPadX: "<< padSize.X();
       LOG(debug1)<<setprecision(5)<<"fPadY: "<< padSize.Y();
       LOG(debug1)<<setprecision(5)<<"fPosX: "<< posHit.X();
       LOG(debug1)<<setprecision(5)<<"fPosY: "<< posHit.Y();
       LOG(debug1)<<setprecision(5)<<"fPosZ: "<< posHit.Z();
       
       if ( fActiveLayers[layerId]) {
	 q->AddBox(X,Y,posHit.Z(), 
                   padSize.X(), padSize.Y(), 0.);
	 refCounter++;
       }
     }
     
   }
   gEve->AddElement(q);
   gEve->Redraw3D(kFALSE); 
   fq=q;
   }
   
}
// -----   Destructor   ----------------------------------------------------
CbmTrdDigiDraw::~CbmTrdDigiDraw() 
{ 
}
// -------------------------------------------------------------------------
// -------------------------------------------------------------------------
  /** Action after each event**/
void CbmTrdDigiDraw::Finish()
{
}
// -------------------------------------------------------------------------
void CbmTrdDigiDraw::Reset()
{
     if(fq!=0){
	   fq->Reset();
       gEve->RemoveElement(fq, fEventManager );
	 }
}

void CbmTrdDigiDraw::SetActiveLayer(Bool_t Layer1, Bool_t Layer2,
				    Bool_t Layer3, Bool_t Layer4,
				    Bool_t Layer5, Bool_t Layer6,
				    Bool_t Layer7, Bool_t Layer8,
				    Bool_t Layer9, Bool_t Layer10,
				    Bool_t Layer11, Bool_t Layer12,
				    Bool_t Layer13, Bool_t Layer14,
				    Bool_t Layer15) 

{
  fActiveLayers[0]=Layer1;
  fActiveLayers[1]=Layer2;
  fActiveLayers[2]=Layer3;
  fActiveLayers[3]=Layer4;
  fActiveLayers[4]=Layer5;
  fActiveLayers[5]=Layer6;
  fActiveLayers[6]=Layer7;
  fActiveLayers[7]=Layer8;
  fActiveLayers[8]=Layer9;
  fActiveLayers[9]=Layer10;
  fActiveLayers[10]=Layer11;
  fActiveLayers[11]=Layer12;
  fActiveLayers[12]=Layer13;
  fActiveLayers[13]=Layer14;
  fActiveLayers[14]=Layer15;
}


ClassImp(CbmTrdDigiDraw)