Application: gvSIG desktop » gvSIG legends: base legends

package org.gvsig.complexlegend.swing.impl;

import java.awt.BorderLayout;

import java.awt.Dimension;

import java.awt.FlowLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.Iterator;

import java.util.List;

import javax.swing.BorderFactory;

import javax.swing.DefaultComboBoxModel;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.JDialog;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.event.ListDataEvent;

import javax.swing.event.ListDataListener;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.app.project.documents.view.legend.gui.AbstractParentPanel;

import org.gvsig.app.project.documents.view.legend.gui.Features;

import org.gvsig.app.project.documents.view.legend.gui.ILegendPanel;

import org.gvsig.complexlegend.ComplexLegendItem;

import org.gvsig.complexlegend.VectorComplexLegend;

import org.gvsig.complexlegend.impl.DefaultVectorComplexLegend;

import org.gvsig.complexlegend.swing.ComplexLegendPanel;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.layers.FLayer;

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

import org.gvsig.fmap.mapcontext.rendering.legend.ILegend;

import org.gvsig.symbology.swing.SymbologySwingLocator;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.observer.Observable;

import org.gvsig.tools.observer.Observer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultComplexLegendPanel extends JPanel implements ComplexLegendPanel, Observer{

        /**

         * 

         */

        private static final long serialVersionUID = -312336406359200569L;

        private static Logger logger = LoggerFactory.getLogger(DefaultComplexLegendPanel.class);

        private VectorComplexLegend auxLegend;

        private FLayer layer;

        private JComboBox cmbValue;

        private JPanel symbolPanel;

        private JButton addButton;

        private JButton removeButton;

        private DefaultComboBoxModel model;

        private ILegendPanel legendPanel;

        private JDialog dlg;

        private ComplexLegendItem currentItem;

        private I18nManager i18nManager;

        private int currentIndex;

        public DefaultComplexLegendPanel(){

                super();

                initialize();

                refreshButtons();

        }

        private void initialize() {

                this.i18nManager = ToolsLocator.getI18nManager();

                setLayout(new BorderLayout());

                JPanel aux = new JPanel(new BorderLayout());

                JPanel fieldsPanel = new JPanel(new FlowLayout());

                fieldsPanel.setBorder(BorderFactory.createTitledBorder(null,

                                     i18nManager.getTranslation("select_scale")));

                fieldsPanel.setPreferredSize(new Dimension(300,60));

                fieldsPanel.add(new JLabel());

                fieldsPanel.add(getComboBox());

                fieldsPanel.add(getAddButton());

                fieldsPanel.add(getRemoveButton());

                symbolPanel = new JPanel();

                symbolPanel.setBorder(BorderFactory.createTitledBorder(null,

                                     i18nManager.getTranslation("legend_definition")));

                aux.add(fieldsPanel, BorderLayout.NORTH);

                aux.add(symbolPanel, BorderLayout.CENTER);

                this.add(aux, BorderLayout.CENTER);

        }

        private JComboBox getComboBox() {

                if(cmbValue == null){

                        cmbValue = new JComboBox(getComboBoxModel());

                        cmbValue.setPreferredSize(new Dimension(350, 20));

                        cmbValue.addActionListener(new ActionListener() {

                                public void actionPerformed(ActionEvent e) {

                                        refreshLegendPanel();

                                }

                        });

                }

                return cmbValue;

        }

        private DefaultComboBoxModel getComboBoxModel(){

                if(this.model == null){

                        this.model = new DefaultComboBoxModel();

                        this.model.addListDataListener(new ListDataListener() {

                                public void intervalRemoved(ListDataEvent arg0) {

                                }

                                public void intervalAdded(ListDataEvent arg0) {

                                }

                                public void contentsChanged(ListDataEvent arg0) {

                                        refreshButtons();

                                }

                        });

                }

                return this.model;

        }

        protected void refreshButtons() {

                boolean status = (getComboBoxModel().getSize() != 0);

                getComboBox().setEnabled(status);

                getRemoveButton().setEnabled(status);

                if(legendPanel != null){

                        legendPanel.getPanel().setVisible(status);

                }

        }

        private JButton getAddButton() {

                if(addButton == null){

                        addButton = new JButton();

                        addButton.setIcon(PluginServices.getIconTheme().get("legend-add"));

                        addButton.addActionListener(new ActionListener() {

                                public void actionPerformed(ActionEvent arg0) {

                                        createComplexLegendItemWindow();

                                }

                        });

                }

                return addButton;

        }

        private JButton getRemoveButton() {

                if(removeButton == null){

                        removeButton = new JButton();

                        removeButton.setIcon(PluginServices.getIconTheme().get("legend-delete"));

                        removeButton.addActionListener(new ActionListener() {

                                public void actionPerformed(ActionEvent arg0) {

                                        removeComplexLegendItem();

                                }

                        });

                }

                return removeButton;

        }

        protected void removeComplexLegendItem() {

                ComplexLegendItem item = (ComplexLegendItem) getComboBox().getSelectedItem();

                int index = auxLegend.getLegends().indexOf(item);

                if(index != -1){

                        auxLegend.remove(item);

                        getComboBoxModel().removeElementAt(index);

                }

                refreshButtons();

        }

        protected void createComplexLegendItemWindow() {

                if(layer != null){

                        ComplexLegendItemPanel legendItemPanel = 

                                        new ComplexLegendItemPanel(layer, (VectorComplexLegend) getLegend());

                        legendItemPanel.addObserver(this);

                        dlg = new JDialog();

                        dlg.setTitle( i18nManager.getTranslation("SelectItem"));

                        dlg.setResizable(true);

                        //legendItemPanel.setParentDialog(dlg);

                        dlg.getContentPane().add(legendItemPanel);

                        dlg.setDefaultCloseOperation(JDialog.DISPOSE_ON_CLOSE);

                        dlg.setSize(510, 510);

                        dlg.setModal(true);

                        dlg.pack();

                        dlg.setLocationRelativeTo(null);

                        dlg.setVisible(true);

                }

        }

        public String getDescription() {

                return  i18nManager.getTranslation(

                            "legend_by_scale");

        }

        public ImageIcon getIcon() {

                 return IconThemeHelper.getImageIcon(

                                "scaled-symbols");

        }

        public ILegend getLegend() {

                if (auxLegend == null) {

                        auxLegend = (DefaultVectorComplexLegend) MapContextLocator.getMapContextManager()

                                .createLegend(DefaultVectorComplexLegend.COMPLEX_LEGEND_NAME);

                }

                saveCurrentPanel();

//                DefaultComplexLegendItem scaleLegend1 = new DefaultComplexLegendItem();

//                

//                SimpleFillSymbol fillsymbol1 = new SimpleFillSymbol();

//                fillsymbol1.setFillColor(Color.WHITE);

//                SingleSymbolLegend legend1 = new SingleSymbolLegend(fillsymbol1);

//                scaleLegend1.setLegend(legend1);

//                scaleLegend1.setMinScale(10000000);

//                scaleLegend1.setMaxScale(20000000);

//                

//                DefaultComplexLegendItem scaleLegend2 = new DefaultComplexLegendItem();

//                SimpleFillSymbol fillsymbol2 = new SimpleFillSymbol();

//                fillsymbol2.setFillColor(Color.BLACK);

//                SingleSymbolLegend legend2 = new SingleSymbolLegend(fillsymbol2);

//                scaleLegend2.setLegend(legend2);

//                scaleLegend2.setMinScale(0);

//                scaleLegend2.setMaxScale(9999999);

//                

//                auxLegend.add(scaleLegend2);

//                auxLegend.add(scaleLegend1);

                return auxLegend;

        }

        public Class getLegendClass() {

                return VectorComplexLegend.class;

        }

        public JPanel getPanel() {

                return this;

        }

        public Class getParentClass() {

                return Features.class;

        }

        public String getTitle() {

                return  i18nManager.getTranslation("complex_symbology");

        }

        public boolean isSuitableFor(FLayer layer) {

                return (layer instanceof FLyrVect);

        }

        public void setData(FLayer layer, ILegend data) {

                this.layer = layer;

                if(data instanceof DefaultVectorComplexLegend){

                        auxLegend = (DefaultVectorComplexLegend) data;

                        updateComboModel();

                }

        }

        private void saveCurrentPanel(){

                if(legendPanel != null && currentItem != null){

                        currentItem.setLegend(legendPanel.getLegend());

                }

        }

        public void refreshLegendPanel(){

                ComplexLegendItem item = (ComplexLegendItem) getComboBox().getSelectedItem();

                if(item != null){

                        ILegendPanel aux = getLegendPanel(item.getLegend()); 

                        if(aux != null){

                                saveCurrentPanel();

                                symbolPanel.removeAll();

                                legendPanel = aux;

                                legendPanel.setData(layer, item.getLegend());

                                symbolPanel.add(legendPanel.getPanel());

                                symbolPanel.repaint();

                                symbolPanel.revalidate();

                                currentItem = item;

                                currentIndex = getComboBox().getSelectedIndex();

                        }

                }

        }

        private ILegendPanel getLegendPanel(ILegend legend){

                List panels = SymbologySwingLocator.getSwingManager().getLegendEditorClasses();

                Iterator it = panels.iterator();

                while(it.hasNext()){

                        Class panelClass = (Class) it.next();

                        ILegendPanel panel;

                        try {

                                panel = (ILegendPanel) panelClass.newInstance();

                                if(!(panel instanceof AbstractParentPanel)){

                                        if(panel.getLegendClass().equals(legend.getClass())){

                                                panel.setData(layer, legend);

                                                return panel;

                                        }

                                }

                        } catch (Exception e) {

                                logger.warn("Error creating ILegendPanel. ", e);

                                return null;

                        }

                }

                return null;

        }

        public void update(Observable observable, Object isAccepted) {

                if(observable instanceof ComplexLegendItemPanel && isAccepted instanceof Boolean){

                        ComplexLegendItemPanel itemPanel = (ComplexLegendItemPanel) observable;

                        boolean acceptForm = (Boolean) isAccepted;

                        if(!acceptForm){

                                /* Si se ha pinchado en cancelar, se cierra sin m?s */

                                dlg.dispose();

                        }else{

                                if(itemPanel.validateValues()){

                                        ComplexLegendItem item = itemPanel.getLegendItem();

                                        /* Guardo el panel actual*/

                                        saveCurrentPanel();

                                        /* A?ado y ordeno la nueva entrada */

                                        auxLegend.add(item);

                                        updateComboModel();

                                        /* Cargo el nuevo panel */

                                        symbolPanel.removeAll();

                                        legendPanel = itemPanel.getLegendPanel();

                                        legendPanel.setData(layer, item.getLegend());

                                        symbolPanel.add(legendPanel.getPanel());

                                        symbolPanel.repaint();

                                        /* Selecciono la nueva entrada */

                                        currentItem = item;

                                        getComboBox().setSelectedItem(item);

                                        currentIndex = getComboBox().getSelectedIndex();

                                        /* Cierro el di?logo */

                                        dlg.dispose();

                                }

                        }

                }

        }

        private void updateComboModel() {

                getComboBoxModel().removeAllElements();

                if(auxLegend != null){

                        Iterator<ComplexLegendItem> it = auxLegend.getLegends().iterator();

                        while(it.hasNext()){

                                ComplexLegendItem item = (ComplexLegendItem) it.next();

                                getComboBoxModel().addElement(item);

                        }

                }

        }

}