PndMvdDigiPixelDraw.cxx

Go to the documentation of this file.

/*
 * PndMvdDigiPixelDraw.cpp
 *
 *  Created on: Apr 20, 2009
 *      Author: stockman
 */

#include "PndMvdDigiPixelDraw.h"

#include "PndSdsCalcFePixel.h"
#include "PndSdsPixelDigiPar.h"
#include "PndSdsHit.h"
#include "PndSdsCluster.h"

#include "FairRootManager.h"

#include "TGeoMatrix.h"
#include "TClonesArray.h"
#include "TEveTrans.h"
#include "TEveManager.h"
#include "TEveBoxSet.h"

PndMvdDigiPixelDraw::PndMvdDigiPixelDraw() : fPixelSize(0.01), fBoxSize(1)
{}

PndMvdDigiPixelDraw::PndMvdDigiPixelDraw(const char* name, Int_t iVerbose) : FairBoxSetDraw(name, iVerbose), fPixelSize(0.01), fBoxSize(1)
{}

PndMvdDigiPixelDraw::~PndMvdDigiPixelDraw()
{
        delete(fGeoH);
}

InitStatus PndMvdDigiPixelDraw::Init()
{
        FairBoxSetDraw::Init();
        fGeoH = PndGeoHandling::Instance();
        fClusterCands = (TClonesArray *)fManager->GetObject("MVDPixelClusterCand");
        fRecoHits     = (TClonesArray *)fManager->GetObject("MVDHitsPixel");

        if(fClusterCands != 0 && fRecoHits != 0) fUseCluster = kTRUE;
        return kSUCCESS;
}

void PndMvdDigiPixelDraw::Exec(Option_t*)
{
        PndSdsPixelDigiPar digipar;
        digipar.SetFECols(104);
        digipar.SetFERows(104);
        digipar.SetMaxFEperCol(10);
        digipar.SetMaxFEperRow(10);
        PndSdsCalcFePixel calc(digipar);
        Int_t col, row, fe;
        Reset();
        if (fUseCluster == kTRUE){
                for (int hitNr = 0; hitNr < fRecoHits->GetEntriesFast(); hitNr++){
                        PndSdsHit*              myRecoHit = (PndSdsHit*)fRecoHits->At(hitNr);
                        TVector3 recoVector;
                        myRecoHit->Position(recoVector);
                        TVector3 recoLocal = fGeoH->MasterToLocalShortId(recoVector, myRecoHit->GetSensorID());
                        TVector3 SensDim = fGeoH->GetSensorDimensionsShortId(myRecoHit->GetSensorID());
                        TString detName = fGeoH->GetPath(myRecoHit->GetSensorID());
                        Int_t sensorId = myRecoHit->GetSensorID();
                        if (fModules[sensorId] == 0)
                                fModules[sensorId] = CreateNewBoxSet(detName);

                        PndSdsCluster* myCluster = (PndSdsCluster*)fClusterCands->At(myRecoHit->GetRefIndex());
                        for (int clusterNr = 0; clusterNr < myCluster->GetClusterSize(); clusterNr++){
                                PndSdsDigiPixel* p = (PndSdsDigiPixel*)fList->At(myCluster->GetDigiIndex(clusterNr));
                                col = p->GetPixelColumn();
                                row = p->GetPixelRow();
                                fe  = p->GetFE();
                                std::cout << "Fe, Col, Row: " << fe << " " << col << "/" << row << std::endl;
                                calc.CalcSensorColRow(col, row, p->GetFE());
                                std::cout << "Col, Row " << p->GetSensorID() << " : " << col << "/" << row << std::endl;
                                TVector3 digiLocal(((col+0.5) - (SensDim.X()*100))*fPixelSize, ((row+0.5)-(SensDim.Y()*100))*fPixelSize, 0);
                                std::cout << "DigiLocal: " << digiLocal.X() << " " << digiLocal.Y() << " " << digiLocal.Z() << std::endl;
                                std::cout << "RecoLocal: " << recoLocal.X() << " " << recoLocal.Y() << " " << recoLocal.Z() << std::endl;
                                TVector3 diff = digiLocal - recoLocal;
                                std::cout << "Diff: " << diff.X() << " " << diff.Y() << " " << diff.Z() << std::endl;
                                TVector3 final = diff * (fBoxSize/fPixelSize);
                                final += recoLocal;
                                std::cout << "Final: " << final.X() << " " << final.Y() << " " << final.Z() << std::endl;

                                fModules[p->GetSensorID()]->AddBox(final.X(), final.Y(), final.Z());
                                fModules[p->GetSensorID()]->DigitValue(p->GetCharge());
                        }
                }
        }
        else {
                for (Int_t i=0; i<fList->GetEntriesFast(); ++i) {
                        PndSdsDigiPixel* p = (PndSdsDigiPixel*)fList->At(i);
                        TString detName = fGeoH->GetPath(p->GetSensorID());
                        Int_t sensorId = p->GetSensorID();

                        if (fModules[sensorId] == 0)
                                fModules[sensorId] = CreateNewBoxSet(detName);

                        col = p->GetPixelColumn();
                        row = p->GetPixelRow();
                        fe  = p->GetFE();
                        std::cout << "Fe, Col, Row: " << fe << " " << col << "/" << row << std::endl;
                        calc.CalcSensorColRow(col, row, p->GetFE());
                        std::cout << "Col, Row " << p->GetSensorID() << " : " << col << "/" << row << std::endl;
                        TVector3 SensDim = fGeoH->GetSensorDimensionsShortId(p->GetSensorID());
                        std::cout << "SensDim: " << SensDim.X() << " " << SensDim.Y() << " " << SensDim.Z() << std::endl;
                        fModules[p->GetSensorID()]->AddBox(((col+0.5) - (SensDim.X()*100))*0.01, ((row+0.5)-(SensDim.Y()*100))*0.01, 0.01);
                        fModules[p->GetSensorID()]->DigitValue(p->GetCharge());
                }
        }

        for (boxSetMapIter it = fModules.begin(); it != fModules.end(); it++){
                   TGeoHMatrix testMatrix = *(fGeoH->GetMatrixShortId(it->first));
                   //TGeoHMatrix invMatrix = testMatrix.Inverse();
                   //Double_t scale[] = {0.01,0.01,0.01};               //Dim is 0.1 mm //[R.K. 01/2017] unused variable
                   //testMatrix.SetScale(scale);
                   TEveTrans& t = it->second->RefMainTrans();
                   t.SetFrom(testMatrix);
                   //TEveElement* el = (TEveElement*)it->second; //[R.K. 01/2017] unused variable
                   TEveElement* man = (TEveElement*)fEventManager;
//                 gEve->AddElement(it->second, fEventManager);
                   gEve->AddElement(it->second, man);
        }

        gEve->Redraw3D(kFALSE);
}

TEveBoxSet* PndMvdDigiPixelDraw::CreateNewBoxSet(TString& name)
{
        TEveBoxSet* bs = new TEveBoxSet(name);
        bs->Reset(TEveBoxSet::kBT_AABoxFixedDim, kFALSE, 32);
        bs->SetDefWidth(fBoxSize);
        bs->SetDefHeight(fBoxSize);
        bs->SetDefDepth(fBoxSize);

        return bs;
}

void PndMvdDigiPixelDraw::Reset()
{
        for (boxSetMapIter it = fModules.begin(); it != fModules.end(); it++){
                //TEveElement* el = (TEveElement*)it->second; //[R.K. 01/2017] unused variable
                TEveElement* man = (TEveElement*)fEventManager;
//                 gEve->AddElement(it->second, fEventManager);
                gEve->AddElement(it->second, man);
        }
        fModules.clear();
}

ClassImp(PndMvdDigiPixelDraw);