#include "AliEveTRDModuleImp.h"
#include "AliEveTRDData.h"
-#include "TMath.h"
-#include <TGListTree.h>
+#include "TGListTree.h"
+#include "TClonesArray.h"
+#include "TGeoManager.h"
+#include "TGeoMatrix.h"
#include "TEveManager.h"
#include "TEveTrack.h"
+#include "TEveGeoNode.h"
+#include "TEveTrans.h"
+
#include "AliLog.h"
-#include "AliRun.h"
-#include "AliTRDv1.h"
#include "AliTRDgeometry.h"
#include "AliTRDCommonParam.h"
#include "AliTRDpadPlane.h"
-#include "AliTRDdigit.h"
#include "AliTRDhit.h"
#include "AliTRDcluster.h"
-#include "AliTRDcalibDB.h"
-#include "AliTRDdataArrayI.h"
#include "AliTRDmcmTracklet.h"
ClassImp(AliEveTRDChamber)
AliEveTRDNode::AliEveTRDNode(const char *typ, Int_t det) :
TEveElement(), AliEveTRDModule(typ, det)
{
+ // Xonstructor.
}
//______________________________________________________________________________
void AliEveTRDNode::Paint(Option_t* option)
{
+ // Paint object.
+
List_i iter = fChildren.begin();
while(iter != fChildren.end()){
(dynamic_cast<AliEveTRDModule*>(*iter))->Paint(option);
//______________________________________________________________________________
void AliEveTRDNode::Reset()
{
+ // Reset.
+
List_i iter = fChildren.begin();
while(iter != fChildren.end()){
(dynamic_cast<AliEveTRDModule*>(*iter))->Reset();
//______________________________________________________________________________
void AliEveTRDNode::Collapse()
{
+ // Collapse.
+
TGListTree *list = gEve->GetListTree();
AliEveTRDNode *node = 0x0;
List_i iter = fChildren.begin();
//______________________________________________________________________________
void AliEveTRDNode::Expand()
{
+ // Expand.
+
TGListTree *list = gEve->GetListTree();
AliEveTRDNode *node = 0x0;
List_i iter = fChildren.begin();
//______________________________________________________________________________
void AliEveTRDNode::EnableListElements()
{
+ // Enable list elements.
+
SetRnrSelf(kTRUE);
AliEveTRDNode *node = 0x0;
AliEveTRDChamber *chmb = 0x0;
//______________________________________________________________________________
void AliEveTRDNode::DisableListElements()
{
+ // Disable list elements.
+
SetRnrSelf(kFALSE);
AliEveTRDNode *node = 0x0;
AliEveTRDChamber *chmb = 0x0;
//______________________________________________________________________________
void AliEveTRDNode::UpdateLeaves()
{
+ // Update leaves.
+
AliEveTRDModule *module;
List_i iter = fChildren.begin();
while(iter != fChildren.end()){
//______________________________________________________________________________
void AliEveTRDNode::UpdateNode()
{
+ // Update node.
+
// Info("UpdateNode()", Form("%s", GetName()));
AliEveTRDNode *node = 0x0;
List_i iter = fChildren.begin();
//______________________________________________________________________________
AliEveTRDChamber::AliEveTRDChamber(Int_t det) :
- TEveElement(), AliEveTRDModule("Chmb", det),
- fDigits(0), fHits(0), fRecPoints(0), fTracklets(0),
- fRowMax(-1), fColMax(-1), fTimeMax(22), fSamplingFrequency(0),
- fX0(0.), fPla(-1),
- fPadPlane (0),
- fGeo (0)
+ TEveElement()
+ ,AliEveTRDModule("Chmb", det)
+ ,fDigits(0x0)
+ ,fHits(0x0)
+ ,fRecPoints(0x0)
+ ,fTracklets(0x0)
+ ,fGeo(0x0)
+ ,fShape(0x0)
+ ,fNrows(-1)
+ ,fNcols(-1)
+ ,fNtime(22)
{
//
// Constructor
//
-
- AliTRDCommonParam* parCom = AliTRDCommonParam::Instance();
- fSamplingFrequency = parCom->GetSamplingFrequency();
}
-//______________________________________________________________________________
-Int_t AliEveTRDChamber::GetSM() const
-{
- if(!fGeo){
- AliWarning("Fail. No TRD geometry defined.");
- return -1;
- }
- return fGeo->GetSector(fDet);
-}
-
-//______________________________________________________________________________
-Int_t AliEveTRDChamber::GetSTK() const
-{
- if(!fGeo){
- AliWarning("Fail. No TRD geometry defined.");
- return -1;
- }
- return fGeo->GetChamber(fDet);
-}
//______________________________________________________________________________
void AliEveTRDChamber::LoadClusters(TObjArray *clusters)
return;
}
- if(!fRecPoints){
- fRecPoints = new AliEveTRDClusters(this);
- fRecPoints->SetMarkerSize(1.);
- fRecPoints->SetMarkerStyle(24);
- fRecPoints->SetMarkerColor(6);
- fRecPoints->SetOwnIds(kTRUE);
- } else fRecPoints->Reset();
+ if(!fRecPoints){
+ AddElement(fRecPoints = new AliEveTRDClusters());
+ fRecPoints->SetTitle(Form("Clusters for Det %d", GetID()));
+ }
+ fRecPoints->Reset();
Float_t q;
- Double_t cloc[3], cglo[3];
-
+ Float_t g[3]; //global coordinates
AliTRDcluster *c=0x0;
- for(int iclus=0; iclus<clusters->GetEntriesFast(); iclus++){
+ Int_t nc = clusters->GetEntriesFast();
+ for(int iclus=0; iclus<nc; iclus++){
c = (AliTRDcluster*)clusters->UncheckedAt(iclus);
- cloc[0] = c->GetX();
- cloc[1] = c->GetY();
- cloc[2] = c->GetZ();
+ c->GetGlobalXYZ(g);
q = c->GetQ();
- fGeo->RotateBack(fDet,cloc,cglo);
- fRecPoints->SetNextPoint(cglo[0], cglo[1], cglo[2]);
- fRecPoints->SetPointId(c);
+ Int_t id = fRecPoints->SetNextPoint(g[0], g[1], g[2]);
+ fRecPoints->SetPointId(id, new AliTRDcluster(*c));
}
+ fRecPoints->StampObjProps();
fLoadRecPoints = kTRUE;
}
AliError(Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
return;
}
- // Info("LoadDigits()", Form("digits =0x%x", digits));
- if(!fDigits) fDigits = new AliEveTRDDigits(this);
- else fDigits->Reset();
+ if(!fDigits) AddElement(fDigits = new AliEveTRDDigits(this));
+ fDigits->Reset();
fDigits->SetData(digits);
+ fDigits->StampObjProps();
+ fDigitsNeedRecompute = kTRUE;
fLoadDigits = kTRUE;
}
//______________________________________________________________________________
-void AliEveTRDChamber::AddHit(AliTRDhit *hit)
+void AliEveTRDChamber::LoadHits(TClonesArray *hits, Int_t &idx)
{
//
// Draw hits
//
- // Info("AddHit()", Form("%s", GetName()));
- if(!fHits){
- fHits = new AliEveTRDHits(this);
- fHits->SetMarkerSize(.1);
- fHits->SetMarkerColor(2);
- fHits->SetOwnIds(kTRUE);
+ if(!fHits){
+ AddElement(fHits = new AliEveTRDHits());
+ fHits->SetTitle(Form("Hits for Det %d", GetID()));
}
-
- fHits->SetNextPoint(hit->X(), hit->Y(), hit->Z());
- fHits->SetPointId(hit);
fLoadHits = kTRUE;
+ Int_t nhits = hits->GetEntriesFast();
+
+ AliTRDhit *hit = 0x0;
+ while(idx<nhits){
+ hit = (AliTRDhit*)hits->UncheckedAt(idx);
+ if(hit->GetDetector() != fDet) return;
+
+ Int_t id = fHits->SetNextPoint(hit->X(), hit->Y(), hit->Z());
+ fHits->SetPointId(id, new AliTRDhit(*hit));
+ fHits->StampObjProps();
+ idx++;
+ }
+ return;
}
//______________________________________________________________________________
Error("LoadTracklets()", Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
return;
}
- // Info("LoadTracklets()", Form("tracks = 0x%x", tracks));
if(!fTracklets){
fTracklets = new std::vector<TEveTrack*>;
//____________________________________________________
void AliEveTRDChamber::Paint(Option_t* option)
{
- /* Info("Paint()", Form("%s", GetName()));*/
+ // Paint object.
+
if(!fRnrSelf) return;
if(fDigits && fRnrDigits){
if(fDigitsNeedRecompute){
}
fDigits->Paint(option);
}
+
if(fRecPoints && fRnrRecPoints) fRecPoints->GetObject()->Paint(option);
+
if(fHits && fRnrHits) fHits->GetObject()->Paint(option);
+
if(fTracklets && fRnrTracklets){
for(std::vector<TEveTrack*>::iterator i=fTracklets->begin(); i != fTracklets->end(); ++i) (*i)->Paint(option);
}
//______________________________________________________________________________
void AliEveTRDChamber::Reset()
{
+ // Reset.
+
if(fHits){
fHits->Reset();
fLoadHits = kFALSE;
if(fDigits){
fDigits->Reset();
fLoadDigits = kFALSE;
+ fDigitsNeedRecompute = kTRUE;
}
if(fRecPoints){
fRecPoints->Reset();
//______________________________________________________________________________
void AliEveTRDChamber::SetGeometry(AliTRDgeometry *geo)
{
- fGeo = geo;
+ // Set geometry.
- fPla = fGeo->GetPlane(fDet);
- fX0 = fGeo->GetTime0(fPla);
+ fGeo = geo;
- fPadPlane = fGeo->GetPadPlane(fPla,fGeo->GetChamber(fDet));
- fRowMax = fPadPlane->GetNrows();
- fColMax = fPadPlane->GetNcols();
+ Int_t ism = geo->GetSector(fDet);
+ Int_t istk = geo->GetStack(fDet);
+ Int_t ilyr = geo->GetLayer(fDet);
+ Int_t icha = istk*6+ilyr;
+
+ // define pad plane size in pads
+ AliTRDpadPlane *pp = fGeo->GetPadPlane(ilyr, istk);
+ fNrows = pp->GetNrows();
+ fNcols = pp->GetNcols();
+
+// this version for setting the rendarable object is not working very nice
+// Int_t shape_offset = TEveGeoShape::Class()->GetDataMemberOffset("fShape");
+// TEveGeoShape* eg_shape = new TEveGeoShape("geometry");
+// eg_shape->RefMainTrans().SetFrom(* gGeoManager->GetCurrentMatrix());
+// * (TGeoShape**) (((char*)eg_shape) + shape_offset) = gGeoManager->GetCurrentVolume()->GetShape();
+//
+// eg_shape->StampColorSelection();
+
+ // define rendarable volumes
+ gGeoManager->cd(Form("/B077_1/BSEGMO%d_1/BTRD%d_1/UTR1_1/UTS1_1/UTI1_1/UT%02d_1", ism, ism, icha));
+ fShape = new TEveGeoTopNode(gGeoManager, gGeoManager->GetCurrentNode());
+ fShape->RefMainTrans().SetFrom(*gGeoManager->GetCurrentMatrix());
+ fShape->DisableListElements();
+ fShape->SetRnrSelf(kFALSE);
+// try to set the properties but it is crashing !!
+// TEveGeoNode *node = 0x0;
+// if((node = (TEveGeoNode*)fShape->FindChild(Form("UA%02d_1", icha)))) node->SetRnrState(kTRUE);
+// else AliWarning(Form("Can not retrieve geo node UA%02d_1", icha));
+
+ AddElement(fShape);
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff