// Copyright (C) 2010, 2011, 2012, 2013 GlavSoft LLC.
// All rights reserved.
//
//-------------------------------------------------------------------------
// This file is part of the TightVNC software.  Please visit our Web site:
//
//                       http://www.tightvnc.com/
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//-------------------------------------------------------------------------
//

package com.glavsoft.rfb.encoding.decoder;

import com.glavsoft.drawing.Renderer;
import com.glavsoft.exceptions.TransportException;
import com.glavsoft.rfb.encoding.EncodingType;
import com.glavsoft.transport.Reader;

import java.io.ByteArrayInputStream;
import java.util.logging.Logger;
import java.util.zip.DataFormatException;
import java.util.zip.Inflater;

public class ZlibDecoder extends Decoder {
	private Inflater decoder;

    @Override
	public void decode(Reader reader, Renderer renderer,
                       FramebufferUpdateRectangle rect) throws TransportException {
		int zippedLength = (int) reader.readUInt32();
		if (0 == zippedLength) return;
		int length = rect.width * rect.height * renderer.getBytesPerPixel();
		byte[] bytes = unzip(reader, zippedLength, length, rect.getEncodingType());
		Reader unzippedReader =
			new Reader(
					new ByteArrayInputStream(bytes, zippedLength, length));
		RawDecoder.getInstance().decode(unzippedReader, renderer, rect);
	}

	protected byte[] unzip(Reader reader, int zippedLength, int length,EncodingType encodingType)
			throws TransportException {
		byte [] bytes = ByteBuffer.getInstance().getBuffer(zippedLength + length);
		reader.readBytes(bytes, 0, zippedLength);
		if (null == decoder || encodingType == EncodingType.ZRLEE) {
            decoder = new Inflater();
		} 
		decoder.setInput(bytes, 0, zippedLength);
		try {
			messageDump(bytes, "infrate: ");
			decoder.inflate(bytes, zippedLength, length);
		} catch (DataFormatException e) {
			throw new TransportException("cannot inflate Zlib data", e);
		}
		return bytes;
	}

	@Override
	public void reset() {
		decoder = null;
	}

	 private void messageDump(Reader reader, String msg) {
		 System.out.print(msg);
		 reader.mark(30);
		 try {
			 for (int i = 0; i < 24; i++) {   // 20 + 4
				 System.out.print(String.format("%02x ", reader.readByte()));
			 }
			 reader.reset();
		 } catch (TransportException e) {

		 }
		 System.out.println();
	 }

	 private void messageDump(byte[] bytes, String msg) {
		 System.out.print(msg);
		 for (int i = 0; i < 24; i++) {   // 20 + 4
			 System.out.print(String.format("%02x ", bytes[i]));
		 }
		 System.out.println();
	 }
}