611 lines
17 KiB
C++
611 lines
17 KiB
C++
/****************************************************************************
|
|
|
|
PmuppAdmin.cpp
|
|
|
|
Author: Andreas Suter
|
|
e-mail: andreas.suter@psi.ch
|
|
|
|
*****************************************************************************/
|
|
|
|
/***************************************************************************
|
|
* Copyright (C) 2010-2017 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 <cstdlib>
|
|
#include <iostream>
|
|
using namespace std;
|
|
|
|
#include <QMessageBox>
|
|
#include <QString>
|
|
#include <QFile>
|
|
#include <QTextStream>
|
|
#include <QVector>
|
|
#include <QDir>
|
|
#include <QProcessEnvironment>
|
|
#include <QSysInfo>
|
|
|
|
#include <QtDebug>
|
|
|
|
#include "PmuppAdmin.h"
|
|
|
|
//--------------------------------------------------------------------------
|
|
// implementation of PmuppColor class
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppColor::PmuppColor
|
|
*/
|
|
PmuppColor::PmuppColor()
|
|
{
|
|
fName = "UnDef";
|
|
fRed = -1;
|
|
fGreen = -1;
|
|
fBlue = -1;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppColor::setRGB
|
|
* @param r
|
|
* @param g
|
|
* @param b
|
|
*/
|
|
void PmuppColor::setRGB(const int r, const int g, const int b)
|
|
{
|
|
if ((r>=0) && (r<=255))
|
|
fRed = r;
|
|
if ((g>=0) && (g<=255))
|
|
fGreen = g;
|
|
if ((b>=0) && (b<=255))
|
|
fBlue = b;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
// implementation of PmuppAdminXMLParser class
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>XML Parser class for the mupp administration file.
|
|
*
|
|
* \param admin pointer to an admin class instance.
|
|
*/
|
|
PmuppAdminXMLParser::PmuppAdminXMLParser(PmuppAdmin *admin) : fAdmin(admin)
|
|
{
|
|
fKeyWord = eEmpty;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Routine called at the beginning of the XML parsing process.
|
|
*/
|
|
bool PmuppAdminXMLParser::startDocument()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Routine called when a new XML tag is found. Here it is used
|
|
* to set a tag for filtering afterwards the content.
|
|
*
|
|
* \param qName name of the XML tag.
|
|
*/
|
|
bool PmuppAdminXMLParser::startElement( const QString&, const QString&,
|
|
const QString& qName,
|
|
const QXmlAttributes& )
|
|
{
|
|
if (qName == "path_file_name") {
|
|
fKeyWord = eRecentFile;
|
|
} else if (qName == "marker") {
|
|
fKeyWord = eMarker;
|
|
} else if (qName == "color") {
|
|
fKeyWord = eColor;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Routine called when the end XML tag is found. It is used to
|
|
* put the filtering tag to 'empty'. It also resets the fFunc flag in case
|
|
* the entry was a theory function.
|
|
*
|
|
* \param qName name of the element.
|
|
*/
|
|
bool PmuppAdminXMLParser::endElement( const QString&, const QString&, const QString& )
|
|
{
|
|
fKeyWord = eEmpty;
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>This routine delivers the content of an XML tag. It fills the
|
|
* content into the load data structure.
|
|
*
|
|
* \param str keeps the content of the XML tag.
|
|
*/
|
|
bool PmuppAdminXMLParser::characters(const QString& str)
|
|
{
|
|
bool ok;
|
|
int ival, r, g, b;
|
|
double dval;
|
|
QString name("");
|
|
QStringList tok;
|
|
|
|
switch (fKeyWord) {
|
|
case eRecentFile:
|
|
fAdmin->addRecentFile(QString(str.toLatin1()).trimmed());
|
|
break;
|
|
case eMarker:
|
|
tok = str.split(",", QString::SkipEmptyParts);
|
|
|
|
if ((tok.count() != 1) && (tok.count() != 2)) {
|
|
return false;
|
|
}
|
|
|
|
ival = tok[0].toInt(&ok);
|
|
if (!ok)
|
|
return false;
|
|
|
|
dval = 1.0;
|
|
if (tok.count() == 2) {
|
|
dval = tok[1].toDouble(&ok);
|
|
if (!ok)
|
|
return false;
|
|
}
|
|
fAdmin->setMarker(ival, dval);
|
|
break;
|
|
case eColor:
|
|
tok = str.split(",", QString::SkipEmptyParts);
|
|
|
|
if ((tok.count() != 3) && (tok.count() != 4)) {
|
|
return false;
|
|
}
|
|
ival = tok[0].toInt(&ok);
|
|
if (!ok)
|
|
return false;
|
|
r = ival;
|
|
ival = tok[1].toInt(&ok);
|
|
if (!ok)
|
|
return false;
|
|
g = ival;
|
|
ival = tok[2].toInt(&ok);
|
|
if (!ok)
|
|
return false;
|
|
b = ival;
|
|
if (tok.count() == 4)
|
|
name = tok[3];
|
|
fAdmin->setColor(r, g, b, name);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Called at the end of the XML parse process. It checks if default paths
|
|
* contain system variables, and if so expand them for the further use.
|
|
*/
|
|
bool PmuppAdminXMLParser::endDocument()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Report XML warnings.
|
|
*
|
|
* \param exception holds the information of the XML warning
|
|
*/
|
|
bool PmuppAdminXMLParser::warning( const QXmlParseException & exception )
|
|
{
|
|
QString msg;
|
|
|
|
msg = QString("**WARNING** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
|
|
msg += QString("**WARNING MESSAGE** ") + exception.message();
|
|
|
|
qWarning() << endl << msg << endl;
|
|
|
|
QMessageBox::warning(0, "WARNING", msg, QMessageBox::Ok, QMessageBox::NoButton);
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Report recoverable XML errors.
|
|
*
|
|
* \param exception holds the information of the XML recoverable errors.
|
|
*/
|
|
bool PmuppAdminXMLParser::error( const QXmlParseException & exception )
|
|
{
|
|
QString msg;
|
|
|
|
msg = QString("**ERROR** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
|
|
msg += QString("**ERROR MESSAGE** ") + exception.message();
|
|
|
|
qWarning() << endl << msg << endl;
|
|
|
|
QMessageBox::critical(0, "ERROR", msg, QMessageBox::Ok, QMessageBox::NoButton);
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Report fatal XML errors.
|
|
*
|
|
* \param exception holds the information of the XML fatal errors.
|
|
*/
|
|
bool PmuppAdminXMLParser::fatalError( const QXmlParseException & exception )
|
|
{
|
|
QString msg;
|
|
|
|
msg = QString("**FATAL ERROR** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
|
|
msg += QString("**FATAL ERROR MESSAGE** ") + exception.message();
|
|
|
|
qWarning() << endl << msg << endl;
|
|
|
|
QMessageBox::critical(0, "FATAL ERROR", msg, QMessageBox::Ok, QMessageBox::NoButton);
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
// implementation of PmuppAdmin class
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Initializes that PmuppAdmin object, and calls the XML parser which feeds
|
|
* the object variables.
|
|
*/
|
|
PmuppAdmin::PmuppAdmin() : QObject()
|
|
{
|
|
// XML Parser part
|
|
// 1st: check local directory
|
|
QString path = QString("./");
|
|
QString fln = QString("mupp_startup.xml");
|
|
QString pathFln = path + fln;
|
|
QProcessEnvironment procEnv = QProcessEnvironment::systemEnvironment();
|
|
if (!QFile::exists(pathFln)) {
|
|
// 2nd: check $HOME/.musrfit/mupp/mupp_startup.xml
|
|
path = procEnv.value("HOME", "");
|
|
pathFln = path + "/.musrfit/mupp/" + fln;
|
|
if (!QFile::exists(pathFln)) {
|
|
// 3rd: check $MUSRFITPATH/mupp_startup.xml
|
|
path = procEnv.value("MUSRFITPATH", "");
|
|
pathFln = path + "/" + fln;
|
|
if (!QFile::exists(pathFln)) {
|
|
// 4th: check $ROOTSYS/bin/mupp_startup.xml
|
|
path = procEnv.value("ROOTSYS", "");
|
|
pathFln = path + "/bin/" + fln;
|
|
if (!QFile::exists(pathFln)) {
|
|
// 5th: not found anywhere hence create it
|
|
path = procEnv.value("HOME", "");
|
|
pathFln = path + "/.musrfit/mupp/" + fln;
|
|
createMuppStartupFile();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (QFile::exists(pathFln)) { // administration file present
|
|
PmuppAdminXMLParser handler(this);
|
|
QFile xmlFile(pathFln);
|
|
QXmlInputSource source( &xmlFile );
|
|
QXmlSimpleReader reader;
|
|
reader.setContentHandler( &handler );
|
|
reader.setErrorHandler( &handler );
|
|
if (!reader.parse( source )) {
|
|
QMessageBox::critical(0, "ERROR",
|
|
"Error parsing mupp_startup.xml settings file.\nProbably a few things will not work porperly.\nPlease fix this first.",
|
|
QMessageBox::Ok, QMessageBox::NoButton);
|
|
return;
|
|
}
|
|
} else {
|
|
QMessageBox::critical(0, "ERROR",
|
|
"Couldn't find the mupp_startup.xml settings file.\nProbably a few things will not work porperly.\nPlease fix this first.",
|
|
QMessageBox::Ok, QMessageBox::NoButton);
|
|
return;
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Destructor
|
|
*/
|
|
PmuppAdmin::~PmuppAdmin()
|
|
{
|
|
saveRecentFiles();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Add recent path-file name to the internal ring-buffer.
|
|
*
|
|
* \param str recent path-file name to be added
|
|
*/
|
|
void PmuppAdmin::addRecentFile(const QString str)
|
|
{
|
|
// check if file name is not already present
|
|
for (int i=0; i<fRecentFile.size(); i++) {
|
|
if (str == fRecentFile[i])
|
|
return;
|
|
}
|
|
|
|
fRecentFile.push_front(str);
|
|
if (fRecentFile.size() > MAX_RECENT_FILES)
|
|
fRecentFile.resize(MAX_RECENT_FILES);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::getRecentFile
|
|
* @param idx
|
|
* @return
|
|
*/
|
|
QString PmuppAdmin::getRecentFile(int idx)
|
|
{
|
|
QString str("");
|
|
|
|
if ((idx >= 0) && (idx < fRecentFile.size()))
|
|
str = fRecentFile[idx];
|
|
|
|
return str;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::getMarker
|
|
* @param idx
|
|
* @return
|
|
*/
|
|
PmuppMarker PmuppAdmin::getMarker(int idx) {
|
|
PmuppMarker marker;
|
|
|
|
if (idx >= fMarker.size())
|
|
return marker;
|
|
|
|
return fMarker[idx];
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::getColor
|
|
* @param name
|
|
* @param r
|
|
* @param g
|
|
* @param b
|
|
*/
|
|
void PmuppAdmin::getColor(QString name, int &r, int &g, int &b)
|
|
{
|
|
int idx=-1;
|
|
for (int i=0; i<fColor.size(); i++) {
|
|
if (fColor[i].getName() == name) {
|
|
idx = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (idx == -1) {
|
|
r = -1;
|
|
g = -1;
|
|
b = -1;
|
|
} else {
|
|
fColor[idx].getRGB(r, g, b);
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::getColor
|
|
* @param idx
|
|
* @param r
|
|
* @param g
|
|
* @param b
|
|
*/
|
|
void PmuppAdmin::getColor(int idx, int &r, int &g, int &b)
|
|
{
|
|
if ((idx<=0) || (idx>fColor.size())) {
|
|
r = -1;
|
|
g = -1;
|
|
b = -1;
|
|
} else {
|
|
fColor[idx].getRGB(r, g, b);
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::setMaker
|
|
* @param marker
|
|
* @param size
|
|
*/
|
|
void PmuppAdmin::setMarker(int marker, double size)
|
|
{
|
|
PmuppMarker markerObj;
|
|
|
|
// make sure marker is in proper range
|
|
if ((marker<1) || (marker>49)) {
|
|
QMessageBox::warning(0, "WARNING", QString("Found Marker (%1) not in the expected range.\nWill ignore it.").arg(marker));
|
|
return;
|
|
}
|
|
markerObj.setMarker(marker);
|
|
markerObj.setMarkerSize(size);
|
|
|
|
fMarker.push_back(markerObj);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::setColor
|
|
* @param r
|
|
* @param g
|
|
* @param b
|
|
* @param name
|
|
*/
|
|
void PmuppAdmin::setColor(int r, int g, int b, QString name)
|
|
{
|
|
if (((r<0) || (r>255)) ||
|
|
((g<0) || (g>255)) ||
|
|
((b<0) || (b>255))) {
|
|
QMessageBox::warning(0, "WARNING", QString("Found Color (%1,%2,%3) not in the expected range.\nWill ignore it.").arg(r).arg(g).arg(b));
|
|
return;
|
|
}
|
|
|
|
PmuppColor color;
|
|
color.setName(name);
|
|
color.setRGB(r,g,b);
|
|
|
|
fColor.push_back(color);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* <p>Merges the recent file ring buffer into mupp_startup.xml and saves it.
|
|
* If a local copy is present it will be saved there, otherwise the master file
|
|
* will be used.
|
|
*/
|
|
void PmuppAdmin::saveRecentFiles()
|
|
{
|
|
// check if mupp_startup.xml is present in the current directory, and if yes, use this file to
|
|
// save the recent file names otherwise use the "master" mupp_startup.xml
|
|
|
|
QString str("");
|
|
QString fln = QString("./mupp_startup.xml");
|
|
if (!QFile::exists(fln)) {
|
|
QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
|
|
fln = QString("%1/.musrfit/mupp/mupp_startup.xml").arg(env.value("HOME"));
|
|
}
|
|
|
|
if (QFile::exists(fln)) { // administration file present
|
|
QVector<QString> data;
|
|
QFile file(fln);
|
|
if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
|
cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** Cannot open " << fln.toLatin1().data() << " for reading." << endl;
|
|
return;
|
|
}
|
|
QTextStream fin(&file);
|
|
while (!fin.atEnd()) {
|
|
data.push_back(fin.readLine());
|
|
}
|
|
file.close();
|
|
|
|
// remove <path_file_name> from data
|
|
for (QVector<QString>::iterator it = data.begin(); it != data.end(); ++it) {
|
|
if (it->contains("<path_file_name>")) {
|
|
it = data.erase(it);
|
|
--it;
|
|
}
|
|
}
|
|
|
|
// add recent files
|
|
int i;
|
|
for (i=0; i<data.size(); i++) {
|
|
if (data[i].contains("<recent_files>"))
|
|
break;
|
|
}
|
|
|
|
if (i == data.size()) {
|
|
cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** " << fln.toLatin1().data() << " seems to be corrupt." << endl;
|
|
return;
|
|
}
|
|
i++;
|
|
for (int j=0; j<fRecentFile.size(); j++) {
|
|
str = " <path_file_name>" + fRecentFile[j] + "</path_file_name>";
|
|
data.insert(i++, str);
|
|
}
|
|
|
|
if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
|
|
cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** Cannot open " << fln.toLatin1().data() << " for reading." << endl;
|
|
return;
|
|
}
|
|
fin.setDevice(&file);
|
|
for (int i=0; i<data.size(); i++)
|
|
fin << data[i] << endl;
|
|
file.close();
|
|
} else {
|
|
QString msg("Failed to write mupp_startup.xml. Neither a local nor a global copy found.");
|
|
QMessageBox::warning(0, "WARNING", msg, QMessageBox::Ok, QMessageBox::NoButton);
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
/**
|
|
* @brief PmuppAdmin::createMuppStartupFile
|
|
*/
|
|
void PmuppAdmin::createMuppStartupFile()
|
|
{
|
|
// get $HOME
|
|
QProcessEnvironment procEnv = QProcessEnvironment::systemEnvironment();
|
|
QString pathName = procEnv.value("HOME", "");
|
|
pathName += "/.musrfit/mupp";
|
|
|
|
// check if the directory $HOME/.musrfit/mupp exists if not create it
|
|
QDir dir(pathName);
|
|
if (!dir.exists()) {
|
|
// directory $HOME/.musrfit/mupp does not exist hence create it
|
|
dir.mkpath(pathName);
|
|
}
|
|
|
|
// create default mupp_startup.xml file in $HOME/.musrfit/mupp
|
|
pathName += "/mupp_startup.xml";
|
|
|
|
// get the default mupp_startup.xml.in from the internal resources
|
|
QFile fres(":/mupp_startup.xml.in");
|
|
if (!fres.exists()) {
|
|
QString msg = QString("Neither couldn't find nor create mupp_startup.xml. Things are likely not to work.");
|
|
QMessageBox::critical(0, "ERROR", msg);
|
|
return;
|
|
}
|
|
|
|
if (!fres.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
|
QString msg = QString("Couldn't open internal resource file mupp_startup.xml.in. Things are likely not to work.");
|
|
QMessageBox::critical(0, "ERROR", msg);
|
|
return;
|
|
}
|
|
// text stream for fres
|
|
QTextStream fin(&fres);
|
|
|
|
// mupp_startup.xml default file
|
|
QFile file(pathName);
|
|
|
|
if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
|
|
return;
|
|
|
|
// text stream for file
|
|
QTextStream fout(&file);
|
|
|
|
QString line;
|
|
while (!fin.atEnd()) {
|
|
line = fin.readLine();
|
|
fout << line << endl;
|
|
}
|
|
|
|
file.close();
|
|
fres.close();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
// END
|
|
//--------------------------------------------------------------------------
|