qgsplotregistry.h

Go to the documentation of this file.

/***************************************************************************
                            qgsplotregistry.h
                            -----------------
    begin                : June 2025
    copyright            : (C) 2025 by Mathieu Pellerin
    email                : mathieu at opengis dot ch
 ***************************************************************************/
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSPLOTREGISTRY_H
#define QGSPLOTREGISTRY_H
 
#include "qgis_core.h"
#include "qgis_sip.h"
#include "qgsapplication.h"
#include "qgsplot.h"
#include "qgsvectorlayerplotdatagatherer.h"
 
class QgsPlotWidget SIP_EXTERNAL;

class CORE_EXPORT QgsPlotAbstractMetadata
{
  public:
    QgsPlotAbstractMetadata( const QString &type, const QString &visibleName )
      : mType( type )
      , mVisibleName( visibleName )
    {}
 
    virtual ~QgsPlotAbstractMetadata() = default;

    QString type() const { return mType; }

    QString visibleName() const { return mVisibleName; }
 
    /*
     * IMPORTANT: While it seems like /Factory/ would be the correct annotations here, that's not
     * the case.
     * As per Phil Thomson's advice on https://www.riverbankcomputing.com/pipermail/pyqt/2017-July/039450.html:
     *
     * "
     * /Factory/ is used when the instance returned is guaranteed to be new to Python.
     * In this case it isn't because it has already been seen when being returned by createPlot()
     * (However for a different sub-class implemented in C++ then it would be the first time it was seen
     * by Python so the /Factory/ on create() would be correct.)
     *
     * You might try using /TransferBack/ on create() instead - that might be the best compromise.
     * "
     */

    virtual QgsPlot *createPlot() = 0 SIP_TRANSFERBACK;

    virtual QgsVectorLayerAbstractPlotDataGatherer *createPlotDataGatherer( QgsPlot *plot = nullptr ) = 0 SIP_TRANSFERBACK;

    virtual QgsPlotWidget *createPlotWidget( QWidget *parent = nullptr ) = 0 SIP_TRANSFERBACK;
 
  private:
    QString mType;
    QString mVisibleName;
};

typedef std::function<QgsPlot *()> QgsPlotCreateFunc SIP_SKIP;
typedef std::function<QgsVectorLayerAbstractPlotDataGatherer *( QgsPlot *plot )> QgsPlotDataGathererCreateFunc SIP_SKIP;
typedef std::function<QgsPlotWidget *( QWidget *parent )> QgsPlotWidgetCreateFunc SIP_SKIP;
 
#ifndef SIP_RUN

class CORE_EXPORT QgsPlotMetadata : public QgsPlotAbstractMetadata
{
  public:
    QgsPlotMetadata(
      const QString &type, const QString &visibleName, const QgsPlotCreateFunc &pfCreate, const QgsPlotDataGathererCreateFunc &pdgfCreate = nullptr, const QgsPlotWidgetCreateFunc &pwfCreate = nullptr
    )
      : QgsPlotAbstractMetadata( type, visibleName )
      , mCreateFunc( pfCreate )
      , mDataGathererCreateFunc( pdgfCreate )
      , mWidgetCreateFunc( pwfCreate )
    {}

    QgsPlotCreateFunc createFunction() const { return mCreateFunc; }

    QgsPlotDataGathererCreateFunc createPlotDataGathererFunction() const { return mDataGathererCreateFunc; }

    QgsPlotWidgetCreateFunc widgetCreateFunction() const { return mWidgetCreateFunc; }

    void setWidgetCreateFunction( QgsPlotWidgetCreateFunc function ) SIP_SKIP { mWidgetCreateFunc = std::move( function ); }
 
    QgsPlot *createPlot() override { return mCreateFunc ? mCreateFunc() : nullptr; }
    QgsVectorLayerAbstractPlotDataGatherer *createPlotDataGatherer( QgsPlot *plot = nullptr ) override { return mDataGathererCreateFunc ? mDataGathererCreateFunc( plot ) : nullptr; }
    QgsPlotWidget *createPlotWidget( QWidget *parent = nullptr ) override { return mWidgetCreateFunc ? mWidgetCreateFunc( parent ) : nullptr; }
 
  protected:
    QgsPlotCreateFunc mCreateFunc = nullptr;
    QgsPlotDataGathererCreateFunc mDataGathererCreateFunc = nullptr;
    QgsPlotWidgetCreateFunc mWidgetCreateFunc = nullptr;
};
 
#endif

class CORE_EXPORT QgsPlotRegistry : public QObject
{
    Q_OBJECT
 
  public:
    QgsPlotRegistry( QObject *parent = nullptr );
    ~QgsPlotRegistry() override;

    bool populate();
 
    QgsPlotRegistry( const QgsPlotRegistry &rh ) = delete;
    QgsPlotRegistry &operator=( const QgsPlotRegistry &rh ) = delete;

    QgsPlotAbstractMetadata *plotMetadata( const QString &type ) const;

    bool addPlotType( QgsPlotAbstractMetadata *metadata SIP_TRANSFER );

    bool removePlotType( const QString &type );
 
    /*
     * IMPORTANT: While it seems like /Factory/ would be the correct annotations here, that's not
     * the case.
     * As per Phil Thomson's advice on https://www.riverbankcomputing.com/pipermail/pyqt/2017-July/039450.html:
     *
     * "
     * /Factory/ is used when the instance returned is guaranteed to be new to Python.
     * In this case it isn't because it has already been seen when being returned by createPlot()
     * (However for a different sub-class implemented in C++ then it would be the first time it was seen
     * by Python so the /Factory/ on create() would be correct.)
     *
     * You might try using /TransferBack/ on create() instead - that might be the best compromise.
     * "
     */

    QgsPlot *createPlot( const QString &type ) const SIP_TRANSFERBACK;

    QMap<QString, QString> plotTypes() const;
 
  signals:

    void plotAdded( const QString &type, const QString &name );

    void plotAboutToBeRemoved( const QString &type );
 
  private:
#ifdef SIP_RUN
    QgsPlotRegistry( const QgsPlotRegistry &rh );
#endif
 
    QMap<QString, QgsPlotAbstractMetadata *> mMetadata;
};
 
#endif //QGSPLOTREGISTRY_H