Application: gvSIG desktop » gvSIG scripting

package org.gvsig.scripting.swing.impl;

import org.gvsig.scripting.swing.api.JScriptingConsole;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Font;

import java.awt.event.ComponentEvent;

import java.awt.event.ComponentListener;

import java.io.IOException;

import java.io.OutputStream;

import java.io.PrintStream;

import javax.swing.JComponent;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTextPane;

import javax.swing.text.AttributeSet;

import javax.swing.text.SimpleAttributeSet;

import javax.swing.text.StyleConstants;

import javax.swing.text.StyleContext;

public class DefaultJScriptingConsole extends JPanel implements JScriptingConsole {

    private JTextAreaOutputStream stdout = null;

    private JTextAreaOutputStream stderr = null;

    private JTextPane console;

    public DefaultJScriptingConsole() {

        this.initComponents();

        this.stdout = new JTextAreaOutputStream(System.out, Color.BLACK);

        this.stderr = new JTextAreaOutputStream(System.err, Color.RED.darker());

        this.addComponentListener(new ComponentListener() {

            @Override

            public void componentResized(ComponentEvent e) {

            }

            @Override

            public void componentMoved(ComponentEvent e) {

            }

            @Override

            public void componentShown(ComponentEvent e) {

                if( System.out != stdout.getPrintStream() ) {

                    System.setOut(stdout.getPrintStream(System.out));

                }

                if( System.err != stderr.getPrintStream() ) {

                    System.setErr(stderr.getPrintStream(System.err));

                }

            }

            @Override

            public void componentHidden(ComponentEvent e) {

                System.setOut(stdout.getOriginalPS());

                System.setErr(stderr.getOriginalPS());

            }

        });

    }

    private void initComponents(){

        this.setLayout(new BorderLayout());

        console = new JTextPane();

        console.setFont(new Font("monospaced", Font.PLAIN, 13));

        JScrollPane scrollConsole = new JScrollPane(console);

        this.add(scrollConsole, BorderLayout.CENTER);

    }

    @Override

    public void clear() {

        this.console.setText("");

    }

    @Override

    public void append(String s) {

        append(s, stdout.color);

    }

    @Override

    public void append(String msg, Color color) {

        StyleContext sc = StyleContext.getDefaultStyleContext();

        AttributeSet aset = sc.addAttribute(SimpleAttributeSet.EMPTY, StyleConstants.Foreground, color);

        int len = console.getDocument().getLength();

        console.setCaretPosition(len);

        console.setCharacterAttributes(aset, false);

        console.replaceSelection(msg);        

    }

    @Override

    public PrintStream getStdout() {

        return this.stdout.getPrintStream();

    }

    @Override

    public PrintStream getStderr() {

        return this.stderr.getPrintStream();

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private class JTextAreaOutputStream extends OutputStream {

        PrintStream printStream = null;

        PrintStream originalPS;

        private final Color color;

        public JTextAreaOutputStream(PrintStream originalPS, Color color) {

            super();

            this.originalPS = originalPS;

            this.color = color;

        }

        @Override

        public void write(int i) throws IOException {

            append(Character.toString((char) i), color);

            originalPS.write(i);

        }

        @Override

        public void write(byte[] buf, int off, int len) throws IOException {

            String s = new String(buf, off, len);

            append(s, color);

            originalPS.write(buf, off, len);

        }

        public PrintStream getPrintStream() {

            if( this.printStream == null ) {

                this.printStream = new PrintStream(this);

            }

            return this.printStream;

        }

        public PrintStream getPrintStream(PrintStream originalPS) {

            this.originalPS = originalPS;

            return this.getPrintStream();

        }

        public PrintStream getOriginalPS() {

            return this.originalPS;

        }

    }

}