| gvSIG desktop 1

/* 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.

 *

 */

package org.gvsig.app.project.documents.layout.fframes;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.event.MouseEvent;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

/**

 * DOCUMENT ME!

 * 

 * @author Vicente Caballero Navarro

 */

public class FFrameTable extends FFrameGroup implements IFFrameGroupSelectable {

    public static final String PERSISTENCE_DEFINITION_NAME = "FFrameTable";

    private static final String NUMCOLUMNS_FIELD = "numColumns";

    private static final String NUMROWS_FIELD = "numRows";

    private int numColumns = 2;

    private int numRows = 2;

    private static AffineTransform identity = new AffineTransform();

    private boolean selectFFrameBasic = false;

    private static Image iMove = null;

    /**

     * DOCUMENT ME!

     * 

     * @param g

     *            DOCUMENT ME!

     * @param at

     *            DOCUMENT ME!

     * @param rv

     *            DOCUMENT ME!

     * @param imgBase

     *            DOCUMENT ME!

     */

    public void draw(Graphics2D g, AffineTransform at, Rectangle2D rv,

        BufferedImage imgBase) {

        Rectangle2D.Double r = getBoundingBox(at);

        g.setColor(Color.black);

        g.rotate(Math.toRadians(getRotation()), r.x + (r.width / 2), r.y

            + (r.height / 2));

        drawRectangles(g, at, rv, imgBase);

        g.rotate(Math.toRadians(-getRotation()), r.x + (r.width / 2), r.y

            + (r.height / 2));

        super.draw(g, at, rv, imgBase);

    }

    private void drawRectangles(Graphics2D g, AffineTransform at,

        Rectangle2D rv, BufferedImage imgBase) {

        IFFrame[] fframes = getFFrames();

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

            FFrameBasic basic = (FFrameBasic) fframes[i];

            basic.draw(g, at, rv, imgBase);

        }

    }

    /**

     * DOCUMENT ME!

     * 

     * @param r

     *            DOCUMENT ME!

     * @param g

     *            DOCUMENT ME!

     */

    public void drawBox(Rectangle2D r, Graphics2D g) {

        calculateTable(r);

        IFFrame[] fframes = getFFrames();

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

            FFrameBasic basic = (FFrameBasic) fframes[i];

            basic.draw(g, identity, null, null);

        }

    }

    /**

     * DOCUMENT ME!

     * 

     * @param g

     *            DOCUMENT ME!

     * @param at

     *            DOCUMENT ME!

     * 

     * @throws ReadDriverException

     */

    public void print(Graphics2D g, AffineTransform at) throws ReadException {

        draw(g, at, null, null);

    }

    /**

     * DOCUMENT ME!

     * 

     * @return DOCUMENT ME!

     */

    public String getNameFFrame() {

        return PluginServices.getText(this, "box") + num;

    }

    public String getName() {

        return PERSISTENCE_DEFINITION_NAME;

    }

    /**

     * DOCUMENT ME!

     * 

     * @return DOCUMENT ME!

     */

    public int getNumColumns() {

        return numColumns;

    }

    /**

     * DOCUMENT ME!

     * 

     * @param numColumns

     *            DOCUMENT ME!

     */

    public void setNumColumns(int numColumns) {

        this.numColumns = numColumns;

    }

    /**

     * DOCUMENT ME!

     * 

     * @return DOCUMENT ME!

     */

    public int getNumRows() {

        return numRows;

    }

    /**

     * DOCUMENT ME!

     * 

     * @param numRows

     *            DOCUMENT ME!

     */

    public void setNumRows(int numRows) {

        this.numRows = numRows;

    }

    public void calculateTable(Rectangle2D r) {

        double wC = r.getWidth() / numColumns;

        double hR = r.getHeight() / numRows;

        int rows = numRows;

        clearFFrames();

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

            for (int j = 0; j < rows; j++) {

                double x = r.getX() + (wC * i);

                double y = r.getY() + (hR * j);

                double w = wC;

                double h = hR;

                Rectangle2D rBasic = new Rectangle2D.Double(x, y, w, h);

                FFrameBasic basic =

                    (FFrameBasic) layoutManager

                        .createFrame(FFrameBasic.PERSISTENCE_DEFINITION_NAME);

                basic.setRectangle(rBasic);

                addFFrame(basic);

            }

        }

    }

    public void selectFFrame(boolean b) {

        selectFFrameBasic = b;

    }

    public IFFrame joinFFrame() {

        IFFrame[] fframes = this.getFFrames();

        Rectangle2D r = null;

        for (int i = fframes.length - 1; i >= 0; i--) {

            if (fframes[i].getSelected() != IFFrame.NOSELECT) {

                if (r == null) {

                    r = (Rectangle2D) fframes[i].getBoundBox().clone();

                } else {

                    r.add(fframes[i].getBoundBox());

                }

                this.removeFFrame(i);

            }

        }

        if (r != null) {

            // Layout

            FFrameBasic basic =

                (FFrameBasic) layoutManager

                    .createFrame(FFrameBasic.PERSISTENCE_DEFINITION_NAME);

            basic.setRectangle(r);

            this.addFFrame(basic);

            return this;

        }

        return null;

    }

    public void setSelected(Point2D p, MouseEvent e) {

        if (!selectFFrameBasic) {

            super.setSelected(p, e);

        } else {

            setSelectedGroup(p, e);

        }

    }

    public boolean contains(Point2D p) {

        if (!selectFFrameBasic) {

            return super.contains(p);

        }

        return contains(p);

    }

    public void drawHandlers(Graphics2D g) {

        if (!selectFFrameBasic) {

            super.drawHandlers(g);

        } else {

            drawHandlersGroup(g);

        }

    }

    public int getContains(Point2D p) {

        if (!selectFFrameBasic) {

            return super.getContains(p);

        }

        return getContainsGroup(p);

    }

    public Rectangle2D getMovieRect(int difx, int dify) {

        // TODO Esto de momento lo dejo que no se pueda mover

        // cuando se tiene seleccionada la herramienta de selecci�n de un

        // FFrameBasic.

        Rectangle2D r = super.getMovieRect(difx, dify);

        if (!selectFFrameBasic) {

            return r;

        }

        r = super.getMovieRect(0, 0);

        return r;

    }

    public Rectangle2D getMovieRectGroup(int difX, int difY) {

        Rectangle2D r = null;

        IFFrame[] fframes = getFFrames();

        List<IFFrame> selected = new ArrayList<IFFrame>();

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

            IFFrame fframe = fframes[i];

            if (fframe.getSelected() != IFFrame.NOSELECT) {

                selected.add(fframe);

            }

        }

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

            if (i == 0) {

                r = selected.get(i).getMovieRect(difX, difY);

            } else {

                r.add(selected.get(i).getMovieRect(difX, difY));

            }

        }

        return r;

    }

    public int getContainsGroup(Point2D p) {

        List<IFFrame> selected = new ArrayList<IFFrame>();

        IFFrame[] fframes = getFFrames();

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

            IFFrame fframe = fframes[i];

            int contains = fframe.getContains(p);

            if (contains != IFFrame.NOSELECT) {

                if (contains == IFFrame.RECT) {

                    selected.add(fframe);

                } else {

                    return contains;

                }

            }

        }

        if (selected.size() > 0) {

            return selected.get(0).getContains(p);

        }

        return 0;

    }

    /**

     * Dibuja los handlers sobre el boundingBox en el graphics que se pasa como

     * par�metro.

     * 

     * @param g

     *            Graphics sobre el que dibujar.

     */

    public void drawHandlersGroup(Graphics2D g) {

        int size = 8;

        g.setColor(Color.gray);

        IFFrame[] fframes = getFFrames();

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

            IFFrame fframe = fframes[i];

            if (fframe.getSelected() != IFFrame.NOSELECT) {

                Rectangle2D r = fframe.getBoundingBox(null);

                Point2D p = new Point2D.Double();

                g.rotate(Math.toRadians(getRotation()),

                    r.getX() + (r.getWidth() / 2), r.getY()

                        + (r.getHeight() / 2));

                AffineTransform atRotate = new AffineTransform();

                atRotate.rotate(Math.toRadians(getRotation()),

                    r.getX() + (r.getWidth() / 2), r.getY()

                        + (r.getHeight() / 2));

                g.fillRect((int) r.getX() - size / 2,

                    (int) r.getY() - size / 2, size, size);

                atRotate

                    .transform(new Point2D.Double(r.getX() - size / 2, r.getY()

                        - size / 2), p);

                no.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getMaxX() - size / 2, (int) r.getY() - size

                    / 2, size, size);

                atRotate.transform(

                    new Point2D.Double(r.getMaxX() - size / 2, r.getY() - size

                        / 2), p);

                ne.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getX() - size / 2, (int) r.getMaxY() - size

                    / 2, size, size);

                atRotate.transform(

                    new Point2D.Double(r.getX() - size / 2, r.getMaxY() - size

                        / 2), p);

                so.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getMaxX() - size / 2, (int) r.getMaxY()

                    - size / 2, size, size);

                atRotate.transform(

                    new Point2D.Double(r.getMaxX() - size / 2, r.getMaxY()

                        - size / 2), p);

                se.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getCenterX() - (size / 2), (int) r.getY()

                    - size / 2, size, size);

                atRotate.transform(new Point2D.Double(r.getCenterX()

                    - (size / 2), r.getY() - size / 2), p);

                n.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getCenterX() - (size / 2), (int) r.getMaxY()

                    - size / 2, size, size);

                atRotate.transform(new Point2D.Double(r.getCenterX()

                    - (size / 2), r.getMaxY() - size / 2), p);

                s.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getX() - size / 2, (int) r.getCenterY()

                    - (size / 2), size, size);

                atRotate.transform(

                    new Point2D.Double(r.getX() - size / 2, r.getCenterY()

                        - (size / 2)), p);

                o.setRect((int) p.getX(), (int) p.getY(), size, size);

                g.fillRect((int) r.getMaxX() - size / 2, (int) r.getCenterY()

                    - (size / 2), size, size);

                atRotate.transform(

                    new Point2D.Double(r.getMaxX() - size / 2, r.getCenterY()

                        - (size / 2)), p);

                e.setRect((int) p.getX() - size / 2, (int) p.getY() - size / 2,

                    size, size);

                g.rotate(Math.toRadians(-getRotation()),

                    r.getX() + (r.getWidth() / 2), r.getY()

                        + (r.getHeight() / 2));

            }

        }

    }

    public boolean containsGroup(Point2D p) {

        IFFrame[] fframes = getFFrames();

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

            IFFrame fframe = fframes[i];

            if (fframe.contains(p)) {

                return true;

            }

        }

        return false;

    }

    public void setSelectedGroup(Point2D p, MouseEvent e) {

        IFFrame[] fframes = getFFrames();

        if (!e.isShiftDown()) {

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

                fframes[j].setSelected(false);

            }

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

                IFFrame fframe = fframes[i];

                fframe.setSelected(p, e);

            }

        } else {

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

                IFFrame fframe = fframes[i];

                if (fframe.contains(p)) {

                    if ((fframe.getSelected() == IFFrame.NOSELECT)) {

                        fframe.setSelected(p, e);

                    } else {

                        fframe.setSelected(false);

                    }

                }

            }

        }

    }

    public int getSelectedGroup() {

        return 0;

    }

    public Image getMapCursor(Point2D p) {

        if (!selectFFrameBasic) {

            return super.getMapCursor(p);

        }

        return getMapCursorGroup(p);

    }

    public Image getMapCursorGroup(Point2D p) {

        int select = getContains(p);

        switch (select) {

        case (RECT):

            return iMove;

        }

        return null;

    }

    /**

     * Initilizes the static icons

     */

    public static void initializeIcons() {

        iMove = PluginServices.getIconTheme().get("move-icon").getImage();

    }

    public static void registerPersistent() {

        PersistenceManager manager = ToolsLocator.getPersistenceManager();

        if (manager.getDefinition(PERSISTENCE_DEFINITION_NAME) == null) {

            DynStruct definition =

                manager.addDefinition(FFrameTable.class,

                    PERSISTENCE_DEFINITION_NAME,

                    "FFrameTable persistence definition", null, null);

            definition.extend(manager

                .getDefinition(FFrameGroup.PERSISTENCE_DEFINITION_NAME));

            definition.addDynFieldInt(NUMCOLUMNS_FIELD).setMandatory(true);

            definition.addDynFieldInt(NUMROWS_FIELD).setMandatory(true);

        }

    }

    @Override

    public void loadFromState(PersistentState state)

        throws PersistenceException {

        super.loadFromState(state);

        numColumns = state.getInt(NUMCOLUMNS_FIELD);

        numRows = state.getInt(NUMROWS_FIELD);

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        super.saveToState(state);

        state.set(NUMCOLUMNS_FIELD, numColumns);

        state.set(NUMROWS_FIELD, numRows);

    }

}