*    GeoTools - The Open Source Java GIS Toolkit

*    http://geotools.org

*

*    (C) 2006-2013, Open Source Geospatial Foundation (OSGeo)

*

*    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;

*    version 2.1 of the License.

*

*    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.

*/

* Testing {@link ImageMosaicReader}.

*

* @author Simone Giannecchini, GeoSolutions

* @author Stefan Alfons Krueger (alfonx), Wikisquare.de

* @since 2.3

*

*

*

* @source $URL$

*/

    private final static double DELTA = 1E-4;

   

    private final static Logger LOGGER = Logger.getLogger(ImageMosaicReaderTest.class.toString());

   

  public static junit.framework.Test suite() {

      return new JUnit4TestAdapter(ImageMosaicReaderTest.class);

  }

  private URL rgbURL;

 

  private URL heterogeneousGranulesURL;

  private URL indexURL;

  private URL indexAlphaURL;

  private URL grayURL;

  private URL index_unique_paletteAlphaURL;

  private URL rgbAURL;

  private URL rgbAURLTiff;

  private URL overviewURL;

  static boolean INTERACTIVE;

  private URL timeURL;

  private URL timeAdditionalDomainsURL;

 

  private URL timeAdditionalDomainsRangeURL;

 

  private URL timeRangesURL;

  private URL imposedEnvelopeURL;

 

    private final static String H2_SAMPLE_PROPERTIES = "SPI=org.geotools.data.h2.H2DataStoreFactory\n"

            + "dbtype=h2\n" + "Loose\\ bbox=true #important for performances\n"

            + "Estimated\\ extends=false #important for performances\n" + "user=gs\n"

            + "passwd=gs\n" + "validate \\connections=true #important for avoiding errors\n"

            + "Connection\\ timeout=3600\n"

            + "max \\connections=10 #important for performances, internal pooling\n"

            + "min \\connections=5  #important for performances, internal pooling\n";

    private URL timeFormatURL;

    private URL oneBitURL;

 

  /**

   * Testing crop capabilities.

   *

   * @throws MismatchedDimensionException

   * @throws IOException

   * @throws FactoryException

   */

  @Test

  public void crop() throws Exception {

    imageMosaicCropTest(rgbURL, "crop-rgbURL");

   

    imageMosaicCropTest(indexURL, "crop-indexURL");

   

    imageMosaicCropTest(grayURL, "crop-grayURL");

   

    imageMosaicCropTest(overviewURL, "crop-overviewURL");

   

    imageMosaicCropTest(indexAlphaURL, "crop-indexAlphaURL");

   

    imageMosaicCropTest(rgbAURL, "crop-rgbAURL");

   

    imageMosaicCropTest(index_unique_paletteAlphaURL,"crop-index_unique_paletteAlphaURL");

  }

  /**

   * Tests the {@link ImageMosaicReader} with default parameters for the

   * various input params.

   *

   * @throws IOException

   * @throws MismatchedDimensionException

   * @throws FactoryException

   */

  @Test

  public void alpha() throws Exception{

   

    final String testName="alpha-";

    if (INTERACTIVE)

      imageMosaicSimpleParamsTest(rgbURL, null, null,testName+rgbURL.getFile()+"-original", false);

    GridCoverage2D coverage=imageMosaicSimpleParamsTest(rgbURL, Color.black,Color.black,testName+rgbURL.getFile(), false);

                ColorModel colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

                assertTrue(colorModel instanceof ComponentColorModel);

               

    if (INTERACTIVE)

      // the input images have transparency and they do overlap, we need

      // to ask for blending mosaic.

      imageMosaicSimpleParamsTest(rgbAURL, null, null,testName+rgbAURL.getFile()+"-original", true);

    coverage=imageMosaicSimpleParamsTest(rgbAURL, Color.black,Color.black,testName+rgbAURL.getFile(), false);

                colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

                assertTrue(colorModel instanceof ComponentColorModel);

    // //

    //

    // This images have borders that are black and have a color model that

    // is IndexColorModel but all with different palette hence a color

    // conversion will be applied to go to RGB.

    //

    // When we do the input transparent color we will add transparency to

    // the images but only where the transparent color resides. Moreover the

    // background will be transparent.

    //

    // //

    if (INTERACTIVE)

      imageMosaicSimpleParamsTest(indexURL, null, null,testName+indexURL.getFile()+"-original", false);

    coverage = imageMosaicSimpleParamsTest(indexURL, new Color(58, 49, 8),Color.black,testName+indexURL.getFile(), false);

    colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

    assertTrue(colorModel instanceof ComponentColorModel);

   

   

   

    if (INTERACTIVE)

      imageMosaicSimpleParamsTest(overviewURL, null, null,testName+overviewURL.getFile()+"-original", false);

    coverage = imageMosaicSimpleParamsTest(overviewURL, new Color(58, 49, 8),Color.black,testName+overviewURL.getFile()+"-indexURL", false);

    colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

                assertTrue(colorModel instanceof ComponentColorModel);

               

    if (INTERACTIVE)

      imageMosaicSimpleParamsTest(indexAlphaURL, null, null,testName+indexAlphaURL.getFile()+"-original", false);

    coverage = imageMosaicSimpleParamsTest(indexAlphaURL, new Color(41,41, 33), Color.black,testName+indexAlphaURL.getFile(), false);

    colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

                assertTrue(colorModel instanceof ComponentColorModel);

               

   

   

    if (INTERACTIVE)

      imageMosaicSimpleParamsTest(grayURL, null, null,testName+grayURL.getFile()+"-original", false);

    coverage = imageMosaicSimpleParamsTest(grayURL, Color.black,Color.black, testName+grayURL.getFile(), false);

    colorModel = coverage.getRenderedImage().getColorModel();

                assertTrue(colorModel.hasAlpha());

                assertTrue(colorModel instanceof ComponentColorModel);;

   

  }

 

  /**

   * Tests the {@link ImageMosaicReader} with default parameters for the

   * various input params.

   *

   * @throws IOException

   * @throws MismatchedDimensionException

   * @throws FactoryException

   */

  @Test

  public void overviews() throws Exception {

    final AbstractGridFormat format = TestUtils.getFormat(overviewURL);

    final ImageMosaicReader reader = TestUtils.getReader(overviewURL, format);

    // limit yourself to reading just a bit of it

    final ParameterValue<GridGeometry2D> gg =  AbstractGridFormat.READ_GRIDGEOMETRY2D.createValue();

    final GeneralEnvelope envelope = reader.getOriginalEnvelope();

    final Dimension dim= new Dimension();

    dim.setSize(reader.getOriginalGridRange().getSpan(0)/2.0, reader.getOriginalGridRange().getSpan(1)/2.0);

    final Rectangle rasterArea=(( GridEnvelope2D)reader.getOriginalGridRange());

    rasterArea.setSize(dim);

    final GridEnvelope2D range= new GridEnvelope2D(rasterArea);

    gg.setValue(new GridGeometry2D(range,envelope));

   

    // use imageio with defined tiles

    final ParameterValue<Boolean> useJai = AbstractGridFormat.USE_JAI_IMAGEREAD.createValue();

    useJai.setValue(false);

   

    final ParameterValue<String> tileSize = AbstractGridFormat.SUGGESTED_TILE_SIZE.createValue();

    tileSize.setValue("128,128");

   

    // Test the output coverage

    TestUtils.checkCoverage(reader, new GeneralParameterValue[] {gg,useJai ,tileSize}, "overviews test");

  }

 

    /**

     *

     * @throws IOException

     * @throws MismatchedDimensionException

     * @throws FactoryException

     */

    @Test

    // @Ignore

    public void readingResolutions() throws Exception {

        final AbstractGridFormat format = TestUtils.getFormat(overviewURL);

        final ImageMosaicReader reader = TestUtils.getReader(overviewURL, format);

        double[] result = reader.getReadingResolutions(OverviewPolicy.QUALITY, new double[] { 32, 32 });

        assertEquals(16.0714285714285, result[0], DELTA);

        assertEquals(16.0427807486631, result[1], DELTA);

    }

 

  @Test

      //@Ignore

  public void timeElevationH2() throws Exception {

     

      final File workDir=new File(TestData.file(this, "."),"water_temp3");

        if(!workDir.mkdir()){

            FileUtils.deleteDirectory(workDir);

            assertTrue("Unable to create workdir:"+workDir,workDir.mkdir());

        }

      FileUtils.copyFile(TestData.file(this, "watertemp.zip"), new File(workDir,"watertemp.zip"));

      TestData.unzipFile(this, "water_temp3/watertemp.zip");

      final URL timeElevURL = TestData.url(this, "water_temp3");

     

      //place H2 file in the dir

      FileWriter out=null;

      try{

        out = new FileWriter(new File(TestData.file(this, "."),"/water_temp3/datastore.properties"));

        out.write("SPI=org.geotools.data.h2.H2DataStoreFactory\n");

        out.write("database=imagemosaic\n");

        out.write("dbtype=h2\n");

        out.write("Loose\\ bbox=true #important for performances\n");

        out.write("Estimated\\ extends=false #important for performances\n");

        out.write("user=geosolutions\n");

        out.write("passwd=geosolutions\n");

        out.write("validate \\connections=true #important for avoiding errors\n");

        out.write("Connection\\ timeout=3600\n");

        out.write("max \\connections=10 #important for performances, internal pooling\n");

        out.write("min \\connections=5  #important for performances, internal pooling\n");

        out.flush();

      } finally {

        if(out!=null){

          IOUtils.closeQuietly(out);

        }

      }

     

     

      // now start the test

    final AbstractGridFormat format = TestUtils.getFormat(timeElevURL);

    assertNotNull(format);

    ImageMosaicReader reader = TestUtils.getReader(timeElevURL, format);

    assertNotNull(reader);

   

    final String[] metadataNames = reader.getMetadataNames();

    assertNotNull(metadataNames);

    assertEquals(metadataNames.length,12);

   

    assertEquals("true", reader.getMetadataValue("HAS_TIME_DOMAIN"));

    final String timeMetadata = reader.getMetadataValue("TIME_DOMAIN");

    assertNotNull(timeMetadata);

    assertEquals(2,timeMetadata.split(",").length);

    assertEquals(timeMetadata.split(",")[0],reader.getMetadataValue("TIME_DOMAIN_MINIMUM"));

    assertEquals(timeMetadata.split(",")[1],reader.getMetadataValue("TIME_DOMAIN_MAXIMUM"));

    assertEquals("java.sql.Timestamp", reader.getMetadataValue("TIME_DOMAIN_DATATYPE"));

   

    assertEquals("true", reader.getMetadataValue("HAS_ELEVATION_DOMAIN"));

    final String elevationMetadata = reader.getMetadataValue("ELEVATION_DOMAIN");

    assertNotNull(elevationMetadata);

                assertEquals("0,100",elevationMetadata);

    assertEquals(2,elevationMetadata.split(",").length);

          assertEquals(Double.parseDouble(elevationMetadata.split(",")[0]),Double.parseDouble(reader.getMetadataValue("ELEVATION_DOMAIN_MINIMUM")),1E-6);

          assertEquals(Double.parseDouble(elevationMetadata.split(",")[1]),Double.parseDouble(reader.getMetadataValue("ELEVATION_DOMAIN_MAXIMUM")),1E-6);

          assertEquals("java.lang.Integer", reader.getMetadataValue("ELEVATION_DOMAIN_DATATYPE"));

   

   

    // limit yourself to reading just a bit of it

    final ParameterValue<GridGeometry2D> gg =  AbstractGridFormat.READ_GRIDGEOMETRY2D.createValue();

    final GeneralEnvelope envelope = reader.getOriginalEnvelope();

    final Dimension dim= new Dimension();

    dim.setSize(reader.getOriginalGridRange().getSpan(0)/2.0, reader.getOriginalGridRange().getSpan(1)/2.0);

    final Rectangle rasterArea=(( GridEnvelope2D)reader.getOriginalGridRange());

    rasterArea.setSize(dim);

    final GridEnvelope2D range= new GridEnvelope2D(rasterArea);

    gg.setValue(new GridGeometry2D(range,envelope));

   

   

    // use imageio with defined tiles

    final ParameterValue<List> time = ImageMosaicFormat.TIME.createValue();

    final List<Date> timeValues= new ArrayList<Date>();

    final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.sss'Z'");

    sdf.setTimeZone(TimeZone.getTimeZone("GMT+0"));

    Date date = sdf.parse("2008-10-31T00:00:00.000Z");

    timeValues.add(date);

    time.setValue(timeValues);

   

    final ParameterValue<double[]> bkg = ImageMosaicFormat.BACKGROUND_VALUES.createValue();

    bkg.setValue(new double[]{-9999.0});

   

   

    final ParameterValue<Boolean> direct= ImageMosaicFormat.USE_JAI_IMAGEREAD.createValue();

    direct.setValue(false);

   

    final ParameterValue<List> elevation = ImageMosaicFormat.ELEVATION.createValue();

    elevation.setValue(Arrays.asList(100.0));

                 

    // Test the output coverage

    TestUtils.checkCoverage(reader, new GeneralParameterValue[] {gg,time,bkg ,elevation ,direct}, "Time-Elevation Test");

    reader.dispose();

   

    reader= TestUtils.getReader(timeElevURL, format);

                elevation.setValue(Arrays.asList(NumberRange.create(0.0,10.0)));

       

                // Test the output coverage

                TestUtils.checkCoverage(reader, new GeneralParameterValue[] { gg, time, bkg, elevation,direct },"Time-Elevation Test");

               

         // clean up

         if (!INTERACTIVE){         

           FileUtils.deleteDirectory( TestData.file(this, "water_temp3"));

         }

  } 

 

  /**

   * This test is used to check backward compatibility with old imagemosaics wich does not include

   * the TypeName=MOSAICNAME into the generated MOSAICNAME.properties file

   *

   * @throws Exception

   */

    @Test

    // @Ignore

    public void testTypeNameBackwardsCompatibility() throws Exception {

        final File workDir = new File(TestData.file(this, "."), "water_temp5");

        if (!workDir.mkdir()) {

            FileUtils.deleteDirectory(workDir);

            assertTrue("Unable to create workdir:" + workDir, workDir.mkdir());

        }

        FileUtils