PSubTextEdit.cpp

Go to the documentation of this file.

/****************************************************************************
 
  PSubTextEdit.cpp
 
  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch
 
*****************************************************************************/
 
/***************************************************************************
 *   Copyright (C) 2009-2026 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

 
#include <memory>
 
#include <QAction>
#include <QMenu>
#include <QDateTime>
#include <QLineEdit>
#include <QComboBox>
#include <QMessageBox>
#include <QIcon>
#include <QPixmap>
#include <QImage>
 
#include "PAdmin.h"
#include "PSubTextEdit.h"
#include "PGetTitleBlockDialog.h"
#include "PGetParameterBlockDialog.h"
#include "PGetTheoryBlockDialog.h"
#include "PGetFunctionsBlockDialog.h"
#include "PGetAsymmetryRunBlockDialog.h"
#include "PGetSingleHistoRunBlockDialog.h"
#include "PGetNonMusrRunBlockDialog.h"
#include "PGetFourierBlockDialog.h"
#include "PGetPlotBlockDialog.h"
#include "PMsrHighlighter.h"
 
//----------------------------------------------------------------------------------------------------
PSubTextEdit::PSubTextEdit(PAdmin *admin, QWidget *parent) :
                           QPlainTextEdit(parent),
                           fAdmin(admin)
{
  // Pass the current text document to the highlighter. This ensures that every
  // time a new .msr file or tab is spawned, the document registers the parser
  // formatting.
  PMsrHighlighter *highlight = new PMsrHighlighter(this->document());
}
 
//----------------------------------------------------------------------------------------------------
int PSubTextEdit::getFitType()
{
  QString str = toPlainText();
  int idx = str.indexOf("fittype");
  if (idx == -1)
    return -1;
 
  bool ok;
  for (int i=idx+7; i<str.length(); i++) {
    if (str[i] != ' ') {
      idx = QString(str[i]).toInt(&ok);
      if (!ok)
        idx = -1;
      break;
    }
  }
 
  return idx;
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertTitle()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetTitleBlockDialog> dlg = std::make_unique<PGetTitleBlockDialog>(fAdmin->getHelpUrl("title"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    QString title = dlg->getTitle();
    insertPlainText(title+"\n");
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertParameterBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetParameterBlockDialog> dlg = std::make_unique<PGetParameterBlockDialog>(fAdmin->getHelpUrl("parameters"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    insertPlainText(dlg->getParams());
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertTheoryFunction(QString name)
{
  QString str = "????";
 
  int idx = -1;
  for (unsigned int i=0; i<fAdmin->getTheoryCounts(); i++) {
    if (name == fAdmin->getTheoryItem(i)->label) {
      idx = i;
      break;
    }
  }
 
  if (idx == -1)
    return;
 
  PTheory *theoItem = fAdmin->getTheoryItem(idx);
  if (theoItem == 0)
    return;
 
  // add theory function name
  str = theoItem->name + "   ";
  if (theoItem->name == "userFcn") {
    str += "libMyLibrary.so  TMyFunction  ";
  }
 
  // add pseudo parameters
  for (int i=0; i<theoItem->params; i++) {
    str += QString("%1").arg(i+1) + "   ";
  }
 
  // add comment
  str += theoItem->comment;
 
  // add newline
  str += "\n";
 
  insertPlainText(str);
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertTheoryBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetTheoryBlockDialog> dlg = std::make_unique<PGetTheoryBlockDialog>(fAdmin, fAdmin->getHelpUrl("theory"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    insertPlainText(dlg->getTheoryBlock());
    insertPlainText("\n");
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertFunctionBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetFunctionsBlockDialog> dlg = std::make_unique<PGetFunctionsBlockDialog>(fAdmin->getHelpUrl("functions"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    insertPlainText(dlg->getFunctionsBlock());
    insertPlainText("\n\n");
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertAsymRunBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetAsymmetryRunBlockDialog> dlg = std::make_unique<PGetAsymmetryRunBlockDialog>(fAdmin->getHelpUrl("run"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    QString str, workStr;
    bool valid = true, present = true;
    // check if there is already a run block present, necessary because of the '####' line
    // STILL MISSING
 
    // add run line
    str += dlg->getRunHeaderInfo();
 
    // add fittype
    str += "fittype         2         (asymmetry fit)\n";
 
    // add alpha if present
    workStr = dlg->getAlphaParameter(present);
    if (present) {
      str += workStr;
    }
 
    // add beta if present
    workStr = dlg->getBetaParameter(present);
    if (present) {
      str += workStr;
    }
 
    // add map
    workStr = dlg->getMap(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Given map not valid, will add a default map line",
         QMessageBox::Ok, QMessageBox::NoButton);
      str += "map             0  0  0  0  0  0  0  0  0  0\n";
    }
 
    // add forward
    str += dlg->getForward();
 
    // add backward
    str += dlg->getBackward();
 
    // add background or backgr.fix
    workStr = dlg->getBackground(valid);
    str += workStr;
    if (!valid) {
      QMessageBox::critical(this, "ERROR",
         "Either <b>background</b> or <b>backgr.fix</b> is needed!\nWill set <b>background</b> to 0..10, please correct!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add data
    workStr = dlg->getData(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Not all Data entries are present.Fix is needed!\nWill not set anything!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add t0 if present
    workStr = dlg->getT0(present);
    if (present) {
      str += workStr;
    } else {
      QMessageBox::warning(this, "ERROR",
         "T0's not given, assume that they are present in the data file!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add fit range
    workStr = dlg->getFitRange(valid);
    str += workStr;
    if (!valid) {
      QMessageBox::critical(this, "ERROR",
         "No valid fit range is given.Fix is needed!\nWill add a default one!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add packing
    workStr = dlg->getPacking(present);
    str += workStr;
    if (!present) {
      QMessageBox::critical(this, "ERROR",
         "No valid packing/binning is given.Fix is needed!\nWill add a default one!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // insert Asymmetry Run Block at the current cursor position
    insertPlainText(str);
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertSingleHistRunBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetSingleHistoRunBlockDialog> dlg = std::make_unique<PGetSingleHistoRunBlockDialog>(fAdmin->getHelpUrl("run"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    QString str, workStr;
    bool valid = true, present = true;
    // check if there is already a run block present, necessary because of the '####' line
    // STILL MISSING
 
    // add run line
    str += dlg->getRunHeaderInfo();
 
    // add fittype
    str += "fittype         0         (single histogram fit)\n";
 
    // add map
    workStr = dlg->getMap(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Given map not valid, will add a default map line",
         QMessageBox::Ok, QMessageBox::NoButton);
      str += "map             0  0  0  0  0  0  0  0  0  0\n";
    }
 
    // add forward
    str += dlg->getForward();
 
    // add norm
    str += dlg->getNorm();
 
    // add lifetime parameter
    workStr = dlg->getMuonLifetimeParam(present);
    if (present) {
      str += workStr;
    }
 
    // add lifetime correction flag if present
    workStr = dlg->getLifetimeCorrection(present);
    if (present) {
      str += workStr;
    }
 
    // add background, backgr.fix or backgr.fit
    workStr = dlg->getBackground(valid);
    str += workStr;
    if (!valid) {
      QMessageBox::critical(this, "ERROR",
         "Either <b>background</b>, <b>backgr.fix</b>, or <b>backgr.fit</b> is needed!\nWill set <b>background</b> to 0..10, please correct!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add t0 if present
    workStr = dlg->getT0(present);
    if (present) {
      str += workStr;
    } else {
      QMessageBox::warning(this, "ERROR",
         "T0's not given, assume that they are present in the data file!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add data
    workStr = dlg->getData(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Not all Data entries are present.Fix is needed!\nWill not set anything!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add fit range
    workStr = dlg->getFitRange(valid);
    str += workStr;
    if (!valid) {
      QMessageBox::critical(this, "ERROR",
         "No valid fit range is given.Fix is needed!\nWill add a default one!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add packing
    workStr = dlg->getPacking(present);
    str += workStr;
    if (!present) {
      QMessageBox::critical(this, "ERROR",
         "No valid packing/binning is given.Fix is needed!\nWill add a default one!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // insert Single Histogram Run Block at the current cursor position
    insertPlainText(str);
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertNonMusrRunBlock()
{
  std::unique_ptr<PGetNonMusrRunBlockDialog> dlg = std::make_unique<PGetNonMusrRunBlockDialog>(fAdmin->getHelpUrl("run"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    QString str, workStr;
    bool valid = true;
    // check if there is already a run block present, necessary because of the '####' line
    // STILL MISSING
 
    // add run line
    str += dlg->getRunHeaderInfo();
 
    // add fittype
    str += "fittype         8         (non musr fit)\n";
 
    // add map
    workStr = dlg->getMap(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Given map not valid, will add a default map line",
         QMessageBox::Ok, QMessageBox::NoButton);
      str += "map             0  0  0  0  0  0  0  0  0  0\n";
    }
 
    // add xy-data
    workStr = dlg->getXYData(valid);
    if (valid) {
      str += workStr;
    } else {
      QMessageBox::critical(this, "ERROR",
         "Not all xy-data entries are present.Fix is needed!\nWill not set anything!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add fit range
    workStr = dlg->getFitRange(valid);
    str += workStr;
    if (!valid) {
      QMessageBox::critical(this, "ERROR",
         "No valid fit range is given.Fix is needed!\nWill add a default one!",
         QMessageBox::Ok, QMessageBox::NoButton);
    }
 
    // add packing
    str += "packing         1\n";
 
    // insert NonMusr Run Block at the current cursor position
    insertPlainText(str);
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertCommandBlock()
{
  insertPlainText("###############################################################\n");
  insertPlainText("COMMANDS\n");
  insertPlainText("SET BATCH\n");
  insertPlainText("STRATEGY 1\n");
  insertPlainText("MINIMIZE\n");
  insertPlainText("#MINOS\n");
  insertPlainText("SAVE\n");
  insertPlainText("END RETURN\n\n");
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertFourierBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetFourierBlockDialog> dlg = std::make_unique<PGetFourierBlockDialog>(fAdmin->getHelpUrl("fourier"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    insertPlainText(dlg->getFourierBlock()+"\n");
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertPlotBlock()
{
  // for the time being the url's are hard coded but should be transfered to the XML startup
  std::unique_ptr<PGetPlotBlockDialog> dlg = std::make_unique<PGetPlotBlockDialog>(fAdmin->getHelpUrl("plot"));
 
  if (dlg == nullptr)
    return;
 
  if (dlg->exec() == QDialog::Accepted) {
    insertPlainText(dlg->getPlotBlock()+"\n");
  }
}
 
//----------------------------------------------------------------------------------------------------
void PSubTextEdit::insertStatisticBlock()
{
  QDateTime dt = QDateTime::currentDateTime();
  insertPlainText("###############################################################\n");
  insertPlainText("STATISTIC --- " + dt.toString("yyyy-MM-dd hh:mm:ss") + "\n");
  insertPlainText("chisq = ????, NDF = ????, chisq/NDF = ????\n\n");
}
 
//----------------------------------------------------------------------------------------------------
// END
//----------------------------------------------------------------------------------------------------