Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.app.extension.geometriespalette;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.Graphics2D;

import java.awt.Image;

import java.awt.image.BufferedImage;

import java.util.List;

import javax.swing.AbstractListModel;

import javax.swing.DefaultListCellRenderer;

import javax.swing.ImageIcon;

import javax.swing.JLabel;

import javax.swing.JList;

import javax.swing.ListCellRenderer;

import javax.swing.SpinnerNumberModel;

import javax.swing.SwingConstants;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ListSelectionEvent;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import org.gvsig.fmap.dal.swing.featuretype.AttributeDescriptorPickerController;

import org.gvsig.fmap.dal.swing.featuretype.FeatureAttributeListCellRenderer;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryUtils;

import org.gvsig.fmap.geom.type.GeometryType;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.fill.impl.SimpleFillSymbol;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.line.impl.SimpleLineSymbol;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.marker.impl.SimpleMarkerSymbol;

import org.gvsig.tools.swing.api.FilteredListController;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.ToolsSwingManager;

import org.gvsig.tools.swing.api.ToolsSwingUtils;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

/**

 *

 * @author gvSIG Team

 */

public class GeometriesPalettePanel extends GeometriesPalettePanelView {

    private AttributeDescriptorPickerController labelPicker;

    private FeatureAttributeDescriptor attributeLabel;

    private ISymbol symbol;

    private FeatureAttributeDescriptor geomattr;

    private FeatureStore store;

    private FilteredListController dataController;

    private AttributeDescriptorPickerController colorPicker;

    private AttributeDescriptorPickerController groupPicker;

    private class FeratureWithGeomCellRenderer extends DefaultListCellRenderer {

        @Override

        public Component getListCellRendererComponent(

                JList<?> list,

                Object value,

                int index,

                boolean isSelected,

                boolean cellHasFocus) {

            JLabel label = (JLabel) super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);

            LabeledValue<Feature> x = (LabeledValue<Feature>) value;

            if ( x!=null && x.getValue() != null && attributeLabel!=null && symbol!=null ) {

                Image img = drawGeometry(x.getValue());

                label.setHorizontalAlignment(SwingConstants.CENTER);

                label.setHorizontalTextPosition(JLabel.CENTER);

                label.setVerticalTextPosition(JLabel.BOTTOM);

                label.setIcon(new ImageIcon(img));

                label.setText(x.getLabel());

            }

            return label;

        }

    }

    public GeometriesPalettePanel() {

        initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        FeatureAttributeListCellRenderer renderer = dataSwingManager.createDefaultFeatureAttributeListCellRenderer();

        renderer.setUseLabeld(DataManager.USE_LABELS_NO);

        this.cboLabel.setRenderer(renderer);

        labelPicker = dataSwingManager.createAttributeDescriptorPickerController(this.cboLabel);

        labelPicker.addChangeListener((ChangeEvent e) -> {

            doChangeAttrLabel();

        });

        renderer = dataSwingManager.createDefaultFeatureAttributeListCellRenderer();

        renderer.setUseLabeld(DataManager.USE_LABELS_NO);

        this.cboColor.setRenderer(renderer);

        colorPicker = dataSwingManager.createAttributeDescriptorPickerController(this.cboColor);

        colorPicker.addChangeListener((ChangeEvent e) -> {

            doChangeAttrColor();

        });

        renderer = dataSwingManager.createDefaultFeatureAttributeListCellRenderer();

        renderer.setUseLabeld(DataManager.USE_LABELS_NO);

        this.cboGroup.setRenderer(renderer);

        groupPicker = dataSwingManager.createAttributeDescriptorPickerController(this.cboGroup);

        groupPicker.addChangeListener((ChangeEvent e) -> {

            doChangeAttrGroup();

        });

        SpinnerNumberModel spnSizeModel = new SpinnerNumberModel(100, 30, 600, 10);

        this.spnSize.setModel(spnSizeModel);

        this.spnSize.addChangeListener((ChangeEvent e) -> { doChangeSize(); });

        this.lstData.setCellRenderer(new FeratureWithGeomCellRenderer());

        this.lstData.addListSelectionListener((ListSelectionEvent e) -> {

            if( e.getValueIsAdjusting() ) {

                return;

            }

            doCopyFeature();

        });

        this.dataController = toolsSwingManager.createFilteredListController(lstData, txtDataFilter, btnDataFilter);

        ToolsSwingUtils.ensureRowsCols(this, 25,40);

        ToolsSwingUtils.ensureMaxRows(this, 25);

    }

    private void doChangeSize() {

        this.lstData.updateUI();

    }

    private void doChangeAttrGroup() {

    }

    private void doChangeAttrColor() {

    }

    private void doCopyFeature() {

        try {

            LabeledValue<Feature> x = (LabeledValue<Feature>) this.lstData.getSelectedValue();

            if( x == null ) {

                return;

            }

            DataManager dataManager = DALLocator.getDataManager();

            FeatureSelection selection = this.store.createFeatureSelection();

            selection.select(x.getValue());

            dataManager.putFeaturesInClipboard(selection);

        } catch (DataException ex) {

            // TODO: LOGGER.warn

        }

    }

    private Image drawGeometry(Feature feature) {

        Geometry geom = (Geometry) feature.get(geomattr.getName());

        int pixels = (int) this.spnSize.getValue();

        Dimension size = new Dimension(pixels, pixels);

        BufferedImage img = new BufferedImage(size.width, size.height, BufferedImage.TYPE_INT_ARGB);

        ViewPort viewPort = new ViewPort();

        viewPort.setImageSize(size);

        viewPort.setProjection(geomattr.getSRS());

        viewPort.setEnvelope(geom.getEnvelope());

        this.symbol.draw(

                (Graphics2D) img.getGraphics(), 

                viewPort.getAffineTransform(),

                geom,

                feature, 

                null

        );

        return img;

    }

    private void doChangeAttrLabel() {

        this.attributeLabel = this.labelPicker.get();

        this.lstData.setModel(new AbstractListModel<LabeledValue<Feature>>() {

            List<Feature> features = store.getFeatures((String)null, attributeLabel.getName());

            @Override

            public int getSize() {

                return features.size();

            }

            @Override

            public LabeledValue<Feature> getElementAt(int index) {

                Feature feature = features.get(index);

                return new LabeledValueImpl(

                        feature.getString(attributeLabel.getName()),

                        feature

                );

            }

        });

    }

    public void setFeatureStore(FeatureStore store) {

        this.store = store;

        this.lstData.setModel(new AbstractListModel<LabeledValue<Feature>>() {

            List<Feature> features = store.getFeatures();

            @Override

            public int getSize() {

                return features.size();

            }

            @Override

            public LabeledValue<Feature> getElementAt(int index) {

                Feature feature = features.get(index);

                return new LabeledValueImpl(

                        feature.getString(attributeLabel.getName()),

                        feature

                );

            }

        });

        FeatureType featureType = store.getDefaultFeatureTypeQuietly();

        this.labelPicker.setFeatureType(featureType);

        this.colorPicker.setFeatureType(featureType);

        this.groupPicker.setFeatureType(featureType);

        this.geomattr = featureType.getDefaultGeometryAttribute();

        GeometryType geomtype = geomattr.getGeomType();

        if( GeometryUtils.isSubtype(Geometry.TYPES.POINT, geomtype.getType()) ||

                GeometryUtils.isSubtype(Geometry.TYPES.MULTIPOINT, geomtype.getType())) {

            this.symbol = new SimpleMarkerSymbol();

        } else if( GeometryUtils.isSubtype(Geometry.TYPES.SURFACE, geomtype.getType()) ||

                GeometryUtils.isSubtype(Geometry.TYPES.MULTISURFACE, geomtype.getType())) {

            this.symbol = new SimpleFillSymbol();

        } else if( GeometryUtils.isSubtype(Geometry.TYPES.CURVE, geomtype.getType()) ||

                GeometryUtils.isSubtype(Geometry.TYPES.MULTICURVE, geomtype.getType())) {

            this.symbol = new SimpleLineSymbol();

        }

    }

    public static void showPalette(FLyrVect layer) {

        WindowManager winmanager = ToolsSwingLocator.getWindowManager();

        GeometriesPalettePanel panel = new GeometriesPalettePanel();

        winmanager.showWindow(

                panel,

                "Paleta",

                WindowManager.MODE.TOOL

        );

        panel.setFeatureStore(layer.getFeatureStore());

    }

}