QGCanvasWindow.cc

Go to the documentation of this file.

#include "QGCanvasWindow.hh"
#include <algorithm>
#include <iostream>
#include "QGCanvasSetupFrame.hh"
#include "QGCanvasWindowButtonsFrame.hh"
#include "QGFileHandler.hh"
#include "QStyle.hh"
#include "QGHistogram.hh"
#include "QGPlot.hh"
#include "QGPlotEditWindow.hh"
#include "QGPlotFrame.hh"
#include "QGSessionHandler.hh"
#include "TCanvas.h"
#include "TGButton.h"
#include "TGLabel.h"
#include "TRootEmbeddedCanvas.h"
#include "TInterpreter.h"
 
ClassImp(QGCanvasWindow)
 
using std::cout;
using std::endl;
using std::list;
using std::string;
using std::vector;
 
TStyle* QStyle = 0;
 
QGCanvasWindow::QGCanvasWindow(const TGWindow *p, UInt_t w, UInt_t h) : 
QGWindow(p, w, h),
fFileHandler(0), fPlotFrame(0), fSetupFrame(0) 
{
    if(!QStyle) new QStyles;
 
    gInterpreter->ProcessLine("using namespace Diana;");
    fFileHandler = new QGFileHandler();
    
    fSetupFrame = new QGCanvasSetupFrame(this, w, h);
    AddFrame(fSetupFrame, new TGLayoutHints(kLHintsCenterX | kLHintsCenterY, 2, 2, 2, 0));
    
    fPlotFrame = new QGPlotFrame(this, w, h);
    AddFrame(fPlotFrame, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY, 2, 2, 2, 0));
    
    fFilenameLabel = new TGLabel(this, "File:  ");
    AddFrame(fFilenameLabel, new TGLayoutHints(kLHintsLeft, 2, 2, 2, 0));
    
    fButtonsFrame = new QGCanvasWindowButtonsFrame(this, w);
    AddFrame(fButtonsFrame, new TGLayoutHints(kLHintsBottom, 2, 2, 2, 2));
    
    Resize(w, h);
    MapSubwindows();
    fSetupFrame->SetDisplay();
    fButtonsFrame->ShowDefaultDisplay();
    DisplaySetupFrame();
    MapWindow();
    
    UpdateWindowDropDownBoxes();
}
 
QGCanvasWindow::~QGCanvasWindow()
{
    // disassociate plots from this canvas window
    while (fPlots.begin() != fPlots.end()) {
        (*fPlots.begin())->SetCanvasWindow(0); // removes this plot from the list
    }
    delete fFileHandler;
}
 
void QGCanvasWindow::AddPlot(QGPlot *plot)
{
    if (plot == 0) {
        cout << "QGCanvasWindow::AddPlot(QGPlot *plot)        Warning:  plot == 0" << endl;
    } else if (find(fPlots.begin(), fPlots.end(), plot) == fPlots.end()) {
        fPlots.push_back(plot);
    }
}
 
void QGCanvasWindow::AddToDrawnPlots(QGPlot *plot)
{
    if (plot == 0) {
        cout << "QGCanvasWindow::AddToDrawnPlots(QGPlot *plot)        Warning:  plot == 0" << endl;
    } else if (find(fDrawnPlots.begin(), fDrawnPlots.end(), plot) == fDrawnPlots.end()) {
        fDrawnPlots.push_back(plot);
    }
}
 
void QGCanvasWindow::ClearDrawnPlots()
{
    fDrawnPlots.clear();
}
 
void QGCanvasWindow::DisplayPlotFrame()
{
    HideFrame(fSetupFrame);    
    ShowFrame(fPlotFrame);
    ShowFrame(fFilenameLabel);
    if (fFileHandler->IsFileOpen()) {
        ShowFrame(fFilenameLabel);
    } else {
        HideFrame(fFilenameLabel);
    }
    fButtonsFrame->ShowDefaultDisplay();
}
 
void QGCanvasWindow::DisplaySetupFrame()
{
    HideFrame(fPlotFrame);
    fSetupFrame->SetDisplay();
    ShowFrame(fSetupFrame);
    if (fFileHandler->IsFileOpen()) {
        ShowFrame(fFilenameLabel);
    } else {
        HideFrame(fFilenameLabel);
    }
    fButtonsFrame->ShowSetupWindowDisplay();
}
 
void QGCanvasWindow::RemoveFromDrawnPlots(QGPlot *plot)
{
    if (find(fDrawnPlots.begin(), fDrawnPlots.end(), plot) != fDrawnPlots.end()) {
        fDrawnPlots.erase(find(fDrawnPlots.begin(), fDrawnPlots.end(), plot));
    }
}
 
void QGCanvasWindow::RemovePlot(QGPlot *plot)
{
    if (find(fPlots.begin(), fPlots.end(), plot) != fPlots.end()) {
        fPlots.erase(find(fPlots.begin(), fPlots.end(), plot));
    }
    RemoveFromDrawnPlots(plot);
}
 
void QGCanvasWindow::SetPlotsToRegenerate()
{
    list<QGPlot*>::const_iterator plotIter;
    for (plotIter = fPlots.begin(); plotIter != fPlots.end(); ++plotIter) {
        (*plotIter)->SetRegenerate();
    }
}
 
void QGCanvasWindow::SetProperty(string key, string value)
{
  if (key == FILE_KEY) {
    GetFileHandler()->OpenFile(value);
    if (GetFileHandler()->IsFileOpen()) {
      DisplayPlotFrame();
      UpdateFileLabel();
    }
    SetPlotsToRegenerate();
    UpdateVariablesDropDownBoxes();
  }
}
 
void QGCanvasWindow::UpdateFileLabel()
{
    if (GetFileHandler()->IsFileOpen()) {
        string label = "File:  ";
        vector<string> filenames = GetFileHandler()->GetFilenames();
        if (filenames.size() == 1) {
            string::size_type pos = filenames[0].find("/");
            if (pos == string::npos) {
                label += filenames[0];
            } else {
                label += filenames[0].substr(filenames[0].rfind("/") + 1);
            }
        } else if (filenames.size() > 1) {
            label += "Multiple files";
        }
        fFilenameLabel->SetText(label.c_str());
        fFilenameLabel->Resize();
    }
}
 
void QGCanvasWindow::UpdateVariablesDropDownBoxes()
{
    list<QGPlot*>::const_iterator plotIter;
    for (plotIter = fPlots.begin(); plotIter != fPlots.end(); ++plotIter) {
        if ((*plotIter)->GetEditWindow()) {
            (*plotIter)->GetEditWindow()->FillVariablesDropDownBoxes();
        }
    }
}