/*
* GeoTools - The Open Source Java GIS Toolkit
* http://geotools.org
*
* (C) 2002-2008, 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.feature.visitor;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;
import org.geotools.filter.FidFilter;
import org.geotools.filter.visitor.DefaultFilterVisitor;
import org.opengis.filter.Id;
/**
* Gather up all FeatureId strings into a provided HashSet.
* <p>
* Example:<code>Set<String> fids = (Set<String>) filter.accept( IdCollectorFilterVisitor.ID_COLLECTOR, new HashSet() );</code>
*
* @source $URL$
*/
public class IdCollectorFilterVisitor extends DefaultFilterVisitor {
public static final IdCollectorFilterVisitor ID_COLLECTOR = new IdCollectorFilterVisitor(true);
public static final IdCollectorFilterVisitor IDENTIFIER_COLLECTOR = new IdCollectorFilterVisitor(false);
private final boolean mCollectStringIds;
/**
* @deprecated use {@link #IdCollectorFilterVisitor(boolean)}
*/
protected IdCollectorFilterVisitor(){
mCollectStringIds = true;
}
protected IdCollectorFilterVisitor(boolean collectStringIds){
mCollectStringIds = collectStringIds;
}
@SuppressWarnings("unchecked")
@Override
public Object visit( Id filter, Object data ) {
if( mCollectStringIds){
Set set = (Set) data;
set.addAll( filter.getIDs() );
return set;
}else{
Set set = (Set) data;
set.addAll( filter.getIdentifiers());
return set;
}
}
}