[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTRDTrackListEditor.cxx

#include "TTreeStream.h"
#include <EveDet/AliEveTRDTrackList.h>
#include "AliEveTRDTrackListEditor.h"

#include <TGFileDialog.h>
#include <TFile.h>
#include <TGButton.h>
#include <TGedEditor.h>     ////// MAYBE THIS CAN BE REMOVED
#include <TGTextEntry.h>
#include <TGTextView.h>
#include <TGListBox.h>
#include <TGMsgBox.h>
#include <TGLabel.h>
#include <TG3DLine.h>
#include <TEveMacro.h>
#include <TEveManager.h>
#include <TObjString.h>
#include <TSystem.h>
#include <TROOT.h>
#include <AliTRDtrackV1.h>
#include <EveDet/AliEveTRDData.h>

ClassImp(AliEveTRDTrackListEditor)

///////////////////////////////////////////////////////////
/////////////   AliEveTRDTrackListEditor //////////////////
///////////////////////////////////////////////////////////
AliEveTRDTrackListEditor::AliEveTRDTrackListEditor(const TGWindow* p, Int_t width, Int_t height,
				                   UInt_t options, Pixel_t back) :
  TGedFrame(p, width, height, options, back),
  fM(0),
  fMainFrame(0),
  fHistoFrame(0),
  fHistoSubFrame(0),
  fBrowseFrame(0),
  fbBrowse(0),
  fbApplyMacros(0),
  fbRemoveMacros(0),
  fbDrawHisto(0),
  fteField(0),
  ftlMacroList(0),
  ftlMacroSelList(0),
  fFileInfo(0),
  fFileTypes(0),
  fLabel1(0), fLabel2(0), fLabel3(0), fLabel4(0),
  fLine1(0), fLine2(0), fLine3(0), fLine4(0),
  fCheckButtons(0)
{  
  fMainFrame = CreateEditorTabSubFrame("Apply macros");
 
  // Functionality for adding macros 
  fLabel1 = new TGLabel(fMainFrame,"Add macro(s):");
  fMainFrame->AddFrame(fLabel1);
  fBrowseFrame = new TGHorizontalFrame(fMainFrame);

  fteField = new TGTextEntry(fBrowseFrame);
  fteField->Connect("ReturnPressed()","AliEveTRDTrackListEditor", this, "HandleMacroPathSet()"); 
  fBrowseFrame->AddFrame(fteField);
  
  fbBrowse = new TGTextButton(fBrowseFrame, "Browse");
  fbBrowse->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "BrowseMacros()");
  fBrowseFrame->AddFrame(fbBrowse);
  fMainFrame->AddFrame(fBrowseFrame);

  fLine1 = new TGHorizontal3DLine(fMainFrame, 194, 8);
  fMainFrame->AddFrame(fLine1, new TGLayoutHints(kLHintsLeft  | kLHintsTop, 2, 2, 8, 2));
  fLabel2 = new TGLabel(fMainFrame,"Selection macros:");
  fMainFrame->AddFrame(fLabel2);

  ftlMacroSelList = new TGListBox(fMainFrame);
  ftlMacroSelList->Resize(194, 94);
  ftlMacroSelList->SetMultipleSelections(kTRUE);
  fMainFrame->AddFrame(ftlMacroSelList);

  fLine2 = new TGHorizontal3DLine(fMainFrame, 194, 8);
  fMainFrame->AddFrame(fLine2, new TGLayoutHints(kLHintsLeft  | kLHintsTop, 2, 2, 8, 2));
  fLabel3 = new TGLabel(fMainFrame,"Process macros:");
  fMainFrame->AddFrame(fLabel3);

  ftlMacroList = new TGListBox(fMainFrame);
  ftlMacroList->Resize(194, 94);
  ftlMacroList->SetMultipleSelections(kTRUE);
  fMainFrame->AddFrame(ftlMacroList);

  fLine3 = new TGHorizontal3DLine(fMainFrame, 194, 8);
  fMainFrame->AddFrame(fLine3, new TGLayoutHints(kLHintsLeft  | kLHintsTop, 2, 2, 8, 2));  

  fbApplyMacros = new TGTextButton(fMainFrame, "Apply selected macro(s)");
  fbApplyMacros->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "ApplyMacros()");
  fbApplyMacros->SetRightMargin(12);
  fMainFrame->AddFrame(fbApplyMacros);

  fbRemoveMacros = new TGTextButton(fMainFrame, "Remove selected macro(s)");
  fbRemoveMacros->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "RemoveMacros()");
  fMainFrame->AddFrame(fbRemoveMacros);

  // Stuff for displaying histograms
  fHistoFrame = CreateEditorTabSubFrame("Histograms");  
  fHistoFrame->SetMapSubwindows(kTRUE);
  fLabel4 = new TGLabel(fHistoFrame,"Data from applied macros:");
  fHistoFrame->AddFrame(fLabel4);

  fHistoSubFrame = new TGVerticalFrame(fHistoFrame);
  fHistoSubFrame->SetMapSubwindows(kTRUE);
  fHistoSubFrame->Resize(194, 200);
  fHistoFrame->AddFrame(fHistoSubFrame);

  fLine4 = new TGHorizontal3DLine(fHistoFrame, 194, 8);
  fHistoFrame->AddFrame(fLine4, new TGLayoutHints(kLHintsLeft  | kLHintsTop, 2, 2, 8, 2));  

  fbDrawHisto = new TGTextButton(fHistoFrame, "Draw histogram");
  fbDrawHisto->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "DrawHistos()");
  fHistoFrame->AddFrame(fbDrawHisto);

  // Set up file dialog
  fFileInfo = new TGFileInfo();
  fFileInfo->SetMultipleSelection(kTRUE);

  fFileTypes = new Char_t*[6];
  fFileTypes[0] = (Char_t*)"All files"; fFileTypes[1] = (Char_t*)"*";
  fFileTypes[2] = (Char_t*)"ROOT macros"; fFileTypes[3] = (Char_t*)"*.C";
  fFileTypes[4] = 0; fFileTypes[5] = 0;
  fFileInfo->fFileTypes = (const Char_t**)fFileTypes;
  fFileInfo->fFileTypeIdx = 2;
  fFileInfo->fMultipleSelection = kTRUE;

  // Set focus on "Apply macros" tab
  //fMainFrame->TGWindow::RequestFocus();
  //Update();
}

//______________________________________________________
AliEveTRDTrackListEditor::~AliEveTRDTrackListEditor()
{
  if (fFileTypes != 0)
  {
    delete [] fFileTypes;
    fFileTypes = 0;
  }

  if (fFileInfo != 0)
  {
    delete fFileInfo; 
    fFileInfo = 0;
  }
}

//______________________________________________________
void AliEveTRDTrackListEditor::AddMacro(const Char_t* path, const Char_t* name)
{
  Int_t result = fM->AddMacro(path, name);

  switch (result)
  {
  case SUCCESS:
    UpdateMacroList();
    break;
  case WARNING:
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Warning", "Macro is already in list (won't be added again)!",
                 kMBIconExclamation, kMBOk);
    break;
  case SIGNATURE_ERROR:
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error", 
                 "Macro has not the signature of...\n...a selection macro: Bool_t YourMacro(AliTRDtrackV1*)\n...a process macro: void YourMacro(AliTRDtrackV1*, Double_t*&, Int_t&)", 
                 kMBIconExclamation, kMBOk);
    break;               
  case NOT_EXIST_ERROR:
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error", 
                 "File does not exist or you do not have read permission!", kMBIconExclamation, kMBOk);
    break;
  default:
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error", 
                 Form("AliEveTRDTrackList::AddMacro exited with unknown return value: %d", result),
                 kMBIconExclamation, kMBOk);
    break;
  }
}

//______________________________________________________
void AliEveTRDTrackListEditor::ApplyMacros()
{
  // First apply the selection macros
  TList* iterator = new TList();
  ftlMacroSelList->GetSelectedEntries(iterator);
  fM->ApplySelectionMacros(iterator);
  
  // Update view
  gEve->Redraw3D();

  if (iterator != 0) delete iterator;  

  // Now apply the process macros
  iterator = new TList();
  ftlMacroList->GetSelectedEntries(iterator);
  fM->ApplyProcessMacros(iterator);
  
  // Now the data is stored in the following file and can easily be accessed
  TFile* file = new TFile("TRD.TrackListMacroData.root", "READ");
  if (!file)  
  {
    Error("Apply macros", "Cannot open file \"TRD.TrackListMacroData.root\"");
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Apply macros", 
                 "Cannot open file \"TRD.TrackListMacroData.root\"", kMBIconExclamation, kMBOk);
    
    if (iterator != 0)  delete iterator;  
    iterator = 0;

    return;  
  }
    
  TTree* t = 0;
  for (Int_t i = 0; i < iterator->GetEntries(); i++) {
    t = (TTree*)file->Get(Form("TrackData%d", i));
    if (t != 0) {
      gEve->AddCanvasTab(Form("Macro%d", i));
      t->Draw(Form("Macro%d", i), "1");
 
      delete t;
      t = 0;

      // ONLY DISPLAY ONE MACRO (the first one possible) -> Remove the next line to display all
      break;
    } else {
      Error("Apply macros", Form("No data for macro%d found!", i));
      new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Apply macros", 
                   Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);  
    }
  }
  file->Close("R");
  delete file;
  file = 0;

  if (iterator != 0)  delete iterator;  
  iterator = 0;  

  // Update histogram tab (data has to be reloaded)
  //fHistoFrame->TGWindow::RequestFocus();
  //fGedEditor->TGCompositeFrame::ShowFrame(fHistoFrame);
  SetModel(fM);
  Update();
}


//______________________________________________________
void AliEveTRDTrackListEditor::BrowseMacros()
{
  new TGFileDialog(gClient->GetRoot(), GetMainFrame(), kFDOpen, fFileInfo);
  
  if (fFileInfo->fIniDir != 0 && fFileInfo->fFileNamesList != 0)
  {       
    // Extract filenames
    TObject* iter = fFileInfo->fFileNamesList->First();
 
    Char_t* name = 0;

    while (iter != 0)
    {
      // NOTE: fileInfo->fFileNamesList will be changed by that, too!
      name = strrchr(iter->GetName(), '/');
      // Delete '"' at the end
      name[strlen(name)] = '\0';
              
      AddMacro(fFileInfo->fIniDir, name + 1); 
      iter = (TObjString*)fFileInfo->fFileNamesList->After(iter);
    }
  }

  // -> The following problem has been fixed (trunk -> Changes according to 03 September 2008):
  // Some error occurs, when one ends the filedialog with "cancel": fileInfo->fFileNamesList is set to 0x0, but
  // in the next launch no new memory is allocated. So do this manually.
  //if (fileInfo->fFileNamesList == 0)  fileInfo->fFileNamesList = new TList();
}

//______________________________________________________
void AliEveTRDTrackListEditor::DrawHistos()
{
  Int_t nHistograms = GetNSelectedHistograms();
  if (nHistograms <= 0)
  {
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error", 
                 "No data selected. Please select the data you want to plot!", kMBIconExclamation, kMBOk);
    return;
  }
  if (nHistograms > 3)
  {
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), 
                 "Error", "Only histograms with up to 3 dimensions supported. Please select 1,2 or 3 data macros!",
                 kMBIconExclamation, kMBOk);
    return;
  }

  TFile* file = new TFile("TRD.TrackListMacroData.root", "READ");
  if (!file)  
  {
    Error("Draw histograms", "Cannot open file \"TRD.TrackListMacroData.root\"");
    new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms", 
                 "Cannot open file \"TRD.TrackListMacroData.root\"", kMBIconExclamation, kMBOk);
    return;
  }
  
  TTree* t = 0;
  TTree* tFriend1 = 0;
  TTree* tFriend2 = 0;

  Int_t indexOfMacro1 = 0;
  Int_t indexOfMacro2 = 0;
  Int_t indexOfMacro3 = 0;

  // Load the trees in succession and remember the entries
  for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries(); i++)
  {
    if (fCheckButtons[i]->TGButton::GetState() == kButtonDown)
    {
      if (t == 0)
      {
        indexOfMacro1 = i;
        if (!(t = (TTree*)file->Get(Form("TrackData%d", i))))
        { 
          Error("Draw histograms", Form("No data for macro%d found!", i));
          new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms", 
                       Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
          break;   
        }
     
        // 1d histogram   
        if (nHistograms == 1) 
        {
          gEve->AddCanvasTab(Form("Macro%d", indexOfMacro1));
          t->Draw(Form("Macro%d", indexOfMacro1), "1");

          break;     
        }
      }
      else if (tFriend1 == 0)
      {
        indexOfMacro2 = i;
        if (!(tFriend1 = (TTree*)file->Get(Form("TrackData%d", i))))
        { 
          Error("Draw histograms", Form("No data for macro%d found!", i));
          new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms", 
                       Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
          break;   
        }
        
        // 2d histogram
        if (nHistograms == 2) 
        {
          gEve->AddCanvasTab(Form("Macro%d - Macro%d", indexOfMacro1, indexOfMacro2));
          t->AddFriend(tFriend1);
          t->Draw(Form("Macro%d:Macro%d", indexOfMacro1, indexOfMacro2), "1");
 
          break;     
        }
      }    
      // 3d histogram
      else
      {
        indexOfMacro3 = i;
        if (!(tFriend2 = (TTree*)file->Get(Form("TrackData%d", i))))
        { 
          Error("Draw histograms", Form("No data for macro%d found!", i));
          new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms", 
                       Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
          break;   
        }
        
        gEve->AddCanvasTab(Form("Macro%d - Macro%d - Macro%d", indexOfMacro1, indexOfMacro2, indexOfMacro3));
        t->AddFriend(tFriend1);
        t->AddFriend(tFriend2);
        t->Draw(Form("Macro%d:Macro%d:Macro%d", indexOfMacro1, indexOfMacro2, indexOfMacro3), "1");
 
        break;     
      }
    }
  }

  if (t != 0) delete t;
  t = 0;
  if (tFriend1 != 0)  delete tFriend1;
  tFriend1 = 0;
  if (tFriend2 != 0)  delete tFriend2;
  tFriend2 = 0;

  file->Close("R");
  delete file;
  file = 0;
}

//______________________________________________________
Int_t AliEveTRDTrackListEditor::GetNSelectedHistograms()
{
  Int_t count = 0;
  
  for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries(); i++)
  {
    if (fCheckButtons[i]->TGButton::GetState() == kButtonDown)  count++;
  }

  return count;
}

//______________________________________________________
void AliEveTRDTrackListEditor::HandleMacroPathSet()
{
  if (strlen(fteField->GetText()) != 0)
  {         			
    // Check if file exists
    FILE* fp = NULL;

    fp = fopen(fteField->GetText(), "rb");
    if (fp != NULL)
    {
      fclose(fp);

      // Extract filename
      Char_t* name = strrchr(fteField->GetText(), '/');

      // Current path
      if (name == NULL)
      {
        name = new Char_t[100];
        memset(name, '\0', sizeof(Char_t) * 100);
        sprintf(name, "%s", fteField->GetText());

        // Add path to textfield
        Char_t pathname[100];
        memset(pathname, '\0', sizeof(Char_t) * 100);
        sprintf(pathname, "./%s", fteField->GetText());
        fteField->SetText(pathname);

        AddMacro(".", name);  
        if (name != 0)  delete name;
        name = 0;
      }
      // Different path
      else
      {
        // Extract path
        Char_t* path = new Char_t[240];
        memset(path, '\0', sizeof(Char_t) * 240);
        strncpy(path, fteField->GetText(), strlen(fteField->GetText()) - strlen(name));
        
        // Ignore the slash "/" in name
        AddMacro(path, name + 1);  
  
        if (path != 0)  delete path;
        path = 0;
      }       
    }
    else
    {
      new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error", 
                   "File does not exist or you do not have read permission!", kMBIconExclamation, kMBOk);
    }
  }
}

//______________________________________________________
void AliEveTRDTrackListEditor::RemoveMacros()
{
  TList* iterator = new TList();
  
  ftlMacroList->GetSelectedEntries(iterator);
  fM->RemoveProcessMacros(iterator);

  if (iterator != 0)  delete iterator;

  iterator = new TList();
  ftlMacroSelList->GetSelectedEntries(iterator);
  fM->RemoveSelectionMacros(iterator);

  UpdateMacroList();

  if (iterator != 0)  delete iterator;
  iterator = 0;
}


//______________________________________________________
void AliEveTRDTrackListEditor::SetModel(TObject* obj)
{
  // Set model object
  fM = dynamic_cast<AliEveTRDTrackList*>(obj);

  UpdateMacroList();
  UpdateHistoList();  
}

//______________________________________________________
void AliEveTRDTrackListEditor::UpdateHistoList()
{
  fHistoSubFrame->TGCompositeFrame::RemoveAll();
  
  // Set buttons for histograms
  if (fCheckButtons != 0) delete fCheckButtons;
  fCheckButtons = new TGCheckButton*[fM->fDataFromMacroList->GetEntries()];
  
  TObjString* iter = (TObjString*)fM->fDataFromMacroList->First();
  for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries() && iter != 0; i++)
  {
    fCheckButtons[i] = new TGCheckButton(fHistoSubFrame, iter->GetName());
    fHistoSubFrame->AddFrame(fCheckButtons[i]);
    
    fCheckButtons[i]->SetState(kButtonUp, kFALSE);
    fCheckButtons[i]->MapRaised();
            
    iter = (TObjString*)fM->fDataFromMacroList->After(iter);
  }  
}


//______________________________________________________
void AliEveTRDTrackListEditor::UpdateMacroList()
{
  ftlMacroList->RemoveAll();
 
  TObjString* iter = (TObjString*)fM->fMacroList->First();

  while (iter != 0)
  {
    ftlMacroList->AddEntry(iter->GetName(), -1);
    iter = (TObjString*)fM->fMacroList->After(iter);
  }

  ftlMacroList->SortByName();


  ftlMacroSelList->RemoveAll();
 
  iter = (TObjString*)fM->fMacroSelList->First();

  while (iter != 0)
  {
    ftlMacroSelList->AddEntry(iter->GetName(), -1);
    iter = (TObjString*)fM->fMacroSelList->After(iter);
  }

  ftlMacroSelList->SortByName(); 
}
Commit	Line	Data
caaf90d2	1	#include "TTreeStream.h"
caaf90d2	2	#include <EveDet/AliEveTRDTrackList.h>
2ef0687e	3	#include "AliEveTRDTrackListEditor.h"
2ef0687e	4
caaf90d2	5	#include <TGFileDialog.h>
	6	#include <TFile.h>
	7	#include <TGButton.h>
c04e790a	8	#include <TGedEditor.h> ////// MAYBE THIS CAN BE REMOVED
caaf90d2	9	#include <TGTextEntry.h>
	10	#include <TGTextView.h>
	11	#include <TGListBox.h>
	12	#include <TGMsgBox.h>
	13	#include <TGLabel.h>
	14	#include <TG3DLine.h>
	15	#include <TEveMacro.h>
	16	#include <TEveManager.h>
	17	#include <TObjString.h>
	18	#include <TSystem.h>
	19	#include <TROOT.h>
	20	#include <AliTRDtrackV1.h>
	21	#include <EveDet/AliEveTRDData.h>
	22
2ef0687e	23	ClassImp(AliEveTRDTrackListEditor)
	24
	25	///////////////////////////////////////////////////////////
	26	///////////// AliEveTRDTrackListEditor //////////////////
	27	///////////////////////////////////////////////////////////
	28	AliEveTRDTrackListEditor::AliEveTRDTrackListEditor(const TGWindow* p, Int_t width, Int_t height,
	29	UInt_t options, Pixel_t back) :
	30	TGedFrame(p, width, height, options, back),
	31	fM(0),
	32	fMainFrame(0),
caaf90d2	33	fHistoFrame(0),
caaf90d2	34	fHistoSubFrame(0),
2ef0687e	35	fBrowseFrame(0),
caaf90d2	36	fbBrowse(0),
	37	fbApplyMacros(0),
	38	fbRemoveMacros(0),
	39	fbDrawHisto(0),
	40	fteField(0),
	41	ftlMacroList(0),
	42	ftlMacroSelList(0),
	43	fFileInfo(0),
	44	fFileTypes(0),
	45	fLabel1(0), fLabel2(0), fLabel3(0), fLabel4(0),
	46	fLine1(0), fLine2(0), fLine3(0), fLine4(0),
	47	fCheckButtons(0)
	48	{
2ef0687e	49	fMainFrame = CreateEditorTabSubFrame("Apply macros");
3f797131	50
2ef0687e	51	// Functionality for adding macros
3f797131	52	fLabel1 = new TGLabel(fMainFrame,"Add macro(s):");
3f797131	53	fMainFrame->AddFrame(fLabel1);
2ef0687e	54	fBrowseFrame = new TGHorizontalFrame(fMainFrame);
2ef0687e	55
caaf90d2	56	fteField = new TGTextEntry(fBrowseFrame);
	57	fteField->Connect("ReturnPressed()","AliEveTRDTrackListEditor", this, "HandleMacroPathSet()");
	58	fBrowseFrame->AddFrame(fteField);
2ef0687e	59
caaf90d2	60	fbBrowse = new TGTextButton(fBrowseFrame, "Browse");
	61	fbBrowse->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "BrowseMacros()");
	62	fBrowseFrame->AddFrame(fbBrowse);
2ef0687e	63	fMainFrame->AddFrame(fBrowseFrame);
2ef0687e	64
3f797131	65	fLine1 = new TGHorizontal3DLine(fMainFrame, 194, 8);
caaf90d2	66	fMainFrame->AddFrame(fLine1, new TGLayoutHints(kLHintsLeft \| kLHintsTop, 2, 2, 8, 2));
3f797131	67	fLabel2 = new TGLabel(fMainFrame,"Selection macros:");
	68	fMainFrame->AddFrame(fLabel2);
	69
caaf90d2	70	ftlMacroSelList = new TGListBox(fMainFrame);
	71	ftlMacroSelList->Resize(194, 94);
	72	ftlMacroSelList->SetMultipleSelections(kTRUE);
	73	fMainFrame->AddFrame(ftlMacroSelList);
3f797131	74
3f797131	75	fLine2 = new TGHorizontal3DLine(fMainFrame, 194, 8);
caaf90d2	76	fMainFrame->AddFrame(fLine2, new TGLayoutHints(kLHintsLeft \| kLHintsTop, 2, 2, 8, 2));
3f797131	77	fLabel3 = new TGLabel(fMainFrame,"Process macros:");
	78	fMainFrame->AddFrame(fLabel3);
	79
caaf90d2	80	ftlMacroList = new TGListBox(fMainFrame);
	81	ftlMacroList->Resize(194, 94);
	82	ftlMacroList->SetMultipleSelections(kTRUE);
	83	fMainFrame->AddFrame(ftlMacroList);
2ef0687e	84
3f797131	85	fLine3 = new TGHorizontal3DLine(fMainFrame, 194, 8);
caaf90d2	86	fMainFrame->AddFrame(fLine3, new TGLayoutHints(kLHintsLeft \| kLHintsTop, 2, 2, 8, 2));
3f797131	87
caaf90d2	88	fbApplyMacros = new TGTextButton(fMainFrame, "Apply selected macro(s)");
	89	fbApplyMacros->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "ApplyMacros()");
	90	fbApplyMacros->SetRightMargin(12);
	91	fMainFrame->AddFrame(fbApplyMacros);
2ef0687e	92
caaf90d2	93	fbRemoveMacros = new TGTextButton(fMainFrame, "Remove selected macro(s)");
	94	fbRemoveMacros->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "RemoveMacros()");
	95	fMainFrame->AddFrame(fbRemoveMacros);
2ef0687e	96
caaf90d2	97	// Stuff for displaying histograms
	98	fHistoFrame = CreateEditorTabSubFrame("Histograms");
	99	fHistoFrame->SetMapSubwindows(kTRUE);
	100	fLabel4 = new TGLabel(fHistoFrame,"Data from applied macros:");
	101	fHistoFrame->AddFrame(fLabel4);
	102
	103	fHistoSubFrame = new TGVerticalFrame(fHistoFrame);
	104	fHistoSubFrame->SetMapSubwindows(kTRUE);
	105	fHistoSubFrame->Resize(194, 200);
	106	fHistoFrame->AddFrame(fHistoSubFrame);
	107
	108	fLine4 = new TGHorizontal3DLine(fHistoFrame, 194, 8);
	109	fHistoFrame->AddFrame(fLine4, new TGLayoutHints(kLHintsLeft \| kLHintsTop, 2, 2, 8, 2));
	110
	111	fbDrawHisto = new TGTextButton(fHistoFrame, "Draw histogram");
	112	fbDrawHisto->Connect("Clicked()", "AliEveTRDTrackListEditor", this, "DrawHistos()");
	113	fHistoFrame->AddFrame(fbDrawHisto);
2ef0687e	114
2ef0687e	115	// Set up file dialog
caaf90d2	116	fFileInfo = new TGFileInfo();
	117	fFileInfo->SetMultipleSelection(kTRUE);
	118
	119	fFileTypes = new Char_t*[6];
	120	fFileTypes[0] = (Char_t)"All files"; fFileTypes[1] = (Char_t)"*";
	121	fFileTypes[2] = (Char_t)"ROOT macros"; fFileTypes[3] = (Char_t)"*.C";
	122	fFileTypes[4] = 0; fFileTypes[5] = 0;
	123	fFileInfo->fFileTypes = (const Char_t**)fFileTypes;
	124	fFileInfo->fFileTypeIdx = 2;
	125	fFileInfo->fMultipleSelection = kTRUE;
	126
	127	// Set focus on "Apply macros" tab
caaf90d2	128	//fMainFrame->TGWindow::RequestFocus();
caaf90d2	129	//Update();
2ef0687e	130	}
2ef0687e	131
c04e790a	132	//______________________________________________________
caaf90d2	133	AliEveTRDTrackListEditor::~AliEveTRDTrackListEditor()
3f797131	134	{
caaf90d2	135	if (fFileTypes != 0)
39b891ff	136	{
caaf90d2	137	delete [] fFileTypes;
caaf90d2	138	fFileTypes = 0;
2ef0687e	139	}
3f797131	140
caaf90d2	141	if (fFileInfo != 0)
39b891ff	142	{
caaf90d2	143	delete fFileInfo;
caaf90d2	144	fFileInfo = 0;
39b891ff	145	}
caaf90d2	146	}
caaf90d2	147
c04e790a	148	//______________________________________________________
caaf90d2	149	void AliEveTRDTrackListEditor::AddMacro(const Char_t* path, const Char_t* name)
	150	{
	151	Int_t result = fM->AddMacro(path, name);
	152
	153	switch (result)
39b891ff	154	{
caaf90d2	155	case SUCCESS:
	156	UpdateMacroList();
	157	break;
	158	case WARNING:
39b891ff	159	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Warning", "Macro is already in list (won't be added again)!",
39b891ff	160	kMBIconExclamation, kMBOk);
caaf90d2	161	break;
	162	case SIGNATURE_ERROR:
	163	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error",
	164	"Macro has not the signature of...\n...a selection macro: Bool_t YourMacro(AliTRDtrackV1)\n...a process macro: void YourMacro(AliTRDtrackV1, Double_t*&, Int_t&)",
	165	kMBIconExclamation, kMBOk);
	166	break;
	167	case NOT_EXIST_ERROR:
	168	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error",
	169	"File does not exist or you do not have read permission!", kMBIconExclamation, kMBOk);
	170	break;
	171	default:
	172	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error",
	173	Form("AliEveTRDTrackList::AddMacro exited with unknown return value: %d", result),
	174	kMBIconExclamation, kMBOk);
	175	break;
2ef0687e	176	}
	177	}
	178
c04e790a	179	//______________________________________________________
2ef0687e	180	void AliEveTRDTrackListEditor::ApplyMacros()
2ef0687e	181	{
3f797131	182	// First apply the selection macros
caaf90d2	183	TList* iterator = new TList();
	184	ftlMacroSelList->GetSelectedEntries(iterator);
	185	fM->ApplySelectionMacros(iterator);
	186
39b891ff	187	// Update view
39b891ff	188	gEve->Redraw3D();
39b891ff	189
caaf90d2	190	if (iterator != 0) delete iterator;
2ef0687e	191
3f797131	192	// Now apply the process macros
3f797131	193	iterator = new TList();
caaf90d2	194	ftlMacroList->GetSelectedEntries(iterator);
	195	fM->ApplyProcessMacros(iterator);
	196
	197	// Now the data is stored in the following file and can easily be accessed
	198	TFile* file = new TFile("TRD.TrackListMacroData.root", "READ");
	199	if (!file)
	200	{
	201	Error("Apply macros", "Cannot open file \"TRD.TrackListMacroData.root\"");
	202	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Apply macros",
	203	"Cannot open file \"TRD.TrackListMacroData.root\"", kMBIconExclamation, kMBOk);
	204
	205	if (iterator != 0) delete iterator;
	206	iterator = 0;
2ef0687e	207
caaf90d2	208	return;
	209	}
	210
	211	TTree* t = 0;
c04e790a	212	for (Int_t i = 0; i < iterator->GetEntries(); i++) {
caaf90d2	213	t = (TTree*)file->Get(Form("TrackData%d", i));
c04e790a	214	if (t != 0) {
caaf90d2	215	gEve->AddCanvasTab(Form("Macro%d", i));
caaf90d2	216	t->Draw(Form("Macro%d", i), "1");
39b891ff	217
caaf90d2	218	delete t;
caaf90d2	219	t = 0;
c04e790a	220
	221	// ONLY DISPLAY ONE MACRO (the first one possible) -> Remove the next line to display all
	222	break;
	223	} else {
caaf90d2	224	Error("Apply macros", Form("No data for macro%d found!", i));
	225	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Apply macros",
	226	Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
	227	}
2ef0687e	228	}
caaf90d2	229	file->Close("R");
	230	delete file;
	231	file = 0;
2ef0687e	232
caaf90d2	233	if (iterator != 0) delete iterator;
	234	iterator = 0;
	235
	236	// Update histogram tab (data has to be reloaded)
c04e790a	237	//fHistoFrame->TGWindow::RequestFocus();
c04e790a	238	//fGedEditor->TGCompositeFrame::ShowFrame(fHistoFrame);
caaf90d2	239	SetModel(fM);
caaf90d2	240	Update();
2ef0687e	241	}
2ef0687e	242
c04e790a	243
c04e790a	244	//______________________________________________________
2ef0687e	245	void AliEveTRDTrackListEditor::BrowseMacros()
2ef0687e	246	{
caaf90d2	247	new TGFileDialog(gClient->GetRoot(), GetMainFrame(), kFDOpen, fFileInfo);
2ef0687e	248
caaf90d2	249	if (fFileInfo->fIniDir != 0 && fFileInfo->fFileNamesList != 0)
3f797131	250	{
2ef0687e	251	// Extract filenames
caaf90d2	252	TObject* iter = fFileInfo->fFileNamesList->First();
2ef0687e	253
	254	Char_t* name = 0;
	255
	256	while (iter != 0)
	257	{
caaf90d2	258	// NOTE: fileInfo->fFileNamesList will be changed by that, too!
2ef0687e	259	name = strrchr(iter->GetName(), '/');
	260	// Delete '"' at the end
	261	name[strlen(name)] = '\0';
caaf90d2	262
	263	AddMacro(fFileInfo->fIniDir, name + 1);
	264	iter = (TObjString*)fFileInfo->fFileNamesList->After(iter);
2ef0687e	265	}
2ef0687e	266	}
3f797131	267
caaf90d2	268	// -> The following problem has been fixed (trunk -> Changes according to 03 September 2008):
3f797131	269	// Some error occurs, when one ends the filedialog with "cancel": fileInfo->fFileNamesList is set to 0x0, but
3f797131	270	// in the next launch no new memory is allocated. So do this manually.
caaf90d2	271	//if (fileInfo->fFileNamesList == 0) fileInfo->fFileNamesList = new TList();
	272	}
	273
c04e790a	274	//______________________________________________________
caaf90d2	275	void AliEveTRDTrackListEditor::DrawHistos()
	276	{
	277	Int_t nHistograms = GetNSelectedHistograms();
	278	if (nHistograms <= 0)
	279	{
	280	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error",
	281	"No data selected. Please select the data you want to plot!", kMBIconExclamation, kMBOk);
	282	return;
	283	}
	284	if (nHistograms > 3)
	285	{
	286	new TGMsgBox(gClient->GetRoot(), GetMainFrame(),
	287	"Error", "Only histograms with up to 3 dimensions supported. Please select 1,2 or 3 data macros!",
	288	kMBIconExclamation, kMBOk);
	289	return;
	290	}
	291
	292	TFile* file = new TFile("TRD.TrackListMacroData.root", "READ");
	293	if (!file)
	294	{
	295	Error("Draw histograms", "Cannot open file \"TRD.TrackListMacroData.root\"");
	296	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms",
	297	"Cannot open file \"TRD.TrackListMacroData.root\"", kMBIconExclamation, kMBOk);
	298	return;
	299	}
	300
	301	TTree* t = 0;
	302	TTree* tFriend1 = 0;
	303	TTree* tFriend2 = 0;
	304
	305	Int_t indexOfMacro1 = 0;
	306	Int_t indexOfMacro2 = 0;
	307	Int_t indexOfMacro3 = 0;
	308
	309	// Load the trees in succession and remember the entries
	310	for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries(); i++)
	311	{
	312	if (fCheckButtons[i]->TGButton::GetState() == kButtonDown)
	313	{
	314	if (t == 0)
	315	{
	316	indexOfMacro1 = i;
	317	if (!(t = (TTree*)file->Get(Form("TrackData%d", i))))
	318	{
	319	Error("Draw histograms", Form("No data for macro%d found!", i));
	320	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms",
	321	Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
	322	break;
	323	}
	324
	325	// 1d histogram
	326	if (nHistograms == 1)
	327	{
	328	gEve->AddCanvasTab(Form("Macro%d", indexOfMacro1));
	329	t->Draw(Form("Macro%d", indexOfMacro1), "1");
	330
	331	break;
	332	}
	333	}
	334	else if (tFriend1 == 0)
	335	{
	336	indexOfMacro2 = i;
	337	if (!(tFriend1 = (TTree*)file->Get(Form("TrackData%d", i))))
	338	{
339	Error("Draw histograms", Form("No data for macro%d found!", i));
340	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms",
341	Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
342	break;
343	}
344
345	// 2d histogram
346	if (nHistograms == 2)
347	{
348	gEve->AddCanvasTab(Form("Macro%d - Macro%d", indexOfMacro1, indexOfMacro2));
349	t->AddFriend(tFriend1);
350	t->Draw(Form("Macro%d:Macro%d", indexOfMacro1, indexOfMacro2), "1");
351
352	break;
353	}
354	}
355	// 3d histogram
356	else
357	{
358	indexOfMacro3 = i;
359	if (!(tFriend2 = (TTree*)file->Get(Form("TrackData%d", i))))
360	{
361	Error("Draw histograms", Form("No data for macro%d found!", i));
362	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error - Draw histograms",
363	Form("No data for macro%d found!", i), kMBIconExclamation, kMBOk);
364	break;
365	}
366
367	gEve->AddCanvasTab(Form("Macro%d - Macro%d - Macro%d", indexOfMacro1, indexOfMacro2, indexOfMacro3));
368	t->AddFriend(tFriend1);
369	t->AddFriend(tFriend2);
370	t->Draw(Form("Macro%d:Macro%d:Macro%d", indexOfMacro1, indexOfMacro2, indexOfMacro3), "1");
371
372	break;
373	}
374	}
375	}
376
377	if (t != 0) delete t;
378	t = 0;
379	if (tFriend1 != 0) delete tFriend1;
380	tFriend1 = 0;
381	if (tFriend2 != 0) delete tFriend2;
382	tFriend2 = 0;
383
384	file->Close("R");
385	delete file;
386	file = 0;
387	}
388
c04e790a	389	//______________________________________________________
caaf90d2	390	Int_t AliEveTRDTrackListEditor::GetNSelectedHistograms()
	391	{
	392	Int_t count = 0;
	393
	394	for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries(); i++)
	395	{
	396	if (fCheckButtons[i]->TGButton::GetState() == kButtonDown) count++;
	397	}
	398
	399	return count;
2ef0687e	400	}
2ef0687e	401
c04e790a	402	//______________________________________________________
2ef0687e	403	void AliEveTRDTrackListEditor::HandleMacroPathSet()
2ef0687e	404	{
caaf90d2	405	if (strlen(fteField->GetText()) != 0)
39b891ff	406	{
2ef0687e	407	// Check if file exists
	408	FILE* fp = NULL;
	409
caaf90d2	410	fp = fopen(fteField->GetText(), "rb");
39b891ff	411	if (fp != NULL)
39b891ff	412	{
3f797131	413	fclose(fp);
caaf90d2	414
3f797131	415	// Extract filename
caaf90d2	416	Char_t* name = strrchr(fteField->GetText(), '/');
3f797131	417
3f797131	418	// Current path
39b891ff	419	if (name == NULL)
39b891ff	420	{
caaf90d2	421	name = new Char_t[100];
	422	memset(name, '\0', sizeof(Char_t) * 100);
	423	sprintf(name, "%s", fteField->GetText());
3f797131	424
	425	// Add path to textfield
	426	Char_t pathname[100];
	427	memset(pathname, '\0', sizeof(Char_t) * 100);
caaf90d2	428	sprintf(pathname, "./%s", fteField->GetText());
	429	fteField->SetText(pathname);
	430
	431	AddMacro(".", name);
	432	if (name != 0) delete name;
	433	name = 0;
39b891ff	434	}
	435	// Different path
	436	else
	437	{
3f797131	438	// Extract path
	439	Char_t* path = new Char_t[240];
	440	memset(path, '\0', sizeof(Char_t) * 240);
caaf90d2	441	strncpy(path, fteField->GetText(), strlen(fteField->GetText()) - strlen(name));
	442
	443	// Ignore the slash "/" in name
	444	AddMacro(path, name + 1);
	445
	446	if (path != 0) delete path;
	447	path = 0;
	448	}
39b891ff	449	}
	450	else
	451	{
3f797131	452	new TGMsgBox(gClient->GetRoot(), GetMainFrame(), "Error",
39b891ff	453	"File does not exist or you do not have read permission!", kMBIconExclamation, kMBOk);
2ef0687e	454	}
	455	}
	456	}
	457
c04e790a	458	//______________________________________________________
2ef0687e	459	void AliEveTRDTrackListEditor::RemoveMacros()
	460	{
	461	TList* iterator = new TList();
	462
caaf90d2	463	ftlMacroList->GetSelectedEntries(iterator);
caaf90d2	464	fM->RemoveProcessMacros(iterator);
3f797131	465
caaf90d2	466	if (iterator != 0) delete iterator;
3f797131	467
caaf90d2	468	iterator = new TList();
	469	ftlMacroSelList->GetSelectedEntries(iterator);
	470	fM->RemoveSelectionMacros(iterator);
2ef0687e	471
2ef0687e	472	UpdateMacroList();
caaf90d2	473
	474	if (iterator != 0) delete iterator;
	475	iterator = 0;
2ef0687e	476	}
2ef0687e	477
c04e790a	478
c04e790a	479	//______________________________________________________
2ef0687e	480	void AliEveTRDTrackListEditor::SetModel(TObject* obj)
2ef0687e	481	{
caaf90d2	482	// Set model object
2ef0687e	483	fM = dynamic_cast<AliEveTRDTrackList*>(obj);
2ef0687e	484
2ef0687e	485	UpdateMacroList();
caaf90d2	486	UpdateHistoList();
	487	}
	488
c04e790a	489	//______________________________________________________
caaf90d2	490	void AliEveTRDTrackListEditor::UpdateHistoList()
	491	{
	492	fHistoSubFrame->TGCompositeFrame::RemoveAll();
	493
	494	// Set buttons for histograms
	495	if (fCheckButtons != 0) delete fCheckButtons;
	496	fCheckButtons = new TGCheckButton*[fM->fDataFromMacroList->GetEntries()];
	497
	498	TObjString* iter = (TObjString*)fM->fDataFromMacroList->First();
	499	for (Int_t i = 0; i < fM->fDataFromMacroList->GetEntries() && iter != 0; i++)
	500	{
	501	fCheckButtons[i] = new TGCheckButton(fHistoSubFrame, iter->GetName());
	502	fHistoSubFrame->AddFrame(fCheckButtons[i]);
	503
	504	fCheckButtons[i]->SetState(kButtonUp, kFALSE);
	505	fCheckButtons[i]->MapRaised();
	506
	507	iter = (TObjString*)fM->fDataFromMacroList->After(iter);
	508	}
2ef0687e	509	}
2ef0687e	510
c04e790a	511
c04e790a	512	//______________________________________________________
2ef0687e	513	void AliEveTRDTrackListEditor::UpdateMacroList()
2ef0687e	514	{
caaf90d2	515	ftlMacroList->RemoveAll();
2ef0687e	516
caaf90d2	517	TObjString* iter = (TObjString*)fM->fMacroList->First();
2ef0687e	518
	519	while (iter != 0)
	520	{
caaf90d2	521	ftlMacroList->AddEntry(iter->GetName(), -1);
caaf90d2	522	iter = (TObjString*)fM->fMacroList->After(iter);
2ef0687e	523	}
2ef0687e	524
caaf90d2	525	ftlMacroList->SortByName();
3f797131	526
3f797131	527
caaf90d2	528	ftlMacroSelList->RemoveAll();
3f797131	529
caaf90d2	530	iter = (TObjString*)fM->fMacroSelList->First();
3f797131	531
	532	while (iter != 0)
	533	{
caaf90d2	534	ftlMacroSelList->AddEntry(iter->GetName(), -1);
caaf90d2	535	iter = (TObjString*)fM->fMacroSelList->After(iter);
3f797131	536	}
3f797131	537
caaf90d2	538	ftlMacroSelList->SortByName();
2ef0687e	539	}