src/core/raster/qgsrasterprojector.h

Go to the documentation of this file.

/***************************************************************************
    qgsrasterprojector.h - Raster projector
     --------------------------------------
    Date                 : Jan 16, 2011
    Copyright            : (C) 2005 by Radim Blazek
    email                : radim dot blazek 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.                                   *
 *                                                                         *
 ***************************************************************************/

/* This code takes ideas from WarpBuilder in Geotools.
 * Thank to Ing. Andrea Aime, Ing. Simone Giannecchini and GeoSolutions S.A.S.
 * See : http://geo-solutions.blogspot.com/2011/01/developers-corner-improving.html
 */

#ifndef QGSRASTERPROJECTOR_H
#define QGSRASTERPROJECTOR_H

#include <QVector>
#include <QList>

#include "qgsrectangle.h"
#include "qgscoordinatereferencesystem.h"
#include "qgscoordinatetransform.h"
#include "qgsrasterinterface.h"

#include <cmath>

class QgsPoint;

class CORE_EXPORT QgsRasterProjector : public QgsRasterInterface
{
  public:
    QgsRasterProjector(
      QgsCoordinateReferenceSystem theSrcCRS,
      QgsCoordinateReferenceSystem theDestCRS,
      QgsRectangle theDestExtent,
      int theDestRows, int theDestCols,
      double theMaxSrcXRes, double theMaxSrcYRes,
      QgsRectangle theExtent
    );
    QgsRasterProjector(
      QgsCoordinateReferenceSystem theSrcCRS,
      QgsCoordinateReferenceSystem theDestCRS,
      double theMaxSrcXRes, double theMaxSrcYRes,
      QgsRectangle theExtent
    );
    QgsRasterProjector();
    // To avoid synthesized which fails on copy of QgsCoordinateTransform
    // (QObject child) in Python bindings
    QgsRasterProjector( const QgsRasterProjector &projector );

    ~QgsRasterProjector();

    QgsRasterProjector & operator=( const QgsRasterProjector &projector );

    QgsRasterInterface *clone() const;

    int bandCount() const;

    QGis::DataType dataType( int bandNo ) const;

    void setCRS( const QgsCoordinateReferenceSystem & theSrcCRS, const QgsCoordinateReferenceSystem & theDestCRS );

    QgsCoordinateReferenceSystem srcCrs() const  { return mSrcCRS; }

    QgsCoordinateReferenceSystem destCrs() const  { return mDestCRS; }

    void setMaxSrcRes( double theMaxSrcXRes, double theMaxSrcYRes )
    {
      mMaxSrcXRes = theMaxSrcXRes; mMaxSrcYRes = theMaxSrcYRes;
    }

    QgsRasterBlock *block( int bandNo, const QgsRectangle & extent, int width, int height );

  private:
    QgsRectangle srcExtent() { return mSrcExtent; }

    int srcRows() { return mSrcRows; }
    int srcCols() { return mSrcCols; }
    void setSrcRows( int theRows ) { mSrcRows = theRows; mSrcXRes = mSrcExtent.height() / mSrcRows; }
    void setSrcCols( int theCols ) { mSrcCols = theCols; mSrcYRes = mSrcExtent.width() / mSrcCols; }

    void srcRowCol( int theDestRow, int theDestCol, int *theSrcRow, int *theSrcCol );

    int dstRows() const { return mDestRows; }
    int dstCols() const { return mDestCols; }

    void destPointOnCPMatrix( int theRow, int theCol, double *theX, double *theY );

    int matrixRow( int theDestRow );
    int matrixCol( int theDestCol );

    QgsPoint srcPoint( int theRow, int theCol );

    inline void preciseSrcRowCol( int theDestRow, int theDestCol, int *theSrcRow, int *theSrcCol );

    inline void approximateSrcRowCol( int theDestRow, int theDestCol, int *theSrcRow, int *theSrcCol );

    void calc();

    void insertRows();

    void insertCols();

    /* calculate single control point in current matrix */
    void calcCP( int theRow, int theCol );

    bool calcRow( int theRow );

    bool calcCol( int theCol );

    void calcSrcExtent();

    void calcSrcRowsCols();

    bool checkCols();

    bool checkRows();

    void calcHelper( int theMatrixRow, QgsPoint *thePoints );

    void nextHelper();

    QString cpToString();

    QgsCoordinateReferenceSystem mSrcCRS;

    QgsCoordinateReferenceSystem mDestCRS;

    QgsCoordinateTransform mCoordinateTransform;

    QgsRectangle mDestExtent;

    QgsRectangle mSrcExtent;

    QgsRectangle mExtent;

    int mDestRows;

    int mDestCols;

    double mDestXRes;

    double mDestYRes;

    int mSrcRows;

    int mSrcCols;

    double mSrcXRes;

    double mSrcYRes;

    double mDestRowsPerMatrixRow;

    double mDestColsPerMatrixCol;

    QList< QList<QgsPoint> > mCPMatrix;

    /* Same size as mCPMatrix */
    QList< QList<bool> > mCPLegalMatrix;

    /* Warning: using QList is slow on access */
    QgsPoint *pHelperTop;

    /* Warning: using QList is slow on access */
    QgsPoint *pHelperBottom;

    int mHelperTopRow;

    int mCPCols;
    int mCPRows;

    double mSqrTolerance;

    double mMaxSrcXRes;
    double mMaxSrcYRes;

    bool mApproximate;
};

#endif