// ClassicalSharp copyright 2014-2016 UnknownShadow200 | Licensed under MIT
using System;
using System.Drawing;
using System.Drawing.Imaging;
#if ANDROID
using Android.Graphics;
using Java.Nio;
#endif

namespace ClassicalSharp {

	/// <summary> Wrapper around a bitmap that allows extremely fast manipulation of 32bpp images. </summary>
	public unsafe class FastBitmap : IDisposable {
		
		public FastBitmap() { }
		
		[Obsolete( "You should always specify whether the bitmap is readonly or not." )]
		public FastBitmap( Bitmap bmp, bool lockBits ) : this( bmp, lockBits, false ) {
		}
		
		public FastBitmap( Bitmap bmp, bool lockBits, bool readOnly ) {
			SetData( bmp, lockBits, readOnly );
		}
		
		public void SetData( Bitmap bmp, bool lockBits, bool readOnly ) {
			Bitmap = bmp;
			if( lockBits )
				LockBits();
			ReadOnly = readOnly;
		}
		
		public void SetData( int width, int height, int stride, IntPtr scan0, bool readOnly ) {
			Width = width;
			Height = height;
			Stride = stride;
			Scan0 = scan0;
			scan0Byte = (byte*)scan0;
			ReadOnly = readOnly;
		}
		
		public Bitmap Bitmap;
		public bool ReadOnly;
		BitmapData data;
		byte* scan0Byte;
		
		public bool IsLocked { get { return data != null; } }
		
		public IntPtr Scan0;
		public int Stride;
		public int Width, Height;
		
		public static bool CheckFormat( PixelFormat format ) {
			return format == PixelFormat.Format32bppRgb || format == PixelFormat.Format32bppArgb;
		}
		
		/// <summary> Returns a pointer to the start of the y'th scanline. </summary>
		public int* GetRowPtr( int y ) {
			return (int*)(scan0Byte + (y * Stride));
		}
		
		public static void MovePortion( int srcX, int srcY, int dstX, int dstY, FastBitmap src, FastBitmap dst, int size ) {
			for( int y = 0; y < size; y++ ) {
				int* srcRow = src.GetRowPtr( srcY + y );
				int* dstRow = dst.GetRowPtr( dstY + y );
				for( int x = 0; x < size; x++ )
					dstRow[dstX + x] = srcRow[srcX + x];
			}
		}
		
		public void Dispose() { UnlockBits(); }
		
		#if !ANDROID
		
		public void LockBits() {
			if( Bitmap == null ) throw new InvalidOperationException( "Underlying bitmap is null." );
			if( data != null ) return;
			
			PixelFormat format = Bitmap.PixelFormat;
			if( !CheckFormat( format ) )
				throw new NotSupportedException( "Unsupported bitmap pixel format: " + format );
			
			Rectangle rec = new Rectangle( 0, 0, Bitmap.Width, Bitmap.Height );
			data = Bitmap.LockBits( rec, ImageLockMode.ReadWrite, format );
			Scan0 = data.Scan0;
			scan0Byte = (byte*)Scan0;			
			Stride = data.Stride;
			Width = data.Width;
			Height = data.Height;
		}
		
		public void UnlockBits() {
			if( Bitmap == null || data == null )
				return;
			Bitmap.UnlockBits( data );
			data = null;
			scan0Byte = (byte*)IntPtr.Zero;
			Scan0 = IntPtr.Zero;
			Width = Height = Stride = 0;
		}
		
		#else
		
		public void LockBits() {
			if( Bitmap == null ) throw new InvalidOperationException( "Underlying bitmap is null." );
			if( data != null ) return;

			data = ByteBuffer.AllocateDirect( Bitmap.Width * Bitmap.Height * 4 );
			Bitmap.CopyPixelsToBuffer( data );
			Scan0 = data.GetDirectBufferAddress();
			scan0Byte = (byte*)Scan0;			
			Stride = Bitmap.Width * 4;
			Width = Bitmap.Width;
			Height = Bitmap.Height;
		}
		
		public void UnlockBits() {
			if( Bitmap == null || data == null )
				return;
			data.Rewind();
			if( !ReadOnly )
				Bitmap.CopyPixelsFromBuffer( data );
			data.Dispose();

			data = null;
			scan0Byte = (byte*)IntPtr.Zero;
			Scan0 = IntPtr.Zero;
			Width = Height = Stride = 0;
		}
		#endif
	}
}