using System;
using System.Collections.Generic;
using System.Reflection;
using System.Reflection.Emit;
using FreeImageAPI;

namespace PDFPatcher.Processor.Imaging
{
	static class FreeImageHelper
	{
		public static bool UseDib(this FreeImageBitmap bmp, Func<FIBITMAP, FIBITMAP> handler) {
			if (handler != null) {
				var t = bmp.Tag; // ensure not disposed
				var dib = GetDib(bmp);
				dib = handler(dib);
				return ReplaceDib(bmp, dib);
			}
			return false;
		}

		static readonly Func<FreeImageBitmap, FIBITMAP> GetDib = CreateGetDibMethod();
		static readonly Func<FreeImageBitmap, FIBITMAP, bool> ReplaceDib = CreateReplaceDibMethod();

		static Func<FreeImageBitmap, FIBITMAP> CreateGetDibMethod() {
			var m = new DynamicMethod("GetDib", typeof(FIBITMAP), new[] { typeof(FreeImageBitmap) }, true);
			var il = m.GetILGenerator();
			il.Emit(OpCodes.Ldarg_0);
			il.Emit(OpCodes.Ldfld, typeof(FreeImageBitmap).GetField("dib", BindingFlags.Instance | BindingFlags.NonPublic));
			il.Emit(OpCodes.Ret);
			return (Func<FreeImageBitmap, FIBITMAP>)m.CreateDelegate(typeof(Func<FreeImageBitmap, FIBITMAP>));
		}

		static Func<FreeImageBitmap, FIBITMAP, bool> CreateReplaceDibMethod() {
			var m = new DynamicMethod("ReplaceDib", typeof(bool), new[] { typeof(FreeImageBitmap), typeof(FIBITMAP) }, true);
			var il = m.GetILGenerator();
			il.Emit(OpCodes.Ldarg_0);
			il.Emit(OpCodes.Ldarg_1);
			il.EmitCall(OpCodes.Callvirt, typeof(FreeImageBitmap).GetMethod("ReplaceDib", BindingFlags.Instance | BindingFlags.NonPublic), null);
			il.Emit(OpCodes.Ret);
			return (Func<FreeImageBitmap, FIBITMAP, bool>)m.CreateDelegate(typeof(Func<FreeImageBitmap, FIBITMAP, bool>));
		}
	}
}