QGCutsEditorWindow.cc

Go to the documentation of this file.

#include "QGCutsEditorWindow.hh"
#include <list>
#include <sstream>
#include <string>
#include <vector>
#include "QGCutsFrame.hh"
#include "QGDefaultsHandler.hh"
#include "QGGraphicalCut.hh"
#include "QGSessionHandler.hh"
#include "QGTextFileHandler.hh"
#include "TGButton.h"
#include "TGFileDialog.h"
#include "TGPicture.h"
#include "TGTextEdit.h"
#include "TString.h"
 
// Images
#include "ImagePaste.xpm"
#include "ImageSave.xpm"
 
ClassImp(QGCutsEditorWindow)
 
using std::flush;
using std::list;
using std::string;
using std::stringstream;
using std::vector;
 
QGCutsEditorWindow::QGCutsEditorWindow(const TGWindow *p, UInt_t w, UInt_t h) :
QGWindow(p, w, h)
{
    SetWindowName("Cuts Editor");
    
    fTextEditor = new TGTextEdit(this);
    AddFrame(fTextEditor, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY, 5, 5, 5, 0));
    
    fButtonsFrame = new TGHorizontalFrame(this);
    AddFrame(fButtonsFrame, new TGLayoutHints(kLHintsCenterX, 5, 5, 5, 5));
 
    fPasteButton = new TGPictureButton(fButtonsFrame, gClient->GetPicturePool()->GetPicture("Paste", (char**)ImagePaste));
    fPasteButton->Connect("Clicked()", "QGCutsEditorWindow", this, "HandlePasteButton()");
    fPasteButton->SetToolTipText("Paste cuts");
    fButtonsFrame->AddFrame(fPasteButton, new TGLayoutHints(kLHintsCenterY, 0, 0, 0, 0));
 
    fSaveButton = new TGPictureButton(fButtonsFrame, gClient->GetPicturePool()->GetPicture("Save", (char**)ImageSave));
    fSaveButton->Connect("Clicked()", "QGCutsEditorWindow", this, "HandleSaveButton()");
    fSaveButton->SetToolTipText("Save these cuts as well as any graphical cuts to file");
    fButtonsFrame->AddFrame(fSaveButton, new TGLayoutHints(kLHintsCenterY, 0, 0, 0, 0));
    
    MapSubwindows();
    Resize(GetDefaultSize());
    Resize(w, h);
    MapWindow();
}
 
QGCutsEditorWindow::~QGCutsEditorWindow()
{
}
 
void QGCutsEditorWindow::HandlePasteButton()
{
    const vector<string>& cuts = QGCutsFrame::GetCopiedCuts();
    if (cuts.size() != 0) {
        if (fTextEditor->ReturnLineCount() > 1) {
            fTextEditor->AddLineFast("");
            fTextEditor->AddLineFast(" || ");
            fTextEditor->AddLineFast("");
        }
        fTextEditor->AddLineFast("(");
        vector<string>::const_iterator cutIter;
        for (cutIter = cuts.begin(); cutIter != cuts.end(); ++cutIter) {
            string cut = "   (" + *cutIter + ")";
            fTextEditor->AddLineFast(cut.c_str());
            if (cutIter + 1 != cuts.end()) {
                fTextEditor->AddLineFast("    && ");
            }
        }
        fTextEditor->AddLineFast(")");
        fTextEditor->Update();
    }
}
 
void QGCutsEditorWindow::HandleSaveButton()
{
    // open file selection dialog box
    string cutsDirectory = QGDefaultsHandler::Instance()->GetCutsDirectory();
    TString dir(cutsDirectory.c_str());
    TGFileInfo fi;
    fi.fIniDir = StrDup(dir);
    const char *filetypes[] = {"GUI session files" , "*.gui",
                                0,                    0};
    fi.fFileTypes = filetypes;
    new TGFileDialog(gClient->GetRoot(), 0, kFDSave, &fi);
    if (fi.fFilename) {
        TString filename(fi.fFilename);
        TString extension(fi.fFileTypes[fi.fFileTypeIdx+1]);
        extension.Remove(TString::kLeading, '*');
        if (!filename.EndsWith(extension)) {
            filename.Append(extension);
        }
        Save(filename.Data());
        string currentDirectory = fi.fIniDir;
        if (currentDirectory != cutsDirectory) {
            QGDefaultsHandler::Instance()->SetCutsDirectory(currentDirectory);
        }
    }
}
 
void QGCutsEditorWindow::Save(const string& filename)
{
    // open file handler for output
    QGTextFileHandler *textFileHandler = new QGTextFileHandler();
    
    // save cuts string
    textFileHandler->AppendLine(CUTS_STRING_KEY);
    textFileHandler->AppendLine(fTextEditor->GetText()->AsString().Data());
    textFileHandler->AppendLine();
    
    // save graphical cuts
    const list<QGPlot*>& plots = QGPlot::GetPlots();
    list<QGPlot*>::const_iterator plotIter;
    for (plotIter = plots.begin(); plotIter != plots.end(); ++plotIter) {
        if (QGGraphicalCut *graphicalCut = dynamic_cast<QGGraphicalCut*>(*plotIter)) {
            string name = graphicalCut->GetName();
            string xVariable = graphicalCut->GetXVariable();
            string yVariable = graphicalCut->GetYVariable();
            textFileHandler->AppendLine(GRAPHICAL_CUT_KEY);
            textFileHandler->AppendLine(NAME_KEY, name);
            textFileHandler->AppendLine(X_VARIABLE_KEY, xVariable);
            textFileHandler->AppendLine(Y_VARIABLE_KEY, yVariable);
            for (Int_t vertex = 0; vertex < graphicalCut->GetN() - 1; ++vertex) {
                stringstream vertexStream;
                vertexStream.precision(16);
                vertexStream << graphicalCut->GetX()[vertex] << " , " << graphicalCut->GetY()[vertex] << flush;
                textFileHandler->AppendLine(VERTEX_KEY, vertexStream.str());
            }
            textFileHandler->AppendLine();
        }
    }
    
    // write to file
    textFileHandler->WriteFile(filename.c_str());
    delete textFileHandler;
}