/**
    VIDIX driver for Hauppauge PVR 350.

    Copyright 2007 Lutz Koschorreck.

    Based on genfb_vid.c and ivtv_xv.c

    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 2 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, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    09.05.2007 Lutz Koschorreck
    First version: Tested with ivtv-0.10.1, xine-ui-0.99.5, xine-lib-1.1.6

**/

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <math.h>
#include <inttypes.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <linux/types.h>
#include <linux/ivtv.h>

#include "vidix.h"
#include "vidixlib.h"
#include "fourcc.h"
#include "dha.h"
#include "pci_ids.h"
#include "pci_names.h"

#define VIDIX_STATIC ivtv_

#define IVTV_MSG "[ivtv-vid] "
#define MAXLINE 128
#define IVTVMAXWIDTH 720
#define IVTVMAXHEIGHT 576

static int yuvdev = 0;
static void *memBase = 0;
static int frameSize = 0;
static int probed = 0;
static int ivtv_verbose;
static vidix_rect_t destVideo;
static vidix_rect_t srcVideo;
static unsigned char *outbuf = NULL;

/* VIDIX exports */

static vidix_capability_t ivtv_cap =
{
  "Hauppauge PVR 350 YUV Video",
  "Lutz Koschorreck",
  TYPE_OUTPUT,
  { 0, 0, 0, 0 },
  IVTVMAXHEIGHT,
  IVTVMAXWIDTH,
  4,
  4,
  -1,
  FLAG_UPSCALER|FLAG_DOWNSCALER,
  -1,
  -1,
  { 0, 0, 0, 0 }
};

static void de_macro_y (unsigned char *src, unsigned char *dst,
                        unsigned int w, unsigned int h,
                        int src_x, int src_y,
                        int height __attribute__ ((unused)), int width)
{
  unsigned int x, y, i;
  unsigned char *dst_2;
  unsigned int h_tail, w_tail;
  unsigned int h_size, w_size;

  // Always round the origin, but compensate by increasing the size
  if (src_x & 15)
  {
    w += src_x & 15;
    src_x &= ~15;
  }

  if (src_y & 15)
  {
    h += src_y & 15;
    src_y &= ~15;
  }

  // The right / bottom edge might not be a multiple of 16
  h_tail = h & 15;
  w_tail = w & 15;

  // One block is 16 pixels high
  h_size = 16;

  // descramble Y plane
  for (y = 0; y < h; y += 16)
  {
    // Clip if we've reached the bottom & the size isn't a multiple of 16
    if (y + 16 > h) h_size = h_tail;

    for (x = 0; x < w; x += 16)
    {
      if (x + 16 > w)
        w_size = w_tail;
      else
        w_size = 16;

      dst_2 = dst + (720 * y) + (720 * src_y) + (256 * (src_x>>4)) + (x * 16);

      for (i = 0; i < h_size; i++)
      {
        memcpy (dst_2, src + src_x + x + (y + i) * width + (src_y * width),
                w_size);
        dst_2 += 16;
      }
    }
  }
}

static void de_macro_uv (unsigned char *srcu, unsigned char *srcv,
                         unsigned char *dst, unsigned int w, unsigned int h,
                         int src_x, int src_y, int height, int width)
{
  unsigned int x, y, i, f;
  unsigned char *dst_2;
  unsigned int h_tail, w_tail;
  unsigned int h_size;

  /* The uv plane is half the size of the y plane,
     so 'correct' all dimensions. */
  w /= 2;
  h /= 2;
  src_x /= 2;
  src_y /= 2;
  height /= 2;
  width /= 2;

  // Always round the origin, but compensate by increasing the size
  if (src_x & 7)
  {
    w += src_x & 7;
    src_x &= ~7;
  }

  if (src_y & 15)
  {
    h += src_y & 15;
    src_y &= ~15;
  }

  // The right / bottom edge may not be a multiple of 16
  h_tail = h & 15;
  w_tail = w & 7;

  h_size = 16;

  // descramble U/V plane
  for (y = 0; y < h; y += 16)
  {
    if (y + 16 > h)
      h_size = h_tail;

    for (x = 0; x < w; x += 8)
    {
      dst_2 = dst + (720 * y) + (720 * src_y) + (256 * (src_x>>3)) + (x * 32);
      if (x + 8 <= w)
      {
        for (i = 0; i < h_size; i++)
        {
          int idx = src_x + x + ((y + i) * width) + (src_y * width);
          dst_2[0] = srcu[idx + 0];
          dst_2[1] = srcv[idx + 0];
          dst_2[2] = srcu[idx + 1];
          dst_2[3] = srcv[idx + 1];
          dst_2[4] = srcu[idx + 2];
          dst_2[5] = srcv[idx + 2];
          dst_2[6] = srcu[idx + 3];
          dst_2[7] = srcv[idx + 3];
          dst_2[8] = srcu[idx + 4];
          dst_2[9] = srcv[idx + 4];
          dst_2[10] = srcu[idx + 5];
          dst_2[11] = srcv[idx + 5];
          dst_2[12] = srcu[idx + 6];
          dst_2[13] = srcv[idx + 6];
          dst_2[14] = srcu[idx + 7];
          dst_2[15] = srcv[idx + 7];
          dst_2 += 16;
        }
      }
      else
      {
        for (i = 0; i < h_size; i ++)
        {
          int idx = src_x + x + ((y + i) * width) + (src_y * width);
          for (f = 0; f < w_tail; f++)
          {
            dst_2[0] = srcu[idx + f];
            dst_2[1] = srcv[idx + f];
            dst_2 += 2;
          }

          dst_2 += 16 - (w_tail << 1);
        }
      }
    }
  }
}

static int ivtv_probe (int verbose, int force __attribute__ ((unused)))
{
  pciinfo_t lst[MAX_PCI_DEVICES];
  unsigned int i, num_pci;
  int err;
  FILE *procFb;
  unsigned char fb_number = 0;
  unsigned char yuv_device_number, yuv_device;
  char yuv_device_name[] = "/dev/videoXXX\0";
  
  if (verbose)
    printf (IVTV_MSG"probe\n");

  ivtv_verbose = verbose;

  err = pci_scan (lst, &num_pci);
  if (err)
  {
    printf (IVTV_MSG"Error occured during pci scan: %s\n", strerror (err));
    return err;
  }

  if (ivtv_verbose)
    printf (IVTV_MSG"Found %d pci devices\n", num_pci);

  for (i = 0; i < num_pci; i++)
  {
    if (ivtv_verbose == 2)
      printf (IVTV_MSG"Found chip [%04X:%04X] '%s' '%s'\n",
              lst[i].vendor, lst[i].device, pci_vendor_name (lst[i].vendor),
              pci_device_name(lst[i].vendor,lst[i].device));
    
    if (VENDOR_INTERNEXT == lst[i].vendor)
    {
      switch (lst[i].device)
      {
      case DEVICE_INTERNEXT_ITVC15_MPEG_2_ENCODER:
        if (ivtv_verbose)
          printf (IVTV_MSG"Found PVR 350\n");
        goto card_found;
      }
    }
  }
	
  if (ivtv_verbose)
    printf (IVTV_MSG"Can't find chip\n");

  return (ENXIO);

 card_found:
	
  /* Try to find framebuffer device */
  procFb = fopen ("/proc/fb", "r");
  if (procFb)
  {
    char procEntry[MAXLINE] = {0};
    while (NULL != fgets(procEntry, MAXLINE, procFb))
    {
      char *pos = NULL;
      if (ivtv_verbose)
        printf (IVTV_MSG" %s", procEntry);
      
      if (NULL != (pos = strstr(procEntry, " cx23415 TV out")))
      {
        *pos = '\0';
        fb_number = atoi (procEntry);
        if (ivtv_verbose)
          printf (IVTV_MSG"Framebuffer found #%u\n", fb_number);
        goto fb_found;
      }
    }
  }
  else
  {
    if (ivtv_verbose)
      printf (IVTV_MSG"Framebuffer device not found\n");
    return (ENXIO);
  }

 fb_found:
  fclose (procFb);
	
  /* Try to find YUV device */
  yuv_device_number = 48;
  yuv_device = 48 + fb_number;
	
  do {
    sprintf (yuv_device_name, "/dev/video%u", yuv_device);
    yuvdev = open (yuv_device_name, O_RDWR);
    if (-1 != yuvdev)
    {
      if (ivtv_verbose)
        printf (IVTV_MSG"YUV device found /dev/video%u\n", yuv_device);
      goto yuv_found;
    }
    else
    {
      if (ivtv_verbose)
        printf (IVTV_MSG"YUV device not found: /dev/video%u\n", yuv_device);
    }
  } while (yuv_device-- > yuv_device_number);
	
  return (ENXIO);

 yuv_found:
  probed = 1;
  return 0;
}

static int ivtv_init (void)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"init\n");
    
  if (!probed)
  {
    if (ivtv_verbose)
      printf (IVTV_MSG"Driver was not probed but is being initialized\n");

    return (EINTR);
  }
  
  outbuf = malloc ((IVTVMAXHEIGHT * IVTVMAXWIDTH)
                   + (IVTVMAXHEIGHT * IVTVMAXWIDTH / 2));

  if (!outbuf)
  {
    if (ivtv_verbose)
      printf (IVTV_MSG"Not enough memory availabe!\n");
    return (EINTR);
  }
	
  return 0;
}

static void ivtv_destroy (void)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"destory\n");
  close (yuvdev);	
  free (outbuf);
}

static int ivtv_get_caps (vidix_capability_t *to)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"GetCap\n");
  memcpy (to, &ivtv_cap, sizeof (vidix_capability_t));

  return 0;
}

static int ivtv_query_fourcc (vidix_fourcc_t *to)
{
  int supports = 0;
  
  if (ivtv_verbose)
    printf (IVTV_MSG"query fourcc (%x)\n", to->fourcc);
	
  switch (to->fourcc)
  {
  case IMGFMT_YV12:
    supports = 1;
    break;
  default:
    supports = 0;
  }
	
  if (!supports)
  {
    to->depth = to->flags = 0;
    return (ENOTSUP);
  }
  
  to->depth = VID_DEPTH_12BPP |
    VID_DEPTH_15BPP | VID_DEPTH_16BPP |
    VID_DEPTH_24BPP | VID_DEPTH_32BPP;
  to->flags = 0;

  return 0;
}

static int ivtv_config_playback (vidix_playback_t *info)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"config playback\n");

  if (2 == ivtv_verbose)
  {
    printf (IVTV_MSG"src : x:%d y:%d w:%d h:%d\n", 
            info->src.x, info->src.y, info->src.w, info->src.h);
    printf (IVTV_MSG"dest: x:%d y:%d w:%d h:%d\n", 
            info->dest.x, info->dest.y, info->dest.w, info->dest.h);
  }
	
  memcpy (&destVideo, &info->dest, sizeof (vidix_rect_t));
  memcpy (&srcVideo, &info->src, sizeof (vidix_rect_t));

  info->num_frames = 2;
  info->frame_size = frameSize =
    info->src.w*info->src.h+(info->src.w*info->src.h)/2;
  info->dest.pitch.y = 16;
  info->dest.pitch.u = info->dest.pitch.v = 16;
  info->offsets[0] = 0;
  info->offsets[1] = info->frame_size;
  info->offset.y = 0;
  info->offset.u = IVTVMAXWIDTH * IVTVMAXHEIGHT;
  info->offset.v = IVTVMAXWIDTH * IVTVMAXHEIGHT
     + (IVTVMAXWIDTH / 2) * (IVTVMAXHEIGHT / 2);    

  info->dga_addr = memBase = malloc (info->num_frames*info->frame_size);   

  if (ivtv_verbose)
    printf (IVTV_MSG"frame_size: %d, dga_addr: %p\n",
            info->frame_size, info->dga_addr);

  return 0;
}

static int ivtv_playback_on (void)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"playback on\n");

  return 0;
}

static int ivtv_playback_off (void)
{
  if (ivtv_verbose)
    printf (IVTV_MSG"playback off\n");

  return 0;
}

static int ivtv_frame_sel (unsigned int frame)
{
  struct ivtvyuv_ioctl_dma_host_to_ivtv_args args;

  de_macro_y ((memBase + (frame * frameSize)),
              outbuf, srcVideo.w, srcVideo.h,
              srcVideo.x, srcVideo.y, destVideo.h, destVideo.w);

  de_macro_uv ((memBase + (frame * frameSize))
               + (srcVideo.w * srcVideo.h) + srcVideo.w * srcVideo.h / 4,
               (memBase + (frame * frameSize)) + (srcVideo.w * srcVideo.h),
               outbuf + IVTVMAXWIDTH * IVTVMAXHEIGHT,
               srcVideo.w, srcVideo.h, srcVideo.x, srcVideo.y,
               destVideo.h, destVideo.w);

  args.y_source = outbuf;
  args.uv_source = outbuf + (IVTVMAXWIDTH * IVTVMAXHEIGHT);
  args.src_x = srcVideo.x;
  args.src_y = srcVideo.y;
  args.dst_x = destVideo.x;
  args.dst_y = destVideo.y;
  args.src_w = srcVideo.w;
  args.dst_w = destVideo.w;
  args.srcBuf_width = srcVideo.w;
  args.src_h = srcVideo.h;
  args.dst_h = destVideo.h;
  args.srcBuf_height = srcVideo.h;
  args.yuv_type = 0;
	
  if(ioctl(yuvdev, IVTV_IOC_PREP_FRAME_YUV, &args) == -1)
    printf ("Ioctl IVTV_IOC_PREP_FRAME_YUV returned failed Error\n");

  return 0;
}

VDXDriver ivtv_drv = {
  "ivtv",
  NULL,
  .probe = ivtv_probe,
  .get_caps = ivtv_get_caps,
  .query_fourcc = ivtv_query_fourcc,
  .init = ivtv_init,
  .destroy = ivtv_destroy,
  .config_playback = ivtv_config_playback,
  .playback_on = ivtv_playback_on,
  .playback_off = ivtv_playback_off,
  .frame_sel = ivtv_frame_sel,
};