package jp.ac.u_ryukyu.alicevnc;

import java.util.ArrayList;

import jp.ac.u_ryukyu.ie.cr.DecodeInfomation;

import com.glavsoft.exceptions.TransportException;
import com.glavsoft.rfb.encoding.EncodingType;
import com.glavsoft.rfb.encoding.decoder.FramebufferUpdateRectangle;
import com.glavsoft.rfb.encoding.decoder.RichCursorDecoder;
import com.glavsoft.rfb.encoding.decoder.ZRLEDecoder;

import alice.codesegment.CodeSegment;
import alice.datasegment.CommandType;
import alice.datasegment.ReceiveData;
import alice.datasegment.Receiver;

public class ReceiveUpdateRectangle extends CodeSegment{

    private Receiver info = ids.create(CommandType.PEEK);
    private Receiver info1 = ids.create(CommandType.TAKE); // AliceVNC Message
    private Receiver info2 = ids.create(CommandType.TAKE);
    private Receiver info3 = ids.create(CommandType.PEEK);

    public ReceiveUpdateRectangle(){
        info.setKey("holder");
        info1.setKey("updateRectangle");
        info2.setKey("pixelData");
        info3.setKey("_CLIST");
    }

    @Override
    public void run() {
        // for measurement;
        long time = System.currentTimeMillis();
        MeasurementData md = null;
        if (info2.getReceiveData().setTime) {
            ReceiveData r = info2.getReceiveData();
            md = new MeasurementData();
            md.depth = r.depth;
            md.time = time - r.time;
            md.size = ((byte[])info2.getObj()).length;
        }
        // send other node
        @SuppressWarnings("unchecked")
        ArrayList<String> clist = info3.asClass(ArrayList.class);
        for (String node : clist){
            if (!info1.from.equals(node)) {
                // not need send data comes from
                // Data Segment from other node is already serialized
                ods.put(node, "updateRectangle", info1.getReceiveData()); //TODO need timeout counter
                //ods.setCompressFlag(true);

                // for measurement
                ReceiveData r = info2.getReceiveData();
                if (!r.setTime) {
                    r.setTime = true;
                    r.time = System.currentTimeMillis();
                }
                ods.put(node, "pixelData",  info2.getReceiveData());
                ods.setCompressFlag(false);
            }
        }


        try {
            RenderAndControllerHolder holder = info.asClass(RenderAndControllerHolder.class);
            DecodeInfomation message = info1.asClass(DecodeInfomation.class);

            if (message.getEncodingType() == EncodingType.ZRLEE){ //TODO
                ZRLEDecoder decoder = new ZRLEDecoder();
                FramebufferUpdateRectangle rect = new FramebufferUpdateRectangle(
                        message.x, message.y, message.width, message.height);
                rect.encodingType = message.getEncodingType();
                byte[] buf = info2.asClass(byte[].class);

                // this method name is wrong.
                decoder.decode(holder.getRenderer(), rect, buf, 0);
                holder.getRepaintController().repaintBitmap(rect);

            } else if (message.getEncodingType() == EncodingType.DESKTOP_SIZE) {
                holder.setRenderer(
                        holder.getRepaintController().createRenderer(null, message.width,
                                message.height, message.getPixelFormat())
                                );
                boolean visible = true;
                if ("local".equals(info2.from)) {
                    visible = false;
                }
                holder.getViewerWindow().setVisible(visible);
            } else if (message.getEncodingType() == EncodingType.RICH_CURSOR) {
                byte[] buf = info2.asClass(byte[].class);
                RichCursorDecoder.getInstance().decode(holder.getRenderer(), message, buf);
                holder.getRepaintController().repaintCursor();

            }
            // for measurement
            if (md !=null)
                ods.put("parent", "measurement", md);

            this.recycle(); //TODO

        } catch (TransportException e) {
            e.printStackTrace();
        }
    }

}