Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2006 Prodevelop and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *   Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *   +34 963862235

 *   gvsig@gva.es

 *   www.gvsig.gva.es

 *

 *    or

 *

 *   Prodevelop Integraci?n de Tecnolog?as SL

 *   Conde Salvatierra de ?lava , 34-10

 *   46004 Valencia

 *   Spain

 *

 *   +34 963 510 612

 *   +34 963 510 968

 *   gis@prodevelop.es

 *   http://www.prodevelop.es

 */

package es.prodevelop.cit.gvsig.arcims.fmap.drivers;

import com.hardcode.driverManager.Driver;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.engine.data.DataSource;

import com.hardcode.gdbms.engine.data.DataSourceFactory;

import com.hardcode.gdbms.engine.data.IDataSourceListener;

import com.hardcode.gdbms.engine.data.SourceInfo;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

import com.hardcode.gdbms.engine.data.edition.DataWare;

import com.hardcode.gdbms.engine.data.object.ObjectSourceInfo;

import com.hardcode.gdbms.engine.data.persistence.DataSourceLayerMemento;

import com.hardcode.gdbms.engine.data.persistence.Memento;

import com.hardcode.gdbms.engine.data.persistence.MementoException;

import com.hardcode.gdbms.engine.instruction.EvaluationException;

import com.hardcode.gdbms.engine.instruction.SemanticException;

import com.hardcode.gdbms.engine.values.IntValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueCollection;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.hardcode.gdbms.parser.ParseException;

import com.iver.cit.gvsig.fmap.layers.FBitSet;

import es.prodevelop.cit.gvsig.arcims.gui.wizards.ArcImsWizard;

import org.apache.log4j.Logger;

import org.gvsig.remoteClient.arcims.exceptions.ArcImsException;

import org.gvsig.remoteClient.arcims.utils.FieldInformation;

import org.gvsig.remoteClient.arcims.utils.GetFeaturesTags;

import java.io.File;

import java.io.IOException;

import java.sql.SQLException;

import java.sql.Types;

import java.util.ArrayList;

import java.util.BitSet;

import java.util.HashMap;

public class ArcImsAttributesDataSourceAdapter

    implements ArcImsAlphanumericDataSource, ObjectDriver {

    private static Logger logger = Logger.getLogger(ArcImsAttributesDataSourceAdapter.class.getName());

    private ArcImsInMemoryAttsTableDriver dataDriver;

    // private File csvFile;

    private String tableName;

    private DataSourceFactory dsFactory = null;

    private ObjectSourceInfo sourceInfo;

    private FMapFeatureArcImsDriver parentDriver;

    private int rowsPerRequest = 10;

    // private int privateKeyIndex = 0;

    // array of Value[]

    // private AttributesDataa attsData; 

    public ArcImsAttributesDataSourceAdapter(ArrayList col_Names,

        ArrayList col_Types, Value[] ids, DataSourceFactory dsf,

        FMapFeatureArcImsDriver drv, String forcedName) {

        // attsData = new AttributesDataa(col_Names, col_Types, ids, drv);

        parentDriver = drv;

        dsFactory = dsf;

        if (forcedName == null) {

            long timeid = System.currentTimeMillis();

            tableName = "table_" + timeid;

        }

        else {

            tableName = forcedName;

        }

        dataDriver = new ArcImsInMemoryAttsTableDriver(tableName, col_Names,

                col_Types);

        sourceInfo = new ObjectSourceInfo();

        sourceInfo.driver = this;

        sourceInfo.name = tableName;

        sourceInfo.driverName = dataDriver.getName();

        //                dsFactory = new DataSourceFactory();

        //                dsFactory.initialize();

        //                dsFactory.addDataSource(this, tableName);

        try {

            // dataDriver.open(dataDriver.getConnection(), tableName);

            //                        String sqlStr = ArcImsSqlUtils.getSqlSentenceCreateTableWithFieldsAndTypes(tableName, col_Names, col_Types, dataDriver.getRowIndexFieldName());

            //                        dataDriver.open(sqlStr);

            String idcolname = getIdColumnName(col_Names, col_Types);

            for (int i = 0; i < ids.length; i++) {

                Value id = ids[i];

                int rwcount = (int) dataDriver.getRowCount();

                //                                sqlStr = ArcImsSqlUtils.getSqlSentenceSetEmptyRowWithId(

                //                                                tableName, id, idcolname,

                //                                                rwcount,

                //                                                dataDriver.getRowIndexFieldName());

                dataDriver.addEmptyRowWithId(id, idcolname);

                // dataDriver.addInIndexToIdHashMap(rwcount, ((IntValue) id).intValue());

                // logger.debug("Adding id = " + ((IntValue) id).intValue() + " in row = " + rwcount);

                dataDriver.addInIdToIndexHashMap(((IntValue) id).intValue(),

                    rwcount);

                // dsFactory.executeSQL(sqlStr, DataSourceFactory.DATA_WARE_DIRECT_MODE);

                //                                dataDriver.execute(sqlStr);

                //                                dataDriver.increaseRowCount(false);

            }

            logger.info("Stored " + ids.length + " IDs");

            setRowsPerRequest();

        }

        catch (Exception e) {

            logger.error("While creating in-memory database ", e);

        }

    }

    public void deleteRow(long rowInd) throws DriverException {

        // DriverException de = new DriverException("cannot_delete");

        logger.warn("So far, you cannot delete ");

    }

    public void insertFilledRowAt(long index, Value[] values)

        throws DriverException {

        insertFilledRowAt(index, values, true);

    }

    public void insertFilledRowAt(long index, Value[] values, boolean req)

        throws DriverException {

        // dataDriver.setRequested((int) dataDriver.getRowCount(), req);

        // start();

        dataDriver.addRow(values, req);

        // stop();

        logger.warn("Row was added at the end (not inserted at position " +

            index + ")");

        // this.

        // dataDriver.addRow(dsFactory, values);

        // dataDriver.insertRow((int) index, values);

    }

    public void insertEmptyRowAt(long index) throws DriverException {

        // start();

        int count = dataDriver.getFieldCount();

        // stop();

        Value[] empty = new Value[count];

        for (int i = 0; i < count; i++) {

            empty[i] = ValueFactory.createNullValue();

        }

        insertFilledRowAt(index, empty, false);

    }

    public void start() throws DriverException {

    }

    public void stop() throws DriverException {

    }

    public String getName() {

        return tableName;

    }

    public String getAlias() {

        return tableName;

        // return tableAlias;

    }

    public long[] getWhereFilter() throws IOException {

        // dataDriver.

        return null;

        //                String message = "'Where' filter not used in method: ArcImsAttributesDataSourceAdapter.getWhereFilter()";

        //                IOException ioe = new IOException(message);

        //                throw ioe;

    }

    public DataSourceFactory getDataSourceFactory() {

        return dsFactory;

    }

    public Memento getMemento() throws MementoException {

        DataSourceLayerMemento m = new DataSourceLayerMemento(tableName,

                tableName);

        return m;

    }

    public void setDataSourceFactory(DataSourceFactory arg0) {

        dsFactory = arg0;

    }

    public void setSourceInfo(SourceInfo arg0) {

        sourceInfo = (ObjectSourceInfo) arg0;

    }

    public SourceInfo getSourceInfo() {

        return sourceInfo;

    }

    public String getAsString() throws DriverException {

        //                return attsData.getAsString();

        return dataDriver.toString();

    }

    public void remove() throws DriverException {

        if (dsFactory == null) {

            logger.error(

                "dsFactory is null (?) and cannot receive the remove method ");

            return;

        }

        try {

            dsFactory.remove(this);

        }

        catch (DriverException e) {

            String message = "Driver Exception in method: " +

                "ArcImsAttributesDataSourceAdapter.remove()\n" +

                e.getMessage();

            DriverException de = new DriverException(message);

            throw de;

        }

    }

    public int[] getPrimaryKeys() throws DriverException {

        int[] resp = new int[1];

        resp[0] = dataDriver.getIdFieldInd();

        return resp;

    }

    public ValueCollection getPKValue(long arg0) throws DriverException {

        ValueCollection vc = new ValueCollection();

        vc.add(ValueFactory.createValue(getRowId((int) arg0)));

        return vc;

    }

    public String getPKName(int arg0) throws DriverException {

        if (arg0 == 0) {

            return dataDriver.getIdFieldName();

        }

        else {

            DriverException de = new DriverException("pk_error");

            logger.error("Multiple primary key are never used. ", de);

            throw de;

        }

    }

    public String[] getPKNames() throws DriverException {

        String[] resp = new String[1];

        resp[0] = dataDriver.getIdFieldName();

        return resp;

    }

    public int getPKType(int arg0) throws DriverException {

        if (arg0 == 0) {

            //                        return attsData.getFieldType(attsData.getIdIndex());

            return Types.BIGINT;

        }

        else {

            DriverException de = new DriverException("pk_error");

            logger.error("Multiple primary key are never used. ", de);

            throw de;

        }

    }

    public int getPKCardinality() throws DriverException {

        // Multiple primary key are never used.

        return 1;

    }

    public Value[] getRowWithoutDownloading(long row) throws DriverException {

        // start();

        Value[] resp = dataDriver.getRow(dsFactory, (int) row);

        // stop();

        return resp;

    }

    public Value[] getRow(long arg0) throws DriverException {

        //                Value[] resp = attsData.getRow((int) arg0);

        //                return resp;

        // return null;

        FBitSet fbs = dataDriver.getThisAndNonRequestedUpTo((int) arg0,

                rowsPerRequest);

        try {

            parentDriver.requestFeatureAttributesWithoutChecking(fbs);

        }

        catch (ArcImsException e) {

            DriverException de = new DriverException(e.getMessage());

            logger.error("While getting row ", de);

            throw de;

        }

        // start();

        Value[] resp = dataDriver.getRow(dsFactory, (int) arg0);

        // stop();

        return resp;

    }

    public String[] getFieldNames() throws DriverException {

        //                String[] resp = attsData.getFieldNames();

        //                return resp;

        // return null;

        // start();

        int count = dataDriver.getFieldCount();

        String[] resp = new String[count];

        for (int i = 0; i < count; i++) {

            // resp[i] = dataDriver.getFieldName(i);

            resp[i] = dataDriver.getSqlFieldName(i);

        }

        // stop();

        return resp;

        // dataDriver.ge

    }

    public int getFieldIndexByName(String arg0) throws DriverException {

        //                int resp = attsData.getFieldIndexByName(arg0);

        //                return resp;

        // start();

        int resp = dataDriver.getFieldIndexByName(arg0);

        // stop();

        return resp;

        // return 0;

    }

    public DataWare getDataWare(int arg0) {

        //                DataWare resp = attsData.getDataWare(arg0);

        //                return resp;

        return null;

    }

    public Value getFieldValue(long row, int colind) throws DriverException {

        if (dataDriver.isNonRequestedRow((int) row)) {

            int[] lim = getBlockLimits((int) row);

            logger.debug("Se ha pedido fila no disponible: " + row);

            logger.debug("Limites: [ " + lim[0] + " , " + lim[1] + " ]");

            // FBitSet fbs = dataDriver.getThisAndNonRequestedUpTo((int) row, rowsPerRequest);

            try {

                parentDriver.requestBlockWithoutChecking(lim);

            }

            catch (ArcImsException e) {

                DriverException de = new DriverException(e.getMessage());

                logger.error("While getting row ", de);

                throw de;

            }

        }

        //                Value v = dataDriver.getCachedValue((int) row, colind);

        //                if (v != null) {

        //                        return v;

        //                }

        Value resp = dataDriver.getFieldValue(row, colind);

        //                stop();

        //                dataDriver.setCachedValue((int) row, colind, resp);

        return resp;

    }

    public int getFieldCount() throws DriverException {

        //                int resp = attsData.getFieldCount();

        //                return resp;

        // start();

        int resp = dataDriver.getFieldCount();

        // stop();

        return resp;

        // return 0;

    }

    public String getFieldName(int arg0) throws DriverException {

        //                String resp = attsData.getFieldName(arg0);

        //                return resp;

        // start();

        String resp = dataDriver.getSqlFieldName(arg0);

        // stop();

        return resp;

        // return "";

    }

    public long getRowCount() throws DriverException {

        //                long resp = (int) attsData.getRowCount();

        //                return resp;

        // start();

        long resp = (long) dataDriver.getRowCount();

        // stop();

        return resp;

        // return 0;

    }

    public int getFieldType(int arg0) throws DriverException {

        //                int resp = attsData.getFieldType(arg0);

        //                return resp;

        // start();

        int resp = dataDriver.getFieldType(arg0);

        // This is done because gvSIG doesnt know about

        // FieldInformation.ID and FieldInformation.SHAPE

        if (resp == FieldInformation.ID) {

            resp = FieldInformation.INTEGER;

        }

        if (resp == FieldInformation.SHAPE) {

            resp = FieldInformation.STRING;

        }

        // stop();

        return resp;

        // return 0;

    }

    // **************************

    // new_row[0] must contain ID

    // **************************

    // returns added row index 

    public void updateRow(Value[] new_row, String[] fld_query, int rowind)

        throws DriverException {

        // dataDriver.

        //                start();

        // int resp =

        dataDriver.updateRow(new_row, fld_query, rowind);

    }

    public String getIdColName() throws DriverException {

        String resp = dataDriver.getIdFieldName();

        return resp;

        //                return attsData.getIdColName();

        // return "";

    }

    public int getIdIndex() throws DriverException {

        int resp = dataDriver.getIdFieldInd();

        return resp;

        //                return attsData.getIdIndex();

        // return 0;

    }

    public int getRowId(int row) throws DriverException {

        //                start();

        //                int resp = dataDriver.getRowId(dsFactory, row);

        //                stop();

        int resp = dataDriver.getRowId(row);

        return resp;

    }

    public int getRowIndex(int id) throws DriverException {

        //                start();

        //                int resp = dataDriver.getRowIndex(dsFactory, id);

        //                stop();

        int resp = dataDriver.getRowIndex(id);

        return resp;

    }

    public boolean isNonRequestedRow(int rowind) {

        //                return attsData.isNonRequestedRow(rowind);

        return dataDriver.isNonRequestedRow(rowind);

    }

    public void write(DataWare arg0) throws DriverException {

        // TODO Auto-generated method stub

    }

    public String getTableName() {

        return tableName;

    }

    //        public String getOriginalColumnName(int idcolindex) {

    //                return dataDriver.getOriginalColumnName(idcolindex);

    //        }

    public void addAsRequested(FBitSet fbs) {

        dataDriver.addAsRequested(fbs);

    }

    public BitSet getNonRequestedFromHere(int start, int total) {

        return dataDriver.getThisAndNonRequestedUpTo(start, total);

    }

    public String getIdColumnName(ArrayList col_Names, ArrayList col_Types) {

        int coltype;

        int idindex = -1;

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

            coltype = ((Integer) col_Types.get(i)).intValue();

            if (coltype == FieldInformation.ID) {

                idindex = i;

                break;

            }

        }

        if (idindex == -1) {

            logger.error("ID not found ");

        }

        return (String) col_Names.get(idindex);

    }

    public int getRowsPerRequest() {

        return rowsPerRequest;

    }

    private void setRowsPerRequest() {

        int rowcount = (int) dataDriver.getRowCount();

        int sqroot = Math.round((float) Math.sqrt(1.0 * rowcount));

        rowsPerRequest = sqroot;

        if (rowsPerRequest > GetFeaturesTags.MAX_ROWS_PER_REQUEST) {

            rowsPerRequest = GetFeaturesTags.MAX_ROWS_PER_REQUEST;

        }

        if (rowsPerRequest < GetFeaturesTags.MIN_ROWS_PER_REQUEST) {

            rowsPerRequest = GetFeaturesTags.MIN_ROWS_PER_REQUEST;

        }

        if (rowsPerRequest > rowcount) {

            rowsPerRequest = rowcount;

        }

        logger.info("Rows per request was set to " + rowsPerRequest +

            " (rowcount = " + rowcount + ")");

    }

    public String getOriginalFieldName(int idcolindex) {

        return dataDriver.getOriginalFieldName(idcolindex);

    }

    public boolean isVirtualField(int fieldId) throws DriverException {

        // TODO Auto-generated method stub

        return false;

    }

    public int getFieldWidth(int i) throws DriverException {

        // TODO Auto-generated method stub

        return 10;

    }

    private int[] getBlockLimits(int row) {

        int[] resp = new int[2];

        int rcount = 0;

        try {

            rcount = (int) getRowCount();

        }

        catch (DriverException e) {

            logger.error("While getting rowcount ", e);

        }

        int rperreq = getRowsPerRequest();

        int bloque = (row / rperreq);

        int inicio = rperreq * bloque;

        int fin = (inicio + rperreq) - 1;

        if (fin > (rcount - 1)) {

            fin = (rcount - 1);

        }

        resp[0] = inicio;

        resp[1] = fin;

        return resp;

    }

    public Driver getDriver() {

        // TODO Auto-generated method stub

        return null;

    }

    public void reload() throws DriverException, IOException {

        // TODO Auto-generated method stub

    }

    public void addDataSourceListener(IDataSourceListener arg0) {

        // TODO Auto-generated method stub

    }

    public void removeDataSourceListener(IDataSourceListener arg0) {

        // TODO Auto-generated method stub

    }

}