TransferFunctionPlugin.h

#pragma once

#include <ViewPlugin.h>

#include <actions/HorizontalToolbarAction.h>
#include <graphics/Vector2f.h>

#include "SettingsAction.h"
#include "MaterialSettings.h"

#include <QPointer>
#include <QTimer>

using namespace mv::plugin;
using namespace mv::util;
using namespace mv::gui;

class Points;
class TransferFunctionWidget;

namespace mv
{
    namespace gui {
        class DropWidget;
    }
}

class TransferFunctionPlugin : public ViewPlugin
{
    Q_OBJECT

public:
    TransferFunctionPlugin(const PluginFactory* factory);
    ~TransferFunctionPlugin() override;

    void init() override;

    /**
     * Load one (or more datasets in the view)
     * @param datasets Dataset(s) to load
     */
    void loadData(const Datasets& datasets) override;

    /** Get number of points in the position dataset */
    std::uint32_t getNumberOfPoints() const;

public:
    void createSubset(const bool& fromSourceData = false, const QString& name = "");

protected: // Data loading

    /** Invoked when the position points dataset changes */
    void positionDatasetChanged();


public: // Miscellaneous

    /** Get smart pointer to points dataset for point position */
    Dataset<Points>& getPositionDataset();

public:

    /** Get reference to the scatter plot widget */
    TransferFunctionWidget& getTransferFunctionWidget();

    SettingsAction& getSettingsAction() { return *(_settingsAction).get(); }

private:
    void updateVolumeData();
    void updateSelection();

public: // Serialization

    /**
     * Load plugin from variant map
     * @param variantMap Variant map representation of the plugin
     */
    void fromVariantMap(const QVariantMap& variantMap) override;

    /**
     * Save plugin to variant map
     * @return Variant map representation of the plugin
     */
    QVariantMap toVariantMap() const override;

private:
    QPointer<TransferFunctionWidget>    _transferFunctionWidget;    /** The visualization widget */
    mv::gui::DropWidget*                _dropWidget;                /** Widget for dropping datasets */

    Dataset<Points>                     _positionDataset;           /** Smart pointer to points dataset for point position */
    std::vector<mv::Vector2f>           _positions;                 /** Point positions */
    unsigned int                        _numPoints;                 /** Number of point positions */

    QPointer<SettingsAction>            _settingsAction;            /** Group action for all settings */
    QPointer<MaterialSettings>			_materialSettings;          /** Material settings action */
    QPointer<HorizontalToolbarAction>   _primaryToolbarAction;      /** Horizontal toolbar for primary content */

    static const std::int32_t LAZY_UPDATE_INTERVAL = 2;

};

// =============================================================================
// Factory
// =============================================================================

class TransferFunctionPluginFactory : public ViewPluginFactory
{
    Q_INTERFACES(mv::plugin::ViewPluginFactory mv::plugin::PluginFactory)
    Q_OBJECT
    Q_PLUGIN_METADATA(IID   "studio.manivault.TransferFunctionPlugin"
                      FILE  "PluginInfo.json")

public:
    TransferFunctionPluginFactory();

    ViewPlugin* produce() override;

    /**
     * Get plugin trigger actions given \p datasets
     * @param datasets Vector of input datasets
     * @return Vector of plugin trigger actions
     */
    PluginTriggerActions getPluginTriggerActions(const mv::Datasets& datasets) const override;

    /**
     * Get the URL of the GitHub repository
     * @return URL of the GitHub repository (or readme markdown URL if set)
     */
    QUrl getRepositoryUrl() const override;
};