File indexing completed on 2024-05-12 11:32:35

 /*
     SPDX-License-Identifier: GPL-2.0-or-later
     SPDX-FileCopyrightText: 2018 Nikita Sirgienko <warquark@gmail.com>
 */
 
 #include "rvariablemodel.h"
 #include "rkeywords.h"
 #include "rsession.h"
 
 #include <result.h>
 
 using namespace Cantor;
 
 RVariableModel::RVariableModel(RSession* session) : DefaultVariableModel(session)
 {
 }
 
 RVariableModel::~RVariableModel()
 {
     if (m_expression)
         m_expression->setFinishingBehavior(Expression::FinishingBehavior::DeleteOnFinish);
 }
 
 void RVariableModel::update()
 {
     if (m_expression)
         return;
 
     m_expression = session()->evaluateExpression(QLatin1String("%model update"), Expression::FinishingBehavior::DoNotDelete, true);
     connect(m_expression, &Expression::statusChanged, this, &RVariableModel::parseResult);
 }
 
 void RVariableModel::parseResult(Cantor::Expression::Status status)
 {
     switch(status)
     {
         case Expression::Status::Done:
         {
             if (!m_expression->result())
                 break;
 
             const QChar recordSep(30);
             const QChar unitSep(31);
 
             const QString output = m_expression->result()->data().toString();
 
             const QStringList& names = output.section(unitSep, 0, 0).split(recordSep, QString::SkipEmptyParts);
             const QStringList& values = output.section(unitSep, 1, 1).split(recordSep, QString::SkipEmptyParts);
             QStringList funcs = output.section(unitSep, 2, 2).split(recordSep, QString::SkipEmptyParts);
             const QStringList& constants = output.section(unitSep, 3, 3).split(recordSep, QString::SkipEmptyParts);
 
             QList<Variable> vars;
             if (!values.isEmpty()) // Variables management disabled
                 for (int i = 0; i < names.size(); i++)
                 {
                     if (i < values.size())
                         vars.append(Variable{names.at(i), values.at(i)});
                     else
                         vars.append(Variable{names.at(i), QString()});
                 }
             else
                 for (int i = 0; i < names.size(); i++)
                     vars.append(Variable{names.at(i), QString()});
 
             setVariables(vars);
 
             // Remove primitive function "(" because it not function for user calling (i guess)
             // And the function with name like this make highlighting worse actually
             funcs.removeOne(QLatin1String("("));
 
             // Also removes syntax keywords from functions list, like "function"
             for (const QString& keyword: RKeywords::instance()->keywords())
                 funcs.removeOne(keyword);
 
             setFunctions(funcs);
             setConstants(constants);
             break;
         }
         case Expression::Status::Error:
             qWarning() << "R code for update variable model finishs with error message: " << m_expression->errorMessage();
             break;
 
         case Expression::Status::Interrupted:
             break;
 
         default:
             return;
     }
 
     m_expression->deleteLater();
     m_expression = nullptr;
 }
 
 void RVariableModel::setConstants(QStringList newConstants)
 {
     QStringList addedConstants;
     QStringList removedConstants;
 
     //remove the old variables
     int i = 0;
     while (i < m_constants.size())
     {
         //check if this var is present in the new variables
         bool found = false;
         for (const QString& constant : newConstants)
             if(m_constants[i] == constant)
             {
                 found=true;
                 break;
             }
 
         if(!found)
         {
             removedConstants << m_constants[i];
             m_constants.removeAt(i);
         }
         else
             i++;
     }
 
     for (const QString& constant : newConstants)
     {
         if (!m_constants.contains(constant))
         {
             addedConstants << constant;
             m_constants.append(constant);
         }
     }
 
     emit constantsAdded(addedConstants);
     emit constantsRemoved(removedConstants);
 }