text: /*
   * Copyright (C) 2010-2013 Bluecherry, LLC <http://www.bluecherrydvr.com>
   *
   * Original author:
   * Ben Collins <bcollins@ubuntu.com>
   *
   * Additional work by:
   * John Brooks <john.brooks@bluecherry.net>
   *
   * 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.
   */
  
  #include <linux/kernel.h>
  #include <linux/font.h>
  #include <linux/bitrev.h>
  #include <linux/slab.h>
  
  #include "solo6x10.h"
  
  #define VI_PROG_HSIZE			(1280 - 16)
  #define VI_PROG_VSIZE			(1024 - 16)
  
  #define IRQ_LEVEL			2
  
  static void solo_capture_config(struct solo_dev *solo_dev)
  {
  	unsigned long height;
  	unsigned long width;
  	void *buf;
  	int i;
  
  	solo_reg_write(solo_dev, SOLO_CAP_BASE,
  		       SOLO_CAP_MAX_PAGE((SOLO_CAP_EXT_SIZE(solo_dev)
  					  - SOLO_CAP_PAGE_SIZE) >> 16)
  		       | SOLO_CAP_BASE_ADDR(SOLO_CAP_EXT_ADDR(solo_dev) >> 16));
  
  	/* XXX: Undocumented bits at b17 and b24 */
  	if (solo_dev->type == SOLO_DEV_6110) {
  		/* NOTE: Ref driver has (62 << 24) here as well, but it causes
  		 * wacked out frame timing on 4-port 6110. */
  		solo_reg_write(solo_dev, SOLO_CAP_BTW,
  			       (1 << 17) | SOLO_CAP_PROG_BANDWIDTH(2) |
  			       SOLO_CAP_MAX_BANDWIDTH(36));
  	} else {
  		solo_reg_write(solo_dev, SOLO_CAP_BTW,
  			       (1 << 17) | SOLO_CAP_PROG_BANDWIDTH(2) |
  			       SOLO_CAP_MAX_BANDWIDTH(32));
  	}
  
  	/* Set scale 1, 9 dimension */
  	width = solo_dev->video_hsize;
  	height = solo_dev->video_vsize;
  	solo_reg_write(solo_dev, SOLO_DIM_SCALE1,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 8) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Set scale 2, 10 dimension */
  	width = solo_dev->video_hsize / 2;
  	height = solo_dev->video_vsize;
  	solo_reg_write(solo_dev, SOLO_DIM_SCALE2,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 8) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Set scale 3, 11 dimension */
  	width = solo_dev->video_hsize / 2;
  	height = solo_dev->video_vsize / 2;
  	solo_reg_write(solo_dev, SOLO_DIM_SCALE3,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 8) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Set scale 4, 12 dimension */
  	width = solo_dev->video_hsize / 3;
  	height = solo_dev->video_vsize / 3;
  	solo_reg_write(solo_dev, SOLO_DIM_SCALE4,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 8) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Set scale 5, 13 dimension */
  	width = solo_dev->video_hsize / 4;
  	height = solo_dev->video_vsize / 2;
  	solo_reg_write(solo_dev, SOLO_DIM_SCALE5,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 8) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Progressive */
  	width = VI_PROG_HSIZE;
  	height = VI_PROG_VSIZE;
  	solo_reg_write(solo_dev, SOLO_DIM_PROG,
  		       SOLO_DIM_H_MB_NUM(width / 16) |
  		       SOLO_DIM_V_MB_NUM_FRAME(height / 16) |
  		       SOLO_DIM_V_MB_NUM_FIELD(height / 16));
  
  	/* Clear OSD */
  	solo_reg_write(solo_dev, SOLO_VE_OSD_CH, 0);
  	solo_reg_write(solo_dev, SOLO_VE_OSD_BASE, SOLO_EOSD_EXT_ADDR >> 16);
  	solo_reg_write(solo_dev, SOLO_VE_OSD_CLR,
  		       0xF0 << 16 | 0x80 << 8 | 0x80);
  
  	if (solo_dev->type == SOLO_DEV_6010)
  		solo_reg_write(solo_dev, SOLO_VE_OSD_OPT,
  			       SOLO_VE_OSD_H_SHADOW | SOLO_VE_OSD_V_SHADOW);
  	else
  		solo_reg_write(solo_dev, SOLO_VE_OSD_OPT, SOLO_VE_OSD_V_DOUBLE
  			       | SOLO_VE_OSD_H_SHADOW | SOLO_VE_OSD_V_SHADOW);
  
  	/* Clear OSG buffer */
  	buf = kzalloc(SOLO_EOSD_EXT_SIZE(solo_dev), GFP_KERNEL);
  	if (!buf)
  		return;
  
  	for (i = 0; i < solo_dev->nr_chans; i++) {
  		solo_p2m_dma(solo_dev, 1, buf,
  			     SOLO_EOSD_EXT_ADDR +
  			     (SOLO_EOSD_EXT_SIZE(solo_dev) * i),
  			     SOLO_EOSD_EXT_SIZE(solo_dev), 0, 0);
  	}
  	kfree(buf);
  }
  
  #define SOLO_OSD_WRITE_SIZE (16 * OSD_TEXT_MAX)
  
  /* Should be called with enable_lock held */
  int solo_osd_print(struct solo_enc_dev *solo_enc)
  {
  	struct solo_dev *solo_dev = solo_enc->solo_dev;
  	unsigned char *str = solo_enc->osd_text;
  	u8 *buf = solo_enc->osd_buf;
  	u32 reg;
  	const struct font_desc *vga = find_font("VGA8x16");
  	const unsigned char *vga_data;
  	int i, j;
  
  	if (WARN_ON_ONCE(!vga))
  		return -ENODEV;
  
  	reg = solo_reg_read(solo_dev, SOLO_VE_OSD_CH);
  	if (!*str) {
  		/* Disable OSD on this channel */
  		reg &= ~(1 << solo_enc->ch);
  		goto out;
  	}
  
  	memset(buf, 0, SOLO_OSD_WRITE_SIZE);
  	vga_data = (const unsigned char *)vga->data;
  
  	for (i = 0; *str; i++, str++) {
  		for (j = 0; j < 16; j++) {
  			buf[(j << 1) | (i & 1) | ((i & ~1) << 4)] =
  			    bitrev8(vga_data[(*str << 4) | j]);
  		}
  	}
  
  	solo_p2m_dma(solo_dev, 1, buf,
  		     SOLO_EOSD_EXT_ADDR_CHAN(solo_dev, solo_enc->ch),
  		     SOLO_OSD_WRITE_SIZE, 0, 0);
  
  	/* Enable OSD on this channel */
  	reg |= (1 << solo_enc->ch);
  
  out:
  	solo_reg_write(solo_dev, SOLO_VE_OSD_CH, reg);
  	return 0;
  }
  
  /**
   * Set channel Quality Profile (0-3).
   */
  void solo_s_jpeg_qp(struct solo_dev *solo_dev, unsigned int ch,
  		    unsigned int qp)
  {
  	unsigned long flags;
  	unsigned int idx, reg;
  
  	if ((ch > 31) || (qp > 3))
  		return;
  
  	if (solo_dev->type == SOLO_DEV_6010)
  		return;
  
  	if (ch < 16) {
  		idx = 0;
  		reg = SOLO_VE_JPEG_QP_CH_L;
  	} else {
  		ch -= 16;
  		idx = 1;
  		reg = SOLO_VE_JPEG_QP_CH_H;
  	}
  	ch *= 2;
  
  	spin_lock_irqsave(&solo_dev->jpeg_qp_lock, flags);
  
  	solo_dev->jpeg_qp[idx] &= ~(3 << ch);
  	solo_dev->jpeg_qp[idx] |= (qp & 3) << ch;
  
  	solo_reg_write(solo_dev, reg, solo_dev->jpeg_qp[idx]);
  
  	spin_unlock_irqrestore(&solo_dev->jpeg_qp_lock, flags);
  }
  
  int solo_g_jpeg_qp(struct solo_dev *solo_dev, unsigned int ch)
  {
  	int idx;
  
  	if (solo_dev->type == SOLO_DEV_6010)
  		return 2;
  
  	if (WARN_ON_ONCE(ch > 31))
  		return 2;
  
  	if (ch < 16) {
  		idx = 0;
  	} else {
  		ch -= 16;
  		idx = 1;
  	}
  	ch *= 2;
  
  	return (solo_dev->jpeg_qp[idx] >> ch) & 3;
  }
  
  #define SOLO_QP_INIT 0xaaaaaaaa
  
  static void solo_jpeg_config(struct solo_dev *solo_dev)
  {
  	if (solo_dev->type == SOLO_DEV_6010) {
  		solo_reg_write(solo_dev, SOLO_VE_JPEG_QP_TBL,
  			       (2 << 24) | (2 << 16) | (2 << 8) | 2);
  	} else {
  		solo_reg_write(solo_dev, SOLO_VE_JPEG_QP_TBL,
  			       (4 << 24) | (3 << 16) | (2 << 8) | 1);
  	}
  
  	spin_lock_init(&solo_dev->jpeg_qp_lock);
  
  	/* Initialize Quality Profile for all channels */
  	solo_dev->jpeg_qp[0] = solo_dev->jpeg_qp[1] = SOLO_QP_INIT;
  	solo_reg_write(solo_dev, SOLO_VE_JPEG_QP_CH_L, SOLO_QP_INIT);
  	solo_reg_write(solo_dev, SOLO_VE_JPEG_QP_CH_H, SOLO_QP_INIT);
  
  	solo_reg_write(solo_dev, SOLO_VE_JPEG_CFG,
  		(SOLO_JPEG_EXT_SIZE(solo_dev) & 0xffff0000) |
  		((SOLO_JPEG_EXT_ADDR(solo_dev) >> 16) & 0x0000ffff));
  	solo_reg_write(solo_dev, SOLO_VE_JPEG_CTRL, 0xffffffff);
  	if (solo_dev->type == SOLO_DEV_6110) {
  		solo_reg_write(solo_dev, SOLO_VE_JPEG_CFG1,
  			       (0 << 16) | (30 << 8) | 60);
  	}
  }
  
  static void solo_mp4e_config(struct solo_dev *solo_dev)
  {
  	int i;
  	u32 cfg;
  
  	solo_reg_write(solo_dev, SOLO_VE_CFG0,
  		       SOLO_VE_INTR_CTRL(IRQ_LEVEL) |
  		       SOLO_VE_BLOCK_SIZE(SOLO_MP4E_EXT_SIZE(solo_dev) >> 16) |
  		       SOLO_VE_BLOCK_BASE(SOLO_MP4E_EXT_ADDR(solo_dev) >> 16));
  
  
  	cfg = SOLO_VE_BYTE_ALIGN(2) | SOLO_VE_INSERT_INDEX
  		| SOLO_VE_MOTION_MODE(0);
  	if (solo_dev->type != SOLO_DEV_6010) {
  		cfg |= SOLO_VE_MPEG_SIZE_H(
  			(SOLO_MP4E_EXT_SIZE(solo_dev) >> 24) & 0x0f);
  		cfg |= SOLO_VE_JPEG_SIZE_H(
  			(SOLO_JPEG_EXT_SIZE(solo_dev) >> 24) & 0x0f);
  	}
  	solo_reg_write(solo_dev, SOLO_VE_CFG1, cfg);
  
  	solo_reg_write(solo_dev, SOLO_VE_WMRK_POLY, 0);
  	solo_reg_write(solo_dev, SOLO_VE_VMRK_INIT_KEY, 0);
  	solo_reg_write(solo_dev, SOLO_VE_WMRK_STRL, 0);
  	if (solo_dev->type == SOLO_DEV_6110)
  		solo_reg_write(solo_dev, SOLO_VE_WMRK_ENABLE, 0);
  	solo_reg_write(solo_dev, SOLO_VE_ENCRYP_POLY, 0);
  	solo_reg_write(solo_dev, SOLO_VE_ENCRYP_INIT, 0);
  
  	solo_reg_write(solo_dev, SOLO_VE_ATTR,
  		       SOLO_VE_LITTLE_ENDIAN |
  		       SOLO_COMP_ATTR_FCODE(1) |
  		       SOLO_COMP_TIME_INC(0) |
  		       SOLO_COMP_TIME_WIDTH(15) |
  		       SOLO_DCT_INTERVAL(solo_dev->type == SOLO_DEV_6010 ? 9 : 10));
  
  	for (i = 0; i < solo_dev->nr_chans; i++) {
  		solo_reg_write(solo_dev, SOLO_VE_CH_REF_BASE(i),
  			       (SOLO_EREF_EXT_ADDR(solo_dev) +
  			       (i * SOLO_EREF_EXT_SIZE)) >> 16);
  		solo_reg_write(solo_dev, SOLO_VE_CH_REF_BASE_E(i),
  			       (SOLO_EREF_EXT_ADDR(solo_dev) +
  			       ((i + 16) * SOLO_EREF_EXT_SIZE)) >> 16);
  	}
  
  	if (solo_dev->type == SOLO_DEV_6110) {
  		solo_reg_write(solo_dev, SOLO_VE_COMPT_MOT, 0x00040008);
  	} else {
  		for (i = 0; i < solo_dev->nr_chans; i++)
  			solo_reg_write(solo_dev, SOLO_VE_CH_MOT(i), 0x100);
  	}
  }
  
  int solo_enc_init(struct solo_dev *solo_dev)
  {
  	int i;
  
  	solo_capture_config(solo_dev);
  	solo_mp4e_config(solo_dev);
  	solo_jpeg_config(solo_dev);
  
  	for (i = 0; i < solo_dev->nr_chans; i++) {
  		solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(i), 0);
  		solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(i), 0);
  	}
  
  	return 0;
  }
  
  void solo_enc_exit(struct solo_dev *solo_dev)
  {
  	int i;
  
  	for (i = 0; i < solo_dev->nr_chans; i++) {
  		solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(i), 0);
  		solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(i), 0);
  	}
  }
  