File indexing completed on 2023-05-30 09:03:12

 /*
     SPDX-License-Identifier: GPL-2.0-or-later
     SPDX-FileCopyrightText: 2012 Filipe Saraiva <filipe@kde.org>
 */
 
 #include "pythonexpression.h"
 
 #include <config-cantorlib.h>
 
 #include "textresult.h"
 #include "imageresult.h"
 #include "helpresult.h"
 #include "session.h"
 #include "settings.h"
 
 #include <QDebug>
 
 #include "pythonsession.h"
 
 PythonExpression::PythonExpression(Cantor::Session* session, bool internal) : Cantor::Expression(session, internal)
 {
 }
 
 PythonExpression::~PythonExpression() {
 }
 
 void PythonExpression::evaluate()
 {
     session()->enqueueExpression(this);
 }
 
 QString PythonExpression::internalCommand()
 {
     QString cmd = command();
 
     if (PythonSettings::integratePlots())
     {
         auto* pySession = static_cast<PythonSession*>(session());
         const QString& filepath = pySession->plotFilePrefixPath() + QString::number(pySession->plotFileCounter()) + QLatin1String(".png");
 
         for(const auto& package : session()->enabledGraphicPackages())
         {
             if (package.isHavePlotCommand())
             {
                 cmd.append(QLatin1String("\n"));
                 cmd.append(package.savePlotCommand(filepath, pySession->plotFileCounter()));
             }
         }
     }
 
     QStringList commandLine = cmd.split(QLatin1String("\n"));
     QString commandProcessing;
 
     for(const QString& command : commandLine){
         const QString firstLineWord = command.trimmed().replace(QLatin1String("("), QLatin1String(" "))
             .split(QLatin1String(" ")).at(0);
 
         // Ignore comments
         if (firstLineWord.length() != 0 && firstLineWord[0] == QLatin1Char('#')){
 
             commandProcessing += command + QLatin1String("\n");
             continue;
         }
 
         if(firstLineWord.contains(QLatin1String("execfile"))){
 
             commandProcessing += command;
             continue;
         }
 
         commandProcessing += command + QLatin1String("\n");
 
     }
 
     return commandProcessing;
 }
 
 void PythonExpression::parseOutput(const QString& output)
 {
     qDebug() << "expression output: " << output;
     if(command().simplified().startsWith(QLatin1String("help(")))
     {
         QString resultStr = output;
         setResult(new Cantor::HelpResult(resultStr.remove(output.lastIndexOf(QLatin1String("None")), 4)));
     }
     else if (!output.isEmpty())
     {
         auto* pySession = static_cast<PythonSession*>(session());
         const QString& plotFilePrefixPath = pySession->plotFilePrefixPath();
         const QString& searchPrefixPath = QLatin1String("INNER PLOT INFO CANTOR: ") + plotFilePrefixPath;
 
         QStringList buffer;
         const QStringList& lines = output.split(QLatin1String("\n"));
         for (const QString& line : lines)
         {
             if (line.startsWith(searchPrefixPath))
             {
                 if (!buffer.isEmpty() && !(buffer.size() == 1 && buffer[0].isEmpty()))
                     addResult(new Cantor::TextResult(buffer.join(QLatin1String("\n"))));
 
                 QString filepath = plotFilePrefixPath + QString::number(pySession->plotFileCounter()) + QLatin1String(".png");
                 pySession->plotFileCounter()++;
                 addResult(new Cantor::ImageResult(QUrl::fromLocalFile(filepath)));
                 buffer.clear();
             }
             else
                 buffer.append(line);
         }
         if (!buffer.isEmpty() && !(buffer.size() == 1 && buffer[0].isEmpty()))
             addResult(new Cantor::TextResult(buffer.join(QLatin1String("\n"))));
     }
 
     setStatus(Cantor::Expression::Done);
 }
 
 void PythonExpression::parseError(const QString& error)
 {
     qDebug() << "expression error: " << error;
     setErrorMessage(error);
 
     setStatus(Cantor::Expression::Error);
 }
 
 void PythonExpression::parseWarning(const QString& warning)
 {
     if (!warning.isEmpty())
     {
         auto* result = new Cantor::TextResult(warning);
         result->setStdErr(true);
         addResult(result);
     }
 }