| gvSIG desktop 1

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

 *

 * Copyright (C) 2005 IVER T.I. 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

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

*

* $Id: SimpleLabeling.java 11104 2007-04-10 16:34:01Z jaume $

* $Log$

* Revision 1.5  2007-04-10 16:34:01  jaume

* towards a styled labeling

*

* Revision 1.4  2007/04/02 16:34:56  jaume

* Styled labeling (start commiting)

*

* Revision 1.3  2007/03/28 16:48:01  jaume

* *** empty log message ***

*

* Revision 1.2  2007/03/26 14:40:38  jaume

* added print method (BUT UNIMPLEMENTED)

*

* Revision 1.1  2007/03/20 16:16:20  jaume

* refactored to use ISymbol instead of FSymbol

*

* Revision 1.2  2007/03/09 11:20:57  jaume

* Advanced symbology (start committing)

*

* Revision 1.1  2007/03/09 08:33:43  jaume

* *** empty log message ***

*

* Revision 1.1.2.5  2007/02/21 07:34:08  jaume

* labeling starts working

*

* Revision 1.1.2.4  2007/02/15 16:23:44  jaume

* *** empty log message ***

*

* Revision 1.1.2.3  2007/02/09 07:47:05  jaume

* Isymbol moved

*

* Revision 1.1.2.2  2007/02/02 16:21:24  jaume

* start commiting labeling stuff

*

* Revision 1.1.2.1  2007/02/01 17:46:49  jaume

* *** empty log message ***

*

*

*/

package com.iver.cit.gvsig.fmap.rendering.styling;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.geom.AffineTransform;

import java.awt.geom.Rectangle2D;

import java.awt.image.BufferedImage;

import java.util.Hashtable;

import java.util.logging.Level;

import java.util.logging.Logger;

import javax.print.attribute.PrintRequestAttributeSet;

import javax.print.attribute.standard.PrintQuality;

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

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

import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;

import com.iver.cit.gvsig.exceptions.visitors.VisitorException;

import com.iver.cit.gvsig.fmap.ViewPort;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.core.TextPath;

import com.iver.cit.gvsig.fmap.core.symbols.ITextSymbol;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

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

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

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

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

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

import com.iver.utiles.XMLEntity;

import com.iver.utiles.swing.threads.Cancellable;

/**

 * This is the default and simplest (therefore fastest) labeling strategy.

 * @author   jaume dominguez faus - jaume.dominguez@iver.es

 */

public class SimpleLabeling implements ILabelingStrategy {

        private static Logger logger = Logger.getAnonymousLogger();

        private ILabelingMethod method;

        private IPlacementConstraints placementConstraints;

        private FLyrVect layer;

        private IZoomConstraints zoomConstraints;

        private Hashtable textPaths = new Hashtable(), texts = new Hashtable();

        protected SimpleLabeling(FLayer layer) throws ReadDriverException {

                FLyrVect l = (FLyrVect) layer;

                this.layer = l;

                if (layer instanceof FLyrVect) {

                        FLyrVect lyrVect = (FLyrVect) layer;

                        placementConstraints = LabelingFactory.createPlacementConstraints(lyrVect);

                } else {

                        throw new Error("Layer type not yet supported");

                }

        }

        public ILabelingMethod getLabelingMethod() {

                return method;

        }

        public void setLabelingMethod(ILabelingMethod method) {

                this.method = method;

        }

        public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort,

                                         Cancellable cancel)

        throws ReadDriverException {

                 Rectangle2D theExtent = viewPort.getAdjustedExtent();

                 if (method == null) {

                         Logger.getAnonymousLogger().warning("Layer '"+layer.getName()+"'. No labeling method was set, labels will not be drawn");

                         return;

                 }

                 if (zoomConstraints != null) {

                         double scale = viewPort.getScale();

                         if (scale < zoomConstraints.getMinScale() ||

                             scale > zoomConstraints.getMaxScale())

                                 return;

                 }

                 FBitSet bs;

                 int i = -1;

                 try {

                         bs = layer.queryByRect(theExtent);

                         LabelClass lc = method.getDefaultLabelClass();

                         ReadableVectorial rv = layer.getSource();

                         SelectableDataSource sds = rv.getRecordset();

                         for(i=bs.nextSetBit(0); i>=0 && !cancel.isCanceled(); i=bs.nextSetBit(i+1)) {

                                 IGeometry geom;

                                 Integer index = new Integer(i);

                                 String[] texts = getText(index, sds);

                                 geom = rv.getShape(i);

                                 FShape labelShape = lc.getShape(g, geom);

                                 if (labelShape != null) {

                                         FShape[] places = placementConstraints.getLocationsFor(

                                                         geom, labelShape,

                                                         null);

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

                                                 places[j].transform(viewPort.getAffineTransform());

                                                 lc.drawInsideRectangle(g, places[j].getBounds(), texts);

                                                 g.setColor(Color.BLUE);

                                                 g.draw(places[j]);

                                         }

                                 }

                         }

                 } catch (ExpansionFileReadException e) {

                         // TODO Auto-generated catch block

                         logger.log(Level.SEVERE, e.getMessage());

                 } catch (ReadDriverException e) {

                         logger.log(Level.SEVERE, "Getting label text for "+i+"th shape of "+layer.getName());

                 } catch (VisitorException e) {

                         logger.log(Level.SEVERE, "Getting label text for "+i+"th shape of "+layer.getName());

                 }

        }

        private String[] getText(Integer index, SelectableDataSource sds) throws ReadDriverException {

                String text = (String) texts.get(index);

                if (text == null) {

                        String expr = method.getDefaultLabelClass().getLabelExpression();

                        // TODO el analizador sint?ctico de momento s?lo pilla un valor del combo por

                        // defecto

                        String fieldName = expr.replaceAll("\\[", "").replaceAll("\\]", "").trim();

                        text = sds.getFieldValue(index.longValue(), sds.getFieldIndexByName(fieldName)).toString();

                        texts.put(index, text);

                }

                return new String[] { text };

        }

        private TextPath getTreePath(Integer index, Graphics2D g, FShape shp, char[] text) {

                TextPath tp = (TextPath) textPaths.get(index);

                if (tp == null) {

                        tp = new TextPath(g, shp, text);

                        textPaths.put(index, tp);

                }

                return tp;

        }

        public String getClassName() {

                return getClass().getName();

        }

        public XMLEntity getXMLEntity() {

                // TODO Implement it

//                throw new Error("Not yet implemented!");

                return new XMLEntity();

        }

        public void setXMLEntity(XMLEntity xml) {

                // TODO Implement it

//                throw new Error("Not yet implemented!");

        }

        public IPlacementConstraints getPlacementConstraints() {

                return placementConstraints;

        }

        public void setPlacementConstraints(IPlacementConstraints constraints) {}

        public IZoomConstraints getZoomConstraints() {

                return zoomConstraints;

        }

        public void setZoomConstraints(IZoomConstraints constraints) {

                this.zoomConstraints = constraints;

        }

        public void print(Graphics2D g, ViewPort viewPort, Cancellable cancel, PrintRequestAttributeSet properties) throws ReadDriverException {

//                double originalScaleFactor = unitFactor;

//                PrintQuality pq = (PrintQuality) properties.get(PrintQuality.class);

//                if (pq.equals(PrintQuality.NORMAL)){

//                        unitFactor *= (double) 300/72;

//                }else if (pq.equals(PrintQuality.HIGH)){

//                        unitFactor *= (double) 600/72;

//                }else if (pq.equals(PrintQuality.DRAFT)){

//                        //        unitFactor *= 72; (which is the same than doing nothing)

//                }

//                draw(null, g, viewPort, cancel);

//                unitFactor = originalScaleFactor;

        }

}