Application: gvSIG desktop

package org.gvsig.fmap.data.vectorial.joinstore;

import java.util.Collection;

import java.util.Iterator;

import java.util.NoSuchElementException;

import org.gvsig.fmap.data.DataCollection;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.vectorial.AbstractFeatureCollection;

import org.gvsig.fmap.data.vectorial.Feature;

import org.gvsig.fmap.data.vectorial.FeatureCollection;

import org.gvsig.fmap.data.vectorial.FeatureID;

import org.gvsig.fmap.data.vectorial.FeatureManager;

import org.gvsig.fmap.data.vectorial.FeatureStore;

import org.gvsig.fmap.data.vectorial.FeatureType;

import org.gvsig.fmap.data.vectorial.JoinFeature;

import org.gvsig.fmap.data.vectorial.expressionevaluator.FeatureFilter;

public class JoinDataCollection extends AbstractFeatureCollection{

//        private DataCollection dataCollection1 = null;

//        private DataCollection dataCollection2 = null;

        private FeatureType featureType;

        private String filter;

        private FeatureFilter parser;

//        private FeatureStore store1;

//        private FeatureStore store2;

        private String order;

        private JoinFeatureStore joinStore;

        public JoinDataCollection(JoinFeatureStore joinStore,FeatureType type, String filter,String order) throws ReadException {

                this.joinStore=joinStore;

//                this.store1=joinStore.getPrimaryStore();

//                this.store2=joinStore.getSecondaryStore();

                this.order=order;

//                driverFeatureCount1=store1.getFeatureCount();

                this.featureType=type;

                this.filter=filter;

                if (this.filter!=null)

                        parser = new FeatureFilter(filter,this.featureType);

        }

        public int size() {

                return 0;

        }

        public boolean contains(Object arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public Iterator iterator() {

                this.checkModified();

                JoinIterator complexIter = null;

                try {

                        FeatureStore store1=joinStore.getPrimaryStore();

                        FeatureStore store2=joinStore.getSecondaryStore();

                        //TODO el featureType y el order hay que modificarlo y reorganizarlo.

                        FeatureCollection coll1=(FeatureCollection)store1.getDataCollection(featureType,filter,order);

                        FeatureCollection coll2=(FeatureCollection)store2.getDataCollection(featureType,filter,order);

                        complexIter=new JoinIterator(this.featureType,coll1,coll2);

                } catch (ReadException e) {

                        e.printStackTrace();

                }

                return complexIter;

        }

        public Object[] toArray() {

                // TODO Auto-generated method stub

                return null;

        }

        public Object[] toArray(Object[] arg0) {

                // TODO Auto-generated method stub

                return null;

        }

        public boolean add(Object arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public boolean remove(Object arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public boolean containsAll(Collection arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public boolean addAll(Collection arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public boolean removeAll(Collection arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public boolean retainAll(Collection arg0) {

                // TODO Auto-generated method stub

                return false;

        }

        public void clear() {

                // TODO Auto-generated method stub

        }

        protected class JoinIterator implements Iterator{

//                protected long position=0;

//                private boolean nextChecked=false;

                private Feature feature;

                private FeatureType featureType;

                private FeatureCollection coll1;

                private FeatureCollection coll2;

                private Iterator iterator1;

                public JoinIterator(FeatureType featureType,FeatureCollection coll1,FeatureCollection coll2){

                        this.featureType=featureType;

                        this.coll1=coll1;

                        this.coll2=coll2;

                        this.iterator1=coll1.iterator();

                }

//                protected FeatureID createCurrentFeatureID(long pos){

//                        if (pos<driverFeatureCount){

//                                return new ComplexFeatureID(store1,store2,pos);

//                        } else {

//                                return null;

//                        }

//                }

                public boolean hasNext(){

                        return iterator1.hasNext();

                }

                public Object next() {

                        Feature feature=(Feature)iterator1.next();

                        JoinFeatureStoreParameters jfsp=joinStore.getJoinParameters();

                        String linkFieldPrimary=jfsp.getLinkFieldPrimary();

                        String linkFieldSecondary=jfsp.getLinkFieldSecondary();

                        Object obj=feature.get(linkFieldPrimary);

                        Iterator iter=null;

                        JoinFeature complexFeature=null;

                        try {

                                iter = joinStore.getSecondaryStore().getDataCollection(featureType,linkFieldSecondary + " = " + obj.toString() + " and "+ filter,null).iterator();

                                Feature fea = null;

                                while (iter.hasNext()) {

                                        fea = (Feature) iter.next();

                                        break;

                                }

                                complexFeature=new JoinFeature(featureType,feature,fea);

                        } catch (ReadException e) {

                                e.printStackTrace();

                        }

                        return complexFeature;

                }

                public void remove() {

                        throw new UnsupportedOperationException();

                }

        }

        public void dispose() {

//                this.observable.deleteObservers();

//                this.observable = null;

//                this.featureIDs.clear();

//                this.featureIDs = null;

//                this.featureType = null;

//                this.store = null;

//                this.parser = null;

//                this.featureManager = null;

//                ;

        }

        public Feature getFeature(int index) {

                FeatureStore store1=joinStore.getPrimaryStore();

                FeatureStore store2=joinStore.getSecondaryStore();

                //TODO el featureType y el order hay que modificarlo y reorganizarlo.

                FeatureCollection coll1=null;

                FeatureCollection coll2=null;

                try {

                        coll1 = (FeatureCollection)store1.getDataCollection(featureType,filter,order);

                        coll2=(FeatureCollection)store2.getDataCollection(featureType,filter,order);

                } catch (ReadException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                //                TODO falta modificar el featureType para pedir solo los atributos que tenga en el primaryStore.

                Feature feature=coll1.getFeature(index);

                Object obj=feature.get(((JoinFeatureStoreParameters)joinStore.getParameters()).getLinkFieldPrimary());

                Iterator iter=null;

                JoinFeature complexFeature=null;

                try {

                        iter = joinStore.getSecondaryStore().getDataCollection(joinStore.getPrimaryStore().getDefaultFeatureType(),((JoinFeatureStoreParameters)joinStore.getParameters()).getLinkFieldSecondary() + " = " + obj.toString(),null).iterator();

                        Feature fea = null;

                        while (iter.hasNext()) {

                                fea = (Feature) iter.next();

                                break;

                        }

                        complexFeature=new JoinFeature(featureType,feature,fea);

                } catch (ReadException e) {

                        e.printStackTrace();

                }

                return complexFeature;

        }

}