Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* This program is free software; you can redistribute it and/or

* modify it under the terms of the GNU General Public License

* as published by the Free Software Foundation; either version 2

* of the License, or (at your option) any later version.

*

* This program 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 General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

* MA  02110-1301, USA.

*

*/

/*

* AUTHORS (In addition to CIT):

* ${year} IVER T.I. S.A.   {{Task}}

*/

package org.gvsig.data.datastores.vectorial.db.sde;

import java.security.KeyException;

import java.util.ArrayList;

import java.util.Vector;

import org.gvsig.data.DataManager;

import org.gvsig.data.IDataExplorerParameters;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.INewDataStoreParameters;

import org.gvsig.data.ResourceManager;

import org.gvsig.data.datastores.vectorial.db.DBAttributeDescriptor;

import org.gvsig.data.datastores.vectorial.db.DBExplorer;

import org.gvsig.data.datastores.vectorial.db.DBFeatureType;

import org.gvsig.data.datastores.vectorial.db.DBParameters;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.DataException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.InitializeWriterException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.INewFeatureStoreParameters;

import com.esri.sde.sdk.client.SeColumnDefinition;

import com.esri.sde.sdk.client.SeConnection;

import com.esri.sde.sdk.client.SeException;

import com.esri.sde.sdk.client.SeLayer;

import com.esri.sde.sdk.client.SeTable;

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class SDEExplorer extends DBExplorer {

    /**

     * DOCUMENT ME!

     */

    public static String DATAEXPLORER_NAME = "SDEExplorer";

    private SDEExplorerParameters parameters;

    private String defaultSchema;

    private SDEResource resource;

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws InitializeException DOCUMENT ME!

     */

    protected SeConnection getConnection() throws ReadException {

        return this.resource.getConnection();//SDEUtils.getConnection(host, port, db, dbuser, dbpass);

    }

    /**

     * DOCUMENT ME!

     *

     * @param ndsp DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws InitializeException DOCUMENT ME!

     * @throws InitializeWriterException DOCUMENT ME!

     */

    public IDataStoreParameters add(INewFeatureStoreParameters ndsp)

        throws InitializeException, InitializeWriterException {

        SeConnection conn;

                try {

                        conn = this.getConnection();

                } catch (ReadException e) {

                        throw new InitializeException(DATAEXPLORER_NAME,e);

                }

        IDataStoreParameters params = this.add(ndsp, conn);

        return params;

    }

    /**

     * DOCUMENT ME!

     *

     * @param ndsp DOCUMENT ME!

     * @param conn DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws InitializeException DOCUMENT ME!

     * @throws InitializeWriterException DOCUMENT ME!

     */

    protected IDataStoreParameters add(INewFeatureStoreParameters ndsp,

        SeConnection conn)

        throws InitializeException, InitializeWriterException {

            try {

            if (!ndsp.isValid()){

                        throw new InitializeException(this.getName(),new Exception("Parameters not valid"));

                }

                SDEStoreParameters sdeParam = (SDEStoreParameters)ndsp.getDataStoreParameters();

                DBFeatureType fType = (DBFeatureType)ndsp.getFeatureType();

                SeTable table = new SeTable(conn,sdeParam.tableID());

                DBAttributeDescriptor[] fads=(DBAttributeDescriptor[])fType.toArray(new DBAttributeDescriptor[0]);

                SeColumnDefinition[] columns=new SeColumnDefinition[fads.length];

                for (int j = 0; j < fads.length; j++) {

                        SeColumnDefinition column=new SeColumnDefinition(fads[j].getName(),SDETypes.fieldTypeToInt(fads[j].getDataType()),fads[j].getSize(),fads[j].getPrecision(),fads[j].isAllowNull());

                        columns[j]=column;

                }

                        table.create(columns);

                        return sdeParam;

            } catch (SeException e) {

                        throw new InitializeWriterException(DATAEXPLORER_NAME,e);

                }

          }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public INewDataStoreParameters createNewDataStoreParameter() {

        return new SDENewStoreParameter((IDataStoreParameters) this.newStoreParamters());

    }

    /**

     * DOCUMENT ME!

     *

     * @param dsp DOCUMENT ME!

     *

     * @throws ReadException DOCUMENT ME!

     */

    public void remove(IDataStoreParameters dsp) throws ReadException {

        SeConnection conn = this.getConnection();

        this.remove(dsp, conn);

    }

    /**

     * DOCUMENT ME!

     *

     * @param dsp DOCUMENT ME!

     * @param conn DOCUMENT ME!

     *

     * @throws ReadException DOCUMENT ME!

     */

    protected void remove(IDataStoreParameters dsp, SeConnection conn)

        throws ReadException {

            try {

                SeTable table = new SeTable(conn,((SDEStoreParameters)dsp).tableID());

                table.delete();

        } catch (SeException e) {

                        throw new ReadException(this.getClass().getName(),e);

                }

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean canCreate() {

        return true;

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public String getName() {

        return DATAEXPLORER_NAME;

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public String getDataStoreName() {

        return SDEStore.DATASTORE_NAME;

    }

    /**

     * DOCUMENT ME!

     *

     * @param parameters DOCUMENT ME!

     *

     * @throws InitializeException DOCUMENT ME!

     */

    public void init(IDataExplorerParameters parameters)

        throws InitializeException {

            SDEResource tmpResource = new SDEResource((SDEExplorerParameters)parameters);

                SDEResource theResource;

                ResourceManager resMan = ResourceManager.getResourceManager();

                try {

                        theResource = (SDEResource)resMan.addResource(tmpResource);

                } catch (DataException e1) {

                        throw new InitializeException(this.getName(),e1);

                }

            this.parameters = (SDEExplorerParameters) parameters;

            this.resource=theResource;

     }

    private SDEStoreParameters newStoreParamters() {

        DataManager manager = DataManager.getManager();

        SDEStoreParameters param = (SDEStoreParameters) manager.createDataStoreParameters(getDataStoreName());

        this.parameters.fillStoreParameters((DBParameters) param);

        return param;

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws ReadException DOCUMENT ME!

     */

    public IDataStoreParameters[] list() throws ReadException {

        return this.list(this.parameters.isShowInformationDBTables());

    }

    /**

     * DOCUMENT ME!

     *

     * @param showInformationDBTables DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws ReadException DOCUMENT ME!

     */

    public IDataStoreParameters[] list(boolean showInformationDBTables)

        throws ReadException {

        SeConnection conn = this.getConnection();

        ArrayList paramList = new ArrayList();

        SDEStoreParameters sdeparam = null;

        try {

            Vector theLayers = conn.getLayers();

            for (int i = 0; i < theLayers.size(); i++) {

                sdeparam = this.newStoreParamters();

                SeLayer layer = (SeLayer) theLayers.elementAt(i);

                sdeparam.setTableName(layer.getTableName());

                sdeparam.setFields(new String[] { "*" });

                sdeparam.setGeometryField(layer.getSpatialColumn());

                paramList.add(sdeparam);

            }

            return (IDataStoreParameters[]) paramList.toArray(new IDataStoreParameters[0]);

        } catch (SeException e) {

            throw new ReadException(this.getClass().getName(), e);

        }

    }

    /**

     * DOCUMENT ME!

     *

     * @param dsp DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     *

     * @throws ReadException DOCUMENT ME!

     */

    public IFeatureType[] getFeatureTypes(IDataStoreParameters dsp)

        throws ReadException {

        return new IFeatureType[] {

            SDEUtils.getFeatureType(this.getConnection(),

                (SDEStoreParameters) dsp)

        };

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public String getDefaultSchema() {

        return this.defaultSchema;

    }

        public void dispose() throws DataException {

                ResourceManager resMan = ResourceManager.getResourceManager();

            try {

                        resMan.remove(this.resource);

                } catch (DataException e1) {

                        throw new CloseException(this.getName(),e1);

                } catch (KeyException e) {

                        throw new CloseException(this.getName(),e);

                }

        }

}