/*
 * PSPLINK
 * -----------------------------------------------------------------------
 * Licensed under the BSD license, see LICENSE in PSPLINK root for details.
 *
 * bitmap.c - PSPLINK kernel module bitmap code
 *
 * Copyright (c) 2005 James F <tyranid@gmail.com>
 *
 * $HeadURL: svn://svn.ps2dev.org/psp/trunk/psplink/psplink/bitmap.c $
 * $Id: bitmap.c 1622 2005-12-27 19:29:57Z tyranid $
 */

#include <pspkernel.h>
#include <pspdisplay.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>

#define BMP_ID "BM"
int screenshotFd;

struct BitmapHeader
{
  char id[2];
  uint32_t filesize;
  uint32_t reserved;
  uint32_t offset;
  uint32_t headsize;
  uint32_t width;
  uint32_t height;
  uint16_t planes;
  uint16_t bpp;
  uint32_t comp;
  uint32_t bitmapsize;
  uint32_t hres;
  uint32_t vres;
  uint32_t colors;
  uint32_t impcolors;
} __attribute__((packed));

int write_8888_data(void *frame, int pixel_data)
{
  int line; /*uint8_t*/
  char lineOutput=0;
  uint8_t *p;
  int i;
  int h;

  line = pixel_data;
  for(h = 271; h >= 0; h--)
  {
    p = frame + (h*512*4);
    for(i = 0; i < 480; i++)
    {
      fileBuffer[((i*3) + 2)]=p[i*4];
      fileBuffer[((i*3) + 1)]=p[(i*4) + 1];
      fileBuffer[((i*3) + 0)]=p[(i*4) + 2];
    }
    sceIoLseek(screenshotFd, line, SEEK_SET); sceIoWrite(screenshotFd, &fileBuffer, 480*3);
    line += 480 * 3;
  }

  return 0;
}

int write_5551_data(void *frame, int pixel_data)
{
  int line; /*uint8_t*/
  char lineOutput=0;
  uint16_t *p;
  int i;
  int h;

  line = pixel_data;
  for(h = 271; h >= 0; h--)
  {
    p = frame;
    p += (h * 512);
    for(i = 0; i < 480; i++)
    {
      fileBuffer[((i*3) + 2)]=(p[i] & 0x1F) << 3;
      fileBuffer[((i*3) + 1)]=((p[i] >> 5) & 0x1F) << 3;
      fileBuffer[((i*3) + 0)]=((p[i] >> 10) & 0x1F) << 3;
    }
    sceIoLseek(screenshotFd, line, SEEK_SET); sceIoWrite(screenshotFd, &fileBuffer, 480*3);
    line += 480*3;
  }

  return 0;
}

int write_565_data(void *frame, int pixel_data)
{
  int line; /*uint8_t*/
  char lineOutput=0;
  uint16_t *p;
  int i;
  int h;

  line = pixel_data;
  for(h = 271; h >= 0; h--)
  {
    p = frame;
    p += (h * 512);
    for(i = 0; i < 480; i++)
    {
      fileBuffer[((i*3) + 2)]=(p[i] & 0x1F) << 3;
      fileBuffer[((i*3) + 1)]=((p[i] >> 5) & 0x3F) << 2;
      fileBuffer[((i*3) + 0)]=((p[i] >> 11) & 0x1F) << 3;
    }
    sceIoLseek(screenshotFd, line, SEEK_SET); sceIoWrite(screenshotFd, &fileBuffer, 480*3);
    line += 480*3;
  }

  return 0;
}

int write_4444_data(void *frame, int pixel_data)
{
  int line; /*uint8_t*/
  char lineOutput=0;
  uint16_t *p;
  int i;
  int h;

  line=pixel_data;
  for(h = 271; h >= 0; h--)
  {
    p = frame;
    p += (h * 512);
    for(i = 0; i < 480; i++)
    {
      fileBuffer[((i*3) + 2)]=(p[i] & 0xF) << 4;
      fileBuffer[((i*3) + 1)]=(p[i] & 0xF) << 4;
      fileBuffer[((i*3) + 0)]=(p[i] & 0xF) << 4;
    }
    sceIoLseek(screenshotFd, line, SEEK_SET); sceIoWrite(screenshotFd, &fileBuffer, 480*3);
    line += 480*3;
  }

  return 0;
}

int bitmapWrite(void *frame_addr, void *a_ignored, int format, const char *file)
{
  struct BitmapHeader bmp;
  int pixel_data=sizeof(struct BitmapHeader);

  memset(&bmp, 0, sizeof(struct BitmapHeader));
  memcpy(&bmp.id, BMP_ID, sizeof(bmp.id));
  bmp.filesize = 480*272*3 + sizeof(struct BitmapHeader);
  bmp.offset = sizeof(struct BitmapHeader);
  bmp.headsize = 0x28;
  bmp.width = 480;
  bmp.height = 272;
  bmp.planes = 1;
  bmp.bpp = 24;
  bmp.bitmapsize = 480*272*3;
  bmp.hres = 2834;
  bmp.vres = 2834;

  screenshotFd = sceIoOpen(file, PSP_O_WRONLY | PSP_O_CREAT | PSP_O_TRUNC, 0777);
  if(screenshotFd < 0)
  {
    //printf("Could not open file '%s' for writing\n", file);
    return -1;
  }

  sceIoWrite(screenshotFd, &bmp, sizeof(struct BitmapHeader));

  switch(format)
  {
    case PSP_DISPLAY_PIXEL_FORMAT_565: write_565_data(frame_addr, pixel_data);
                       break;
    case PSP_DISPLAY_PIXEL_FORMAT_5551: write_5551_data(frame_addr, pixel_data); 
                       break;
    case PSP_DISPLAY_PIXEL_FORMAT_4444: write_4444_data(frame_addr, pixel_data);
                      break;
    case PSP_DISPLAY_PIXEL_FORMAT_8888: write_8888_data(frame_addr, pixel_data);
                      break;
  };

  sceIoClose(screenshotFd);

  return 0;
}