Extra header added to the list
[u/mrichter/AliRoot.git] / EVE / EveBase / AliEveEventManagerEditor.cxx
index 5823489..a582fcd 100644 (file)
@@ -9,6 +9,7 @@
 
 #include "AliEveEventManagerEditor.h"
 #include "AliEveEventManager.h"
+#include "AliEveConfigManager.h"
 
 #include <AliESDEvent.h>
 
 #include <TGTextView.h>
 #include <TGLabel.h>
 
+#include "Riostream.h"
+
+#ifdef ZMQ
+#include "AliStorageAdministratorPanelListEvents.h"
+#include "AliStorageAdministratorPanelMarkEvent.h"
+#endif
+
 //______________________________________________________________________________
 // GUI editor for AliEveEventManager.
 //
 
+using std::ifstream;
+using std::ofstream;
+using std::ios;
+using std::cout;
+using std::endl;
+using std::string;
+
 ClassImp(AliEveEventManagerEditor)
 
 //______________________________________________________________________________
@@ -31,7 +46,7 @@ AliEveEventManagerEditor::AliEveEventManagerEditor(const TGWindow *p, Int_t widt
                                                   UInt_t options, Pixel_t back) :
   TGedFrame(p, width, height, options | kVerticalFrame, back),
   fM(0),
-  fNextEvent(0),
+  fDumpEventInfo(0),
   fEventInfo(0)
 {
   // Constructor.
@@ -40,12 +55,13 @@ AliEveEventManagerEditor::AliEveEventManagerEditor(const TGWindow *p, Int_t widt
 
   {
     TGHorizontalFrame* f = new TGHorizontalFrame(this);
-    fNextEvent = new TGTextButton(f, "Next Event");
-    fNextEvent->SetWidth(100);
-    fNextEvent->ChangeOptions(fNextEvent->GetOptions() | kFixedWidth);
-    f->AddFrame(fNextEvent, new TGLayoutHints(kLHintsNormal, 4,0,0,0));
-    fNextEvent->Connect("Clicked()",
-                       "AliEveEventManagerEditor", this, "DoNextEvent()");
+    fDumpEventInfo = new TGTextButton(f, "Dump Event Info");
+    fDumpEventInfo->SetToolTipText("Append information about current event to event_info.txt file.");
+    fDumpEventInfo->SetWidth(120);
+    fDumpEventInfo->ChangeOptions(fDumpEventInfo->GetOptions() | kFixedWidth);
+    f->AddFrame(fDumpEventInfo, new TGLayoutHints(kLHintsNormal, 4,0,0,0));
+    fDumpEventInfo->Connect("Clicked()",
+                       "AliEveEventManagerEditor", this, "DumpEventInfo()");
     AddFrame(f, new TGLayoutHints(kLHintsExpandX, 8,8,8,0));
   }
   {
@@ -68,7 +84,7 @@ void AliEveEventManagerEditor::SetModel(TObject* obj)
 {
   // Set model object.
 
-  fM = dynamic_cast<AliEveEventManager*>(obj);
+  fM = static_cast<AliEveEventManager*>(obj);
 
   fEventInfo->LoadBuffer(fM->GetEventInfoVertical());
 }
@@ -76,11 +92,17 @@ void AliEveEventManagerEditor::SetModel(TObject* obj)
 /******************************************************************************/
 
 //______________________________________________________________________________
-void AliEveEventManagerEditor::DoNextEvent()
+void AliEveEventManagerEditor::DumpEventInfo()
 {
-  // Load next event
+  // Dump event-info into event_info.txt.
+  // The info is appended into the file.
+
+  ofstream f("event_info.txt", ios::out | ios::app);
 
-  fM->NextEvent();
+  f << "================================================================================\n\n";
+  f << fM->GetEventInfoHorizontal() << std::endl << std::endl;
+
+  f.close();
 }
 
 
@@ -103,11 +125,12 @@ AliEveEventManagerWindow::AliEveEventManagerWindow(AliEveEventManager* mgr) :
   fNextEvent    (0),
   fLastEvent    (0),
   fRefresh      (0),
-  fTrigger      (0),
   fEventId      (0),
   fInfoLabel    (0),
   fAutoLoad     (0),
+  fLoopMarked   (0),
   fAutoLoadTime (0),
+  fTrigSel      (0),
   fEventInfo    (0)
 {
   // Constructor.
@@ -136,7 +159,14 @@ AliEveEventManagerWindow::AliEveEventManagerWindow(AliEveEventManager* mgr) :
     b->Connect("Clicked()", cls, this, "DoNextEvent()");
     fLastEvent = b = MkTxtButton(f, "Last", width);
     b->Connect("Clicked()", cls, this, "DoLastEvent()");
-
+    fMarkEvent = b = MkTxtButton(f, "Mark", width);
+    b->Connect("Clicked()", cls, this, "DoMarkEvent()");
+      fRestartReco = b = MkTxtButton(f, "Restart reco", 2*width);
+      b->Connect("Clicked()", cls, this, "DoRestartReco()");
+      fRestartManager = b = MkTxtButton(f, "Restart manager", 2*width);
+      b->Connect("Clicked()", cls, this, "DoRestartManager()");
+      
+      
     MkLabel(f, "||", 0, 8, 8);
 
     fRefresh = b = MkTxtButton(f, "Refresh", width + 8);
@@ -149,6 +179,11 @@ AliEveEventManagerWindow::AliEveEventManagerWindow(AliEveEventManager* mgr) :
     fAutoLoad->SetToolTipText("Automatic event loading.");
     fAutoLoad->Connect("Toggled(Bool_t)", cls, this, "DoSetAutoLoad()");
 
+    fLoopMarked = new TGCheckButton(f, "Loop Marked");
+    f->AddFrame(fLoopMarked, new TGLayoutHints(kLHintsLeft, 0, 4, 3, 0));
+    fLoopMarked->SetToolTipText("Automatic marked events loading.");
+    fLoopMarked->Connect("Toggled(Bool_t)", cls, this, "DoSetLoopMarked()");
+      
     fAutoLoadTime = new TEveGValuator(f, "Time: ", 110, 0);
     f->AddFrame(fAutoLoadTime);
     fAutoLoadTime->SetShowSlider(kFALSE);
@@ -158,20 +193,25 @@ AliEveEventManagerWindow::AliEveEventManagerWindow(AliEveEventManager* mgr) :
     fAutoLoadTime->SetToolTip("Automatic event loading time in seconds.");
     fAutoLoadTime->Connect("ValueSet(Double_t)", cls, this, "DoSetAutoLoadTime()");
 
-    MkLabel(f, "||", 0, 8, 8);
+    fTrigSel = new TGComboBox(f);
+    fTrigSel->Resize(4*width,b->GetDefaultHeight());
+    fTrigSel->AddEntry("No trigger selection",-1);
+    fTrigSel->Select(-1,kFALSE);
+    f->AddFrame(fTrigSel, new TGLayoutHints(kLHintsNormal, 10, 5, 0, 0));
+    fTrigSel->Connect("Selected(char*)", cls, this, "DoSetTrigSel()");
 
-    MkLabel(f, "TRG select:", 0, 0, 4, 4);
-    fTrigger = new TGComboBox(f);
-    f->AddFrame(fTrigger, new TGLayoutHints(kLHintsNormal));
-    fTrigger->Resize(75,20);
-    //fTrigger->EnableTextInput(kTRUE);
-    fTrigger->Connect("Selected(const char*)", cls, this, "DoSetTriggerType(const char*)");
+    fStorageStatus = MkLabel(f,"Storage: Waiting",0,8,8);
+    fEventServerStatus = MkLabel(f,"Event Server: Waiting",0,10,10);
+      
   }
-
+    
   fEventInfo = new TGTextView(this, 400, 600);
   AddFrame(fEventInfo, new TGLayoutHints(kLHintsNormal | kLHintsExpandX | kLHintsExpandY));
 
-  fM->Connect("NewEventLoaded()", cls, this, "Update()");
+   fM->Connect("NewEventLoaded()", cls, this, "Update(=1)");
+   fM->Connect("NoEventLoaded()", cls, this, "Update(=0)");
+   fM->Connect("StorageManagerOk()",cls,this,"StorageManagerChangedState(=1)");
+   fM->Connect("StorageManagerDown()",cls,this,"StorageManagerChangedState(=0)");
 
   SetCleanup(kDeepCleanup);
   Layout();
@@ -198,14 +238,30 @@ void AliEveEventManagerWindow::DoFirstEvent()
 void AliEveEventManagerWindow::DoPrevEvent()
 {
   // Load previous event
-  fM->PrevEvent();
+  // fM->PrevEvent();
+  if (fM->IsOnlineMode()) {
+    fM->GotoEvent(1);
+  }
+  else {
+    fM->GotoEvent((Int_t) fEventId->GetNumber()-1);
+
+  }
+
 }
 
 //______________________________________________________________________________
 void AliEveEventManagerWindow::DoNextEvent()
 {
   // Load next event
-  fM->NextEvent();
+  // fM->NextEvent();
+  if (fM->IsOnlineMode()) {
+      cout<<"next event, online node"<<endl;
+    fM->GotoEvent(2);
+  }
+  else {
+      cout<<"next event, offline mode"<<endl;
+    fM->GotoEvent((Int_t) fEventId->GetNumber()+1);
+  }
 }
 
 //______________________________________________________________________________
@@ -216,6 +272,79 @@ void AliEveEventManagerWindow::DoLastEvent()
 }
 
 //______________________________________________________________________________
+void AliEveEventManagerWindow::DoMarkEvent()
+{
+  // Mark current event
+  fM->MarkCurrentEvent();
+}
+
+//______________________________________________________________________________
+void AliEveEventManagerWindow::DoRestartReco()
+{
+    ifstream configFile (GetConfigFilePath());
+    string username,hostname;
+    
+    if (configFile.is_open())
+    {
+        string line;
+        int from,to;
+        while(configFile.good())
+        {
+            getline(configFile,line);
+            from = line.find("\"")+1;
+            to = line.find_last_of("\"");
+            if(line.find("EVENT_SERVER=")==0)
+            {
+                hostname=line.substr(from,to-from);
+            }
+            else if(line.find("EVENT_SERVER_USER=")==0)
+            {
+                username=line.substr(from,to-from);
+            }
+        }
+        if(configFile.eof()){configFile.clear();}
+        configFile.close();
+    }
+    else{cout<<"Event Manager Editor -- Unable to open config file"<<endl;}
+
+    // Kill reconstruction server
+    gSystem->Exec(Form("ssh -n -f %s@%s \"killall alieventserver\"",username.c_str(),hostname.c_str()));
+}
+
+void AliEveEventManagerWindow::DoRestartManager()
+{
+    ifstream configFile (GetConfigFilePath());
+    string username,hostname;
+    
+    if (configFile.is_open())
+    {
+        string line;
+        int from,to;
+        while(configFile.good())
+        {
+            getline(configFile,line);
+            from = line.find("\"")+1;
+            to = line.find_last_of("\"");
+            if(line.find("STORAGE_SERVER=")==0)
+            {
+                hostname=line.substr(from,to-from);
+            }
+            else if(line.find("STORAGE_SERVER_USER=")==0)
+            {
+                username=line.substr(from,to-from);
+            }
+        }
+        if(configFile.eof()){configFile.clear();}
+        configFile.close();
+    }
+    else{cout<<"Event Manager Editor -- Unable to open config file"<<endl;}
+
+    
+    // Kill storage manager
+    gSystem->Exec(Form("ssh -n -f %s@%s \"killall alistorage\"",username.c_str(),hostname.c_str()));
+}
+
+//______________________________________________________________________________
 void AliEveEventManagerWindow::DoSetEvent()
 {
   // Set current event
@@ -239,7 +368,15 @@ void AliEveEventManagerWindow::DoSetAutoLoad()
   // Set the auto-load flag
 
   fM->SetAutoLoad(fAutoLoad->IsOn());
-  Update();
+  Update(fM->NewEventAvailable());
+}
+
+//______________________________________________________________________________
+void AliEveEventManagerWindow::DoSetLoopMarked()
+{
+    // Set the auto-load flag
+    fM->SetLoopMarked(fLoopMarked->IsOn());
+//    Update(fM->NewEventAvailable());
 }
 
 //______________________________________________________________________________
@@ -251,50 +388,146 @@ void AliEveEventManagerWindow::DoSetAutoLoadTime()
 }
 
 //______________________________________________________________________________
-void AliEveEventManagerWindow::DoSetTriggerType(const char* type)
+void AliEveEventManagerWindow::DoSetTrigSel()
 {
-  // Slot for setting trigger type.
+  // Set the trigger selection
 
-  TString typestr = type;
-  if (typestr=="")
-  {
-    fM->SetSelectOnTriggerType(kFALSE);
-  }
-  else
-  {
-    fM->SetTriggerType( typestr );
-    fM->SetSelectOnTriggerType(kTRUE);
-  }
+  fM->SetTrigSel(fTrigSel->GetSelectedEntry()->EntryId());
 }
 
 //______________________________________________________________________________
-void AliEveEventManagerWindow::Update()
+void AliEveEventManagerWindow::Update(int state)
 {
-  // Update current event, number of available events, list of active triggers
 
   Bool_t autoLoad = fM->GetAutoLoad();
   Bool_t extCtrl  = fM->IsUnderExternalControl();
   Bool_t evNavOn  = !autoLoad && !extCtrl;
 
-  fFirstEvent->SetEnabled(evNavOn);
-  fPrevEvent ->SetEnabled(evNavOn);
-  fLastEvent ->SetEnabled(evNavOn);
-  fNextEvent ->SetEnabled(!autoLoad);
-  fRefresh   ->SetEnabled(evNavOn);
+#ifdef ZMQ
 
-  fEventId->SetNumber(fM->GetEventId());
-  fEventId->SetState(evNavOn);
-  fInfoLabel->SetText(Form("/ %d", fM->GetMaxEventId()));
+  AliESDEvent*  esd = fM->GetESD();
+  AliStorageAdministratorPanelListEvents* listEventsTab = AliStorageAdministratorPanelListEvents::GetInstance();
+  AliEveConfigManager *configManager = AliEveConfigManager::GetMaster();
 
-  // fAutoLoadTime->SetEnabled(fM->GetAutoLoad());
-  fAutoLoad->SetState(fM->GetAutoLoad() ? kButtonDown : kButtonUp);
-  fAutoLoadTime->SetValue(fM->GetAutoLoadTime());
+  if (!fM->IsOnlineMode()) {
 
-  fEventInfo->LoadBuffer(fM->GetEventInfoHorizontal());
+      listEventsTab->SetOfflineMode(kTRUE);
 
-  SetupTriggerSelect();
+      fFirstEvent->SetEnabled(!autoLoad);
+      fPrevEvent ->SetEnabled(!autoLoad);
+      fLastEvent ->SetEnabled(!autoLoad);
+      fNextEvent ->SetEnabled(!autoLoad);
+      fMarkEvent ->SetEnabled(kFALSE);
 
-  Layout();
+      fInfoLabel->SetText(Form("/ %d", fM->GetMaxEventId()));
+
+  }
+  else if (1 == state && fStorageStatus->GetText()->Contains("DOWN")) {
+    fEventServerStatus->SetText("Event Server: OK");
+    fFirstEvent->SetEnabled(kFALSE);
+    fPrevEvent ->SetEnabled(kFALSE);
+    fLastEvent ->SetEnabled(!autoLoad);
+    fNextEvent ->SetEnabled(kFALSE);
+    fMarkEvent ->SetEnabled(kFALSE);
+    listEventsTab->SetOfflineMode(kTRUE);
+    fInfoLabel->SetText(Form("/ %d",fM->GetEventId()));
+  }
+  else if (0 == state && fStorageStatus->GetText()->Contains("DOWN")) {
+    fEventServerStatus->SetText("Event Server: Waiting");
+    fFirstEvent->SetEnabled(kFALSE);
+    fPrevEvent ->SetEnabled(kFALSE);
+    fLastEvent ->SetEnabled(!autoLoad);
+    fNextEvent ->SetEnabled(kFALSE);
+    fMarkEvent ->SetEnabled(kFALSE);
+    listEventsTab->SetOfflineMode(kTRUE);
+  }
+  else if (0 == state && fStorageStatus->GetText()->Contains("OK")) {
+    fEventServerStatus->SetText("Event Server: Waiting");
+    fFirstEvent->SetEnabled(!autoLoad);
+    fPrevEvent ->SetEnabled(!autoLoad);
+    fLastEvent ->SetEnabled(!autoLoad);
+    fNextEvent ->SetEnabled(!autoLoad);
+    fMarkEvent ->SetEnabled(kTRUE);
+    listEventsTab->SetOfflineMode(kFALSE);
+  }
+  else {
+    fEventServerStatus->SetText("Event Server: OK");
+    fFirstEvent->SetEnabled(!autoLoad);
+    fPrevEvent ->SetEnabled(!autoLoad);
+    fLastEvent ->SetEnabled(!autoLoad);
+    fNextEvent ->SetEnabled(!autoLoad);
+    fMarkEvent ->SetEnabled(kTRUE);
+    listEventsTab->SetOfflineMode(kFALSE);
+  }
+#endif
+
+  if (1 == state) {
+    fRefresh   ->SetEnabled(evNavOn);
+
+    fEventId->SetNumber(fM->GetEventId());
+    fEventId->SetState(evNavOn);
+    // fInfoLabel->SetText(Form("/ %d", fM->GetMaxEventId()));
+
+    fAutoLoad->SetState(fM->GetAutoLoad() ? kButtonDown : kButtonUp);
+    fAutoLoadTime->SetValue(fM->GetAutoLoadTime());
+
+    // Loop over active trigger classes
+    if (fM->GetESD()) {
+      for(Int_t iTrig = 0; iTrig < AliESDRun::kNTriggerClasses; iTrig++) {
+       TString trigName = fM->GetESD()->GetESDRun()->GetTriggerClass(iTrig);
+       if (trigName.IsNull()) {
+         if (fTrigSel->GetListBox()->GetEntry(iTrig)) {
+           if (fTrigSel->GetSelected() == iTrig) fTrigSel->Select(-1);
+           fTrigSel->RemoveEntry(iTrig);
+         }
+         continue;
+       }
+       if (!fTrigSel->FindEntry(trigName.Data()))
+         fTrigSel->AddEntry(trigName.Data(),iTrig);
+      }
+    }
+    fTrigSel->SetEnabled(!evNavOn);
+
+//    fEventInfo->LoadBuffer(fM->GetEventInfoHorizontal());
+
+    Layout();
+  }
+}
+
+void AliEveEventManagerWindow::StorageManagerChangedState(int state)
+{
+#ifdef ZMQ
+
+  Bool_t autoLoad = fM->GetAutoLoad();
+  AliStorageAdministratorPanelListEvents* listEventsTab = AliStorageAdministratorPanelListEvents::GetInstance();
+  AliEveConfigManager *configManager = AliEveConfigManager::GetMaster();
+
+  if (fM->IsOnlineMode()) {
+    if (state == 0)
+      {
+        fStorageStatus->SetText("Storage: DOWN");
+        fMarkEvent->SetEnabled(kFALSE);
+        fNextEvent->SetEnabled(kFALSE);
+        fLastEvent->SetEnabled(!autoLoad);
+        fPrevEvent->SetEnabled(kFALSE);
+        fFirstEvent->SetEnabled(kFALSE);
+       listEventsTab->SetOfflineMode(kTRUE);
+       fEventId->SetState(kFALSE);
+      }
+    else if(state == 1)
+      {
+       listEventsTab->SetOfflineMode(kFALSE);
+
+        fStorageStatus->SetText("Storage: OK");
+        fMarkEvent->SetEnabled(kTRUE);
+        fNextEvent->SetEnabled(!autoLoad);
+        fLastEvent->SetEnabled(!autoLoad);
+        fPrevEvent->SetEnabled(!autoLoad);
+        fFirstEvent->SetEnabled(!autoLoad);
+       fEventId->SetState(!autoLoad);
+      }
+  }
+#endif
 }
 
 //------------------------------------------------------------------------------
@@ -335,42 +568,3 @@ TGLabel* AliEveEventManagerWindow::MkLabel(TGCompositeFrame* p,
   return l;
 }
 
-void AliEveEventManagerWindow::SetupTriggerSelect()
-{
-  // Do nothing if already enabled.
-  if (fTrigger->GetNumberOfEntries() > 0)
-    return;
-
-  AliESDEvent* esd = fM->GetESD();
-  if (esd && fM->GetESDFile() != 0)
-  {
-    TString activetrg = esd->GetESDRun()->GetActiveTriggerClasses();  //Get list of active classes
-    TObjArray* activetrgarr = activetrg.Tokenize(" "); //break up the classes string, space as separator
-    Int_t entries = activetrgarr->GetEntries();  //how many triggerclasses
-    TString entry;  //to hold the triger class name
-    TObjString* entryobj;
-    if (entries == 0)
-    {
-      fTrigger->SetEnabled(kFALSE);  //no trigger classes
-    }
-    else
-    {
-      fTrigger->RemoveAll(); //some initial cleanup
-      fTrigger->SetEnabled(kTRUE);  //no trigger classes
-      fTrigger->AddEntry("",-1);  //first entry empty - select to not filter by trigger
-      for (Int_t i=0;i<entries;i++)
-      {
-       entryobj = (TObjString*)activetrgarr->At(i);
-       entry = entryobj->GetString();
-       fTrigger->AddEntry(entry.Data(), i);
-      }
-    }
-    fTrigger->Select(-1, kTRUE); //set default no filtering and emit
-    fTrigger->SetEnabled(kTRUE);
-  }
-  else
-  {
-    fTrigger->SetEnabled(kFALSE);
-  }
-}
-