1 // Author: Benjamin Hess 29/01/2010
3 /*************************************************************************
4 * Copyright (C) 2009-2010, Alexandru Bercuci, Benjamin Hess. *
5 * All rights reserved. *
6 *************************************************************************/
8 //////////////////////////////////////////////////////////////////////////
10 // AliEveListAnalyser //
12 // An AliEveListAnalyser is, in principal, a TEveElementList with some //
13 // sophisticated features. You can add macros to this list, which then //
14 // can be applied to the list of analysis objects (these objects can be //
15 // added to the list in the same way as for the TEveElementList, but //
16 // also "by clicking" (cf. AliEveListAnaLyserEditor)). //
17 // In general, please use AddMacro(...) for this purpose. //
18 // Macros that are no longer needed can be removed from the list via //
19 // RemoveSelectedMacros(...). This function takes an iterator of the //
20 // list of macros that are to be removed. //
21 // An entry looks like: //
22 // The data for each macro consists of path, name, type and the command //
23 // that will be used to apply the macro. This stuff is stored in a map //
24 // which takes the macro name for the key and the above mentioned data //
25 // in a TGeneralMacroData-object for the value. //
26 // You can get the macro type via GetMacroType(...). //
27 // To find the type of objects the macro will deal with (corresponds to //
28 // "YourObjectType" in the examples below) please use //
29 // GetMacroObjectType(...). //
30 // With ApplySOSelectionMacros(...) or ApplyProcessMacros(...) //
31 // respectively you can apply the macros to the track list via //
32 // iterators (same style like for RemoveSelectedMacros(...)(cf. above)).//
33 // Selection macros (de-)select macros according to a selection rule //
34 // by setting the rnr-state of the tracks. //
35 // If multiple selection macros are applied, a track is selected, if //
36 // all selection macros select the track. //
37 // Process macros create data or histograms, which will be stored in //
38 // a temporary file. The editor of this class will access this file //
39 // and draw all the stuff within it's DrawHistos() function. The file //
40 // will be deleted by the destructor. //
42 // Currently, the following macro types are supported: //
43 // Selection macros: //
44 // Bool_t YourMacro(const YourObjectType*); //
45 // Bool_t YourMacro(const YourObjectType*, const YourObjectType2*); //
48 // void YourMacro(const YourObjectType*, Double_t*&, Int_t&); //
49 // void YourMacro(const YourObjectType*, const YourObjectType2*, //
50 // Double_t*&, Int_t&); //
51 // TH1* YourMacro(const YourObjectType*); //
52 // TH1* YourMacro(const YourObjectType*, const YourObjectType2*); //
54 // The macros which take 2 tracks are applied to all track pairs //
55 // (whereby BOTH tracks of the pair have to be selected by the single //
56 // track selection macros and have to be unequal, otherwise they will //
57 // be skipped) that have been selected by ALL correlated tracks //
58 // selection macros. The selection macros with 2 tracks do NOT affect //
59 // process macros that process only a single track! //
60 //////////////////////////////////////////////////////////////////////////
63 // Uncomment to display debugging infos
64 //#define AliEveListAnalyser_DEBUG
66 #include <TEveManager.h>
67 #include <TEveSelection.h>
69 #include <TFunction.h>
73 #include <TMethodArg.h>
74 #include <TMethodCall.h>
75 #include <TObjString.h>
80 #include <TTreeStream.h>
82 #include <AliTRDReconstructor.h>
84 #include <EveDet/AliEveListAnalyser.h>
85 #include <EveDet/AliEveListAnalyserEditor.h>
87 ClassImp(AliEveListAnalyser)
89 ///////////////////////////////////////////////////////////
90 ///////////// AliEveListAnalyser ////////////////////////
91 ///////////////////////////////////////////////////////////
92 AliEveListAnalyser::AliEveListAnalyser(const Text_t* n, const Text_t* t, Bool_t doColor):
93 TEveElementList(n, t, doColor),
95 fDataFromMacroList(0x0),
99 fHistoDataSelected(0),
100 fMacroListSelected(0),
101 fSelectedTab(2) // Standard tab: "Apply macros" (index 2)
103 // Creates the AliEveListAnalyser.
105 // Only accept childs of type TEveElement
106 SetChildClass(TEveElement::Class());
108 // Allocate memory for the lists and declare them as owners of their contents
109 fDataFromMacroList = new TList();
110 fDataFromMacroList->TCollection::SetOwner(kTRUE);
112 fMacroList = new TMap();
113 // Set map to owner of it's objects to delete them, if they are removed from the map
114 fMacroList->SetOwnerKeyValue(kTRUE, kTRUE);
116 // Set the build directory for AClic
117 if(gSystem->AccessPathName(Form("%s/.QArec" , gSystem->Getenv("HOME")))) gSystem->Exec("mkdir $HOME/.QArec");
118 gSystem->SetBuildDir(Form("%s/.QArec", gSystem->Getenv("HOME")));
120 AddStandardContent();
123 //______________________________________________________
124 AliEveListAnalyser::~AliEveListAnalyser()
126 // Frees allocated memory (lists etc.).
128 // Stop adding objects
131 // Let the editor know that the list will be destroyed -> The editor will save the data
134 fEditor->SaveMacroList(fMacroList);
138 if (fDataFromMacroList != 0)
140 fDataFromMacroList->Delete();
141 delete fDataFromMacroList;
142 fDataFromMacroList = 0;
151 fMacroList->DeleteAll();
155 // Note: gSystem->AccessPathName(...) returns kTRUE, if the access FAILED!
156 if(!gSystem->AccessPathName(Form("/tmp/ListAnalyserMacroData_%s.root", gSystem->Getenv("USER"))))
157 gSystem->Exec(Form("rm /tmp/ListAnalyserMacroData_%s.root", gSystem->Getenv("USER")));
160 //______________________________________________________
161 Int_t AliEveListAnalyser::AddMacro(const Char_t* path, const Char_t* nameC, Bool_t forceReload)
163 // Checks, if the file exists and if the signature is correct.
164 // If these criteria are fullfilled, the library for this macro is built
165 // and the macro is added to the corresponding list.
166 // Supported macro types:
168 // Bool_t YourMacro(const YourObjectType*)
169 // Bool_t YourMacro(const YourObjectType*, const YourObjectType2*)
172 // void YourMacro(const YourObjectType*, Double_t*&, Int_t&)
173 // void YourMacro(const YourObjectType*, const YourObjectType2*, Double_t*&, Int_t&)
174 // TH1* YourMacro(const YourObjectType*)
175 // TH1* YourMacro(const YourObjectType*, const YourObjectType2*)
177 Char_t pathname[fkMaxMacroPathNameLength];
178 memset(pathname, '\0', sizeof(Char_t) * fkMaxMacroPathNameLength);
180 // Expand the path and create the pathname
181 Char_t* systemPath = gSystem->ExpandPathName(path);
182 snprintf(pathname, fkMaxMacroPathNameLength, "%s/%s", systemPath, nameC);
186 // Delete ".C" from filename
187 Char_t name[fkMaxMacroNameLength];
188 memset(name, '\0', sizeof(Char_t) * fkMaxMacroNameLength);
190 for (UInt_t ind = 0; ind < fkMaxMacroNameLength && ind < strlen(nameC) - 2; ind++) name[ind] = nameC[ind];
192 // Check, if files exists
194 if((fp = fopen(pathname, "rb"))){
197 } else return NOT_EXIST_ERROR;
199 // Clean up root, load the desired macro and then check the type of the macro
200 // A.B. gROOT->Reset();
202 gROOT->ProcessLineSync(Form(".L %s+%c", pathname, forceReload ? '+' : ' '));
207 objectType = GetMacroObjectType(name, 1);
208 objectType2 = GetMacroObjectType(name, 2);
210 // We need this line... otherwise, in some cases, there will be problems concerning ACLIC
211 gROOT->ProcessLineSync(Form(".L %s", pathname));
213 if (!objectType) return UNKNOWN_OBJECT_TYPE_ERROR;
215 // This might be a macro dealing with only 1 object... test this afterwards!
216 Bool_t testSecondObj = kFALSE;
219 objectType2 = TObject::Class();
220 testSecondObj = kTRUE;
222 AliEveListAnalyserMacroType type = GetMacroType(name, objectType->GetName(), objectType2->GetName(), kFALSE);
228 case AliEveListAnalyser::kCorrelObjectSelect:
229 case AliEveListAnalyser::kCorrelObjectAnalyse:
230 case AliEveListAnalyser::kCorrelObjectHisto:
231 // There must be a second type -> Error!
232 return UNKNOWN_OBJECT_TYPE_ERROR;
235 // Ok, single object macro!
242 // A.B. gROOT->Reset();
244 // Has not the correct signature!
245 if (type == kUnknown) return SIGNATURE_ERROR;
247 // Only add macro, if it is not already in the list
248 Int_t returnValue = WARNING;
249 if(fMacroList->GetValue(name) == 0)
251 returnValue = AddMacroFast(path, name, type, objectType, objectType2) ? SUCCESS : ERROR;
257 //______________________________________________________
258 Bool_t AliEveListAnalyser::AddMacroFast(const Char_t* path, const Char_t* name, AliEveListAnalyserMacroType type,
259 TClass* objectType, TClass* objectType2)
261 // Adds a macro (path/name) to the corresponding list. No checks are performed (file exists,
262 // macro already in list/map, signature correct), no libraries are created!
263 // You can use this function only, if the macro has been added successfully before
264 // (and then maybe was removed). The function is very fast. On success kTRUE is returned, otherwise: kFALSE;
265 // Note: If your macro takes only 1 pointer as a parameter, just use "0x0" for objectType2!
267 Bool_t success = kFALSE;
271 case kSingleObjectSelect:
272 case kCorrelObjectSelect:
273 case kSingleObjectAnalyse:
274 case kSingleObjectHisto:
275 case kCorrelObjectAnalyse:
276 case kCorrelObjectHisto:
277 fMacroList->Add(new TObjString(name), new TGeneralMacroData(name, path, type, objectType, objectType2));
279 // We do not know, where the element has been inserted - deselect this list
280 fMacroListSelected = 0;
284 #ifdef AliEveListAnalyser_DEBUG
285 // Successfull add will only be displayed in debug mode
286 printf("AliEveListAnalyser::AddMacroFast: Added macro \"%s/%s\" with object types \"%s\" and \"%s\" to the corresponding list\n",
287 path, name, objectType->GetName(), objectType2->GetName());
293 // Error will always be displayed
294 printf("AliEveListAnalyser::AddMacroFast: ERROR: Could not add macro \"%s/%s\" with object types \"%s\" and \"%s\" to the corresponding list\n", path, name, objectType->GetName(), objectType2->GetName());
305 //______________________________________________________
306 Int_t AliEveListAnalyser::AddPrimSelectedObject(TEveElement* el)
308 // Adds the TEveElement el to the list. If it is already in the list, it is removed.
309 // If the list is the only parent of the clicked object, the object is moved outside the list in the browser (not deleted!).
310 // If you want to delete the object, just select it there and choose "Destroy" in the menu.
311 // This function is designed to be used together with a signal:
312 // It adds the (primarily) selected objects in the viewer to the list (objects that are already in the list are removed!).
313 // Returns "ERROR" (cf. defines) on error, "WARNING" if the element does not contain any user data and else "SUCCESS" (i.e.
314 // the element has been added successfully or the element is the list itself and therefore ignored, or the element is ignored
315 // because it has been added via secondary selection).
319 Error("AliEveListAnalyser::AddPrimSelectedObject", "Zero pointer!\n");
324 // If the clicked element is the list itself, just do nothing.
325 if (el == this) return SUCCESS;
327 if (!this->HasChild(el))
330 // Ignore objects that do not have any user data, since these cannot be used in the analysis!
331 if (el->GetUserData() == 0x0)
333 Warning("AddPrimSelectedObject", "Selected object does not contain any \"user data\" and therefore is ignored!");
338 // Element clicked that is not in the list (and is not the list itself!) -> Add this element to the list
339 this->AddElement(el);
340 this->SetTitle(Form("Objects %d", this->NumChildren()));
345 // Element clicked that is already in the list. Remove it. But: Only take care of objects that have been added
346 // via primary selection (name does not start with "[sec")
347 if (TString(el->GetElementName()).BeginsWith("[sec:")) return SUCCESS;
350 // Element is a child of this list. So, if there are only 2 parents, we know them: list + eve selection. In this case,
351 // the element needs to be destroyed. If there are more parents, just remove the element from the list.
352 // Since the elements editor will be opened, the element is not deleted, but "moved" outside the list (in the browser).
353 if (el->NumParents() > 2)
355 this->RemoveElement(el);
359 // There must be at least 2 parents!
360 if (el->NumParents() <= 1) return ERROR;
362 TEveElement* listObj = 0x0;
363 listObj = this->FindChild(el->GetElementName());
364 if (!listObj) return ERROR;
366 gEve->AddElement(listObj, 0);
367 // Alternatively: Switch on that the element is NOT destroyed, instead of adding it outside the list. Warning: Memory leaks possible.
368 //listObj->SetDestroyOnZeroRefCnt(kFALSE);
369 this->RemoveElement(listObj);
370 //gEve->RemoveElement(listObj, 0);
374 this->SetTitle(Form("Objects %d", this->NumChildren()));
382 //______________________________________________________
383 void AliEveListAnalyser::AddPrimSelectedObjects()
385 // Adds the (primarily) selected objects in the viewer to the list (objects that are already in the list are ignored).
386 // Hold the CTRL-key for multiple selection.
388 TEveSelection* eveSel = gEve->GetSelection();
391 Error("AliEveListAnalyser::AddPrimSelectedObjects", "Failed to get the selection!\n");
395 TEveElement* elem = 0x0;
396 Bool_t changedSomething = kFALSE;
398 for (TEveElement::List_i iter = eveSel->BeginChildren(); iter != eveSel->EndChildren(); ++iter)
400 if(!(elem = dynamic_cast<TEveElement*>(*iter))) continue;
402 if (!this->HasChild(elem) && elem != this)
404 // Element clicked that is not in the list (and is not the list itself!) -> Add this element to list
405 this->AddElement(elem);
406 this->SetTitle(Form("Objects %d", this->NumChildren()));
407 changedSomething = kTRUE;
411 if (changedSomething) gEve->Redraw3D();
415 //______________________________________________________
416 void AliEveListAnalyser::AddSecSelectedSingleObjectToList(Int_t pointId)
418 // This function adds the selected object (secondary selection in the viewer) to the list
419 // of analysis objects. If the object is already in the list, it will be removed from it.
420 // This function is used to add single objects of a TEvePointset, e.g. single clusters.
422 TEvePointSet* ps = dynamic_cast<TEvePointSet*>((TQObject*) gTQSender);
425 Error("AliEveListAnalyser::AddSecSelectedSingleObjectToList", "Zero pointer!\n");
429 // Check, if object is already there. If so, remove it!
431 // 1st possibility: Object of the list clicked. But: Only take care of objects that have been added
432 // via secondary selection (name starts with "[sec"). Note: HasChild will also return kTRUE, if e.g.
433 // the whole TEvePointSet of clusters is in the last (but maybe another point of it has been clicked!)
435 if (this->HasChild(ps))
437 if (TString(ps->GetName()).BeginsWith("[sec:"))
439 // I don't know why, but you get a crash, if you try this->RemoveElement(ps) (in some cases).
440 // So, use this way instead.
441 TEveElement* listObj = this->FindChild(ps->GetName());
444 listObj->SetUserData(0x0);
445 this->RemoveElement(listObj);
446 this->SetTitle(Form("Objects %d", this->NumChildren()));
453 TObject* obj = ps->GetPointId(pointId);
456 // 2nd possibility: Same object clicked again
457 TEveElement* listObj = 0x0;
458 listObj = this->FindChild(Form("[sec:%d] %s%d", obj->GetUniqueID(), obj->GetName(), pointId));
461 listObj->SetUserData(0x0);
462 this->RemoveElement(listObj);
463 this->SetTitle(Form("Objects %d", this->NumChildren()));
467 // Object clicked that is not in the list -> Add this object to list
468 TEvePointSet* newPS = new TEvePointSet(Form("[sec:%d] %s%d", obj->GetUniqueID(), obj->GetName(), pointId));
469 Double_t x = 0, y = 0, z = 0;
470 ps->GetPoint(pointId, x, y, z);
471 newPS->SetPoint(0, x, y, z);
472 newPS->SetUserData(obj);
473 // Choose yellow for the added points and inherit style and size for the marker
474 newPS->SetMarkerColor(5);
475 newPS->SetMarkerStyle(ps->GetMarkerStyle());
476 newPS->SetMarkerSize(ps->GetMarkerSize());
477 // Own points -> Will be cleared, if this object is removed
478 newPS->SetOwnIds(kTRUE);
480 this->AddElement(newPS);
481 this->SetTitle(Form("Objects %d", this->NumChildren()));
486 Error("AliEveListAnalyser::AddSecSelectedSingleObjectToList", "Selected object is NULL and therefore ignored!");
490 //______________________________________________________
491 void AliEveListAnalyser::AddStandardContent()
493 // Adds standard macros to the macro list.
495 // Add your standard macros here, e.g.:
496 // To add a macro use:
497 // AddMacro("$(ALICE_ROOT)/myFolder", "myMacroName.C");
498 // -> If the file does not exist, nothing happens. So if you want to handle this,
499 // use the return value of AddMacro (NOT_EXIST_ERROR is returned, if file does not exist)
500 // (-> You can also check for other return values (see AddMacro(...)))
504 //______________________________________________________
505 Bool_t AliEveListAnalyser::ApplyProcessMacros(const TList* selIterator, const TList* procIterator)
507 // Uses the procIterator (for the selected process macros) to apply the selected macros to the data.
508 // Returns kTRUE on success, otherwise kFALSE. If there no process macros selected, kTRUE is returned
509 // (this is no error!).
510 // The single object process macros are applied to all selected objects.
511 // The selIterator (for the selected selection macros) will be used to apply the correlated objects selection
512 // macros to all object pairs (whereby BOTH objects have to be selected, otherwise they will be skipped).
513 // All object pairs that have been selected by ALL correlated objects selection macros will be processed by
514 // the correlated objects process macros.
516 // No process macros need to be processed
517 if (procIterator->GetEntries() <= 0) return kTRUE;
520 // A.B. gROOT->Reset();
522 // Clear old data and re-allocate
523 if (fDataTree == NULL){
524 TDirectory *cwd = gDirectory;
525 fDataTree = new TTreeSRedirector(Form("/tmp/ListAnalyserMacroData_%s.root", gSystem->Getenv("USER")));
529 Error("Apply process macros", Form("File \"/tmp/ListAnalyserMacroData_%s.root\" could not be accessed properly!",
530 gSystem->Getenv("USER")));
534 if (fDataFromMacroList != 0) {
535 fDataFromMacroList->Delete();
536 delete fDataFromMacroList;
538 fDataFromMacroList = new TList();
539 fDataFromMacroList->TCollection::SetOwner(kTRUE);
541 fHistoDataSelected = 0;
542 TGeneralMacroData* macro(NULL);
544 TString* procCmds = new TString[procIterator->GetEntries()];
545 AliEveListAnalyserMacroType* mProcType = new AliEveListAnalyserMacroType[procIterator->GetEntries()];
546 TClass** mProcObjectType = new TClass*[procIterator->GetEntries()];
547 TClass** mProcObjectType2 = new TClass*[procIterator->GetEntries()];
549 TString* selCmds(NULL);
550 AliEveListAnalyserMacroType* mSelType(NULL);
551 TClass** mSelObjectType(NULL);
552 TClass** mSelObjectType2(NULL);
554 Bool_t selectedByCorrSelMacro = kFALSE;
556 AliEveListAnalyserMacroType macroType = kUnknown;
557 Int_t numHistoMacros = 0;
560 TEveElement* object1(NULL);
561 TEveElement* object2(NULL);
562 TH1* returnedHist(NULL);
564 // Collect the commands for each process macro and add them to "data-from-list"
565 for (Int_t i = 0; i < procIterator->GetEntries(); i++){
566 macro = (TGeneralMacroData*)fMacroList->GetValue(procIterator->At(i)->GetTitle());
569 Error("Apply process macros",
570 Form("Macro list is corrupted: Macro \"%s\" is not registered!",
571 procIterator->At(i)->GetTitle()));
575 #ifdef AliEveListAnalyser_DEBUG
576 printf("AliEveListAnalyser: Checking process macro: %s\n", macro->GetName());
579 // Find the object types of the macro
580 mProcObjectType[i] = macro->GetObjectType();
581 mProcObjectType2[i] = macro->GetObjectType2();
583 // Find the type of the process macro
584 macroType = macro->GetType();
585 if (macroType == kSingleObjectHisto || macroType == kCorrelObjectHisto){
586 mProcType[i] = macroType;
588 // Create the command
589 procCmds[i] = macro->GetCmd();
591 // Add to "data-from-list" -> Mark as a histo macro with the substring "(histo macro)"
592 fDataFromMacroList->Add(new TObjString(Form("%s (histo macro)", macro->GetName())));
593 } else if (macroType == kSingleObjectAnalyse || macroType == kCorrelObjectAnalyse) {
594 mProcType[i] = macroType;
595 // Create the command
596 procCmds[i] = macro->GetCmd();
598 // Add to "data-from-list"
599 fDataFromMacroList->Add(new TObjString(macro->GetName()));
601 Error("Apply process macros",
602 Form("Macro list corrupted: Macro \"%s/%s.C\" is not registered as a process macro!",
603 macro->GetPath(), macro->GetName()));
604 mProcType[i] = kUnknown;
608 // Collect the commands for each selection macro and add them to "data-from-list"
609 if (selIterator->GetEntries() > 0) {
610 selCmds = new TString[selIterator->GetEntries()];
611 mSelType = new AliEveListAnalyserMacroType[selIterator->GetEntries()];
612 mSelObjectType = new TClass*[selIterator->GetEntries()];
613 mSelObjectType2 = new TClass*[selIterator->GetEntries()];
614 for (Int_t i = 0; i < selIterator->GetEntries(); i++){
615 macro = (TGeneralMacroData*)fMacroList->GetValue(selIterator->At(i)->GetTitle());
618 Error("Apply process macros",
619 Form("Macro list is corrupted: Macro \"%s\" is not registered!",
620 selIterator->At(i)->GetTitle()));
624 #ifdef AliEveListAnalyser_DEBUG
625 printf("AliEveListAnalyser: Checking selection macro: %s\n", macro->GetName());
628 // Find the object types of the macro
629 mSelObjectType[i] = macro->GetObjectType();
630 mSelObjectType2[i] = macro->GetObjectType2();
632 // Find the type of the process macro
633 macroType = macro->GetType();
635 // Single Object select macro
636 if (macroType == kSingleObjectSelect) {
637 // Has already been processed by ApplySOSelectionMacros(...)
638 mSelType[i] = macroType;
640 // Correlated Objects select macro
641 else if (macroType == kCorrelObjectSelect) {
642 mSelType[i] = macroType;
644 // Create the command
645 selCmds[i] = macro->GetCmd();
647 Error("Apply process macros",
648 Form("Macro list corrupted: Macro \"%s/%s.C\" is not registered as a selection macro!",
649 macro->GetPath(), macro->GetName()));
650 mSelType[i] = kUnknown;
655 // Allocate memory for the histograms
656 if (numHistoMacros > 0){
657 histos = new TH1*[numHistoMacros];
658 memset(histos, 0, numHistoMacros*sizeof(TH1*));
660 Bool_t secondBeforeFirstObject = kTRUE;
663 //////////////////////////////////////
664 // WALK THROUGH THE LIST OF OBJECTS //
665 //////////////////////////////////////
666 for (TEveElement::List_i iter = this->BeginChildren(); iter != this->EndChildren(); ++iter){
667 if(!(object1 = dynamic_cast<TEveElement*>(*iter))) continue;
669 // Skip objects that have not been selected
670 if (!object1->GetRnrState()) continue;
672 // Cast to the "real" object behind
673 gROOT->ProcessLineSync(Form("TEveElement *automaticEveElement = (TEveElement*)%p;", (void*)object1));
674 gROOT->ProcessLineSync("TObject* automaticObject_1 = (TObject*)automaticEveElement->GetUserData();");
676 // Collect data for each macro
677 for (Int_t i = 0, histoIndex = 0; i < procIterator->GetEntries(); i++){
678 // Find the type of the object and relate it to the macro object type
679 // Only apply macro to this object, if...
680 // ... the macro takes objects of exactly this type.
681 // ... the macro object type is a child of this object's type.
682 // Otherwise: Continue
684 // Finally, via procCmds[i], the automatic objects are casted to the correct type and analysed by each macro!
685 if (((TObject*)object1->GetUserData())->IsA() != mProcObjectType[i] &&
686 !((TObject*)object1->GetUserData())->InheritsFrom(mProcObjectType[i])) continue;
689 // Single object histo
690 if (mProcType[i] == kSingleObjectHisto){
691 returnedHist = (TH1*)gROOT->ProcessLineSync(procCmds[i]);
692 if (histos && returnedHist)
694 if (!histos[histoIndex]) histos[histoIndex] = returnedHist;
697 histos[histoIndex]->Add((const TH1*)returnedHist);
703 // Correlated Objects histo
704 } else if (mProcType[i] == kCorrelObjectHisto) {
705 // To get all pairs, do the second loop over all objects.
706 // But: If a macro takes 2 pointers of the same type, we must take care that one gets the same pair, when we exchange the objects
707 // (this is not true, if we have different types - even if they inherit from the same classes!).
708 // Thus: If the latter case occurs, we ignore an object pair, if the second object is BEFORE the first object in the list.
709 // Since then the pair has already been taken into account.
710 // Furthermore, we have a pair of objects, if and only if both objects of the pair are selected (Rnr-state)
711 // and are not equal.
712 // The correlated objects process macro will be applied to all pairs that will be additionally selected by
713 // all correlated objects selection macros.
715 secondBeforeFirstObject = kTRUE;
716 for (TEveElement::List_i iter2 = this->BeginChildren(); iter2 != this->EndChildren(); ++iter2)
718 // If the objects are the same, it is not a pair -> continue. From now on: 2nd object BEHIND the 1st object in the list!
721 secondBeforeFirstObject = kFALSE;
724 if(!(object2 = dynamic_cast<TEveElement*>(*iter2))) continue;
726 // Skip objects that have not been selected
727 if (!object2->GetRnrState()) continue;
729 // Same check of the macro object type as before
730 if (((TObject*)object2->GetUserData())->IsA() != mProcObjectType2[i] &&
731 !((TObject*)object2->GetUserData())->InheritsFrom(mProcObjectType2[i])) continue;
732 // Do not process object pairs twice
733 if (secondBeforeFirstObject)
735 if (mProcObjectType[i] == mProcObjectType2[i]) continue;
738 // Cast to the "real" object behind
739 gROOT->ProcessLineSync(Form("TEveElement *automaticEveElement = (TEveElement*)%p;", (void*)object2));
740 gROOT->ProcessLineSync("TObject* automaticObject_2 = (TObject*)automaticEveElement->GetUserData();");
742 // Select object by default (so it will be processed, if there are no correlated objects selection macros!)
743 selectedByCorrSelMacro = kTRUE;
744 for (Int_t j = 0; j < selIterator->GetEntries(); j++){
745 if (mSelType[j] == kCorrelObjectSelect){
746 // Check, whether the macro can deal with both objects. If not, skip it.
747 // Note: Again, via selCmds[i], the automatic objects are casted to the correct type!
748 if (((TObject*)object1->GetUserData())->IsA() != mSelObjectType[j] &&
749 !((TObject*)object1->GetUserData())->InheritsFrom(mSelObjectType[j])) continue;
750 if (((TObject*)object2->GetUserData())->IsA() != mSelObjectType2[j] &&
751 !((TObject*)object2->GetUserData())->InheritsFrom(mSelObjectType2[j])) continue;
753 selectedByCorrSelMacro = (Bool_t)gROOT->ProcessLineSync(selCmds[j]);
754 if (!selectedByCorrSelMacro) break;
758 // If the pair has not been selected by the correlated objects selection macros, skip it!
759 if (!selectedByCorrSelMacro) continue;
761 returnedHist = (TH1*)gROOT->ProcessLineSync(procCmds[i]);
764 if (!histos[histoIndex]) histos[histoIndex] = returnedHist;
767 histos[histoIndex]->Add((const TH1*)returnedHist);
776 // Single object analyse
777 else if (mProcType[i] == kSingleObjectAnalyse) {
778 // Create data pointers in CINT, execute the macro and get the data
779 gROOT->ProcessLineSync("Double_t* results = 0;");
780 gROOT->ProcessLineSync("Int_t n = 0;");
781 gROOT->ProcessLineSync(procCmds[i]);
782 Double_t* results = (Double_t*)gROOT->ProcessLineSync("results;");
783 Int_t nResults = (Int_t)gROOT->ProcessLineSync("n;");
786 Error("Apply macros", Form("Error reading data from macro \"%s\"", procIterator->At(i)->GetTitle()));
789 for (Int_t resInd = 0; resInd < nResults; resInd++){
790 (*fDataTree) << Form("ObjectData%d", i) << Form("Macro%d=", i) << results[resInd] << (Char_t*)"\n";
796 // Correlated objects analyse
797 else if (mProcType[i] == kCorrelObjectAnalyse){
798 // To get all pairs, do the second loop over all objects.
799 // But: If a macro takes 2 pointers of the same type, we must take care that one gets the same pair, when we exchange the objects
800 // (this is not true, if we have different types - even if they inherit from the same classes!).
801 // Thus: If the latter case occurs, we ignore an object pair, if the second object is BEFORE the first object in the list.
802 // Since then the pair has already been taken into account.
803 // Furthermore, we have a pair of objects, if and only if both objects of the pair are selected (Rnr-state)
804 // and are not equal.
805 // The correlated objects process macro will be applied to all pairs that will be additionally selected by
806 // all correlated objects selection macros.
808 secondBeforeFirstObject = kTRUE;
809 for (TEveElement::List_i iter2 = this->BeginChildren(); iter2 != this->EndChildren(); ++iter2)
811 // If the objects are the same, it is not a pair -> continue. From now on: 2nd object BEHIND the 1st object in the list!
814 secondBeforeFirstObject = kFALSE;
817 if(!(object2 = dynamic_cast<TEveElement*>(*iter2))) continue;
819 // Skip objects that have not been selected
820 if (!object2->GetRnrState()) continue;
822 // Same check of the macro object type as before
823 if (((TObject*)object2->GetUserData())->IsA() != mProcObjectType2[i] &&
824 !((TObject*)object2->GetUserData())->InheritsFrom(mProcObjectType2[i])) continue;
825 // Do not process object pairs twice
826 if (secondBeforeFirstObject)
828 if (mProcObjectType[i] == mProcObjectType2[i]) continue;
831 // Cast to the "real" object behind
832 gROOT->ProcessLineSync(Form("TEveElement *automaticEveElement = (TEveElement*)%p;", (void*)object2));
833 gROOT->ProcessLineSync("TObject* automaticObject_2 = (TObject*)automaticEveElement->GetUserData();");
835 // Select object by default (so it will be processed, if there are no correlated objects selection macros!)
836 selectedByCorrSelMacro = kTRUE;
837 for (Int_t j = 0; j < selIterator->GetEntries(); j++) {
838 if (mSelType[j] == kCorrelObjectSelect) {
839 // Check, whether the macro can deal with both objects. If not, skip it.
840 // Note: Again, via selCmds[i], the automatic objects are casted to the correct type!
841 if (((TObject*)object1->GetUserData())->IsA() != mSelObjectType[j] &&
842 !((TObject*)object1->GetUserData())->InheritsFrom(mSelObjectType[j])) continue;
843 if (((TObject*)object2->GetUserData())->IsA() != mSelObjectType2[j] &&
844 !((TObject*)object2->GetUserData())->InheritsFrom(mSelObjectType2[j])) continue;
846 selectedByCorrSelMacro = (Bool_t)gROOT->ProcessLineSync(selCmds[j]);
847 if (!selectedByCorrSelMacro) break;
851 // If the pair has not been selected by the correlated objects selection macros, skip it!
852 if (!selectedByCorrSelMacro) continue;
854 // Create data pointers in CINT, execute the macro and get the data
855 gROOT->ProcessLineSync("Double_t* results = 0;");
856 gROOT->ProcessLineSync("Int_t n = 0;");
857 gROOT->ProcessLineSync(procCmds[i]);
858 Double_t* results = (Double_t*)gROOT->ProcessLineSync("results;");
859 Int_t nResults = (Int_t)gROOT->ProcessLineSync("n;");
862 Error("Apply macros", Form("Error reading data from macro \"%s\"", procIterator->At(i)->GetTitle()));
865 for (Int_t resInd = 0; resInd < nResults; resInd++) {
866 (*fDataTree) << Form("ObjectData%d", i) << Form("Macro%d=", i) << results[resInd] << (Char_t*)"\n";
876 for (Int_t i = 0, histoIndex = 0; i < procIterator->GetEntries() && histoIndex < numHistoMacros; i++) {
877 if (mProcType[i] == kSingleObjectHisto || mProcType[i] == kCorrelObjectHisto) {
878 // Might be empty (e.g. no objects have been selected)!
879 if (histos[histoIndex]) {
880 (*fDataTree) << Form("ObjectData%d", i) << Form("Macro%d=", i) << histos[histoIndex] << (Char_t*)"\n";
886 if (fDataTree != 0) delete fDataTree;
889 if (procCmds != 0) delete [] procCmds;
891 if (mProcObjectType != 0) delete [] mProcObjectType;
893 if (mProcObjectType2 != 0) delete [] mProcObjectType2;
894 mProcObjectType2 = 0;
895 if (mProcType != 0) delete [] mProcType;
898 if (selCmds != 0) delete [] selCmds;
900 if (mSelObjectType != 0) delete [] mSelObjectType;
902 if (mSelObjectType2 != 0) delete [] mSelObjectType2;
904 if (mSelType != 0) delete [] mSelType;
907 if (histos != 0) delete [] histos;
911 // A.B. gROOT->Reset();
913 // If there is data, select the first data set
914 if (procIterator->GetEntries() > 0) SETBIT(fHistoDataSelected, 0);
916 // Now the data is stored in "/tmp/ListAnalyserMacroData_$USER.root"
917 // The editor will access this file to display the data
921 //______________________________________________________
922 void AliEveListAnalyser::ApplySOSelectionMacros(const TList* iterator)
924 // Uses the iterator (for the selected selection macros) to apply the selected macros to the data.
925 // The rnr-states of the objects are set according to the result of the macro calls (kTRUE, if all
926 // macros return kTRUE for this object, otherwise: kFALSE).
927 // "SO" stands for "single object". This means that only single object selection macros are applied.
928 // Correlated objects selection macros will be used inside the call of ApplyProcessMacros(...)!
930 TGeneralMacroData* macro = 0;
931 AliEveListAnalyserMacroType macroType = kUnknown;
932 TEveElement* object1 = 0;
933 Bool_t selectedByMacro = kFALSE;
936 // A.B. gROOT->Reset();
938 // Select all objecs at first. A object is then deselected, if at least one selection macro
939 // returns kFALSE for this object.
940 // Enable all objects (Note: EnableListElements(..) will call "ElementChanged", which will cause unforeseen behaviour!)
941 for (TEveElement::List_i iter = this->BeginChildren(); iter != this->EndChildren(); ++iter) ((TEveElement*)(*iter))->SetRnrState(kTRUE);
944 for (Int_t i = 0; i < iterator->GetEntries(); i++){
945 macro = (TGeneralMacroData*)fMacroList->GetValue(iterator->At(i)->GetTitle());
948 Error("Apply selection macros",
949 Form("Macro list is corrupted: Macro \"%s\" is not registered!", iterator->At(i)->GetTitle()));
953 #ifdef AliEveListAnalyser_DEBUG
954 printf("AliEveListAnalyser: Applying selection macro: %s\n", macro->GetName());
957 // Determine macro type
958 macroType = macro->GetType();
960 // Single object select macro
961 if (macroType == kSingleObjectSelect){
962 // Walk through the list of objects
963 for (TEveElement::List_i iter = this->BeginChildren(); iter != this->EndChildren(); ++iter)
965 object1 = dynamic_cast<TEveElement*>(*iter);
967 if (!object1) continue;
969 // If the object has already been deselected, nothing is to do here
970 if (!object1->GetRnrState()) continue;
972 // Find the type of the object and relate it to the macro object type
973 // Only apply macro to this object, if...
974 // ... the macro takes objects of exactly this type.
975 // ... the macro object type is a child of this object's type.
976 // Otherwise: Continue
977 if (((TObject*)object1->GetUserData())->IsA() != macro->GetObjectType() &&
978 !((TObject*)object1->GetUserData())->InheritsFrom(macro->GetObjectType())) continue;
980 // Cast to the "real" object behind
981 gROOT->ProcessLineSync(Form("TEveElement *automaticEveElement = (TEveElement*)%p;", (void*)object1));
982 gROOT->ProcessLineSync("TObject* automaticObject_1 = (TObject*)automaticEveElement->GetUserData();");
984 // GetCmd() will cast the automatic objects to the correct type for each macro!
985 selectedByMacro = (Bool_t)gROOT->ProcessLineSync(macro->GetCmd());
986 object1->SetRnrState(selectedByMacro && object1->GetRnrState());
989 // Correlated objects select macro
990 else if (macroType == kCorrelObjectSelect){
991 // Will be processed in ApplyProcessMacros(...)
994 Error("Apply selection macros",
995 Form("Macro list corrupted: Macro \"%s/%s.C\" is not registered as a selection macro!",
996 macro->GetPath(), macro->GetName()));
1001 // A.B. gROOT->Reset();
1004 //______________________________________________________
1005 TClass* AliEveListAnalyser::GetMacroObjectType(const Char_t* name, Int_t argNum) const
1007 // Returns the type of object (of argument argNum) the macro with name "name" is dealing with;
1008 // e.g. if you have the signature:
1009 // void MyMacro(const AliTRDtrackV1* track, Double_t* &results, Int_t& nResults)
1010 // the call 'GetMacroObjectType("MyMacro")' yields the AliTRDtrackV1-class.
1011 // If the macro is not found (or there is an error), 0x0 is returned.
1013 if (argNum - 1 < 0) return 0x0;
1015 TFunction* f = gROOT->GetGlobalFunction(name, 0 , kTRUE);
1021 list = f->GetListOfMethodArgs();
1023 if (!list->IsEmpty())
1025 m = (TMethodArg*)list->At(argNum - 1);
1027 if (m) return TClass::GetClass(m->GetTypeName());
1035 //______________________________________________________
1036 AliEveListAnalyser::AliEveListAnalyserMacroType AliEveListAnalyser::GetMacroType(const Char_t* name, const Char_t* objectType,
1037 const Char_t* objectType2, Bool_t UseList) const
1039 // Returns the type of the corresponding macro, that accepts pointers of the classes "objectType" (first pointer) and
1040 // objectType2" (second pointer) as parametres.
1041 // If "UseList" is kTRUE, the type will be looked up in the internal list (very fast). But if this list
1042 // does not exist, you have to use kFALSE for this parameter. Then the type will be determined by the
1043 // prototype! NOTE: It is assumed that the macro has been compiled! If not, the return value is not
1044 // predictable, but normally will be kUnknown.
1045 // Note: AddMacro(Fast) will update the internal list and RemoveMacros respectively.
1047 AliEveListAnalyserMacroType type = kUnknown;
1049 TString* typeStr = 0;
1050 TString* typeStr2 = 0;
1052 if (objectType != 0)
1054 typeStr = new TString(objectType);
1055 // Remove white-spaces
1056 typeStr->ReplaceAll(" ", "");
1060 typeStr = new TString("TObject");
1062 if (objectType2 != 0)
1064 typeStr2 = new TString(objectType2);
1065 // Remove white-spaces
1066 typeStr2->ReplaceAll(" ", "");
1070 typeStr2 = new TString("TObject");
1073 TString* mangled1Str = new TString();
1074 TString* mangled2Str = new TString();
1075 TString* mangled3Str = new TString();
1076 TString* mangled4Str = new TString();
1077 TString* mangledArg1Str = new TString();
1078 TString* mangledArg2Str = new TString();
1080 // We want "const 'OBJECTTYPE'*"
1081 mangled1Str->Form("const %s*", typeStr->Data());
1083 // We want "const 'OBJECTTYPE'*, Double_t*&, Int_t&"
1084 mangled2Str->Form("const %s*, Double_t*&, Int_t&", typeStr->Data());
1086 // We want "const 'OBJECTTYPE'*, const 'OBJECTTYPE2'*"
1087 mangled3Str->Form("const %s*, const %s*", typeStr->Data(), typeStr2->Data());
1089 // We want "const 'OBJECTTYPE'*, const 'OBJECTTYPE2'*, Double_t*&, Int_t&"
1090 mangled4Str->Form("const %s*, const %s*, Double_t*&, Int_t&", typeStr->Data(), typeStr2->Data());
1092 // We want "oPconstsP'OBJECTTYPE'mUsP"
1093 mangledArg1Str->Form("oPconstsP%smUsP", typeStr->Data());
1095 // We want "cOconstsP'OBJECTTYPE2'mUsP"
1096 mangledArg2Str->Form("cOconstsP%smUsP", typeStr2->Data());
1098 // Re-do the check of the macro type
1100 // Single object select macro or single object histo macro?
1101 TFunction* f = gROOT->GetGlobalFunctionWithPrototype(name, mangled1Str->Data(), kTRUE);
1105 // Some additional check (is the parameter EXACTLY of the desired type?)
1106 if (strstr(f->GetMangledName(), mangledArg1Str->Data()) != 0x0)
1108 // Single object select macro?
1109 if (!strcmp(f->GetReturnTypeName(), "Bool_t"))
1111 type = kSingleObjectSelect;
1113 // single object histo macro?
1114 else if (!strcmp(f->GetReturnTypeName(), "TH1*"))
1116 type = kSingleObjectHisto;
1120 // Single object analyse macro?
1121 else if ((f = gROOT->GetGlobalFunctionWithPrototype(name, mangled2Str->Data(), kTRUE))
1124 if (!strcmp(f->GetReturnTypeName(), "void"))
1126 // Some additional check (are the parameters EXACTLY of the desired type?)
1127 if (strstr(f->GetMangledName(), mangledArg1Str->Data()) != 0x0 &&
1128 strstr(f->GetMangledName(), "cODouble_tmUaNsP") != 0x0 &&
1129 strstr(f->GetMangledName(), "cOInt_taNsP") != 0x0)
1131 type = kSingleObjectAnalyse;
1135 // Correlated objects select macro or correlated objects histo macro?
1136 else if ((f = gROOT->GetGlobalFunctionWithPrototype(name, mangled3Str->Data(), kTRUE))
1139 // Some additional check (is the parameter EXACTLY of the desired type?)
1140 if (strstr(f->GetMangledName(), mangledArg1Str->Data()) != 0x0 &&
1141 strstr(f->GetMangledName(), mangledArg2Str->Data()) != 0x0)
1143 // Correlated objects select macro?
1144 if (!strcmp(f->GetReturnTypeName(), "Bool_t"))
1146 type = kCorrelObjectSelect;
1148 // Correlated objects histo macro?
1149 else if (!strcmp(f->GetReturnTypeName(), "TH1*"))
1151 type = kCorrelObjectHisto;
1155 // Correlated objects analyse macro?
1156 else if ((f = gROOT->GetGlobalFunctionWithPrototype(name, mangled4Str->Data(), kTRUE))
1159 if (!strcmp(f->GetReturnTypeName(), "void"))
1161 // Some additional check (is the parameter EXACTLY of the desired type?)
1162 if (strstr(f->GetMangledName(), mangledArg1Str->Data()) != 0x0 &&
1163 strstr(f->GetMangledName(), mangledArg2Str->Data()) != 0x0 &&
1164 strstr(f->GetMangledName(), "cODouble_tmUaNsP") != 0x0 &&
1165 strstr(f->GetMangledName(), "cOInt_taNsP") != 0x0)
1167 type = kCorrelObjectAnalyse;
1172 // Use list to look up the macro type
1175 TGeneralMacroData* macro = 0;
1176 macro = (TGeneralMacroData*)fMacroList->GetValue(name);
1177 if (macro == 0) return kUnknown;
1179 type = macro->GetType();
1182 case kSingleObjectSelect:
1183 case kSingleObjectAnalyse:
1184 case kSingleObjectHisto:
1185 case kCorrelObjectSelect:
1186 case kCorrelObjectAnalyse:
1187 case kCorrelObjectHisto:
1196 if (mangled1Str != 0)
1198 mangled1Str->Clear();
1202 if (mangled2Str != 0)
1204 mangled2Str->Clear();
1208 if (mangled3Str != 0)
1210 mangled3Str->Clear();
1214 if (mangled4Str != 0)
1216 mangled4Str->Clear();
1220 if (mangledArg1Str != 0)
1222 mangledArg1Str->Clear();
1223 delete mangledArg1Str;
1226 if (mangledArg2Str != 0)
1228 mangledArg2Str->Clear();
1229 delete mangledArg2Str;
1251 //______________________________________________________
1252 void AliEveListAnalyser::RemovePrimSelectedObjects()
1254 // Removes the (primarily) selected objects in the viewer from the list (objects that are already in the list are ignored).
1255 // Hold the CTRL-key for multiple selection.
1257 TEveSelection* eveSel = gEve->GetSelection();
1261 Error("AliEveListAnalyser::RemovePrimSelectedObjects", "Failed to get the selection!\n");
1265 TEveElement* elem = 0x0;
1266 Bool_t changedSomething = kFALSE;
1268 for (TEveElement::List_i iter = eveSel->BeginChildren(); iter != eveSel->EndChildren(); ++iter)
1270 if(!(elem = dynamic_cast<TEveElement*>(*iter))) continue;
1272 // Check, if element is already there. If so, remove it!
1273 if (this->HasChild(elem) && elem != this)
1275 this->RemoveElement(elem);
1276 this->SetTitle(Form("Objects %d", this->NumChildren()));
1277 changedSomething = kTRUE;
1281 if (changedSomething) gEve->Redraw3D();
1285 //______________________________________________________
1286 void AliEveListAnalyser::RemoveSelectedMacros(const TList* iterator)
1288 // Uses the iterator (for the selected macros) to remove the selected macros from
1289 // the corresponding list.
1293 for (Int_t i = 0; i < iterator->GetEntries(); i++)
1295 entry = (TPair*)fMacroList->FindObject(iterator->At(i)->GetTitle());
1299 Error("AliEveListAnalyser::RemoveSelectedMacros", Form("Macro \"%s\" not found in list!",
1300 iterator->At(i)->GetTitle()));
1307 Error("AliEveListAnalyser::RemoveSelectedMacros", Form("Key for macro \"%s\" not found in list!",
1308 iterator->At(i)->GetTitle()));
1312 // Key and value will be deleted, too, since fMacroList is the owner of them
1313 Bool_t rem = fMacroList->DeleteEntry(key);
1317 #ifdef AliEveListAnalyser_DEBUG
1318 printf("AliEveListAnalyser::RemoveSelectedMacros(): Removed macro: %s\n", iterator->At(i)->GetTitle());
1323 Error("AliEveListAnalyser::RemoveSelectedMacros", Form("Macro \"%s\" could not be removed from the list!",
1324 iterator->At(i)->GetTitle()));
1329 //______________________________________________________
1330 void AliEveListAnalyser::ResetObjectList()
1332 // Removes all objects from the list.
1335 this->SetTitle(Form("Objects %d", this->NumChildren()));
1338 //______________________________________________________
1339 Bool_t AliEveListAnalyser::StartAddingObjects()
1341 // Starts adding objects for the analysis. Returns kTRUE on success.
1343 if (fConnected == kFALSE)
1345 fConnected = TQObject::Connect("TEvePointSet", "PointSelected(Int_t)", "AliEveListAnalyser", this, "AddSecSelectedSingleObjectToList(Int_t)");
1346 if (fConnected) fConnected = TQObject::Connect(gEve->GetSelection(), "SelectionAdded(TEveElement*)", "AliEveListAnalyser", this, "AddPrimSelectedObject(TEveElement*)");
1348 if (fConnected) return kTRUE;
1350 Error("AliEveListAnalyser::StartAddingObjects", "Connection failed!");
1352 // Connection of 2nd signal failed, but first connection succeeded -> Disconnect 1st signal.
1353 TQObject::Disconnect("TEvePointSet", "PointSelected(Int_t)", this, "AddObjectToList(Int_t)");
1359 //______________________________________________________
1360 Bool_t AliEveListAnalyser::StopAddingObjects()
1362 // Stops adding objects for the analysis. Returns kTRUE on success.
1366 Bool_t dis1 = kFALSE, dis2 = kFALSE;
1367 dis1 = TQObject::Disconnect("TEvePointSet", "PointSelected(Int_t)", this, "AddSecSelectedSingleObjectToList(Int_t)");
1368 dis2 = TQObject::Disconnect(gEve->GetSelection(), "SelectionAdded(TEveElement*)", this, "AddPrimSelectedObject(TEveElement*)");
1370 if (dis1 || dis2) fConnected = kFALSE;
1371 if (dis1 && dis2) return kTRUE;
1374 Error("AliEveListAnalyser::StopAddingObjects", "Disconnection failed!");