add linedensity testdata

qgis · Jan 10, 2020 · 4b8177f · 4b8177f
1 parent ce73da6
commit 4b8177f
Show file tree

Hide file tree

Showing 6 changed files with 146 additions and 6 deletions.
diff --git a/tests/src/analysis/testqgsprocessingalgs.cpp b/tests/src/analysis/testqgsprocessingalgs.cpp
@@ -23,6 +23,7 @@
 #include "qgsprocessingcontext.h"
 #include "qgsprocessingmodelalgorithm.h"
 #include "qgsnativealgorithms.h"
+#include "qgsalgorithmlinedensity.h"
 #include "qgsalgorithmimportphotos.h"
 #include "qgsalgorithmtransform.h"
 #include "qgsalgorithmkmeansclustering.h"
@@ -84,6 +85,9 @@ class TestQgsProcessingAlgs: public QObject
     void densifyGeometries_data();
     void densifyGeometries();
 
+    void lineDensity_data();
+    void lineDensity();
+
     void rasterLogicOp_data();
     void rasterLogicOp();
 
@@ -873,6 +877,128 @@ void TestQgsProcessingAlgs::densifyGeometries()
     QVERIFY2( result.geometry().equals( expectedGeometry ), QStringLiteral( "Result: %1, Expected: %2" ).arg( result.geometry().asWkt(), expectedGeometry.asWkt() ).toUtf8().constData() );
 }
 
+void TestQgsProcessingAlgs::lineDensity_data()
+{
+  QTest::addColumn<QString>( "inputDataset" );
+  QTest::addColumn<QString>( "expectedDataset" );
+  QTest::addColumn<double>( "searchRadius" );
+  QTest::addColumn<double>( "pixelSize" );
+  QTest::addColumn<QString>( "weightField" );
+
+  /*
+   * Testcase 1
+   *
+   * WGS84 data with weights
+   * searchRadius = 3
+   * pixelSize = 2
+   */
+  QTest::newRow( "testcase 1" )
+      << "/linedensity.gml"
+      << QStringLiteral("/linedensity_testcase1.tif")
+      << 3.0
+      << 2.0
+      << QStringLiteral("weight");
+
+  /*
+   * Testcase 2
+   *
+   * WGS84 data without weights
+   * searchRadius = 3
+   * pixelSize = 2
+   */
+  QTest::newRow( "testcase_2" )
+      << "/linedensity.gml"
+      << QStringLiteral( "/linedensity_testcase2.tif" )
+      << 3.0
+      << 2.0
+      << QStringLiteral("");
+
+}
+
+void TestQgsProcessingAlgs::lineDensity()
+{
+  QFETCH( QString, inputDataset );
+  QFETCH( QString, expectedDataset );
+  QFETCH( double, searchRadius );
+  QFETCH( double, pixelSize );
+  QFETCH( QString, weightField );
+
+  //prepare input params
+  QgsProject p;
+  std::unique_ptr< QgsProcessingAlgorithm > alg( QgsApplication::processingRegistry()->createAlgorithmById( QStringLiteral( "native:linedensity" ) ) );
+
+  QString myDataPath( TEST_DATA_DIR ); //defined in CmakeLists.txt
+
+  QgsVectorLayer *layer = new QgsVectorLayer( myDataPath + inputDataset + "|layername=linedensity", QStringLiteral( "layer" ), QStringLiteral( "ogr" ) );
+  p.addMapLayer( layer );
+  QVERIFY( layer->isValid() );
+
+  //set project crs and ellipsoid from input layer
+  p.setCrs( layer->crs(), true);
+
+  //set project after layer has been added so that transform context/ellipsoid from crs is also set
+  std::unique_ptr< QgsProcessingContext > context = qgis::make_unique< QgsProcessingContext >();
+  context->setProject( &p );
+
+  QVariantMap parameters;
+
+  parameters.insert( QStringLiteral( "INPUT" ), QVariant::fromValue( layer ) );
+  parameters.insert( QStringLiteral( "WEIGHT" ), weightField );
+  parameters.insert( QStringLiteral( "RADIUS" ), searchRadius );
+  parameters.insert( QStringLiteral( "PIXEL_SIZE" ), pixelSize );
+  parameters.insert( QStringLiteral( "OUTPUT" ), QgsProcessing::TEMPORARY_OUTPUT);
+
+  //prepare expectedRaster
+  std::unique_ptr<QgsRasterLayer> expectedRaster = qgis::make_unique< QgsRasterLayer >( myDataPath + "/control_images/expected_raster_linedensity" + expectedDataset , "expectedDataset", "gdal");
+  std::unique_ptr< QgsRasterInterface > expectedInterface (expectedRaster->dataProvider()->clone());
+  QgsRasterIterator expectedIter( expectedInterface.get() );
+  expectedIter.startRasterRead( 1, expectedRaster->width(), expectedRaster->height(), expectedInterface->extent() );
+
+  //run alg...
+
+  bool ok = false;
+  QgsProcessingFeedback feedback;
+  QVariantMap results;
+
+  results = alg->run( parameters, *context, &feedback, &ok );
+  QVERIFY( ok );
+
+  //...and check results with expected datasets
+  std::unique_ptr<QgsRasterLayer> outputRaster = qgis::make_unique< QgsRasterLayer >( results.value( QStringLiteral( "OUTPUT" ) ).toString(), "output", "gdal");
+  std::unique_ptr< QgsRasterInterface > outputInterface (outputRaster->dataProvider()->clone());
+
+  QCOMPARE( outputRaster->width(), expectedRaster->width() );
+  QCOMPARE( outputRaster->height(), expectedRaster->height() );
+
+  QgsRasterIterator outputIter ( outputInterface.get() );
+  outputIter.startRasterRead( 1, outputRaster->width(), outputRaster->height(), outputInterface->extent() );
+  int outputIterLeft = 0;
+  int outputIterTop = 0;
+  int outputIterCols = 0;
+  int outputIterRows = 0;
+  int expectedIterLeft = 0;
+  int expectedIterTop = 0;
+  int expectedIterCols = 0;
+  int expectedIterRows = 0;
+
+  std::unique_ptr< QgsRasterBlock > outputRasterBlock;
+  std::unique_ptr< QgsRasterBlock > expectedRasterBlock;
+
+  while ( outputIter.readNextRasterPart( 1, outputIterCols, outputIterRows, outputRasterBlock, outputIterLeft, outputIterTop ) &&
+          expectedIter.readNextRasterPart( 1, expectedIterCols, expectedIterRows, expectedRasterBlock, expectedIterLeft, expectedIterTop ))
+  {
+    for ( int row = 0; row < expectedIterRows; row++ )
+    {
+      for ( int column = 0; column < expectedIterCols; column++ )
+      {
+        double expectedValue = expectedRasterBlock->value( row, column );
+        double outputValue = outputRasterBlock->value( row, column );
+        QCOMPARE(outputValue, expectedValue);
+      }
+    }
+  }
+}
+
 void TestQgsProcessingAlgs::rasterLogicOp_data()
 {
   QTest::addColumn<QVector< double >>( "input1" );

diff --git a/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase1.tif b/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase1.tif
diff --git a/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase1.tif.aux.xml b/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase1.tif.aux.xml
@@ -0,0 +1,10 @@
+<PAMDataset>
+  <PAMRasterBand band="1">
+    <Metadata>
+      <MDI key="STATISTICS_MAXIMUM">1.5624662410119e-05</MDI>
+      <MDI key="STATISTICS_MEAN">6.4060804334076e-06</MDI>
+      <MDI key="STATISTICS_MINIMUM">0</MDI>
+      <MDI key="STATISTICS_STDDEV">4.939512050957e-06</MDI>
+    </Metadata>
+  </PAMRasterBand>
+</PAMDataset>
diff --git a/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase2.tif b/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase2.tif
diff --git a/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase2.tif.aux.xml b/tests/testdata/control_images/expected_raster_linedensity/linedensity_testcase2.tif.aux.xml
@@ -0,0 +1,10 @@
+<PAMDataset>
+  <PAMRasterBand band="1">
+    <Metadata>
+      <MDI key="STATISTICS_MAXIMUM">7.9285882748081e-06</MDI>
+      <MDI key="STATISTICS_MEAN">3.3950790140125e-06</MDI>
+      <MDI key="STATISTICS_MINIMUM">0</MDI>
+      <MDI key="STATISTICS_STDDEV">2.4253179081431e-06</MDI>
+    </Metadata>
+  </PAMRasterBand>
+</PAMDataset>
diff --git a/...processing/tests/testdata/linedensity.gml → tests/testdata/linedensity.gml b/...processing/tests/testdata/linedensity.gml → tests/testdata/linedensity.gml
@@ -41,10 +41,4 @@
       <ogr:weight>3.000</ogr:weight>
     </ogr:linedensity>
   </gml:featureMember>
-  <gml:featureMember>
-    <ogr:linedensity fid="6">
-      <ogr:geometryProperty><gml:LineString srsName="EPSG:4326"><gml:coordinates>1.07042253521127,-2.52394366197183 3.33943661971831,-3.81690140845071 6.34366197183099,-2.09295774647887 8.29577464788733,-2.14366197183099 9.32253521126761,-0.736619718309861</gml:coordinates></gml:LineString></ogr:geometryProperty>
-      <ogr:weight>1.000</ogr:weight>
-    </ogr:linedensity>
-  </gml:featureMember>
 </ogr:FeatureCollection>