Application: gvSIG desktop

package com.iver.cit.gvsig;

import java.math.BigDecimal;

import java.sql.Types;

import java.util.BitSet;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

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

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

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.Extension;

import com.iver.andami.ui.mdiManager.IWindow;

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

import com.iver.cit.gvsig.project.documents.table.gui.Statistics;

import com.iver.cit.gvsig.project.documents.table.gui.Table;

/**

 * @author Fernando Gonz?lez Cort?s

 */

public class TableNumericFieldOperations extends Extension{

        /**

         * @see com.iver.andami.plugins.IExtension#initialize()

         */

        public void initialize() {

        }

        /**

         * @see com.iver.andami.plugins.IExtension#execute(java.lang.String)

         */

        public void execute(String actionCommand) {

                IWindow v = PluginServices.getMDIManager().getActiveWindow();

                if (v != null) {

                        if (v.getClass() == Table.class) {

                                Table table = (Table) v;

                                doExecute(table);

                        }

                }

        }

        /**

         * "execute" method acction

         * @param actionCommand

         * The acction command that executes this method

         * @param table

         * Table to operate

         */

        protected void doExecute(Table table){

                int fieldIndex = table.getSelectedFieldIndices().nextSetBit(0);

                int numRows=0;

                SelectableDataSource ds=null;

                try {

                        ds = table.getModel().getModelo().getRecordset();

                } catch (ReadDriverException e) {

                        e.printStackTrace();

                }

                BitSet selectedRows = (BitSet)ds.getSelection().clone();

                try {

                        int rc = (int) ds.getRowCount();

                        //Si no hay selecci?n se usan todos los registros

                        if (selectedRows.cardinality() == 0){

                                selectedRows.set(0, rc);

                        }else{

                                rc = selectedRows.cardinality();

                        }

                        BigDecimal suma = new BigDecimal(0);

                        BigDecimal min = new BigDecimal(Double.MAX_VALUE);

                        BigDecimal max = new BigDecimal(Double.MIN_VALUE);

                        for(int i=selectedRows.nextSetBit(0); i>=0; i=selectedRows.nextSetBit(i+1)) {

                                if (ds.getFieldValue(i, fieldIndex) instanceof NullValue)

                                        continue;

                                numRows++;

                                double d = ((NumericValue) ds.getFieldValue(i, fieldIndex)).doubleValue();

                                suma = suma.add(new BigDecimal(d));

                                if (d < min.doubleValue()) min = new BigDecimal(d);

                                if (d > max.doubleValue()) max = new BigDecimal(d);

                        }

                        Statistics st = new Statistics();

                        BigDecimal media = new BigDecimal(0);

                        BigDecimal varianza = new BigDecimal(0);

                        double desviacion = 0;

                        if (numRows==0) {

                                suma = new BigDecimal(0);

                                min = new BigDecimal(0);

                                max = new BigDecimal(0);

                        }else {

                                media = suma.divide(new BigDecimal(numRows), BigDecimal.ROUND_HALF_DOWN);

                                varianza = new BigDecimal(0);

                                for(int i=selectedRows.nextSetBit(0); i>=0; i=selectedRows.nextSetBit(i+1)) {

                                        if (ds.getFieldValue(i, fieldIndex) instanceof NullValue)

                                                continue;

                                        double d = ((NumericValue) ds.getFieldValue(i, fieldIndex)).doubleValue();

                                        BigDecimal dif = new BigDecimal(d).subtract(media);

                                        varianza = dif.multiply(dif).add(varianza);

                                }

                                varianza = varianza.divide(new BigDecimal(numRows), BigDecimal.ROUND_HALF_DOWN);

                                desviacion = Math.sqrt(varianza.doubleValue());

                        }

                        st.setStatistics(media.doubleValue(), max.doubleValue(), min.doubleValue(), varianza.doubleValue(), desviacion, numRows, new BigDecimal(max.doubleValue()).subtract(min).doubleValue(), suma.doubleValue());

                        PluginServices.getMDIManager().addWindow(st);

                } catch (ReadDriverException e) {

                        NotificationManager.addError("No se pudo acceder a los datos", e);

                }

        }

        /**

         * @see com.iver.andami.plugins.IExtension#isEnabled()

         */

        public boolean isEnabled() {

                IWindow v = PluginServices.getMDIManager().getActiveWindow();

                if (v == null) {

                        return false;

                }

                if (v instanceof Table) {

                        Table table = (Table) v;

                        return doIsEnabled(table);

                }

                return false;

        }

        protected boolean doIsEnabled(Table table){

                BitSet indices = table.getSelectedFieldIndices();

                // System.out.println("TableNumericFieldOperations.isEnabled: Tabla: " + table.getModel().getModelo().getRecordset().getName() );

                if (indices.cardinality() == 1){

                        try {

                                int index=indices.nextSetBit(0);

                                if (table.getModel().getModelo().getRecordset().getFieldCount()<index+1)

                                        return false;

                                int type = table.getModel().getModelo().getRecordset().getFieldType(index);

                                if ((type == Types.BIGINT) ||

                                                (type == Types.DECIMAL) ||

                                                (type == Types.DOUBLE) ||

                                                (type == Types.FLOAT) ||

                                                (type == Types.INTEGER) ||

                                                (type == Types.SMALLINT) ||

                                                (type == Types.TINYINT) ||

                                                (type == Types.REAL) ||

                                                (type == Types.NUMERIC)){

                                        return true;

                                }

                        } catch (ReadDriverException e) {

                                return false;

                        }

                }

                return false;

        }

        /**

         * @see com.iver.andami.plugins.IExtension#isVisible()

         */

        public boolean isVisible() {

                IWindow v = PluginServices.getMDIManager().getActiveWindow();

                if (v == null) {

                        return false;

                }

                if (v instanceof Table) {

                        return true;

                }

                return false;

        }

}