/*
 * The JTS Topology Suite is a collection of Java classes that
 * implement the fundamental operations required to validate a given
 * geo-spatial data set to a known topological specification.
 *
 * Copyright (C) 2001 Vivid Solutions
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * For more information, contact:
 *
 *     Vivid Solutions
 *     Suite #1A
 *     2328 Government Street
 *     Victoria BC  V8T 5G5
 *     Canada
 *
 *     (250)385-6040
 *     www.vividsolutions.com
 */
package com.revolsys.geometry.test.old.operation;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;

import com.revolsys.geometry.graph.linemerge.LineMerger;
import com.revolsys.geometry.model.Geometry;
import com.revolsys.geometry.model.GeometryFactory;
import com.revolsys.geometry.model.LineString;

import junit.framework.TestCase;

/**
 * @version 1.7
 */
public class LineMergerTest extends TestCase {
  private static final GeometryFactory geometryFactory = GeometryFactory.DEFAULT_3D;

  public static void compare(final Collection expectedGeometries, final Collection actualGeometries,
    final boolean compareDirections) {
    assertEquals("Geometry count, " + actualGeometries, expectedGeometries.size(),
      actualGeometries.size());
    for (final Iterator i = expectedGeometries.iterator(); i.hasNext();) {
      final Geometry expectedGeometry = (Geometry)i.next();
      assertTrue("Not found: " + expectedGeometry + ", " + actualGeometries,
        contains(actualGeometries, expectedGeometry, compareDirections));
    }
  }

  private static boolean contains(final Collection geometries, final Geometry g,
    final boolean exact) {
    for (final Iterator i = geometries.iterator(); i.hasNext();) {
      final Geometry element = (Geometry)i.next();
      if (exact && element.equals(2, g)) {
        return true;
      }
      if (!exact && element.equalsTopo(g)) {
        return true;
      }
    }

    return false;
  }

  public static void doTest(final String[] inputWKT, final String[] expectedOutputWKT,
    final boolean compareDirections) {
    final Collection<Geometry> inputGeometries = toGeometries(inputWKT);
    final List<LineString> outputLines = LineMerger.merge(inputGeometries);
    compare(toGeometries(expectedOutputWKT), outputLines, compareDirections);
  }

  public static void main(final String[] args) {
    junit.textui.TestRunner.run(LineMergerTest.class);
  }

  public static Collection<Geometry> toGeometries(final String[] inputWKT) {
    final List<Geometry> geometries = new ArrayList<>();
    for (final String element : inputWKT) {
      geometries.add(geometryFactory.geometry(element));
    }

    return geometries;
  }

  public LineMergerTest(final String name) {
    super(name);
  }

  private void doTest(final String[] inputWKT, final String[] expectedOutputWKT) {
    doTest(inputWKT, expectedOutputWKT, true);
  }

  public void test1() {
    doTest(new String[] {
      "LINESTRING (120 120, 180 140)", "LINESTRING (200 180, 180 140)",
      "LINESTRING (200 180, 240 180)"
    }, new String[] {
      "LINESTRING (120 120, 180 140, 200 180, 240 180)"
    });
  }

  public void test2() {
    doTest(new String[] {
      "LINESTRING (120 300, 80 340)", "LINESTRING (120 300, 140 320, 160 320)",
      "LINESTRING (40 320, 20 340, 0 320)", "LINESTRING (0 320, 20 300, 40 320)",
      "LINESTRING (40 320, 60 320, 80 340)", "LINESTRING (160 320, 180 340, 200 320)",
      "LINESTRING (200 320, 180 300, 160 320)"
    }, new String[] {
      "LINESTRING (160 320, 180 340, 200 320, 180 300, 160 320)",
      "LINESTRING (40 320, 20 340, 0 320, 20 300, 40 320)",
      "LINESTRING (40 320, 60 320, 80 340, 120 300, 140 320, 160 320)"
    });
  }

  public void test3() {
    doTest(new String[] {
      "LINESTRING (0 0, 100 100)", "LINESTRING (0 100, 100 0)"
    }, new String[] {
      "LINESTRING (0 0, 100 100)", "LINESTRING (0 100, 100 0)"
    });
  }

  public void test4() {
    doTest(new String[] {
      "LINESTRING EMPTY", "LINESTRING EMPTY"
    }, new String[] {});
  }

  public void test5() {
    doTest(new String[] {}, new String[] {});
  }

  public void testSingleUniquePoint() {
    doTest(new String[] {
      "LINESTRING (10642 31441, 10642 31441)", "LINESTRING EMPTY"
    }, new String[] {});
  }
}