Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.swing.impl.pickercontroller;

import java.awt.Cursor;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.net.URL;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JFileChooser;

import javax.swing.text.JTextComponent;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.ArrayUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.folders.FoldersManager;

import org.gvsig.tools.i18n.I18nManager;

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

import org.gvsig.tools.swing.api.pickercontroller.AbstractPickerController;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.gvsig.tools.swing.icontheme.IconThemeManager;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class BytearrayPickerControllerImpl  

        extends AbstractPickerController<byte[]> 

    {

    private final JTextComponent jtext;

    private final JButton btnUpload;

    private final JButton btnDownload;

    private byte[] value;

    private final String fileChooserID;

    private final File initialPath;

    public BytearrayPickerControllerImpl(

            JTextComponent jtext,

            JButton btnUpload,

            JButton btnDownload,

            String fileChooserID, 

            File initialPath

        ) {

        this.jtext = jtext;

        this.btnUpload = btnUpload;

        this.btnDownload = btnDownload;

        this.fileChooserID = fileChooserID;

        this.initialPath = initialPath;

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.btnDownload.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.btnDownload.setText("");

        if( StringUtils.isBlank(this.btnDownload.getToolTipText()) ) {

            this.btnDownload.setToolTipText(i18n.getTranslation("_Download_data"));

        }

        this.btnDownload.setIcon(this.getIcon("picker-bytearray-download"));

        this.btnDownload.addActionListener((ActionEvent e) -> {

          doDownload();

        });

        this.btnUpload.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.btnUpload.setText("");

        if( StringUtils.isBlank(this.btnUpload.getToolTipText()) ) {

            this.btnUpload.setToolTipText(i18n.getTranslation("_Upload_data"));

        }

        this.btnUpload.setIcon(this.getIcon("picker-bytearray-upload"));

        this.btnUpload.addActionListener((ActionEvent e) -> {

          doUpload();

        });

        this.jtext.setEditable(false);

        ToolsSwingLocator.getToolsSwingManager().addClearButton(jtext, (ActionEvent e) -> {

            this.value = null;

            this.jtext.setText("");

            this.btnDownload.setEnabled(false);

        });

    }

    @Override

    public byte[] get() {

        return this.value;

    }

    @Override

    public void set(byte[] value) {

        this.value = value;

        int size = 0;

        if( this.value==null ) {

            this.btnDownload.setEnabled(false);

        } else {

            this.btnDownload.setEnabled(this.btnUpload.isEnabled());

            size = this.value.length;

        }

        this.jtext.setText(String.format("%d bytes", size));

    }

    @Override

    public void coerceAndSet(Object value) {

        this.set((byte[]) this.coerce(DataTypes.BYTEARRAY, value, null));

    }

    @Override

    public void setEnabled(boolean enabled) {

        this.btnDownload.setEnabled(true);

        this.btnUpload.setEnabled(enabled);

    }

    @Override

    public boolean isEnabled() {

        return this.btnUpload.isEnabled();

    }

    private File fileChooser(String title, int mode) {

        ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

        FoldersManager folderManager = ToolsLocator.getFoldersManager();

        File[] f = dialogs.showChooserDialog(

                title, 

                mode, 

                JFileChooser.FILES_ONLY, 

                true, 

                folderManager.getLastPath(

                        this.fileChooserID, 

                        this.initialPath

                ),

                null, 

                true

        );

        if( ArrayUtils.isEmpty(f) ) {

            return null;

        }

        return f[0];

    }

    private void doDownload() {

        File f = this.fileChooser("Save file", JFileChooser.SAVE_DIALOG);

        if( f ==null ) {

            return;

        }

        try {

            FileUtils.writeByteArrayToFile(f, value);

        } catch(Exception ex) {

            LOG.warn("Can't save bytes from file '"+f.getAbsolutePath()+"'.",ex);

        }

    }

    private void doUpload() {

        File f = this.fileChooser("Load file", JFileChooser.OPEN_DIALOG);

        if( f == null ) {

            return;

        }

        try {

            byte[] bytes = FileUtils.readFileToByteArray(f);

            this.set(bytes);

            this.fireChangeEvent();

        } catch (Exception ex) {

            LOG.warn("Can't load bytes from file '"+f.getAbsolutePath()+"'.",ex);

        }

    }

    public static void selfRegister() {

        URL imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-upload.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-upload", icon, imageResource);

        }

        imageResource = BytearrayPickerControllerImpl.class.getClassLoader().getResource("org/gvsig/tools/swing/picker/picker-bytearray-download.png");

        if (imageResource != null) {

            IconThemeManager iconThemeManager = ToolsSwingLocator.getIconThemeManager();

            IconTheme theme = iconThemeManager.getCurrent();

            ImageIcon icon = new ImageIcon(imageResource);

            theme.registerDefault("tools", "picker", "picker-bytearray-download", icon, imageResource);

        }

    }

}