pythonEngineWorker.h

Go to the documentation of this file.

/* Copyright 2018 Iakov Kirilenko, CyberTech Labs Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License. */

#pragma once

#include <QtCore/QString>
#include <QtCore/QThread>
#include <QMutex>

#include <trikControl/brickInterface.h>
#include <trikNetwork/mailboxInterface.h>

#include "PythonQt_QtAll.h"
#include "PyTrikControl0.h"

void PythonQt_init_PyTrikControl(PyObject* module);

namespace trikScriptRunner
{

class PythonEngineWorker : public QObject
{
    Q_OBJECT

public:
    PythonEngineWorker(trikControl::BrickInterface &brick
            , trikNetwork::MailboxInterface * mailbox
            );

    ~PythonEngineWorker();

    void resetBrick();

    void stopScript();

    QStringList knownNames() const;

signals:
    void completed(const QString &error, int scriptId);

    void startedScript(const QString &fileName, int scriptId);

    void startedDirectScript(int scriptId);


    void inited();

    void sendMessage(const QString&);

public slots:
    void run(const QString &script);

    void runDirect(const QString &command);

    void init();

    bool recreateContext();

    void brickBeep();

    void sendStdOutMessage(const QString &text);

private slots:
    void onScriptRequestingToQuit();

    bool initTrik();

    void doRun(const QString &script);

    void doRunDirect(const QString &command);

    void updateErrorMessage(const QString &err);

private:
    enum State {
        ready
        , starting
        , stopping
        , running
    };

    bool evalSystemPy();

    void startScriptEvaluation(int scriptId);

    trikControl::BrickInterface &mBrick;
    trikNetwork::MailboxInterface * const mMailbox;  // Does not have ownership.

    State mState = State::ready;

    QMutex mScriptStateMutex;

    PythonQtObjectPtr mMainContext;

    PyThreadState * mPyInterpreter { nullptr };

    QString mErrorMessage;

    wchar_t *mProgramName { nullptr };
    wchar_t *mPythonPath { nullptr };

    static QAtomicInt initCounter;
};

}