/*
 * Greenshot - a free and open source screenshot tool
 * Copyright (C) 2007-2010  Thomas Braun, Jens Klingen, Robin Krom
 * 
 * For more information see: http://getgreenshot.org/
 * The Greenshot project is hosted on Sourceforge: http://sourceforge.net/projects/greenshot/
 * 
 * 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 1 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, see <http://www.gnu.org/licenses/>.
 */
using System;
using System.Drawing;
using System.Drawing.Imaging;
using System.Runtime.Serialization;

namespace PluginExample {
	/// <summary>
	/// The BitmapBuffer is exactly what it says, it buffers a Bitmap.
	/// And it is possible to Draw on the Bitmap with direct memory access for better performance
	/// </summary>
	[Serializable()] 
	public unsafe class BitmapBuffer : IDisposable {
		private static log4net.ILog LOG = log4net.LogManager.GetLogger(typeof(BitmapBuffer));
		private Bitmap bitmap;
		public Bitmap Bitmap {
			get {return bitmap;}
		}
		[NonSerialized]
		private BitmapData bmData;
		[NonSerialized]
		private Rectangle rect;
		[NonSerialized]
		private byte* pointer;
		[NonSerialized]
		private int stride; /* bytes per pixel row */
		[NonSerialized]
		private int aIndex = -1;
		[NonSerialized]
		private int rIndex = -1;
		[NonSerialized]
		private int gIndex = -1;
		[NonSerialized]
		private int bIndex = -1;
		[NonSerialized]
		private int bytesPerPixel;
		[NonSerialized]
		private bool bitsLocked = false;
		
		public Size Size {
			get {return rect.Size;}
		}
		public int Length {
			get {return rect.Width*rect.Height;}
		}
		public int Width {
			get {return rect.Width;}
		}
		public int Height {
			get {return rect.Height;}
		}
		
		/**
		 * Constructor with just a Bitmap, the Rectangle is the Bitmap itself
		 */
		public BitmapBuffer(Bitmap bmp) : this(bmp, Rectangle.Empty) {
		}
		
		/**
		 * Create a BitmapBuffer from a Bitmap and a Rectangle specifying what part from the Bitmap to take
		 */
		public BitmapBuffer(Bitmap sourceBmp, Rectangle applyRect) {
			Rectangle sourceRect = new Rectangle(applyRect.X, applyRect.Y, applyRect.Width, applyRect.Height);
			Rectangle bitmapRect = new Rectangle(0,0, sourceBmp.Width, sourceBmp.Height);
			
			if(sourceRect.Equals(Rectangle.Empty)) {
				sourceRect = bitmapRect;
			} else {
				sourceRect.Intersect(bitmapRect);
			}
			// Does the rect have any pixels?
			if (sourceRect.Height <= 0 || sourceRect.Width <= 0) {
				return;
			}
			
			if (SupportsPixelFormat(sourceBmp)) {
				// Create copy with supported format
				this.bitmap = sourceBmp.Clone(sourceRect, sourceBmp.PixelFormat);
			} else {
				// When sourceRect is the whole bitmap there is a GDI+ bug in Clone
				// Clone will than return the same PixelFormat as the source
				// a quick workaround is using new Bitmap which uses a default of Format32bppArgb
				if (sourceRect.Equals(bitmapRect)) {
					this.bitmap = new Bitmap(sourceBmp);
				} else {
					this.bitmap = sourceBmp.Clone(sourceRect, PixelFormat.Format24bppRgb);
				}
			}
			// Set "this" rect to location 0,0
			// as the Cloned Bitmap is only the part we want to work with
			this.rect = new Rectangle(0, 0, bitmap.Width, bitmap.Height);
		}

		/**
		 * Destructor
		 */
		~BitmapBuffer() {
			Dispose(false);
		}

		/**
		 * The public accessible Dispose
		 * Will call the GarbageCollector to SuppressFinalize, preventing being cleaned twice
		 */
		public void Dispose() {
			Dispose(true);
			GC.SuppressFinalize(this);
		}

		// The bulk of the clean-up code is implemented in Dispose(bool)

		/**
		 * This Dispose is called from the Dispose and the Destructor.
		 * When disposing==true all non-managed resources should be freed too!
		 */
		protected virtual void Dispose(bool disposing) {
			Unlock();
			if (disposing) {
				if (bitmap != null) {
					bitmap.Dispose();
				}
			}
			bitmap = null;
			bmData = null;
			pointer = null;
		}
		
		/**
		 * This is called when deserializing the object
		 */
		public BitmapBuffer(SerializationInfo info, StreamingContext ctxt) {
			this.bitmap = (Bitmap)info.GetValue("bitmap", typeof(Bitmap));
			this.rect = new Rectangle(0, 0, bitmap.Width, bitmap.Height);
			// The rest will be set when Lock is called
		}

		/**
		 * This is called when serializing the object
		 */
		public void GetObjectData(SerializationInfo info, StreamingContext ctxt) {
			Unlock();
			info.AddValue("bitmap", this.bitmap);
		}

		/**
		 * Lock the bitmap so we have direct access to the memory
		 */
		public void Lock() {
			if(rect.Width > 0 && rect.Height > 0) {
				bmData = bitmap.LockBits(rect, ImageLockMode.ReadWrite, bitmap.PixelFormat); 
				bitsLocked = true;
	
			    System.IntPtr Scan0 = bmData.Scan0; 
		    	pointer = (byte*)(void*)Scan0; 

		    	PrepareForPixelFormat();
			    stride = bmData.Stride;
			} 
		}

		/**
		 * Unlock the System Memory
		 */
		private void Unlock() {
			if(bitsLocked) {
				bitmap.UnlockBits(bmData); 
				bitsLocked = false;
			}
		}
		
		/**
		 * Draw the stored bitmap to the destionation bitmap at the supplied point
		 */
		public void DrawTo(Bitmap destinationBitmap, Point destination) {
			DrawTo(destinationBitmap, null, destination);
		}
		
		/**
		 * Draw the stored Bitmap on the Destination bitmap with the specified rectangle
		 * Be aware that the stored bitmap will be resized to the specified rectangle!!
		 */
		public void DrawTo(Bitmap destinationBitmap, Rectangle destinationRect) {
			DrawTo(destinationBitmap, destinationRect, null);
		}

		/**
		 * private helper to draw the bitmap
		 */
		private void DrawTo(Bitmap destinationBitmap, Rectangle? destinationRect, Point? destination) {
			if (destinationRect.HasValue) {
				// Does the rect have any pixels?
				if (destinationRect.Value.Height <= 0 || destinationRect.Value.Width <= 0) {
					return;
				}
			}
			// Make sure this.bitmap is unlocked
			Unlock();

			using (Graphics graphics = Graphics.FromImage(destinationBitmap)) {
				if (destinationRect.HasValue) {
					LOG.Debug(String.Format("Drawing at: {0},{1} {2},{3}", destinationRect.Value.X, destinationRect.Value.Y, destinationRect.Value.Width, destinationRect.Value.Height));
					graphics.DrawImage(this.bitmap, destinationRect.Value);
				} else if (destination.HasValue) {
					LOG.Debug(String.Format("Drawing at: {0},{1}", destination.Value.X, destination.Value.Y));
					graphics.DrawImage(this.bitmap, destination.Value);
				}
			}
		}
		
		/**
		 * Retrieve the color at location x,y
		 * Before the first time this is called the Lock() should be called once!
		 */
		public Color GetColorAt(int x, int y) {
			if(x>=0 && y>=0 && x<rect.Width && y<rect.Height) {
				int offset = x*bytesPerPixel+y*stride;
				int a = (aIndex==-1) ? 255 : (int)pointer[aIndex+offset];
				return Color.FromArgb(a, pointer[rIndex+offset], pointer[gIndex+offset], pointer[bIndex+offset]);
			} else {
				return Color.Empty;
			}
		}

		/**
		 * Set the color at location x,y
		 * Before the first time this is called the Lock() should be called once!
		 */				
		public void SetColorAt(int x, int y, Color color) {
			if(x>=0 && y>=0 && x<rect.Width && y<rect.Height) {
				int offset = x*bytesPerPixel+y*stride;
				if(aIndex!=-1) pointer[aIndex+offset] = color.A;
				pointer[rIndex+offset] = color.R;
				pointer[gIndex+offset] = color.G;
				pointer[bIndex+offset] = color.B;
			}
		}

		/**
		 * Retrieve the color at location x,y as an array
		 * Before the first time this is called the Lock() should be called once!
		 */
		public int[] GetColorArrayAt(int x, int y) {
			if(x>=0 && y>=0 && x<rect.Width && y<rect.Height) {
				int offset = x*bytesPerPixel+y*stride;
				int a = (aIndex==-1) ? 255 : (int)pointer[aIndex+offset];
				return new int[]{a, pointer[rIndex+offset], pointer[gIndex+offset], pointer[bIndex+offset]};
			} else {
				return new int[]{0,0,0,0};
			}
		}
		
		/**
		 * Set the color at location x,y as an array
		 * Before the first time this is called the Lock() should be called once!
		 */
		public void SetColorArrayAt(int x, int y, int[] colors) {
			if(x>=0 && y>=0 && x<rect.Width && y<rect.Height) {
				int offset = x*bytesPerPixel+y*stride;
				if(aIndex!=-1) pointer[aIndex+offset] = (byte)colors[0];
				pointer[rIndex+offset] = (byte)colors[1];
				pointer[gIndex+offset] = (byte)colors[2];
				pointer[bIndex+offset] = (byte)colors[3];
			}
		}
		
		/**
		 * Checks if the supplied Bitmap has a PixelFormat we support
		 */
		private bool SupportsPixelFormat(Bitmap bitmap) {
			return (bitmap.PixelFormat.Equals(PixelFormat.Format32bppArgb) || 
			        bitmap.PixelFormat.Equals(PixelFormat.Format32bppRgb) || 
			        bitmap.PixelFormat.Equals(PixelFormat.Format24bppRgb));
		}
		
		/**
		 * Set some internal values for accessing the bitmap according to the PixelFormat
		 */
		private void PrepareForPixelFormat() {
			switch(bitmap.PixelFormat) {
				case PixelFormat.Format32bppArgb: 
					bIndex = 0;
					gIndex = 1;
					rIndex = 2;
					aIndex = 3;
					bytesPerPixel = 4;
					break;
				case PixelFormat.Format32bppRgb: 
					bIndex = 0;
					gIndex = 1;
					rIndex = 2;
					bytesPerPixel = 4;
					break;
				case PixelFormat.Format24bppRgb: 
					bIndex = 0;
					gIndex = 1;
					rIndex = 2;
					bytesPerPixel = 3;
					break;
				default: 
					throw new FormatException("Bitmap.Pixelformat."+bitmap.PixelFormat+" is currently not supported. Supported: Format32bpp(A)Rgb, Format24bppRgb");
			}
		}
	}
}