Application: gvSIG desktop

package org.gvsig.app.project.documents.view.legend.gui.legendactions;

import java.awt.event.ActionEvent;

import java.io.File;

import java.util.List;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import org.gvsig.app.ApplicationLocator;

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

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

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.fmap.mapcontext.exceptions.WriteLegendException;

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

import org.gvsig.fmap.mapcontext.rendering.legend.driver.ILegendWriter;

import org.gvsig.i18n.Messages;

public class SaveLegendAction extends AbstractLoadSaveLegendAction {

    private static final long serialVersionUID = -382469306285635077L;

    @Override

    public void actionPerformed(ActionEvent e) {

        if( ! (e.getSource() instanceof LegendsPanel) ) {

            return;

        }

        try {

            saveLegend((LegendsPanel) e.getSource());

        } catch (Exception exc) {

            JOptionPane.showMessageDialog(

                ApplicationLocator.getManager().getRootComponent(),

                Messages.getText("_Unable_to_save_legend")

                + "\n\n" + exc.getMessage(),

                Messages.getText("legend"),

                JOptionPane.WARNING_MESSAGE);

        }

    }

    private void saveLegend(LegendsPanel legends) throws Exception {

        List<LegendFileFilter> filters = getLegendFileFilters(true);

        if( filters.isEmpty() ) {

            JOptionPane.showMessageDialog(

                ApplicationLocator.getManager().getRootComponent(),

                Messages.getText("_No_output_format_available"),

                Messages.getText("legend"),

                JOptionPane.WARNING_MESSAGE);

            return;

        }

        ILegend legend = legends.getLegend();

        MapContextManager mcoman = MapContextLocator.getMapContextManager();

        JFileChooser fileChooser = new JFileChooser();

        fileChooser.setAcceptAllFileFilterUsed(false);

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

            fileChooser.addChoosableFileFilter(filters.get(i));

        }

        if( fileChooser.showSaveDialog(

            ApplicationLocator.getManager().getRootComponent())

            == JFileChooser.APPROVE_OPTION ) {

            LegendFileFilter lff = (LegendFileFilter) fileChooser.getFileFilter();

            File selfile = fileChooser.getSelectedFile();

            if( !selfile.getAbsolutePath().toLowerCase().endsWith(

                "." + lff.getFileExtension().toLowerCase()) ) {

                selfile = new File(selfile.getAbsolutePath()

                    + "." + lff.getFileExtension().toLowerCase());

            }

            ILegendWriter lw = mcoman.createLegendWriter(

                legend.getClass(),

                lff.getFileExtension()

            );

            if( lw == null ) {

                throw new WriteLegendException(legend,

                    new Exception("Unable to create legend writer."));

            }

            lw.write(legend, selfile, lff.getFileExtension());

        }

    }

}