qgspointcloudrgbrenderer.h

Go to the documentation of this file.

/***************************************************************************
                         qgspointcloudrgbrenderer.h
                         --------------------
    begin                : October 2020
    copyright            : (C) 2020 by Nyall Dawson
    email                : nyall dot dawson at gmail dot com
 ***************************************************************************/
 
/***************************************************************************
 *                                                                         *
 *   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 QGSPOINTCLOUDRGBRENDERER_H
#define QGSPOINTCLOUDRGBRENDERER_H
 
#include "qgis_core.h"
#include "qgis_sip.h"
#include "qgscontrastenhancement.h"
#include "qgspointcloudrenderer.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;
 
#ifndef SIP_RUN

class CORE_EXPORT QgsPointCloudRgbRendererPreparedData : public QgsPreparedPointCloudRendererData
{
  public:
    QSet< QString > usedAttributes() const override;
    bool prepareBlock( const QgsPointCloudBlock *block ) override;
    QColor pointColor( const QgsPointCloudBlock *block, int i, double z ) override;
 
    QString redAttribute = u"Red"_s;
    QString greenAttribute = u"Green"_s;
    QString blueAttribute = u"Blue"_s;
    std::unique_ptr< QgsContrastEnhancement > redContrastEnhancement;
    std::unique_ptr< QgsContrastEnhancement > greenContrastEnhancement;
    std::unique_ptr< QgsContrastEnhancement > blueContrastEnhancement;
 
    int redOffset = 0;
    QgsPointCloudAttribute::DataType redType;
    bool useRedContrastEnhancement = false;
    int greenOffset = 0;
    QgsPointCloudAttribute::DataType greenType;
    bool useGreenContrastEnhancement = false;
    int blueOffset = 0;
    QgsPointCloudAttribute::DataType blueType;
    bool useBlueContrastEnhancement = false;
};
#endif

class CORE_EXPORT QgsPointCloudRgbRenderer : public QgsPointCloudRenderer
{
  public:
    QgsPointCloudRgbRenderer();
 
    QString type() const override;
    QgsPointCloudRenderer *clone() const override;
    void renderBlock( const QgsPointCloudBlock *block, QgsPointCloudRenderContext &context ) override;
    QDomElement save( QDomDocument &doc, const QgsReadWriteContext &context ) const override;
    QSet< QString > usedAttributes( const QgsPointCloudRenderContext &context ) const override;
    std::unique_ptr< QgsPreparedPointCloudRendererData > prepare() override SIP_SKIP;

    static QgsPointCloudRenderer *create( QDomElement &element, const QgsReadWriteContext &context ) SIP_FACTORY;

    QString redAttribute() const;

    void setRedAttribute( const QString &attribute );

    QString greenAttribute() const;

    void setGreenAttribute( const QString &attribute );

    QString blueAttribute() const;

    void setBlueAttribute( const QString &attribute );

    const QgsContrastEnhancement *redContrastEnhancement() const;

    void setRedContrastEnhancement( QgsContrastEnhancement *enhancement SIP_TRANSFER );

    const QgsContrastEnhancement *greenContrastEnhancement() const;

    void setGreenContrastEnhancement( QgsContrastEnhancement *enhancement SIP_TRANSFER );

    const QgsContrastEnhancement *blueContrastEnhancement() const;

    void setBlueContrastEnhancement( QgsContrastEnhancement *enhancement SIP_TRANSFER );
 
  private:
    QString mRedAttribute = u"Red"_s;
    QString mGreenAttribute = u"Green"_s;
    QString mBlueAttribute = u"Blue"_s;
 
    std::unique_ptr< QgsContrastEnhancement > mRedContrastEnhancement;
    std::unique_ptr< QgsContrastEnhancement > mGreenContrastEnhancement;
    std::unique_ptr< QgsContrastEnhancement > mBlueContrastEnhancement;
};
 
#endif // QGSPOINTCLOUDRGBRENDERER_H