Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.app.project.documents.view;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import org.apache.commons.lang3.tuple.ImmutablePair;

import org.apache.commons.lang3.tuple.Pair;

import org.gvsig.fmap.dal.AbstractStoresRepository;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.util.UnmodifiableBasicSet;

import org.gvsig.tools.util.UnmodifiableBasicSetAdapter;

/**

 *

 * @author jjdelcerro

 */

class ViewDocumentStoresRepository extends AbstractStoresRepository {

    private final ViewDocument viewdoc;

    public ViewDocumentStoresRepository(String name, ViewDocument viewdoc) {

        super(name);

        this.viewdoc = viewdoc;

    }

    public ViewDocumentStoresRepository(ViewDocument viewdoc) {

        this(viewdoc.getName(), viewdoc);

    }

    private Map<String, Pair<DataStoreParameters, DataStore>> getAll() {

        Map<String, Pair<DataStoreParameters, DataStore>> all = new HashMap<>();

        for (FLayer layer : viewdoc.layers()) {

            if (layer instanceof FLyrVect) {

                FeatureStore store = ((FLyrVect) layer).getFeatureStore();

                if (store != null) {

                    String theName = layer.getName();

                    if (all.containsKey(theName)) {

                        theName = theName + "$";

                        int counter = 1;

                        while (all.containsKey(theName + counter)) {

                            counter++;

                        }

                        theName = theName + counter;

                    }

                    all.put(theName, new ImmutablePair<>(store.getParameters(), store));

                }

            }

        }

        if (all.isEmpty()) {

            return null;

        }

        return all;

    }

    private Map<String, Pair<FeatureType, FeatureStore>> getAllFeatureTypes() {

        Map<String, Pair<FeatureType, FeatureStore>> all = new HashMap<>();

        //            MutablePair<DataStore,String> storeAndName = new MutablePair<>();

        for (FLayer layer : viewdoc.layers()) {

            if (layer instanceof FLyrVect) {

                FeatureStore store = ((FLyrVect) layer).getFeatureStore();

                if (store != null) {

                    String theName = layer.getName();

                    if (all.containsKey(theName)) {

                        theName = theName + "$";

                        int counter = 1;

                        while (all.containsKey(theName + counter)) {

                            counter++;

                        }

                        theName = theName + counter;

                    }

                    //                            storeAndName.setLeft(store);

                    //                            storeAndName.setRight(theName);

                    //                            if( filter == null || filter.test(storeAndName) ) {

                    all.put(theName, new ImmutablePair<>(store.getDefaultFeatureTypeQuietly(), store));

                    //                            }

                }

            }

            //                    if (limit > 0 && all.size() >= limit) {

            //                        break;

            //                    }

        }

        if (all.isEmpty()) {

            return null;

        }

        return all;

    }

    @Override

    protected DataStoreParameters getMyParameters(final String name) {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        if (all == null) {

            return null;

        }

        Pair<DataStoreParameters, DataStore> data = all.get(name);

        if (data == null) {

            return null;

        }

        return data.getLeft();

    }

    @Override

    protected FeatureType getMyFeatureType(String name) {

        Map<String, Pair<FeatureType, FeatureStore>> all = this.getAllFeatureTypes();

        if (all == null) {

            return null;

        }

        Pair<FeatureType, FeatureStore> data = all.get(name);

        if (data == null) {

            return null;

        }

        return all.get(name).getLeft();

    }

    @Override

    public DataStore getStore(String name) {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        if (all == null) {

            return null;

        }

        Pair<DataStoreParameters, DataStore> data = all.get(name);

        if (data == null) {

            return null;

        }

        DataStore store = data.getRight();

        DisposeUtils.bind(store);

        return store;

    }

    @Override

    public boolean containsKey(final String key) {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        if (all == null) {

            return false;

        }

        return all.containsKey(key);

    }

    @Override

    protected UnmodifiableBasicSet<String> getMyKeySet() {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        if (all == null) {

            return null;

        }

        return new UnmodifiableBasicSetAdapter<>(all.keySet());

    }

    @Override

    protected boolean isEmptyMyRepository() {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        return all.isEmpty();

    }

    @Override

    protected int getMySize() {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        return all.size();

    }

    @Override

    public Iterator<DataStoreParameters> iterator() {

        Map<String, Pair<DataStoreParameters, DataStore>> all = this.getAll();

        Iterator<Pair<DataStoreParameters, DataStore>> it = all.values().iterator();

        return new Iterator<DataStoreParameters>() {

            @Override

            public boolean hasNext() {

                return it.hasNext();

            }

            @Override

            public DataStoreParameters next() {

                return it.next().getLeft();

            }

        };

    }

}