Application: gvSIG desktop

package com.iver.cit.gvsig;

import java.io.IOException;

import java.sql.Types;

import java.util.BitSet;

import com.hardcode.driverManager.DriverLoadException;

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

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.drivers.DriverIOException;

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

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

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

/**

 * DOCUMENT ME!

 *

 * @author Vicente Caballero Navarro

 */

public class TableRowsOperations extends Extension {

    /**

     * DOCUMENT ME!

     */

    public void initialize() {

    }

    /**

     * DOCUMENT ME!

     *

     * @param actionCommand DOCUMENT ME!

     */

    public void execute(String actionCommand) {

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

        if (v != null) {

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

                Table table = (Table) v;

                if (actionCommand.compareTo("SELECTIONUP") == 0)

                        showsSelectedRows(table);

                if (actionCommand.compareTo("INVERTSELECTION") == 0)

                        invertSelection(table);

                table.getModel().setModified(true);

            }

        }

    }

    /**

     * Flip the selection (inverts selection)

     * @param table

     */

    private void invertSelection(Table table) {

            SelectableDataSource sds = table.getModel().getModelo().getRecordset();

            FBitSet selectedRows=sds.getSelection();

            try {

                        selectedRows.flip(0, (int)sds.getRowCount());

                } catch (DriverException e) {

                        e.printStackTrace();

                        NotificationManager.addError(e);

                }

            sds.setSelection(selectedRows);

        }

        private void showsSelectedRows(Table table) {

            long[] mapping=null;

                try {

                        mapping = new long[table.getModel().getModelo().getRowCount()];

                FBitSet selectedRows=table.getModel().getModelo().getSelection();

                int m=0;

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

                 i = selectedRows.nextSetBit(i + 1)) {

                        mapping[m]=i;

                        m++;

                }

                for (int i = 0; i < mapping.length;

                     i++) {

                        if (!selectedRows.get(i)) {

                                mapping[m]=i;

                                m++;

                        }

                }

                table.setOrder(mapping);

                } catch (DriverIOException e) {

                        e.printStackTrace();

                } catch (IOException e) {

                        e.printStackTrace();

                }

        }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean isEnabled() {

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

        if (v == null) {

            return false;

        }

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

            Table table = (Table) v;

            return table.getModel().getModelo().getSelection().cardinality()>0;

        }

        return false;

    }

    protected boolean doIsEnabled(Table table) {

        try {

            BitSet indices = table.getSelectedFieldIndices();

            System.out.println("TableNumericFieldOperations.isEnabled: Tabla: " +

                table.getModel().getModelo().getRecordset().getName());

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

                try {

                    int type = table.getModel().getModelo().getRecordset()

                                    .getFieldType(indices.nextSetBit(0));

                    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 (DriverException e) {

                    return false;

                }

            }

        } catch (DriverLoadException e1) {

            // TODO Auto-generated catch block

            e1.printStackTrace();

        }

        return false;

    }

    /**

     * DOCUMENT ME!

     *

     * @return DOCUMENT ME!

     */

    public boolean isVisible() {

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

        if (v == null) {

            return false;

        }

        if (v instanceof Table) {

            return true;

        }

        return false;

    }

}