qgsrastercalculator.h

Go to the documentation of this file.

/***************************************************************************
                          qgsrastercalculator.h  -  description
                          ---------------------
    begin                : September 28th, 2010
    copyright            : (C) 2010 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 QGSRASTERCALCULATOR_H
#define QGSRASTERCALCULATOR_H

#include "qgsfield.h"
#include "qgsrectangle.h"
#include <QString>
#include <QVector>
#include "gdal.h"

class QgsRasterLayer;
class QProgressDialog;


struct ANALYSIS_EXPORT QgsRasterCalculatorEntry
{
  QString ref; //name
  QgsRasterLayer* raster; //pointer to rasterlayer
  int bandNumber; //raster band number
};

class ANALYSIS_EXPORT QgsRasterCalculator
{
  public:
    QgsRasterCalculator( const QString& formulaString, const QString& outputFile, const QString& outputFormat,
                         const QgsRectangle& outputExtent, int nOutputColumns, int nOutputRows, const QVector<QgsRasterCalculatorEntry>& rasterEntries );
    ~QgsRasterCalculator();

    int processCalculation( QProgressDialog* p = 0 );

  private:
    //default constructor forbidden. We need formula, output file, output format and output raster resolution obligatory
    QgsRasterCalculator();

    GDALDriverH openOutputDriver();

    GDALDatasetH openOutputFile( GDALDriverH outputDriver );

    void readRasterPart( double* targetGeotransform,
                         int xOffset, int yOffset,
                         int nCols, int nRows,
                         double* sourceTransform,
                         GDALRasterBandH sourceBand,
                         float* rasterBuffer );

    bool transformationsEqual( double* t1, double* t2 ) const;

    void outputGeoTransform( double* transform ) const;

    QString mFormulaString;
    QString mOutputFile;
    QString mOutputFormat;

    QgsRectangle mOutputRectangle;
    int mNumOutputColumns;
    int mNumOutputRows;

    /***/
    QVector<QgsRasterCalculatorEntry> mRasterEntries;
};

#endif // QGSRASTERCALCULATOR_H