/*
   * Frame Buffer Driver for PKUnity-v3 Unigfx
   * Code specific to PKUnity SoC and UniCore ISA
   *
   *	Maintained by GUAN Xue-tao <gxt@mprc.pku.edu.cn>
   *	Copyright (C) 2001-2010 Guan Xuetao
   *
   * This program is free software; you can redistribute it and/or modify
   * it under the terms of the GNU General Public License version 2 as
   * published by the Free Software Foundation.
   */
  
  #include <linux/module.h>
  #include <linux/kernel.h>
  #include <linux/errno.h>
  #include <linux/platform_device.h>
  #include <linux/clk.h>
  #include <linux/fb.h>
  #include <linux/init.h>
  #include <linux/console.h>
  #include <linux/mm.h>
  
  #include <asm/sizes.h>
  #include <asm/pgtable.h>
  #include <mach/hardware.h>
  
  /* Platform_data reserved for unifb registers. */
  #define UNIFB_REGS_NUM		10
  /* RAM reserved for the frame buffer. */
  #define UNIFB_MEMSIZE		(SZ_4M)		/* 4 MB for 1024*768*32b */
  
  /*
   * cause UNIGFX don not have EDID
   * all the modes are organized as follow
   */
  static const struct fb_videomode unifb_modes[] = {
  	/* 0 640x480-60 VESA */
  	{ "640x480@60",  60,  640, 480,  25175000,  48, 16, 34, 10,  96, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 1 640x480-75 VESA */
  	{ "640x480@75",  75,  640, 480,  31500000, 120, 16, 18,  1,  64, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 2 800x600-60 VESA */
  	{ "800x600@60",  60,  800, 600,  40000000,  88, 40, 26,  1, 128, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 3 800x600-75 VESA */
  	{ "800x600@75",  75,  800, 600,  49500000, 160, 16, 23,  1,  80, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 4 1024x768-60 VESA */
  	{ "1024x768@60", 60, 1024, 768,  65000000, 160, 24, 34,  3, 136, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 5 1024x768-75 VESA */
  	{ "1024x768@75", 75, 1024, 768,  78750000, 176, 16, 30,  1,  96, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 6 1280x960-60 VESA */
  	{ "1280x960@60", 60, 1280, 960, 108000000, 312, 96, 38,  1, 112, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 7 1440x900-60 VESA */
  	{ "1440x900@60", 60, 1440, 900, 106500000, 232, 80, 30,  3, 152, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 8 FIXME 9 1024x600-60 VESA UNTESTED */
  	{ "1024x600@60", 60, 1024, 600,  50650000, 160, 24, 26,  1, 136, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 9 FIXME 10 1024x600-75 VESA UNTESTED */
  	{ "1024x600@75", 75, 1024, 600,  61500000, 176, 16, 23,  1,  96, 1,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  	/* 10 FIXME 11 1366x768-60 VESA UNTESTED */
  	{ "1366x768@60", 60, 1366, 768,  85500000, 256, 58, 18,  1,  112, 3,
  	  0, FB_VMODE_NONINTERLACED, FB_MODE_IS_VESA },
  };
  
  static struct fb_var_screeninfo unifb_default = {
  	.xres =		640,
  	.yres =		480,
  	.xres_virtual =	640,
  	.yres_virtual =	480,
  	.bits_per_pixel = 16,
  	.red =		{ 11, 5, 0 },
  	.green =	{ 5,  6, 0 },
  	.blue =		{ 0,  5, 0 },
  	.activate =	FB_ACTIVATE_NOW,
  	.height =	-1,
  	.width =	-1,
  	.pixclock =	25175000,
  	.left_margin =	48,
  	.right_margin =	16,
  	.upper_margin =	33,
  	.lower_margin =	10,
  	.hsync_len =	96,
  	.vsync_len =	2,
  	.vmode =	FB_VMODE_NONINTERLACED,
  };
  
  static struct fb_fix_screeninfo unifb_fix = {
  	.id =		"UNIGFX FB",
  	.type =		FB_TYPE_PACKED_PIXELS,
  	.visual =	FB_VISUAL_TRUECOLOR,
  	.xpanstep =	1,
  	.ypanstep =	1,
  	.ywrapstep =	1,
  	.accel =	FB_ACCEL_NONE,
  };
  
  static void unifb_sync(struct fb_info *info)
  {
  	/* TODO: may, this can be replaced by interrupt */
  	int cnt;
  
  	for (cnt = 0; cnt < 0x10000000; cnt++) {
  		if (readl(UGE_COMMAND) & 0x1000000)
  			return;
  	}
  
  	if (cnt > 0x8000000)
  		dev_warn(info->device, "Warning: UniGFX GE time out ...
  ");
  }
  
  static void unifb_prim_fillrect(struct fb_info *info,
  				const struct fb_fillrect *region)
  {
  	int awidth = region->width;
  	int aheight = region->height;
  	int m_iBpp = info->var.bits_per_pixel;
  	int screen_width = info->var.xres;
  	int src_sel = 1;	/* from fg_color */
  	int pat_sel = 1;
  	int src_x0 = 0;
  	int dst_x0 = region->dx;
  	int src_y0 = 0;
  	int dst_y0 = region->dy;
  	int rop_alpha_sel = 0;
  	int rop_alpha_code = 0xCC;
  	int x_dir = 1;
  	int y_dir = 1;
  	int alpha_r = 0;
  	int alpha_sel = 0;
  	int dst_pitch = screen_width * (m_iBpp / 8);
  	int dst_offset = dst_y0 * dst_pitch + dst_x0 * (m_iBpp / 8);
  	int src_pitch = screen_width * (m_iBpp / 8);
  	int src_offset = src_y0 * src_pitch + src_x0 * (m_iBpp / 8);
  	unsigned int command = 0;
  	int clip_region = 0;
  	int clip_en = 0;
  	int tp_en = 0;
  	int fg_color = 0;
  	int bottom = info->var.yres - 1;
  	int right = info->var.xres - 1;
  	int top = 0;
  
  	bottom = (bottom << 16) | right;
  	command = (rop_alpha_sel << 26) | (pat_sel << 18) | (src_sel << 16)
  		| (x_dir << 20) | (y_dir << 21) | (command << 24)
  		| (clip_region << 23) | (clip_en << 22) | (tp_en << 27);
  	src_pitch = (dst_pitch << 16) | src_pitch;
  	awidth = awidth | (aheight << 16);
  	alpha_r = ((rop_alpha_code & 0xff) << 8) | (alpha_r & 0xff)
  		| (alpha_sel << 16);
  	src_x0 = (src_x0 & 0x1fff) | ((src_y0 & 0x1fff) << 16);
  	dst_x0 = (dst_x0 & 0x1fff) | ((dst_y0 & 0x1fff) << 16);
  	fg_color = region->color;
  
  	unifb_sync(info);
  
  	writel(((u32 *)(info->pseudo_palette))[fg_color], UGE_FCOLOR);
  	writel(0, UGE_BCOLOR);
  	writel(src_pitch, UGE_PITCH);
  	writel(src_offset, UGE_SRCSTART);
  	writel(dst_offset, UGE_DSTSTART);
  	writel(awidth, UGE_WIDHEIGHT);
  	writel(top, UGE_CLIP0);
  	writel(bottom, UGE_CLIP1);
  	writel(alpha_r, UGE_ROPALPHA);
  	writel(src_x0, UGE_SRCXY);
  	writel(dst_x0, UGE_DSTXY);
  	writel(command, UGE_COMMAND);
  }
  
  static void unifb_fillrect(struct fb_info *info,
  		const struct fb_fillrect *region)
  {
  	struct fb_fillrect modded;
  	int vxres, vyres;
  
  	if (info->flags & FBINFO_HWACCEL_DISABLED) {
  		sys_fillrect(info, region);
  		return;
  	}
  
  	vxres = info->var.xres_virtual;
  	vyres = info->var.yres_virtual;
  
  	memcpy(&modded, region, sizeof(struct fb_fillrect));
  
  	if (!modded.width || !modded.height ||
  	    modded.dx >= vxres || modded.dy >= vyres)
  		return;
  
  	if (modded.dx + modded.width > vxres)
  		modded.width = vxres - modded.dx;
  	if (modded.dy + modded.height > vyres)
  		modded.height = vyres - modded.dy;
  
  	unifb_prim_fillrect(info, &modded);
  }
  
  static void unifb_prim_copyarea(struct fb_info *info,
  				const struct fb_copyarea *area)
  {
  	int awidth = area->width;
  	int aheight = area->height;
  	int m_iBpp = info->var.bits_per_pixel;
  	int screen_width = info->var.xres;
  	int src_sel = 2;	/* from mem */
  	int pat_sel = 0;
  	int src_x0 = area->sx;
  	int dst_x0 = area->dx;
  	int src_y0 = area->sy;
  	int dst_y0 = area->dy;
  
  	int rop_alpha_sel = 0;
  	int rop_alpha_code = 0xCC;
  	int x_dir = 1;
  	int y_dir = 1;
  
  	int alpha_r = 0;
  	int alpha_sel = 0;
  	int dst_pitch = screen_width * (m_iBpp / 8);
  	int dst_offset = dst_y0 * dst_pitch + dst_x0 * (m_iBpp / 8);
  	int src_pitch = screen_width * (m_iBpp / 8);
  	int src_offset = src_y0 * src_pitch + src_x0 * (m_iBpp / 8);
  	unsigned int command = 0;
  	int clip_region = 0;
  	int clip_en = 1;
  	int tp_en = 0;
  	int top = 0;
  	int bottom = info->var.yres;
  	int right = info->var.xres;
  	int fg_color = 0;
  	int bg_color = 0;
  
  	if (src_x0 < 0)
  		src_x0 = 0;
  	if (src_y0 < 0)
  		src_y0 = 0;
  
  	if (src_y0 - dst_y0 > 0) {
  		y_dir = 1;
  	} else {
  		y_dir = 0;
  		src_offset = (src_y0 + aheight) * src_pitch +
  				src_x0 * (m_iBpp / 8);
  		dst_offset = (dst_y0 + aheight) * dst_pitch +
  				dst_x0 * (m_iBpp / 8);
  		src_y0 += aheight;
  		dst_y0 += aheight;
  	}
  
  	command = (rop_alpha_sel << 26) | (pat_sel << 18) | (src_sel << 16) |
  		(x_dir << 20) | (y_dir << 21) | (command << 24) |
  		(clip_region << 23) | (clip_en << 22) | (tp_en << 27);
  	src_pitch = (dst_pitch << 16) | src_pitch;
  	awidth = awidth | (aheight << 16);
  	alpha_r = ((rop_alpha_code & 0xff) << 8) | (alpha_r & 0xff) |
  		(alpha_sel << 16);
  	src_x0 = (src_x0 & 0x1fff) | ((src_y0 & 0x1fff) << 16);
  	dst_x0 = (dst_x0 & 0x1fff) | ((dst_y0 & 0x1fff) << 16);
  	bottom = (bottom << 16) | right;
  
  	unifb_sync(info);
  
  	writel(src_pitch, UGE_PITCH);
  	writel(src_offset, UGE_SRCSTART);
  	writel(dst_offset, UGE_DSTSTART);
  	writel(awidth, UGE_WIDHEIGHT);
  	writel(top, UGE_CLIP0);
  	writel(bottom, UGE_CLIP1);
  	writel(bg_color, UGE_BCOLOR);
  	writel(fg_color, UGE_FCOLOR);
  	writel(alpha_r, UGE_ROPALPHA);
  	writel(src_x0, UGE_SRCXY);
  	writel(dst_x0, UGE_DSTXY);
  	writel(command, UGE_COMMAND);
  }
  
  static void unifb_copyarea(struct fb_info *info, const struct fb_copyarea *area)
  {
  	struct fb_copyarea modded;
  	u32 vxres, vyres;
  	modded.sx = area->sx;
  	modded.sy = area->sy;
  	modded.dx = area->dx;
  	modded.dy = area->dy;
  	modded.width = area->width;
  	modded.height = area->height;
  
  	if (info->flags & FBINFO_HWACCEL_DISABLED) {
  		sys_copyarea(info, area);
  		return;
  	}
  
  	vxres = info->var.xres_virtual;
  	vyres = info->var.yres_virtual;
  
  	if (!modded.width || !modded.height ||
  	    modded.sx >= vxres || modded.sy >= vyres ||
  	    modded.dx >= vxres || modded.dy >= vyres)
  		return;
  
  	if (modded.sx + modded.width > vxres)
  		modded.width = vxres - modded.sx;
  	if (modded.dx + modded.width > vxres)
  		modded.width = vxres - modded.dx;
  	if (modded.sy + modded.height > vyres)
  		modded.height = vyres - modded.sy;
  	if (modded.dy + modded.height > vyres)
  		modded.height = vyres - modded.dy;
  
  	unifb_prim_copyarea(info, &modded);
  }
  
  static void unifb_imageblit(struct fb_info *info, const struct fb_image *image)
  {
  	sys_imageblit(info, image);
  }
  
  static u_long get_line_length(int xres_virtual, int bpp)
  {
  	u_long length;
  
  	length = xres_virtual * bpp;
  	length = (length + 31) & ~31;
  	length >>= 3;
  	return length;
  }
  
  /*
   *  Setting the video mode has been split into two parts.
   *  First part, xxxfb_check_var, must not write anything
   *  to hardware, it should only verify and adjust var.
   *  This means it doesn't alter par but it does use hardware
   *  data from it to check this var.
   */
  static int unifb_check_var(struct fb_var_screeninfo *var,
  			 struct fb_info *info)
  {
  	u_long line_length;
  
  	/*
  	 *  FB_VMODE_CONUPDATE and FB_VMODE_SMOOTH_XPAN are equal!
  	 *  as FB_VMODE_SMOOTH_XPAN is only used internally
  	 */
  
  	if (var->vmode & FB_VMODE_CONUPDATE) {
  		var->vmode |= FB_VMODE_YWRAP;
  		var->xoffset = info->var.xoffset;
  		var->yoffset = info->var.yoffset;
  	}
  
  	/*
  	 *  Some very basic checks
  	 */
  	if (!var->xres)
  		var->xres = 1;
  	if (!var->yres)
  		var->yres = 1;
  	if (var->xres > var->xres_virtual)
  		var->xres_virtual = var->xres;
  	if (var->yres > var->yres_virtual)
  		var->yres_virtual = var->yres;
  	if (var->bits_per_pixel <= 1)
  		var->bits_per_pixel = 1;
  	else if (var->bits_per_pixel <= 8)
  		var->bits_per_pixel = 8;
  	else if (var->bits_per_pixel <= 16)
  		var->bits_per_pixel = 16;
  	else if (var->bits_per_pixel <= 24)
  		var->bits_per_pixel = 24;
  	else if (var->bits_per_pixel <= 32)
  		var->bits_per_pixel = 32;
  	else
  		return -EINVAL;
  
  	if (var->xres_virtual < var->xoffset + var->xres)
  		var->xres_virtual = var->xoffset + var->xres;
  	if (var->yres_virtual < var->yoffset + var->yres)
  		var->yres_virtual = var->yoffset + var->yres;
  
  	/*
  	 *  Memory limit
  	 */
  	line_length =
  	    get_line_length(var->xres_virtual, var->bits_per_pixel);
  	if (line_length * var->yres_virtual > UNIFB_MEMSIZE)
  		return -ENOMEM;
  
  	/*
  	 * Now that we checked it we alter var. The reason being is that the
  	 * video mode passed in might not work but slight changes to it might
  	 * make it work. This way we let the user know what is acceptable.
  	 */
  	switch (var->bits_per_pixel) {
  	case 1:
  	case 8:
  		var->red.offset = 0;
  		var->red.length = 8;
  		var->green.offset = 0;
  		var->green.length = 8;
  		var->blue.offset = 0;
  		var->blue.length = 8;
  		var->transp.offset = 0;
  		var->transp.length = 0;
  		break;
  	case 16:		/* RGBA 5551 */
  		if (var->transp.length) {
  			var->red.offset = 0;
  			var->red.length = 5;
  			var->green.offset = 5;
  			var->green.length = 5;
  			var->blue.offset = 10;
  			var->blue.length = 5;
  			var->transp.offset = 15;
  			var->transp.length = 1;
  		} else {	/* RGB 565 */
  			var->red.offset = 11;
  			var->red.length = 5;
  			var->green.offset = 5;
  			var->green.length = 6;
  			var->blue.offset = 0;
  			var->blue.length = 5;
  			var->transp.offset = 0;
  			var->transp.length = 0;
  		}
  		break;
  	case 24:		/* RGB 888 */
  		var->red.offset = 0;
  		var->red.length = 8;
  		var->green.offset = 8;
  		var->green.length = 8;
  		var->blue.offset = 16;
  		var->blue.length = 8;
  		var->transp.offset = 0;
  		var->transp.length = 0;
  		break;
  	case 32:		/* RGBA 8888 */
  		var->red.offset = 16;
  		var->red.length = 8;
  		var->green.offset = 8;
  		var->green.length = 8;
  		var->blue.offset = 0;
  		var->blue.length = 8;
  		var->transp.offset = 24;
  		var->transp.length = 8;
  		break;
  	}
  	var->red.msb_right = 0;
  	var->green.msb_right = 0;
  	var->blue.msb_right = 0;
  	var->transp.msb_right = 0;
  
  	return 0;
  }
  
  /*
   * This routine actually sets the video mode. It's in here where we
   * the hardware state info->par and fix which can be affected by the
   * change in par. For this driver it doesn't do much.
   */
  static int unifb_set_par(struct fb_info *info)
  {
  	int hTotal, vTotal, hSyncStart, hSyncEnd, vSyncStart, vSyncEnd;
  	int format;
  
  #ifdef CONFIG_PUV3_PM
  	struct clk *clk_vga;
  	u32 pixclk = 0;
  	int i;
  
  	for (i = 0; i <= 10; i++) {
  		if    (info->var.xres         == unifb_modes[i].xres
  		    && info->var.yres         == unifb_modes[i].yres
  		    && info->var.upper_margin == unifb_modes[i].upper_margin
  		    && info->var.lower_margin == unifb_modes[i].lower_margin
  		    && info->var.left_margin  == unifb_modes[i].left_margin
  		    && info->var.right_margin == unifb_modes[i].right_margin
  		    && info->var.hsync_len    == unifb_modes[i].hsync_len
  		    && info->var.vsync_len    == unifb_modes[i].vsync_len) {
  			pixclk = unifb_modes[i].pixclock;
  			break;
  		}
  	}
  
  	/* set clock rate */
  	clk_vga = clk_get(info->device, "VGA_CLK");
  	if (clk_vga == ERR_PTR(-ENOENT))
  		return -ENOENT;
  
  	if (pixclk != 0) {
  		if (clk_set_rate(clk_vga, pixclk)) { /* set clock failed */
  			info->fix = unifb_fix;
  			info->var = unifb_default;
  			if (clk_set_rate(clk_vga, unifb_default.pixclock))
  				return -EINVAL;
  		}
  	}
  #endif
  
  	info->fix.line_length = get_line_length(info->var.xres_virtual,
  						info->var.bits_per_pixel);
  
  	hSyncStart = info->var.xres + info->var.right_margin;
  	hSyncEnd = hSyncStart + info->var.hsync_len;
  	hTotal = hSyncEnd + info->var.left_margin;
  
  	vSyncStart = info->var.yres + info->var.lower_margin;
  	vSyncEnd = vSyncStart + info->var.vsync_len;
  	vTotal = vSyncEnd + info->var.upper_margin;
  
  	switch (info->var.bits_per_pixel) {
  	case 8:
  		format = UDE_CFG_DST8;
  		break;
  	case 16:
  		format = UDE_CFG_DST16;
  		break;
  	case 24:
  		format = UDE_CFG_DST24;
  		break;
  	case 32:
  		format = UDE_CFG_DST32;
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	writel(info->fix.smem_start, UDE_FSA);
  	writel(info->var.yres, UDE_LS);
  	writel(get_line_length(info->var.xres,
  			info->var.bits_per_pixel) >> 3, UDE_PS);
  			/* >> 3 for hardware required. */
  	writel((hTotal << 16) | (info->var.xres), UDE_HAT);
  	writel(((hTotal - 1) << 16) | (info->var.xres - 1), UDE_HBT);
  	writel(((hSyncEnd - 1) << 16) | (hSyncStart - 1), UDE_HST);
  	writel((vTotal << 16) | (info->var.yres), UDE_VAT);
  	writel(((vTotal - 1) << 16) | (info->var.yres - 1), UDE_VBT);
  	writel(((vSyncEnd - 1) << 16) | (vSyncStart - 1), UDE_VST);
  	writel(UDE_CFG_GDEN_ENABLE | UDE_CFG_TIMEUP_ENABLE
  			| format | 0xC0000001, UDE_CFG);
  
  	return 0;
  }
  
  /*
   *  Set a single color register. The values supplied are already
   *  rounded down to the hardware's capabilities (according to the
   *  entries in the var structure). Return != 0 for invalid regno.
   */
  static int unifb_setcolreg(u_int regno, u_int red, u_int green, u_int blue,
  			 u_int transp, struct fb_info *info)
  {
  	if (regno >= 256)	/* no. of hw registers */
  		return 1;
  
  	/* grayscale works only partially under directcolor */
  	if (info->var.grayscale) {
  		/* grayscale = 0.30*R + 0.59*G + 0.11*B */
  		red = green = blue =
  		    (red * 77 + green * 151 + blue * 28) >> 8;
  	}
  
  #define CNVT_TOHW(val, width) ((((val)<<(width))+0x7FFF-(val))>>16)
  	switch (info->fix.visual) {
  	case FB_VISUAL_TRUECOLOR:
  	case FB_VISUAL_PSEUDOCOLOR:
  		red = CNVT_TOHW(red, info->var.red.length);
  		green = CNVT_TOHW(green, info->var.green.length);
  		blue = CNVT_TOHW(blue, info->var.blue.length);
  		transp = CNVT_TOHW(transp, info->var.transp.length);
  		break;
  	case FB_VISUAL_DIRECTCOLOR:
  		red = CNVT_TOHW(red, 8);	/* expect 8 bit DAC */
  		green = CNVT_TOHW(green, 8);
  		blue = CNVT_TOHW(blue, 8);
  		/* hey, there is bug in transp handling... */
  		transp = CNVT_TOHW(transp, 8);
  		break;
  	}
  #undef CNVT_TOHW
  	/* Truecolor has hardware independent palette */
  	if (info->fix.visual == FB_VISUAL_TRUECOLOR) {
  		u32 v;
  
  		if (regno >= 16)
  			return 1;
  
  		v = (red << info->var.red.offset) |
  		    (green << info->var.green.offset) |
  		    (blue << info->var.blue.offset) |
  		    (transp << info->var.transp.offset);
  		switch (info->var.bits_per_pixel) {
  		case 8:
  			break;
  		case 16:
  		case 24:
  		case 32:
  			((u32 *) (info->pseudo_palette))[regno] = v;
  			break;
  		default:
  			return 1;
  		}
  		return 0;
  	}
  	return 0;
  }
  
  /*
   *  Pan or Wrap the Display
   *
   *  This call looks only at xoffset, yoffset and the FB_VMODE_YWRAP flag
   */
  static int unifb_pan_display(struct fb_var_screeninfo *var,
  			   struct fb_info *info)
  {
  	if (var->vmode & FB_VMODE_YWRAP) {
  		if (var->yoffset < 0
  		    || var->yoffset >= info->var.yres_virtual
  		    || var->xoffset)
  			return -EINVAL;
  	} else {
  		if (var->xoffset + info->var.xres > info->var.xres_virtual ||
  		    var->yoffset + info->var.yres > info->var.yres_virtual)
  			return -EINVAL;
  	}
  	info->var.xoffset = var->xoffset;
  	info->var.yoffset = var->yoffset;
  	if (var->vmode & FB_VMODE_YWRAP)
  		info->var.vmode |= FB_VMODE_YWRAP;
  	else
  		info->var.vmode &= ~FB_VMODE_YWRAP;
  	return 0;
  }
  
  int unifb_mmap(struct fb_info *info,
  		    struct vm_area_struct *vma)
  {
  	vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
  
  	return vm_iomap_memory(vma, info->fix.smem_start, info->fix.smem_len);
  }
  
  static struct fb_ops unifb_ops = {
  	.fb_read        = fb_sys_read,
  	.fb_write       = fb_sys_write,
  	.fb_check_var	= unifb_check_var,
  	.fb_set_par	= unifb_set_par,
  	.fb_setcolreg	= unifb_setcolreg,
  	.fb_pan_display	= unifb_pan_display,
  	.fb_fillrect	= unifb_fillrect,
  	.fb_copyarea	= unifb_copyarea,
  	.fb_imageblit   = unifb_imageblit,
  	.fb_mmap	= unifb_mmap,
  };
  
  /*
   *  Initialisation
   */
  static int unifb_probe(struct platform_device *dev)
  {
  	struct fb_info *info;
  	u32 unifb_regs[UNIFB_REGS_NUM];
  	int retval = -ENOMEM;
  	struct resource *iomem;
  	void *videomemory;
  
  	videomemory = (void *)__get_free_pages(GFP_KERNEL | __GFP_COMP,
  				get_order(UNIFB_MEMSIZE));
  	if (!videomemory)
  		goto err;
  
  	memset(videomemory, 0, UNIFB_MEMSIZE);
  
  	unifb_fix.smem_start = virt_to_phys(videomemory);
  	unifb_fix.smem_len = UNIFB_MEMSIZE;
  
  	iomem = platform_get_resource(dev, IORESOURCE_MEM, 0);
  	unifb_fix.mmio_start = iomem->start;
  
  	info = framebuffer_alloc(sizeof(u32)*256, &dev->dev);
  	if (!info)
  		goto err;
  
  	info->screen_base = (char __iomem *)videomemory;
  	info->fbops = &unifb_ops;
  
  	retval = fb_find_mode(&info->var, info, NULL,
  			      unifb_modes, 10, &unifb_modes[0], 16);
  
  	if (!retval || (retval == 4))
  		info->var = unifb_default;
  
  	info->fix = unifb_fix;
  	info->pseudo_palette = info->par;
  	info->par = NULL;
  	info->flags = FBINFO_FLAG_DEFAULT;
  #ifdef FB_ACCEL_PUV3_UNIGFX
  	info->fix.accel = FB_ACCEL_PUV3_UNIGFX;
  #endif
  
  	retval = fb_alloc_cmap(&info->cmap, 256, 0);
  	if (retval < 0)
  		goto err1;
  
  	retval = register_framebuffer(info);
  	if (retval < 0)
  		goto err2;
  	platform_set_drvdata(dev, info);
  	platform_device_add_data(dev, unifb_regs, sizeof(u32) * UNIFB_REGS_NUM);
  
  	fb_info(info, "Virtual frame buffer device, using %dM of video memory
  ",
  		UNIFB_MEMSIZE >> 20);
  	return 0;
  err2:
  	fb_dealloc_cmap(&info->cmap);
  err1:
  	framebuffer_release(info);
  err:
  	return retval;
  }
  
  static int unifb_remove(struct platform_device *dev)
  {
  	struct fb_info *info = platform_get_drvdata(dev);
  
  	if (info) {
  		unregister_framebuffer(info);
  		fb_dealloc_cmap(&info->cmap);
  		framebuffer_release(info);
  	}
  	return 0;
  }
  
  #ifdef CONFIG_PM
  static int unifb_resume(struct platform_device *dev)
  {
  	int rc = 0;
  	u32 *unifb_regs = dev->dev.platform_data;
  
  	if (dev->dev.power.power_state.event == PM_EVENT_ON)
  		return 0;
  
  	console_lock();
  
  	if (dev->dev.power.power_state.event == PM_EVENT_SUSPEND) {
  		writel(unifb_regs[0], UDE_FSA);
  		writel(unifb_regs[1], UDE_LS);
  		writel(unifb_regs[2], UDE_PS);
  		writel(unifb_regs[3], UDE_HAT);
  		writel(unifb_regs[4], UDE_HBT);
  		writel(unifb_regs[5], UDE_HST);
  		writel(unifb_regs[6], UDE_VAT);
  		writel(unifb_regs[7], UDE_VBT);
  		writel(unifb_regs[8], UDE_VST);
  		writel(unifb_regs[9], UDE_CFG);
  	}
  	dev->dev.power.power_state = PMSG_ON;
  
  	console_unlock();
  
  	return rc;
  }
  
  static int unifb_suspend(struct platform_device *dev, pm_message_t mesg)
  {
  	u32 *unifb_regs = dev->dev.platform_data;
  
  	unifb_regs[0] = readl(UDE_FSA);
  	unifb_regs[1] = readl(UDE_LS);
  	unifb_regs[2] = readl(UDE_PS);
  	unifb_regs[3] = readl(UDE_HAT);
  	unifb_regs[4] = readl(UDE_HBT);
  	unifb_regs[5] = readl(UDE_HST);
  	unifb_regs[6] = readl(UDE_VAT);
  	unifb_regs[7] = readl(UDE_VBT);
  	unifb_regs[8] = readl(UDE_VST);
  	unifb_regs[9] = readl(UDE_CFG);
  
  	if (mesg.event == dev->dev.power.power_state.event)
  		return 0;
  
  	switch (mesg.event) {
  	case PM_EVENT_FREEZE:		/* about to take snapshot */
  	case PM_EVENT_PRETHAW:		/* before restoring snapshot */
  		goto done;
  	}
  
  	console_lock();
  
  	/* do nothing... */
  
  	console_unlock();
  
  done:
  	dev->dev.power.power_state = mesg;
  
  	return 0;
  }
  #else
  #define	unifb_resume	NULL
  #define unifb_suspend	NULL
  #endif
  
  static struct platform_driver unifb_driver = {
  	.probe	 = unifb_probe,
  	.remove  = unifb_remove,
  	.resume  = unifb_resume,
  	.suspend = unifb_suspend,
  	.driver  = {
  		.name	= "PKUnity-v3-UNIGFX",
  	},
  };
  
  static int __init unifb_init(void)
  {
  #ifndef MODULE
  	if (fb_get_options("unifb", NULL))
  		return -ENODEV;
  #endif
  
  	return platform_driver_register(&unifb_driver);
  }
  
  module_init(unifb_init);
  
  static void __exit unifb_exit(void)
  {
  	platform_driver_unregister(&unifb_driver);
  }
  
  module_exit(unifb_exit);
  
  MODULE_LICENSE("GPL v2");