Application: gvSIG desktop

package org.gvsig.export.impl;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.ICoordTrans;

import org.cresques.cts.IProjection;

import org.gvsig.export.ExportException;

import org.gvsig.export.ExportLocator;

import org.gvsig.export.ExportParameters;

import org.gvsig.export.ExportParametersGeometry;

import org.gvsig.export.ExportProcess;

import org.gvsig.export.spi.ExportService;

import org.gvsig.export.spi.ExportServiceFactory;

import org.gvsig.export.spi.ExportServiceManager;

import org.gvsig.fmap.dal.OpenDataStoreParameters;

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

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

import org.gvsig.tools.task.TaskStatus;

/**

 *

 * @author jjdelcerro

 */

public class DefaultExportProcess 

        implements ExportProcess 

    {

    private ExportServiceFactory factory;

    private ExportService service;

    private ExportParameters parameters;

    private FeatureStore sourceFeatureStore;

    private IProjection contextProjection;

    private ICoordTrans sourceTransformation;

    private IProjection sourceProjection;

    private Object context;

    public DefaultExportProcess() {

    } 

    @Override

    public void setOutputFormat(String serviceName) throws ExportException {

        if( this.factory!=null && StringUtils.equalsIgnoreCase(serviceName, this.factory.getName())) {

            return;

        }

        ExportServiceManager serviceManager = ExportLocator.getServiceManager();

        this.factory = serviceManager.getServiceFactory(serviceName);

        this.parameters = this.factory.createParameters();

        this.service = this.factory.createService(this.parameters);

        this.parameters.setSourceFeatureStore(this.sourceFeatureStore);

        if( this.parameters instanceof ExportParametersGeometry ) {

            ExportParametersGeometry params = (ExportParametersGeometry) this.parameters;

            params.setContextProjection(this.contextProjection);

            params.setSourceProjection(this.sourceProjection);

            params.setSourceTransformation(this.sourceTransformation);

            params.setContext(this.context);

        }

    }

    @Override

    public void setSourceFeatureStore(FeatureStore store) {

        this.sourceFeatureStore = store;

        if( this.parameters!= null ) {

            this.parameters.setSourceFeatureStore(sourceFeatureStore);

        }

    }

    @Override

    public void setContextProjection(IProjection projection) {

        this.contextProjection = projection;

        if( this.parameters instanceof ExportParametersGeometry ) {

            ((ExportParametersGeometry)this.parameters).setContextProjection(projection);

        }

    }

    @Override

    public void setSourceTransformation(ICoordTrans transformation) {

        this.sourceTransformation = transformation;

        if( this.parameters instanceof ExportParametersGeometry ) {

            ((ExportParametersGeometry)this.parameters).setSourceTransformation(transformation);

        }

    }

    @Override

    public void setSourceProjection(IProjection projection) {

        this.sourceProjection = projection;

        if( this.parameters instanceof ExportParametersGeometry ) {

            ((ExportParametersGeometry)this.parameters).setSourceProjection(projection);

        }

    }

    @Override

    public Object getContext() {

        return context;

    }

    @Override

    public void setContext(Object context) {

        this.context = context;

        if( this.parameters!= null ) {

            this.parameters.setContext(context);

        }

    }

    @Override

    public ExportParameters getParameters() {

        return this.parameters;

    }

    @Override

    public TaskStatus getTaskStatus() {

        if( this.service==null ) {

            throw new IllegalStateException("Can't access task status at this moment, need setting output format.");

        }

        return this.service.getTaskStatus();

    }

    @Override

    public void run() {

        if( this.factory == null ) {

            throw new IllegalStateException("Can't start export process, need setting output format.");

        }

        if( this.parameters == null ) {

            throw new IllegalStateException("Can't start export process, need setting parameters.");

        }

        try {

            FeatureSet featureSet;

            switch(this.parameters.getFeaturesToUse()) {

                case ExportParameters.USE_ALL_FEATURES:

                default:

                    featureSet = this.parameters.getSourceFeatureStore().getFeatureSet();

                    break;

                case ExportParameters.USE_SELECTED_FEATURES:

                    featureSet = this.parameters.getSourceFeatureStore().getFeatureSelection();

                    break;

                case ExportParameters.USE_FILTERED_FEATURES:

                    featureSet = this.parameters.getSourceFeatureStore().getFeatureSet(this.parameters.getFilterExpresion());

                    break;

            }

            this.service.export(featureSet);

        } catch (Exception ex) {

            throw new RuntimeException(ex);

        }

    }

    @Override

    public List<OpenDataStoreParameters> getTargetOpenStoreParameters() {

        if( this.service==null ) {

            return null;

        }

        try {

            return this.service.getTargetOpenStoreParameters();

        } catch (ExportException ex) {

            throw new RuntimeException("Can't create target open store parameters.", ex);

        }

    }

}