Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.sextante.app.extension;

import info.clearthought.layout.TableLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.FileReader;

import java.io.FileWriter;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import javax.swing.JButton;

import javax.swing.JDialog;

import javax.swing.JOptionPane;

import javax.swing.JScrollPane;

import javax.swing.JTable;

import javax.swing.table.DefaultTableCellRenderer;

import javax.swing.table.TableColumnModel;

import javax.swing.table.TableModel;

import org.gvsig.andami.PluginServices;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.gui.core.SextanteGUI;

public class ToolbarConfigDialog extends JDialog {

        private static final long serialVersionUID = 1L;

        private static final String BASE_CONFIG_FILE = "config_base.xml";

        private static final String CONFIG_FILE = "config.xml";

        private JTable jTable;

        private JButton jButtonCancel;

        private JButton jButtonRemove;

        private JButton jButtonAdd;

        private JScrollPane jScrollPane;

        private JButton jButtonOK;

        public ToolbarConfigDialog(){

                super(SextanteGUI.getMainFrame(), "SextanteToolbar", true);

                this.setResizable(true);

                initGUI();

        }

        private void initGUI() {

                try {

                        TableLayout thisLayout = new TableLayout(new double[][] {

                                        {7.0, 50.0, 50.0, TableLayout.FILL, TableLayout.FILL, TableLayout.FILL, 7.0},

                                        {10.0, TableLayout.FILL, TableLayout.FILL, 30.0, 30.0, 30.0, 7.0}});

                        thisLayout.setHGap(5);

                        thisLayout.setVGap(5);

                        getContentPane().setLayout(thisLayout);

                        {

                                jButtonOK = new JButton();

                                getContentPane().add(jButtonOK, "4, 5");

                                jButtonOK.setText(PluginServices.getText(this, "OK"));

                                jButtonOK.addActionListener(new ActionListener() {

                                        public void actionPerformed(ActionEvent evt) {

                                                jButtonOKActionPerformed(evt);

                                        }

                                });

                        }

                        {

                                jButtonCancel = new JButton();

                                getContentPane().add(jButtonCancel, "5, 5");

                                jButtonCancel.setText(PluginServices.getText(this, "Cancel"));

                                jButtonCancel.addActionListener(new ActionListener() {

                                        public void actionPerformed(ActionEvent evt) {

                                                jButtonCancelActionPerformed(evt);

                                        }

                                });

                        }

                        {

                                jScrollPane = new JScrollPane();

                                getContentPane().add(jScrollPane, "1, 0, 5, 2");

                                updateTable();

                        }

                        {

                                jButtonAdd = new JButton();

                                getContentPane().add(jButtonAdd, "1, 3");

                                jButtonAdd.setText("+");

                                jButtonAdd.addActionListener(new ActionListener() {

                                        public void actionPerformed(ActionEvent evt) {

                                                jButtonAddActionPerformed(evt);

                                        }

                                });

                        }

                        {

                                jButtonRemove = new JButton();

                                getContentPane().add(jButtonRemove, "2, 3");

                                jButtonRemove.setText("-");

                                jButtonRemove.addActionListener(new ActionListener() {

                                        public void actionPerformed(ActionEvent evt) {

                                                jButtonRemoveActionPerformed(evt);

                                        }

                                });

                        }

                        pack();

                        this.setSize(519, 255);

                } catch(Exception e) {

                        e.printStackTrace();

                }

        }

        private void updateTable() {

                TableModel tableModel = new ButtonTableModel(SextanteToolbarManager.getButtons());

                jTable = new JTable();

                jScrollPane.setViewportView(jTable);

                jTable.setModel(tableModel);

                TableColumnModel tcm = jTable.getColumnModel();

            tcm.getColumn(0).setPreferredWidth(30);

            DefaultTableCellRenderer  renderer = new IconCellRenderer();

            tcm.getColumn(0).setCellRenderer(renderer);

        }

        private void jButtonAddActionPerformed(ActionEvent evt) {

                ButtonDefinitionDialog dialog = new ButtonDefinitionDialog(this);

                dialog.pack();

                dialog.setVisible(true);

                Button button = dialog.getButton();

                if (button != null){

                        SextanteToolbarManager.addButton(button);

                        updateTable();

                }

        }

        private void jButtonRemoveActionPerformed(ActionEvent evt) {

                int iRow;

                iRow = jTable.getSelectedRow();

                if (iRow != -1){

                        ((ButtonTableModel)jTable.getModel()).removeRow(iRow);

                }

        }

        private void jButtonOKActionPerformed(ActionEvent evt) {

                saveChanges();

                dispose();

                setVisible(false);

                JOptionPane.showMessageDialog(null,

                                PluginServices.getText(this, "You_need_to_restart"),

                                PluginServices.getText(this, "Warning"),

                                JOptionPane.INFORMATION_MESSAGE);

        }

        private void saveChanges() {

                SextanteToolbarManager.save();

                String sPath = System.getProperty("user.dir") + File.separator + "gvSIG"

                                                + File.separator + "extensiones"

                                                + File.separator + "es.unex.sextante";

                File inFile  = new File(sPath + File.separator + BASE_CONFIG_FILE);

                File outFile = new File(sPath + File.separator + CONFIG_FILE);

                try {

                        BufferedReader reader = new BufferedReader(new FileReader(inFile));

                        BufferedWriter writer = new BufferedWriter(new FileWriter(outFile));

                        String line = null;

                        while ((line=reader.readLine()) != null) {

                                writer.write(line);

                                writer.newLine();

                        }

                        reader.close();

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

                                Button button = SextanteToolbarManager.getButton(i);

                                String sIconFilename = button.sIconFilename;

                                if (sIconFilename == null || sIconFilename.trim().equals("")){

                                        sIconFilename = "module2.png";

                                }

                                else{

                                        File iconFile = new File (sIconFilename);

                                        sIconFilename = (iconFile.getName());

                                        File outputFile = new File(sPath + File.separator + "images" + File.separator + iconFile.getName());

                                        InputStream in = new FileInputStream(iconFile);

                                        OutputStream out = new FileOutputStream(outputFile);

                                        byte[] buf = new byte[1024];

                                        int len;

                                        while ((len = in.read(buf)) > 0){

                                                out.write(buf, 0, len);

                                        }

                                        in.close();

                                        out.close();

                                }

                                writer.write("<extension class-name=\"es.unex.sextante.gvsig.extensions.SextanteAlgorithmExecutorExtension\"\n");

                                writer.write("\t description=\"" + button.sName + "\"");

                                writer.write("\t active=\"true\">\n");

                                writer.write("\t <tool-bar name=\"SEXTANTEToolbar\" position=\"3\">\n"

                                +"<action-tool icon=\"images/" + sIconFilename.trim() + "\"  tooltip=\"" + button.sName

                                + "\" position=\"" + Integer.toString(i+2)

                                + "\" action-command=\"" + Integer.toString(i) + "\"/>\n" +

                            "</tool-bar>");

                                writer.write("</extension>\n");

                        }

                        writer.write("</extensions>\n");

                        writer.write("</plugin-config>");

                        writer.close();

                } catch (IOException e) {

                        Sextante.addErrorToLog(e);

                }

        }

        private void jButtonCancelActionPerformed(ActionEvent evt) {

                dispose();

                setVisible(false);

        }

}