package mil.nga.geopackage.test.extension.coverage;

import junit.framework.TestCase;

import org.junit.Test;

import java.util.List;

import mil.nga.geopackage.BoundingBox;
import mil.nga.geopackage.core.srs.SpatialReferenceSystem;
import mil.nga.geopackage.core.srs.SpatialReferenceSystemDao;
import mil.nga.geopackage.extension.coverage.CoverageDataAlgorithm;
import mil.nga.geopackage.extension.coverage.CoverageDataPng;
import mil.nga.geopackage.extension.coverage.CoverageDataResults;
import mil.nga.geopackage.test.ImportCoverageDataGeoPackageTestCase;
import mil.nga.geopackage.tiles.TileBoundingBoxUtils;
import mil.nga.geopackage.tiles.matrix.TileMatrix;
import mil.nga.geopackage.tiles.matrixset.TileMatrixSet;
import mil.nga.geopackage.tiles.matrixset.TileMatrixSetDao;
import mil.nga.sf.proj.Projection;
import mil.nga.sf.proj.ProjectionConstants;
import mil.nga.sf.proj.ProjectionFactory;
import mil.nga.sf.proj.ProjectionTransform;

/**
 * Tiled Gridded Coverage Data Extension PNG Tests from an imported GeoPackage
 *
 * @author osbornb
 */
public class CoverageDataPngImportTest extends
        ImportCoverageDataGeoPackageTestCase {

    private static final boolean PRINT = false;
    private static final boolean allowNulls = false;

    /**
     * Test the coverage data extension with a newly created GeoPackage using the
     * Nearest Neighbor Algorithm
     */
    @Test
    public void testNearestNeighbor() throws Exception {

        CoverageDataPngTestUtils.testCoverageData(geoPackage, null,
                CoverageDataAlgorithm.NEAREST_NEIGHBOR, allowNulls);

    }

    /**
     * Test the coverage data extension with a newly created GeoPackage using the
     * Bilinear Algorithm
     */
    @Test
    public void testBilinear() throws Exception {

        CoverageDataPngTestUtils.testCoverageData(geoPackage, null,
                CoverageDataAlgorithm.BILINEAR, allowNulls);

    }

    /**
     * Test the coverage data extension with a newly created GeoPackage using the
     * Bicubic Algorithm
     */
    @Test
    public void testBicubic() throws Exception {

        CoverageDataPngTestUtils.testCoverageData(geoPackage, null,
                CoverageDataAlgorithm.BICUBIC, allowNulls);

    }

    /**
     * Test a random bounding box using the Nearest Neighbor Algorithm
     */
    @Test
    public void testRandomBoundingBoxNearestNeighbor() throws Exception {

        CoverageDataTestUtils.testRandomBoundingBox(geoPackage,
                CoverageDataAlgorithm.NEAREST_NEIGHBOR, true);

    }

    /**
     * Test a random bounding box using the Bilinear Algorithm
     */
    @Test
    public void testRandomBoundingBoxBilinear() throws Exception {

        CoverageDataTestUtils.testRandomBoundingBox(geoPackage,
                CoverageDataAlgorithm.BILINEAR, true);

    }

    /**
     * Test a random bounding box using the Bicubic Algorithm
     */
    @Test
    public void testRandomBoundingBoxBicubic() throws Exception {

        CoverageDataTestUtils.testRandomBoundingBox(geoPackage,
                CoverageDataAlgorithm.BICUBIC, true);

    }

    /**
     * Test the pixel encoding
     */
    @Test
    public void testPixelEncoding() throws Exception {

        CoverageDataTestUtils.testPixelEncoding(geoPackage, true);

    }

    /**
     * Test a single hard coded location and optional print
     *
     * @throws Exception
     */
    @Test
    public void testLocation() throws Exception {

        double latitude = 61.57941522271581;
        double longitude = -148.96174115565339;

        testLocation(latitude, longitude);
    }

    /**
     * Test 10 random locations and optionally print
     *
     * @throws Exception
     */
    @Test
    public void testRandomLocations() throws Exception {

        BoundingBox projectedBoundingBox = null;

        List<String> coverageDataTables = CoverageDataPng.getTables(geoPackage);
        TileMatrixSetDao dao = geoPackage.getTileMatrixSetDao();

        for (String coverageTable : coverageDataTables) {

            TileMatrixSet tileMatrixSet = dao.queryForId(coverageTable);

            BoundingBox boundingBox = tileMatrixSet.getBoundingBox();
            if (PRINT) {
                System.out.println("Min Latitude: "
                        + boundingBox.getMinLatitude());
                System.out.println("Max Latitude: "
                        + boundingBox.getMaxLatitude());
                System.out.println("Min Longitude: "
                        + boundingBox.getMinLongitude());
                System.out.println("Max Longitude: "
                        + boundingBox.getMaxLongitude());
                System.out.println();
            }
            SpatialReferenceSystemDao srsDao = geoPackage
                    .getSpatialReferenceSystemDao();
            long srsId = tileMatrixSet.getSrsId();
            SpatialReferenceSystem srs = srsDao.queryForId(srsId);
            Projection projection = srs.getProjection();
            Projection requestProjection = ProjectionFactory
                    .getProjection(ProjectionConstants.EPSG_WORLD_GEODETIC_SYSTEM);
            ProjectionTransform coverageToRequest = projection
                    .getTransformation(requestProjection);
            projectedBoundingBox = boundingBox.transform(coverageToRequest);

        }
        if (PRINT) {
            System.out.println("Min Latitude: "
                    + projectedBoundingBox.getMinLatitude());
            System.out.println("Max Latitude: "
                    + projectedBoundingBox.getMaxLatitude());
            System.out.println("Min Longitude: "
                    + projectedBoundingBox.getMinLongitude());
            System.out.println("Max Longitude: "
                    + projectedBoundingBox.getMaxLongitude());
            System.out.println();
        }

        double latDistance = projectedBoundingBox.getMaxLatitude()
                - projectedBoundingBox.getMinLatitude();
        double lonDistance = projectedBoundingBox.getMaxLongitude()
                - projectedBoundingBox.getMinLongitude();

        for (int i = 0; i < 10; i++) {

            // Get a random coordinate
            double latitude = latDistance * .9 * Math.random()
                    + projectedBoundingBox.getMinLatitude()
                    + (.05 * latDistance);
            double longitude = lonDistance * .9 * Math.random()
                    + projectedBoundingBox.getMinLongitude()
                    + (.05 * lonDistance);
            testLocation(latitude, longitude);
            if (PRINT) {
                System.out.println();
            }
        }
    }

    /**
     * Test coverage data requests within the bounds of the tiles and optionally
     * print
     *
     * @throws Exception
     */
    @Test
    public void testBounds() throws Exception {

        long requestEpsg = ProjectionConstants.EPSG_WEB_MERCATOR;

        double widthPixelDistance = 1000;
        double heightPixelDistance = 1000;
        int width = 10;
        int height = 6;
        double minLongitude = -16586000;
        double maxLongitude = minLongitude + (width * widthPixelDistance);
        double minLatitude = 8760000;
        double maxLatitude = minLatitude + (height * heightPixelDistance);

        BoundingBox boundingBox = new BoundingBox(minLongitude, minLatitude,
                maxLongitude, maxLatitude);

        Projection projection = ProjectionFactory.getProjection(requestEpsg);
        Projection printProjection = ProjectionFactory
                .getProjection(ProjectionConstants.EPSG_WORLD_GEODETIC_SYSTEM);
        ProjectionTransform wgs84Transform = projection
                .getTransformation(printProjection);

        if (PRINT) {
            System.out.println();
            System.out.println();
            System.out.println("Bounds Test");
            System.out.println();
            System.out.println("REQUEST");
            System.out.println();
            System.out.println("   Min Lat: " + boundingBox.getMinLatitude());
            System.out.println("   Max Lat: " + boundingBox.getMaxLatitude());
            System.out.println("   Min Lon: " + boundingBox.getMinLongitude());
            System.out.println("   Max Lon: " + boundingBox.getMaxLongitude());
            System.out.println("   Result Width: " + width);
            System.out.println("   Result Height: " + height);

            System.out.println();
            System.out.println();
            System.out.println("WGS84 REQUEST");
            System.out.println();
            BoundingBox wgs84BoundingBox = boundingBox
                    .transform(wgs84Transform);
            System.out.println("   Min Lat: "
                    + wgs84BoundingBox.getMinLatitude());
            System.out.println("   Max Lat: "
                    + wgs84BoundingBox.getMaxLatitude());
            System.out.println("   Min Lon: "
                    + wgs84BoundingBox.getMinLongitude());
            System.out.println("   Max Lon: "
                    + wgs84BoundingBox.getMaxLongitude());

            System.out.println();
            System.out.println();
            System.out.println("WGS84 LOCATIONS");

            for (double lat = maxLatitude - (heightPixelDistance * .5); lat >= minLatitude; lat -= heightPixelDistance) {
                System.out.println();
                for (double lon = minLongitude + (widthPixelDistance * .5); lon <= maxLongitude; lon += widthPixelDistance) {
                    double[] point = wgs84Transform.transform(lon, lat);
                    System.out.print("   (" + point[1] + "," + point[0] + ")");
                }
            }
        }

        for (CoverageDataAlgorithm algorithm : CoverageDataAlgorithm
                .values()) {

            if (PRINT) {
                System.out.println();
                System.out.println();
                System.out.println(algorithm.name() + " SINGLE COVERAGE DATA VALUES");
            }
            for (double lat = maxLatitude - (heightPixelDistance * .5); lat >= minLatitude; lat -= heightPixelDistance) {
                if (PRINT) {
                    System.out.println();
                }
                for (double lon = minLongitude + (widthPixelDistance * .5); lon <= maxLongitude; lon += widthPixelDistance) {
                    Double value = CoverageDataTestUtils.getValue(
                            geoPackage, algorithm, lat, lon, requestEpsg);
                    if (PRINT) {
                        System.out.print("   " + value);
                    }
                    if (!allowNulls) {
                        TestCase.assertNotNull(value);
                    }
                }
            }

            CoverageDataResults results = CoverageDataTestUtils
                    .getValues(geoPackage, algorithm, boundingBox, width,
                            height, requestEpsg);
            if (!allowNulls) {
                TestCase.assertNotNull(results);
            }
            if (results != null) {
                if (PRINT) {
                    System.out.println();
                    System.out.println();
                    System.out.println(algorithm.name());
                }
                Double[][] values = results.getValues();
                TestCase.assertEquals(height, values.length);
                TestCase.assertEquals(width, values[0].length);
                for (int y = 0; y < values.length; y++) {
                    if (PRINT) {
                        System.out.println();
                    }
                    for (int x = 0; x < values[0].length; x++) {
                        Double value = values[y][x];
                        if (PRINT) {
                            System.out.print("   " + value);
                        }
                        if (!allowNulls) {
                            TestCase.assertNotNull(value);
                        }
                    }
                }
            }
        }

    }

    /**
     * Test a full bounding box around tiles and optionally print. Also test the
     * bounds of individual tiles.
     *
     * @throws Exception
     */
    @Test
    public void testFullBoundingBox() throws Exception {

        int width = 10;
        int height = 6;

        List<String> coverageDataTables = CoverageDataPng.getTables(geoPackage);
        TileMatrixSetDao dao = geoPackage.getTileMatrixSetDao();

        for (String coverageTable : coverageDataTables) {

            TileMatrixSet tileMatrixSet = dao.queryForId(coverageTable);

            long geoPackageEpsg = tileMatrixSet.getSrs().getOrganizationCoordsysId();

            Projection projection = ProjectionFactory.getProjection(geoPackageEpsg);
            Projection printProjection = ProjectionFactory
                    .getProjection(ProjectionConstants.EPSG_WORLD_GEODETIC_SYSTEM);
            ProjectionTransform wgs84Transform = projection
                    .getTransformation(printProjection);

            BoundingBox boundingBox = tileMatrixSet.getBoundingBox();

            double minLongitude = boundingBox.getMinLongitude();
            double maxLongitude = boundingBox.getMaxLongitude();
            double minLatitude = boundingBox.getMinLatitude();
            double maxLatitude = boundingBox.getMaxLatitude();

            double widthPixelDistance = (maxLongitude - minLongitude) / width;
            double heightPixelDistance = (maxLatitude - minLatitude) / height;

            if (PRINT) {
                System.out.println();
                System.out.println();
                System.out.println("Full Bounding Box Test");
                System.out.println();
                System.out.println("REQUEST");
                System.out.println();
                System.out.println("   Min Lat: "
                        + boundingBox.getMinLatitude());
                System.out.println("   Max Lat: "
                        + boundingBox.getMaxLatitude());
                System.out.println("   Min Lon: "
                        + boundingBox.getMinLongitude());
                System.out.println("   Max Lon: "
                        + boundingBox.getMaxLongitude());
                System.out.println("   Result Width: " + width);
                System.out.println("   Result Height: " + height);

                System.out.println();
                System.out.println();
                System.out.println("WGS84 REQUEST");
                System.out.println();
                BoundingBox wgs84BoundingBox = boundingBox
                        .transform(wgs84Transform);
                System.out.println("   Min Lat: "
                        + wgs84BoundingBox.getMinLatitude());
                System.out.println("   Max Lat: "
                        + wgs84BoundingBox.getMaxLatitude());
                System.out.println("   Min Lon: "
                        + wgs84BoundingBox.getMinLongitude());
                System.out.println("   Max Lon: "
                        + wgs84BoundingBox.getMaxLongitude());

                System.out.println();
                System.out.println();
                System.out.println("WGS84 LOCATIONS");
                for (double lat = maxLatitude; lat >= minLatitude; lat -= heightPixelDistance) {
                    System.out.println();
                    for (double lon = minLongitude; lon <= maxLongitude; lon += widthPixelDistance) {
                        double[] point = wgs84Transform.transform(lon, lat);
                        System.out.print("   (" + point[1] + "," + point[0]
                                + ")");
                    }
                    double[] point = wgs84Transform
                            .transform(maxLongitude, lat);
                    System.out.print("   (" + point[1] + "," + point[0] + ")");
                }
                System.out.println();
                for (double lon = minLongitude; lon <= maxLongitude; lon += widthPixelDistance) {
                    double[] point = wgs84Transform.transform(lon, minLatitude);
                    System.out.print("   (" + point[1] + "," + point[0] + ")");
                }
                double[] point = wgs84Transform.transform(maxLongitude,
                        minLatitude);
                System.out.print("   (" + point[1] + "," + point[0] + ")");
            }

            for (CoverageDataAlgorithm algorithm : CoverageDataAlgorithm
                    .values()) {

                if (PRINT) {
                    System.out.println();
                    System.out.println();
                    System.out.println(algorithm.name()
                            + " SINGLE COVERAGE DATA VALUES Full Bounding Box");
                }
                for (double lat = maxLatitude; lat >= minLatitude; lat -= heightPixelDistance) {
                    if (PRINT) {
                        System.out.println();
                    }
                    for (double lon = minLongitude; lon <= maxLongitude; lon += widthPixelDistance) {
                        Double value = CoverageDataTestUtils
                                .getValue(geoPackage, algorithm, lat, lon,
                                        geoPackageEpsg);
                        if (PRINT) {
                            System.out.print("   " + value);
                        }
                        if (algorithm == CoverageDataAlgorithm.NEAREST_NEIGHBOR
                                || (lat < maxLatitude && lon > minLongitude
                                && lat > minLatitude && lon < maxLongitude)) {
                            if (!allowNulls) {
                                TestCase.assertNotNull(value);
                            }
                        }
                    }
                    Double value = CoverageDataTestUtils.getValue(
                            geoPackage, algorithm, lat, maxLongitude,
                            geoPackageEpsg);
                    if (PRINT) {
                        System.out.print("   " + value);
                    }
                    if (algorithm == CoverageDataAlgorithm.NEAREST_NEIGHBOR) {
                        if (!allowNulls) {
                            TestCase.assertNotNull(value);
                        }
                    }
                }
                if (PRINT) {
                    System.out.println();
                }
                for (double lon = minLongitude; lon <= maxLongitude; lon += widthPixelDistance) {
                    Double value = CoverageDataTestUtils.getValue(
                            geoPackage, algorithm, minLatitude, lon,
                            geoPackageEpsg);
                    if (PRINT) {
                        System.out.print("   " + value);
                    }
                    if (algorithm == CoverageDataAlgorithm.NEAREST_NEIGHBOR) {
                        if (!allowNulls) {
                            TestCase.assertNotNull(value);
                        }
                    }
                }
                Double value = CoverageDataTestUtils.getValue(
                        geoPackage, algorithm, minLatitude, maxLongitude,
                        geoPackageEpsg);
                if (PRINT) {
                    System.out.print("   " + value);
                }
                if (algorithm == CoverageDataAlgorithm.NEAREST_NEIGHBOR) {
                    if (!allowNulls) {
                        TestCase.assertNotNull(value);
                    }
                }

                CoverageDataResults results = CoverageDataTestUtils
                        .getValues(geoPackage, algorithm, boundingBox,
                                width, height, geoPackageEpsg);
                if (PRINT) {
                    System.out.println();
                    System.out.println();
                    System.out.println(algorithm.name() + " Full Bounding Box");
                }
                Double[][] values = results.getValues();
                for (int y = 0; y < values.length; y++) {
                    if (PRINT) {
                        System.out.println();
                    }
                    for (int x = 0; x < values[0].length; x++) {
                        value = values[y][x];
                        if (PRINT) {
                            System.out.print("   " + values[y][x]);
                        }
                        if (!allowNulls) {
                            TestCase.assertNotNull(value);
                        }
                    }
                }

                TileMatrix tileMatrix = results.getTileMatrix();
                for (int row = 0; row < tileMatrix.getMatrixHeight(); row++) {
                    for (int column = 0; column < tileMatrix.getMatrixWidth(); column++) {

                        BoundingBox boundingBox2 = TileBoundingBoxUtils
                                .getBoundingBox(boundingBox, tileMatrix,
                                        column, row);

                        double minLongitude2 = boundingBox2.getMinLongitude();
                        double maxLongitude2 = boundingBox2.getMaxLongitude();
                        double minLatitude2 = boundingBox2.getMinLatitude();
                        double maxLatitude2 = boundingBox2.getMaxLatitude();

                        if (PRINT) {
                            System.out.println();
                            System.out.println();
                            System.out.println(algorithm.name()
                                    + " SINGLE COVERAGE DATA VALUES Tile row = " + row
                                    + ", column = " + column);
                        }

                        value = CoverageDataTestUtils.getValue(
                                geoPackage, algorithm, maxLatitude2,
                                minLongitude2, geoPackageEpsg);
                        double[] point = wgs84Transform.transform(
                                minLongitude2, maxLatitude2);
                        if (PRINT) {
                            System.out.print("   " + value + " ("
                                    + point[1] + "," + point[0] + ")");
                        }
                        if (algorithm != CoverageDataAlgorithm.NEAREST_NEIGHBOR
                                && (row == 0 || column == 0)) {
                            TestCase.assertNull(value);
                        } else {
                            if (!allowNulls) {
                                TestCase.assertNotNull(value);
                            }
                        }

                        value = CoverageDataTestUtils.getValue(
                                geoPackage, algorithm, maxLatitude2,
                                maxLongitude2, geoPackageEpsg);
                        point = wgs84Transform.transform(maxLongitude2,
                                maxLatitude2);
                        if (PRINT) {
                            System.out.println("   " + value + " ("
                                    + point[1] + "," + point[0] + ")");
                        }
                        if (algorithm != CoverageDataAlgorithm.NEAREST_NEIGHBOR
                                && (row == 0 || column == tileMatrix
                                .getMatrixWidth() - 1)) {
                            TestCase.assertNull(value);
                        } else {
                            if (!allowNulls) {
                                TestCase.assertNotNull(value);
                            }
                        }

                        value = CoverageDataTestUtils.getValue(
                                geoPackage, algorithm, minLatitude2,
                                minLongitude2, geoPackageEpsg);
                        point = wgs84Transform.transform(minLongitude2,
                                minLatitude2);
                        if (PRINT) {
                            System.out.print("   " + value + " ("
                                    + point[1] + "," + point[0] + ")");
                        }
                        if (algorithm != CoverageDataAlgorithm.NEAREST_NEIGHBOR
                                && (row == tileMatrix.getMatrixHeight() - 1 || column == 0)) {
                            TestCase.assertNull(value);
                        } else {
                            if (!allowNulls) {
                                TestCase.assertNotNull(value);
                            }
                        }

                        value = CoverageDataTestUtils.getValue(
                                geoPackage, algorithm, minLatitude2,
                                maxLongitude2, geoPackageEpsg);
                        point = wgs84Transform.transform(maxLongitude2,
                                minLatitude2);
                        if (PRINT) {
                            System.out.println("   " + value + " ("
                                    + point[1] + "," + point[0] + ")");
                        }
                        if (algorithm != CoverageDataAlgorithm.NEAREST_NEIGHBOR
                                && (row == tileMatrix.getMatrixHeight() - 1 || column == tileMatrix
                                .getMatrixWidth() - 1)) {
                            TestCase.assertNull(value);
                        } else {
                            if (!allowNulls) {
                                TestCase.assertNotNull(value);
                            }
                        }

                        results = CoverageDataTestUtils.getValues(
                                geoPackage, algorithm, boundingBox2, width,
                                height, geoPackageEpsg);
                        if (PRINT) {
                            System.out.println();
                            System.out.println();
                            System.out.println(algorithm.name()
                                    + " Tile row = " + row + ", column = "
                                    + column);
                        }
                        if (results == null) {
                            if (PRINT) {
                                System.out.println();
                                System.out.print("null results");
                            }
                        } else {
                            values = results.getValues();
                            for (int y = 0; y < values.length; y++) {
                                if (PRINT) {
                                    System.out.println();
                                }
                                for (int x = 0; x < values[0].length; x++) {
                                    value = values[y][x];
                                    if (PRINT) {
                                        System.out.print("   "
                                                + values[y][x]);
                                    }
                                    if (!allowNulls) {
                                        TestCase.assertNotNull(value);
                                    }
                                }
                            }
                        }

                    }
                }
            }
        }

    }

    /**
     * Test a single location
     *
     * @param latitude
     * @param longitude
     * @throws Exception
     */
    private void testLocation(double latitude, double longitude)
            throws Exception {

        if (PRINT) {
            System.out.println("Latitude: " + latitude);
            System.out.println("Longitude: " + longitude);
        }

        for (CoverageDataAlgorithm algorithm : CoverageDataAlgorithm
                .values()) {
            Double value = CoverageDataTestUtils.getValue(geoPackage,
                    algorithm, latitude, longitude,
                    ProjectionConstants.EPSG_WORLD_GEODETIC_SYSTEM);
            if (PRINT) {
                System.out.println(algorithm.name() + ": " + value);
            }
        }
    }

}