v0.2.48..v0.2.49 changeset RandomWaySplitterTest.cpp

diff --git a/hoot-core-test/src/test/cpp/hoot/core/visitors/RandomWaySplitterTest.cpp b/hoot-core-test/src/test/cpp/hoot/core/visitors/RandomWaySplitterTest.cpp
new file mode 100644
index 0000000..9f8e520
--- /dev/null
+++ b/hoot-core-test/src/test/cpp/hoot/core/visitors/RandomWaySplitterTest.cpp
@@ -0,0 +1,171 @@
+/*
+ * This file is part of Hootenanny.
+ *
+ * Hootenanny 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * --------------------------------------------------------------------
+ *
+ * The following copyright notices are generated automatically. If you
+ * have a new notice to add, please use the format:
+ * " * @copyright Copyright ..."
+ * This will properly maintain the copyright information. DigitalGlobe
+ * copyrights will be updated automatically.
+ *
+ * @copyright Copyright (C) 2014, 2015, 2016, 2017, 2018, 2019 DigitalGlobe (http://www.digitalglobe.com/)
+ */
+
+// CPP Unit
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+#include <cppunit/TestAssert.h>
+#include <cppunit/TestFixture.h>
+
+// Hoot
+#include <hoot/core/elements/OsmMap.h>
+#include <hoot/core/TestUtils.h>
+#include <hoot/core/io/OsmXmlReader.h>
+#include <hoot/core/io/OsmXmlWriter.h>
+#include <hoot/core/util/Log.h>
+#include <hoot/core/util/MapProjector.h>
+#include <hoot/core/visitors/RandomWaySplitter.h>
+
+using namespace std;
+
+namespace hoot
+{
+
+class RandomWaySplitterTest : public HootTestFixture
+{
+  CPPUNIT_TEST_SUITE(RandomWaySplitterTest);
+  CPPUNIT_TEST(runWaySplitTest);
+  CPPUNIT_TEST(runMultiLineStringSplitTest);
+  CPPUNIT_TEST(runInvalidProbabilityTest);
+  CPPUNIT_TEST(runInvalidMinNodeSpacingTest);
+  CPPUNIT_TEST_SUITE_END();
+
+public:
+
+  RandomWaySplitterTest()
+    : HootTestFixture("test-files/visitors/RandomWaySplitterTest/",
+                      "test-output/visitors/RandomWaySplitterTest/")
+  {
+    setResetType(ResetAll);
+  }
+
+  void runWaySplitTest()
+  {
+    OsmXmlReader reader;
+    OsmMapPtr map(new OsmMap());
+    reader.setDefaultStatus(Status::Unknown1);
+    reader.setUseDataSourceIds(true);
+    reader.read(_inputPath + "RandomWaySplitterTest-in-1.osm", map);
+    const int numNodesBeforeSplitting = map->getNodes().size();
+    LOG_VARD(numNodesBeforeSplitting);
+    const int numWaysBeforeSplitting = map->getWays().size();
+    LOG_VARD(numWaysBeforeSplitting)
+
+    MapProjector::projectToPlanar(map);
+    RandomWaySplitter waySplitVisitor;
+    boost::minstd_rand rng;
+    rng.seed(1);
+    waySplitVisitor.setRng(rng);
+    waySplitVisitor.setWaySplitProbability(0.5);
+    waySplitVisitor.setMinNodeSpacing(1.0);
+    map->visitRw(waySplitVisitor);
+    MapProjector::projectToWgs84(map);
+    const int numNewNodesCreatedBySpliting = map->getNodes().size() - numNodesBeforeSplitting;
+    LOG_VARD(numNewNodesCreatedBySpliting);
+    const int numNewWaysCreatedBySpliting = map->getWays().size() - numWaysBeforeSplitting;
+    LOG_VARD(numNewWaysCreatedBySpliting);
+
+    OsmXmlWriter writer;
+    writer.setIncludeHootInfo(true);
+    const QString outFile = _outputPath + "RandomWaySplitterTest-out-1.osm";
+    writer.write(map, outFile);
+
+    HOOT_FILE_EQUALS(_inputPath + "RandomWaySplitterTest-out-1.osm", outFile);
+  }
+
+  void runMultiLineStringSplitTest()
+  {
+    OsmXmlReader reader;
+    OsmMapPtr map(new OsmMap());
+    reader.setDefaultStatus(Status::Unknown1);
+    reader.setUseDataSourceIds(true);
+    reader.read(_inputPath + "RandomWaySplitterTest-in-2.osm", map);
+    const int numNodesBeforeSplitting = map->getNodes().size();
+    LOG_VARD(numNodesBeforeSplitting);
+    const int numWaysBeforeSplitting = map->getWays().size();
+    LOG_VARD(numWaysBeforeSplitting)
+
+    MapProjector::projectToPlanar(map);
+    RandomWaySplitter waySplitVisitor;
+    boost::minstd_rand rng;
+    rng.seed(1);
+    waySplitVisitor.setRng(rng);
+    waySplitVisitor.setWaySplitProbability(0.5);
+    waySplitVisitor.setMinNodeSpacing(1.0);
+    map->visitRw(waySplitVisitor);
+    MapProjector::projectToWgs84(map);
+    const int numNewNodesCreatedBySplitting = map->getNodes().size() - numNodesBeforeSplitting;
+    LOG_VARD(numNewNodesCreatedBySplitting);
+    const int numNewWaysCreatedBySplitting = map->getWays().size() - numWaysBeforeSplitting;
+    LOG_VARD(numNewWaysCreatedBySplitting);
+
+    OsmXmlWriter writer;
+    writer.setIncludeHootInfo(true);
+    const QString outFile = _outputPath + "RandomWaySplitterTest-out-2.osm";
+    writer.write(map, outFile);
+
+    HOOT_FILE_EQUALS(_inputPath + "RandomWaySplitterTest-out-2.osm", outFile);
+  }
+
+  void runInvalidProbabilityTest()
+  {
+    QString exceptionMsg;
+    try
+    {
+      RandomWaySplitter waySplitVisitor;
+      waySplitVisitor.setWaySplitProbability(-0.1);
+    }
+    catch (const HootException& e)
+    {
+      exceptionMsg = e.what();
+    }
+    CPPUNIT_ASSERT_EQUAL(
+      QString("Invalid split probability: -0.1").toStdString(),
+      exceptionMsg.toStdString());
+  }
+
+  void runInvalidMinNodeSpacingTest()
+  {
+    QString exceptionMsg;
+    try
+    {
+      RandomWaySplitter waySplitVisitor;
+      waySplitVisitor.setMinNodeSpacing(-0.1);
+    }
+    catch (const HootException& e)
+    {
+      exceptionMsg = e.what();
+    }
+    CPPUNIT_ASSERT_EQUAL(
+      QString("Invalid minimum node spacing: -0.1").toStdString(),
+      exceptionMsg.toStdString());
+  }
+};
+
+CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(RandomWaySplitterTest, "quick");
+
+}