/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2010, Johann Sorel
 *
 *    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.geotoolkit.data.memory;


import java.util.List;
import java.util.Map;
import org.geotoolkit.data.FeatureIterator;
import org.geotoolkit.data.FeatureStoreRuntimeException;
import org.geotoolkit.data.memory.mapping.DefaultFeatureMapper;
import org.geotoolkit.data.memory.mapping.FeatureMapper;
import org.apache.sis.util.Classes;
import org.opengis.feature.Feature;
import org.opengis.feature.FeatureType;
import org.opengis.feature.PropertyType;

/**
 * Basic support for a FeatureIterator that moves attributs to a new type definition
 * using a mapping objet.
 *
 * @author Johann Sorel (Puzzle-GIS)
 * @module
 */
public class GenericMappingFeatureIterator implements FeatureIterator{

    private final FeatureIterator ite;
    private final FeatureMapper mapper;

    public GenericMappingFeatureIterator(final FeatureIterator ite,
            final FeatureType originalType, final FeatureType newType,
            final Map<PropertyType,List<PropertyType>> mapping,
            final Map<PropertyType,Object> defaults){
        this(ite,new DefaultFeatureMapper(originalType,newType,mapping,defaults));
    }

    public GenericMappingFeatureIterator(final FeatureIterator ite, final FeatureMapper mapper){
        this.ite = ite;
        this.mapper = mapper;
    }

    @Override
    public Feature next() {
        return mapper.transform(ite.next());
    }

    @Override
    public void close() {
        ite.close();
    }

    @Override
    public boolean hasNext() {
        return ite.hasNext();
    }

    @Override
    public void remove() {
        throw new FeatureStoreRuntimeException("Not writable.");
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder(Classes.getShortClassName(this));
        sb.append('\n');
        String subIterator = "\u2514\u2500\u2500" + ite.toString(); //move text to the right
        subIterator = subIterator.replaceAll("\n", "\n\u00A0\u00A0\u00A0"); //move text to the right
        sb.append(subIterator);
        return sb.toString();
    }
}