services/console-proxy/rdpconsole/src/main/java/common/adapter/AwtCanvasAdapter.java

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
package common.adapter;

import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.awt.image.WritableRaster;

import rdpclient.rdp.ServerBitmapUpdate;
import streamer.BaseElement;
import streamer.ByteBuffer;
import streamer.Element;
import streamer.Link;
import streamer.Order;
import streamer.Pipeline;
import streamer.PipelineImpl;
import common.BitmapOrder;
import common.BitmapRectangle;
import common.BufferedImageCanvas;
import common.CopyRectOrder;
import common.OrderType;
import common.ScreenDescription;

public class AwtCanvasAdapter extends BaseElement {

    protected ScreenDescription screen;

    public AwtCanvasAdapter(String id, BufferedImageCanvas canvas, ScreenDescription screen) {
        super(id);
        this.canvas = canvas;
        this.screen = screen;
    }

    protected BufferedImageCanvas canvas;

    @Override
    public String toString() {
        return "AwtRdpAdapter(" + id + ")";
    }

    @Override
    public void handleData(ByteBuffer buf, Link link) {
        if (verbose)
            System.out.println("[" + this + "] INFO: Data received: " + buf + ".");

        Order order = buf.getOrder();
        switch ((OrderType)order.type) {

        case BITMAP_UPDATE:
            handleBitmap((BitmapOrder)order, buf);
            break;

        case COPY_RECT:
            handleCopyRect((CopyRectOrder)order, buf);
            break;

        default:
            throw new RuntimeException("Order is not implemented: " + buf + ".");
        }

        buf.unref();
    }

    private void handleCopyRect(CopyRectOrder order, ByteBuffer buf) {
        // Copy image
        canvas.getOfflineGraphics().copyArea(order.srcX, order.srcY, order.width, order.height, order.x - order.srcX, order.y - order.srcY);

        // Request update of repainted area
        canvas.updateFrameBuffer(order.x, order.y, order.width, order.height);
        canvas.repaint(order.x, order.y, order.width, order.height);


    }

    private void handleBitmap(BitmapOrder order, ByteBuffer buf) {
        // Draw rectangle on offline buffer
        BufferedImage image = canvas.getOfflineImage();
        Graphics2D g = (Graphics2D)image.getGraphics();

        for (BitmapRectangle rectangle : order.rectangles) {

            int x = rectangle.x;
            int y = rectangle.y;
            int width = rectangle.width;
            int height = rectangle.height;
            int bufferWidth = rectangle.bufferWidth;
            int bufferHeight = rectangle.bufferHeight;

            BufferedImage rectImage;
            switch (rectangle.colorDepth) {
            case 8: {
                rectImage = new BufferedImage(bufferWidth, height, BufferedImage.TYPE_BYTE_INDEXED, screen.colorMap);
                WritableRaster raster = rectImage.getRaster();
                raster.setDataElements(0, 0, bufferWidth, bufferHeight, rectangle.bitmapDataStream.toByteArray());
                break;
            }
            case 15: {
                rectImage = new BufferedImage(bufferWidth, height, BufferedImage.TYPE_USHORT_555_RGB);
                WritableRaster raster = rectImage.getRaster();
                raster.setDataElements(0, 0, bufferWidth, bufferHeight, rectangle.bitmapDataStream.toShortArray());
                break;
            }
            case 16: {
                rectImage = new BufferedImage(bufferWidth, height, BufferedImage.TYPE_USHORT_565_RGB);
                WritableRaster raster = rectImage.getRaster();
                raster.setDataElements(0, 0, bufferWidth, bufferHeight, rectangle.bitmapDataStream.toShortArray());
                break;
            }
            case 24:
            case 32: {
                rectImage = new BufferedImage(bufferWidth, height, BufferedImage.TYPE_INT_RGB);
                WritableRaster raster = rectImage.getRaster();
                raster.setDataElements(0, 0, bufferWidth, bufferHeight, rectangle.bitmapDataStream.toIntLEArray());
                break;
            }
            default:
                throw new RuntimeException("Unsupported color depth: " + rectangle.colorDepth + ".");
            }

            g.setClip(x, y, width, height);
            g.drawImage(rectImage, x, y, null);

            // Request update of repainted area
            canvas.updateFrameBuffer(x, y, width, height);
            canvas.repaint(x, y, width, height);
        }

    }

    /**
     * Example.
     */
    public static void main(String args[]) {
        ByteBuffer packet = new ByteBuffer(new byte[] {0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00, 0x10, 0x00, 0x01, 0x00, 0x10, 0x00,
                0x01, 0x04, 0x0a, 0x00, 0x0c, (byte)0x84, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00});

        Pipeline pipeline = new PipelineImpl("test");

        Element bitmap = new ServerBitmapUpdate("bitmap");

        BufferedImageCanvas canvas = new BufferedImageCanvas(1024, 768);
        Element adapter = new AwtCanvasAdapter("test", canvas, null) {
            {
                verbose = true;
            }
        };
        pipeline.addAndLink(bitmap, adapter);

        bitmap.handleData(packet, null);

    }

}