tenmon/scriptengine.h

#ifndef SCRIPTENGINE_H
#define SCRIPTENGINE_H

#include <QObject>
#include <QJSEngine>
#include <QFileInfo>
#include <QThread>
#include <QThreadPool>
#include <QSemaphore>
#include "database.h"
#include "imageinfo.h"

class BatchProcessing;
class Solver;

namespace Script
{

class File;

class ScriptEngine : public QObject
{
    Q_OBJECT
    QJSEngine *_jsEngine;
    Database *_database;
    BatchProcessing *_parent;
    QThreadPool *_pool;
    QSemaphore _semaphore;
    QString _scriptPath;
    QString _outputDir;
    QList<QPair<QString, QString>> _paths;
    Solver *_solver;
public:
    explicit ScriptEngine(BatchProcessing *parent = nullptr);
    void setParams(const QString &scriptPath, const QList<QPair<QString, QString>> &paths, const QString &outputDir);
    void reportError(const QString &message);
    const QString& outputDir() const;
    void interrupt();
    void logError(const QString &message);
    Q_INVOKABLE void log(const QString &message);
    Q_INVOKABLE void mark(File *file);
    Q_INVOKABLE void unmark(File *file);
    Q_INVOKABLE bool isMarked(const File *file) const;
    Q_INVOKABLE void setMaxThread(int maxthread);
    Q_INVOKABLE void sync();
    Q_INVOKABLE QJSValue getString(const QString &label = QString(), const QString &text = QString()) const;
    Q_INVOKABLE QJSValue getInt(const QString &label = QString(), int value = 0);
    Q_INVOKABLE QJSValue getFloat(const QString &label = QString(), double value = 0, int decimals = 3) const;
    Q_INVOKABLE QJSValue getItem(const QStringList &items, const QString &label = "", int current = 0) const;
    Q_INVOKABLE void setStartingSolution(const QJSValue &solution = QJSValue());
    bool convert(File *file, QString &outpath, const QString &format, const QVariantMap &params, bool async);
    QJSValue solveImage(File *file, bool updateHeader);
    QJSValue extractStars(File *file, bool hfr);
    QJSValue newObject();
    QJSValue newArray(uint size);
public slots:
    void run();
signals:
    void newMessage(const QString &message, bool error);
    void finished();
};

class ScriptEngineThread : public QObject
{
    Q_OBJECT
    QThread *_thread;
    ScriptEngine *_engine;
public:
    ScriptEngineThread(BatchProcessing *parent = nullptr);
    ~ScriptEngineThread();
    void setParams(const QString &scriptPath, const QList<QPair<QString, QString>> &paths, const QString &outputDir);
    void start();
    void interrupt();
signals:
    void newMessage(const QString &message, bool error);
    void finished();
};

class FITSRecordModify;

class File : public QObject
{
    Q_OBJECT
    ScriptEngine *_engine = nullptr;
    QString _path;
    QString _root;
    QFileInfo _info;
    bool _fitsKeywordsLoaded = false;
    QStringList _fitsKeywords;
    QMultiHash<QString, FITSRecord> _fitsRecords;
    void loadFitsKeywords();
    bool mkpath(const QString &path) const;
    QJSValue _stats;
    QJSValue _solution;
    QJSValue _stars;
public:
    explicit File(const QString &path, ScriptEngine *engine);
    explicit File(const QString &path, const QString &root, ScriptEngine *engine);
    Q_INVOKABLE QString fileName() const;
    Q_INVOKABLE QString absoluteFilePath() const;
    Q_INVOKABLE QString absolutePath() const;
    Q_INVOKABLE QString relativeFilePath() const;
    Q_INVOKABLE QString relativePath() const;
    Q_INVOKABLE QString baseName() const;
    Q_INVOKABLE QString completeBaseName() const;
    Q_INVOKABLE QString suffix() const;
    Q_INVOKABLE qint64 size() const;
    Q_INVOKABLE QStringList fitsKeywords();
    Q_INVOKABLE QString fitsValue(const QString &key);
    Q_INVOKABLE QJSValue fitsValues(const QString &key);
    Q_INVOKABLE QJSValue fitsRecords();
    Q_INVOKABLE bool modifyFITSRecords(const FITSRecordModify *modify);
    Q_INVOKABLE bool isMarked() const;
    Q_INVOKABLE File* copy(const QString &newpath) const;
    Q_INVOKABLE bool move(const QString &newpath);
    Q_INVOKABLE File* convert(const QString &outpath, const QString &format, const QVariantMap &params = QVariantMap());
    Q_INVOKABLE File* convertAsync(const QString &outpath, const QString &format, const QVariantMap &params = QVariantMap());
    Q_INVOKABLE QJSValue stats();
    Q_INVOKABLE QJSValue solve(bool updateHeader = false);
    Q_INVOKABLE QJSValue extractStars(bool hfr);
};

class FITSRecordModify : public QObject
{
    Q_OBJECT
    QStringList _remove;
    QVector<FITSRecord> _update;
    QVector<FITSRecord> _add;

    friend class File;
public:
    Q_INVOKABLE FITSRecordModify(){};
    Q_INVOKABLE void removeKeyword(const QString &key);
    Q_INVOKABLE void updateKeyword(const QString &key, const QVariant &value, const QString &comment = QString());
    Q_INVOKABLE void addKeyword(const QString &key, const QVariant &value, const QString &comment = QString());
};

}

#endif // SCRIPTENGINE_H