QGIS API Documentation 3.28.0-Firenze (ed3ad0430f)
qgsgridfilewriter.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsgridfilewriter.cpp
3 ---------------------
4 begin : Marco 10, 2008
5 copyright : (C) 2008 by Marco Hugentobler
6 email : marco dot hugentobler at karto dot baug dot ethz dot ch
7 ***************************************************************************/
8
9/***************************************************************************
10 * *
11 * This program is free software; you can redistribute it and/or modify *
12 * it under the terms of the GNU General Public License as published by *
13 * the Free Software Foundation; either version 2 of the License, or *
14 * (at your option) any later version. *
15 * *
16 ***************************************************************************/
17
18#include "qgsgridfilewriter.h"
19#include "qgsinterpolator.h"
20#include "qgsvectorlayer.h"
21#include "qgsfeedback.h"
22#include <QFile>
23#include <QFileInfo>
24
25QgsGridFileWriter::QgsGridFileWriter( QgsInterpolator *i, const QString &outputPath, const QgsRectangle &extent, int nCols, int nRows )
26 : mInterpolator( i )
27 , mOutputFilePath( outputPath )
28 , mInterpolationExtent( extent )
29 , mNumColumns( nCols )
30 , mNumRows( nRows )
31 , mCellSizeX( extent.width() / nCols )
32 , mCellSizeY( extent.height() / nRows )
33{}
34
36{
37 QFile outputFile( mOutputFilePath );
38
39 if ( !outputFile.open( QFile::WriteOnly | QIODevice::Truncate ) )
40 {
41 return 1;
42 }
43
44 if ( !mInterpolator )
45 {
46 outputFile.remove();
47 return 2;
48 }
49
50 QTextStream outStream( &outputFile );
51#if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
52 outStream.setCodec( "UTF-8" );
53#endif
54 outStream.setRealNumberPrecision( 8 );
55 writeHeader( outStream );
56
57 double currentYValue = mInterpolationExtent.yMaximum() - mCellSizeY / 2.0; //calculate value in the center of the cell
58 double currentXValue;
59 double interpolatedValue;
60
61 for ( int i = 0; i < mNumRows; ++i )
62 {
63 currentXValue = mInterpolationExtent.xMinimum() + mCellSizeX / 2.0; //calculate value in the center of the cell
64 for ( int j = 0; j < mNumColumns; ++j )
65 {
66 if ( mInterpolator->interpolatePoint( currentXValue, currentYValue, interpolatedValue, feedback ) == 0 )
67 {
68 outStream << interpolatedValue << ' ';
69 }
70 else
71 {
72 outStream << "-9999 ";
73 }
74 currentXValue += mCellSizeX;
75 }
76
77 outStream << Qt::endl;
78 currentYValue -= mCellSizeY;
79
80 if ( feedback )
81 {
82 if ( feedback->isCanceled() )
83 {
84 outputFile.remove();
85 return 3;
86 }
87 feedback->setProgress( 100.0 * i / static_cast< double >( mNumRows ) );
88 }
89 }
90
91 // create prj file
93 ld = mInterpolator->layerData().at( 0 );
94 QgsFeatureSource *source = ld.source;
95 const QString crs = source->sourceCrs().toWkt();
96 const QFileInfo fi( mOutputFilePath );
97 const QString fileName = fi.absolutePath() + '/' + fi.completeBaseName() + ".prj";
98 QFile prjFile( fileName );
99 if ( !prjFile.open( QFile::WriteOnly | QIODevice::Truncate ) )
100 {
101 return 1;
102 }
103 QTextStream prjStream( &prjFile );
104 prjStream << crs;
105 prjStream << Qt::endl;
106 prjFile.close();
107
108 return 0;
109}
110
111int QgsGridFileWriter::writeHeader( QTextStream &outStream )
112{
113 outStream << "NCOLS " << mNumColumns << Qt::endl;
114 outStream << "NROWS " << mNumRows << Qt::endl;
115 outStream << "XLLCORNER " << mInterpolationExtent.xMinimum() << Qt::endl;
116 outStream << "YLLCORNER " << mInterpolationExtent.yMinimum() << Qt::endl;
117 if ( mCellSizeX == mCellSizeY ) //standard way
118 {
119 outStream << "CELLSIZE " << mCellSizeX << Qt::endl;
120 }
121 else //this is supported by GDAL but probably not by other products
122 {
123 outStream << "DX " << mCellSizeX << Qt::endl;
124 outStream << "DY " << mCellSizeY << Qt::endl;
125 }
126 outStream << "NODATA_VALUE -9999" << Qt::endl;
127 return 0;
128}
QString toWkt(WktVariant variant=WKT1_GDAL, bool multiline=false, int indentationWidth=4) const
Returns a WKT representation of this CRS.
An interface for objects which provide features via a getFeatures method.
virtual QgsCoordinateReferenceSystem sourceCrs() const =0
Returns the coordinate reference system for features in the source.
Base class for feedback objects to be used for cancellation of something running in a worker thread.
Definition: qgsfeedback.h:45
bool isCanceled() const SIP_HOLDGIL
Tells whether the operation has been canceled already.
Definition: qgsfeedback.h:54
void setProgress(double progress)
Sets the current progress for the feedback object.
Definition: qgsfeedback.h:63
QgsGridFileWriter(QgsInterpolator *interpolator, const QString &outputPath, const QgsRectangle &extent, int nCols, int nRows)
Constructor for QgsGridFileWriter, for the specified interpolator.
int writeFile(QgsFeedback *feedback=nullptr)
Writes the grid file.
Interface class for interpolations.
virtual int interpolatePoint(double x, double y, double &result, QgsFeedback *feedback=nullptr)=0
Calculates interpolation value for map coordinates x, y.
QList< LayerData > layerData() const
A rectangle specified with double values.
Definition: qgsrectangle.h:42
double yMaximum() const SIP_HOLDGIL
Returns the y maximum value (top side of rectangle).
Definition: qgsrectangle.h:193
double xMinimum() const SIP_HOLDGIL
Returns the x minimum value (left side of rectangle).
Definition: qgsrectangle.h:188
double yMinimum() const SIP_HOLDGIL
Returns the y minimum value (bottom side of rectangle).
Definition: qgsrectangle.h:198
const QgsCoordinateReferenceSystem & crs
A source together with the information about interpolation attribute / z-coordinate interpolation and...
QgsFeatureSource * source
Feature source.