/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2012 - 2014, 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.
 */
package org.geotools.coverage.io.range.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.measure.Measure;
import javax.measure.quantity.Dimensionless;
import javax.measure.unit.SI;
import javax.measure.unit.Unit;

import org.geotools.coverage.io.range.impl.EnumMeasureTest.Band;
import org.geotools.feature.NameImpl;
import org.geotools.referencing.crs.DefaultEngineeringCRS;
import org.geotools.referencing.cs.DefaultCoordinateSystemAxis;
import org.geotools.referencing.cs.DefaultLinearCS;
import org.geotools.referencing.datum.DefaultEngineeringDatum;
import org.geotools.util.SimpleInternationalString;
import org.junit.Assert;
import org.junit.Test;
import org.opengis.referencing.crs.SingleCRS;
import org.opengis.referencing.cs.AxisDirection;
import org.opengis.referencing.cs.CoordinateSystemAxis;
import org.opengis.referencing.datum.EngineeringDatum;
import org.opengis.util.CodeList;

/**
 * @author Simone Giannecchini, GeoSolutions
 * 
 * @source $URL$
 */
public class CodeMeasureTest extends Assert {

    /** Bands captured as an codelist used as an example below */
    private static List<Code> codes = new ArrayList<Code>();

    static class Code extends CodeList<Code> {
        private static final long serialVersionUID = -1103556579647561536L;

        public Code(String name) {
            super(name, codes);
        }

        public Code[] family() {
            return codes.toArray(new Code[codes.size()]);
        }
    }

    static {
        new Code("BLUE");
        new Code("GREEN");
        new Code("RED");
        new Code("NIR");
        new Code("SWIT");
        new Code("TIR");
        new Code("SWR2");
    }

    /**
     * This test uses use the default implementations to express 7 bands of a landsat image.
     */
    @Test
    public void testLandsatAxis() {
        CoordinateSystemAxis csAxis = new DefaultCoordinateSystemAxis(
                new SimpleInternationalString("light"), "light", AxisDirection.OTHER,
                SI.MICRO(SI.METER));

        DefaultLinearCS lightCS = new DefaultLinearCS("light", csAxis);
        Map<String, Object> datumProperties = new HashMap<String, Object>();
        datumProperties.put("name", "light");

        EngineeringDatum lightDatum = new DefaultEngineeringDatum(datumProperties);
        SingleCRS lightCRS = new DefaultEngineeringCRS("wave length", lightDatum, lightCS);

        List<Measure<Code, Dimensionless>> keys = CodeMeasure.valueOf(codes);

        DefaultAxis<Code, Dimensionless> axis = new DefaultAxis<Code, Dimensionless>(new NameImpl(
                "Bands"), new SimpleInternationalString("Expressed in wavelengths"), keys,
                Unit.ONE, lightCRS);

        assertEquals(lightCRS, axis.getCoordinateReferenceSystem());
        assertEquals(7, axis.getKeys().size());
        
        // Ensure that the equals method is correct
        CodeMeasure<Code> code = (CodeMeasure<Code>) CodeMeasure.valueOf(codes.get(0));
        CodeMeasure<Code> code2 = (CodeMeasure<Code>) CodeMeasure.valueOf(codes.get(1));
        assertFalse(code.equals(code2));
        
        // Check if the ordinal value is correct
        assertNotEquals(code.doubleValue(null), code2.doubleValue(null), 0.01d);
        
        // Ensure the Unit is one
        assertEquals(Unit.ONE, code.getUnit());
        
        // Check if the value is correct
        assertSame(codes.get(0), code.getValue());
        
        // Check if the TO method is correct
        assertSame(code.to(null), code);
    }

}