DisplayWindow.cxx

Go to the documentation of this file.

#include <iostream>
#include <vector>
#include "TROOT.h"
#include "TGClient.h"
#include "TGWindow.h"
#include "TGFrame.h"
#include "TRootEmbeddedCanvas.h"
#include "TCanvas.h"
#include "TH1.h"

#include "nutools/EventDisplayBase/DisplayWindow.h"
#include "nutools/EventDisplayBase/FileMenu.h"
#include "nutools/EventDisplayBase/EditMenu.h"
#include "nutools/EventDisplayBase/MenuBar.h"
#include "nutools/EventDisplayBase/ButtonBar.h"
#include "nutools/EventDisplayBase/Canvas.h"
#include "nutools/EventDisplayBase/StatusBar.h"
#include "nutools/EventDisplayBase/EventHolder.h"

#include "cetlib_except/exception.h"

namespace evdb{

  static std::vector<DisplayWindow*> gsWindows(64);

  //......................................................................

  static std::vector<std::string>     gsName;
  static std::vector<std::string>     gsDescription;
  static std::vector<unsigned int>    gsHeight;
  static std::vector<unsigned int>    gsWidth;
  static std::vector<CanvasCreator_t> gsCanvasCreator;

  void DisplayWindow::SetRunEventAll(int run, int event) 
  {
    for (size_t i=0; i<gsWindows.size(); ++i) {
      if (gsWindows[i]!=0) gsWindows[i]->SetRunEvent(run, event);
    }
  }

  //......................................................................

  void DisplayWindow::DrawAll(const char* opt)
  {
    for (size_t i=0; i<gsWindows.size(); ++i) {
      if (gsWindows[i]!=0) gsWindows[i]->Draw(opt);
    }
  }

  //......................................................................

  void DisplayWindow::SetRunEvent(int run, int event) 
  {
    fButtonBar->SetRunEvent(run, event);
  }

  //......................................................................
  
  void DisplayWindow::SetServicesAll()
  {
    for (size_t i=0; i<gsWindows.size(); ++i) {
      if (gsWindows[i]!=0) gsWindows[i]->SetServices();
    }      
  }
  
  //......................................................................

  void DisplayWindow::SetServices()
  {
    fMenuBar->fEditMenu->SetServices();
  }

  //......................................................................

  const std::vector<std::string>& DisplayWindow::Names() { return gsName; }

  //......................................................................

  void DisplayWindow::Register(const char* name,
                               const char* description,
                               unsigned int h,
                               unsigned int w,
                               CanvasCreator_t creator)
  {
    gsName.push_back(std::string(name));
    gsDescription.push_back(std::string(description));
    gsHeight.push_back(h);
    gsWidth.push_back(w);
    gsCanvasCreator.push_back(creator);
  
    if (gsName.size()>gsWindows.size()) gsWindows.resize(gsName.size());
  }

  //......................................................................

  int DisplayWindow::OpenWindow(int type) 
  {
    unsigned id = 0;
    if (type>0) id = type;
    if (id>=gsName.size()) return 0;

    DisplayWindow* w = gsWindows[id];
    if (w==0) {
      w = gsWindows[id] = new DisplayWindow(id);
    }
    if (w==0) return 0;

    // Update run and event number in newly opened window.
    const art::Event* evt = evdb::EventHolder::Instance()->GetEvent();
    if(evt)
      w->SetRunEvent(evt->id().run(), evt->id().event());

    w->Raise();
    w->Draw();
  
    return 1;
  }


  //......................................................................

  DisplayWindow::DisplayWindow(int id) 
  {
    if(gROOT->IsBatch())
      throw cet::exception("DisplayWindow") << "ROOT is in batch mode"
                                            << " cannot open DisplayWindow";
    if(!gClient)
      throw cet::exception("DisplayWindow") << "No ROOT global TClient";

    const TGWindow* tgw = gClient->GetRoot();
    if(!tgw)
      throw cet::exception("DisplayWindow") << "No TGWindow pointer";

    // Create the main application window. I need a resize to get the
    // window to draw the first time, so create the window slightly
    // smaller than the intended size. Bogus, but so it goes...
    unsigned int w = gsWidth[id];
    unsigned int h = gsHeight[id];
    fMain = new TGMainFrame(tgw, w-1, h-1);

    // Add items to the main window
    fMenuBar   = new MenuBar(fMain);
    fButtonBar = new ButtonBar(fMain);
    fDisplay   = (*gsCanvasCreator[id])(fMain);
    fStatusBar = new StatusBar(fMain);
  
    fMain->SetWindowName(gsName[id].c_str());
  
    // Now that all the subwindows are attached, do the final layout
    fMain->MapSubwindows();
    fMain->MapWindow();
  
    // Don't understand this, but I need a resize to get things to draw
    // the first time...
    fMain->Resize(w,h);

    // Plug the display into its signal/slots
    fDisplay->Connect();

    fMain->Connect("CloseWindow()","evdb::DisplayWindow",this,"CloseWindow()");

    // Add to list of windows open
    gsWindows[id] = this;
  }

  //......................................................................

  void DisplayWindow::Draw(const char* opt) { fDisplay->Draw(opt); }

  //......................................................................

  void DisplayWindow::CloseWindow() { delete this; }

  //......................................................................

  DisplayWindow::~DisplayWindow()
  {
    if (fDisplay)   { delete fDisplay;   fDisplay   = 0; } 
    if (fStatusBar) { delete fStatusBar; fStatusBar = 0; }
    if (fButtonBar) { delete fButtonBar; fButtonBar = 0; }
    if (fMenuBar)   { delete fMenuBar;   fMenuBar   = 0; }
    if (fMain)      { delete fMain;      fMain      = 0; }
    for (unsigned int i=0; i<gsWindows.size(); ++i) {
      if (gsWindows[i] == this) gsWindows[i] = 0;
    }
  }

  //......................................................................

  void DisplayWindow::Raise() { fMain->RaiseWindow(); }

}// namespace