qgsrelief.h

Go to the documentation of this file.

/***************************************************************************
                          qgsrelief.h  -  description
                          ---------------------------
    begin                : November 2011
    copyright            : (C) 2011 by Marco Hugentobler
    email                : marco dot hugentobler at sourcepole 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 QGSRELIEF_H
#define QGSRELIEF_H
 
#include <gdal.h>
 
#include "qgis_analysis.h"
#include "qgsogrutils.h"
 
#include <QColor>
#include <QMap>
#include <QPair>
#include <QString>
 
class QgsAspectFilter;
class QgsSlopeFilter;
class QgsHillshadeFilter;
class QgsFeedback;

class ANALYSIS_EXPORT QgsRelief
{
  public:
    struct ReliefColor
    {
        ReliefColor( const QColor &c, double min, double max )
          : color( c )
          , minElevation( min )
          , maxElevation( max )
        {}
 
        QColor color;
        double minElevation;
        double maxElevation;
    };
 
    QgsRelief( const QString &inputFile, const QString &outputFile, const QString &outputFormat );
    ~QgsRelief();
 
    QgsRelief( const QgsRelief &rh ) = delete;
    QgsRelief &operator=( const QgsRelief &rh ) = delete;

    int processRaster( QgsFeedback *feedback = nullptr );
 
    double zFactor() const { return mZFactor; }
    void setZFactor( double factor ) { mZFactor = factor; }
 
    void clearReliefColors();
    void addReliefColorClass( const QgsRelief::ReliefColor &color );
    QList<QgsRelief::ReliefColor> reliefColors() const { return mReliefColors; }
    void setReliefColors( const QList<QgsRelief::ReliefColor> &c ) { mReliefColors = c; }

    QList<QgsRelief::ReliefColor> calculateOptimizedReliefClasses();

    bool exportFrequencyDistributionToCsv( const QString &file );
 
  private:
#ifdef SIP_RUN
    QgsRelief( const QgsRelief &rh );
#endif
 
    QString mInputFile;
    QString mOutputFile;
    QString mOutputFormat;
 
    double mCellSizeX = 0.0;
    double mCellSizeY = 0.0;
    float mInputNodataValue = -1;
    float mOutputNodataValue = -1;
 
    double mZFactor = 1;
 
    std::unique_ptr<QgsSlopeFilter> mSlopeFilter;
    std::unique_ptr<QgsAspectFilter> mAspectFilter;
    std::unique_ptr<QgsHillshadeFilter> mHillshadeFilter285;
    std::unique_ptr<QgsHillshadeFilter> mHillshadeFilter300;
    std::unique_ptr<QgsHillshadeFilter> mHillshadeFilter315;
 
    //relief colors and corresponding elevations
    QList<ReliefColor> mReliefColors;
 
    bool processNineCellWindow( float *x1, float *x2, float *x3, float *x4, float *x5, float *x6, float *x7, float *x8, float *x9, unsigned char *red, unsigned char *green, unsigned char *blue );

    gdal::dataset_unique_ptr openInputFile( int &nCellsX, int &nCellsY );

    GDALDriverH openOutputDriver();

    gdal::dataset_unique_ptr openOutputFile( GDALDatasetH inputDataset, GDALDriverH outputDriver );

    bool getElevationColor( double elevation, int *red, int *green, int *blue ) const;

    void setDefaultReliefColors();

    int frequencyClassForElevation( double elevation, double minElevation, double elevationClassRange );
    void optimiseClassBreaks( QList<int> &breaks, double *frequencies );

    bool calculateRegression( const QList<QPair<int, double>> &input, double &a, double &b );
};
 
#endif // QGSRELIEF_H