/*
 * Zoran ZR36050 basic configuration functions
 *
 * Copyright (C) 2001 Wolfgang Scherr <scherr@net4you.at>
 *
 * $Id: zr36050.c,v 1.1.2.11 2003/08/03 14:54:53 rbultje Exp $
 *
 * ------------------------------------------------------------------------
 *
 * 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., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 * ------------------------------------------------------------------------
 */

#define ZR050_VERSION "v0.7.1"

#include <linux/module.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/delay.h>

#include <linux/types.h>
#include <linux/wait.h>

/* I/O commands, error codes */
#include <asm/io.h>

/* headerfile of this module */
#include "zr36050.h"

/* codec io API */
#include "videocodec.h"

/* it doesn't make sense to have more than 20 or so,
  just to prevent some unwanted loops */
#define MAX_CODECS 20

/* amount of chips attached via this driver */
static int zr36050_codecs;

/* debugging is available via module parameter */
static int debug;
module_param(debug, int, 0);
MODULE_PARM_DESC(debug, "Debug level (0-4)");

#define dprintk(num, format, args...) \
        do { \
                if (debug >= num) \
                        printk(format, ##args); \
        } while (0)

/* =========================================================================
   Local hardware I/O functions:

   read/write via codec layer (registers are located in the master device)
   ========================================================================= */

/* read and write functions */
static u8
zr36050_read (struct zr36050 *ptr,
              u16             reg)
{
        u8 value = 0;

        // just in case something is wrong...
        if (ptr->codec->master_data->readreg)
                value = (ptr->codec->master_data->readreg(ptr->codec,
                                                          reg)) & 0xFF;
        else
                dprintk(1,
                        KERN_ERR "%s: invalid I/O setup, nothing read!\n",
                        ptr->name);

        dprintk(4, "%s: reading from 0x%04x: %02x\n", ptr->name, reg,
                value);

        return value;
}

static void
zr36050_write (struct zr36050 *ptr,
               u16             reg,
               u8              value)
{
        dprintk(4, "%s: writing 0x%02x to 0x%04x\n", ptr->name, value,
                reg);

        // just in case something is wrong...
        if (ptr->codec->master_data->writereg)
                ptr->codec->master_data->writereg(ptr->codec, reg, value);
        else
                dprintk(1,
                        KERN_ERR
                        "%s: invalid I/O setup, nothing written!\n",
                        ptr->name);
}

/* =========================================================================
   Local helper function:

   status read
   ========================================================================= */

/* status is kept in datastructure */
static u8
zr36050_read_status1 (struct zr36050 *ptr)
{
        ptr->status1 = zr36050_read(ptr, ZR050_STATUS_1);

        zr36050_read(ptr, 0);
        return ptr->status1;
}

/* =========================================================================
   Local helper function:

   scale factor read
   ========================================================================= */

/* scale factor is kept in datastructure */
static u16
zr36050_read_scalefactor (struct zr36050 *ptr)
{
        ptr->scalefact = (zr36050_read(ptr, ZR050_SF_HI) << 8) |
                         (zr36050_read(ptr, ZR050_SF_LO) & 0xFF);

        /* leave 0 selected for an eventually GO from master */
        zr36050_read(ptr, 0);
        return ptr->scalefact;
}

/* =========================================================================
   Local helper function:

   wait if codec is ready to proceed (end of processing) or time is over
   ========================================================================= */

static void
zr36050_wait_end (struct zr36050 *ptr)
{
        int i = 0;

        while (!(zr36050_read_status1(ptr) & 0x4)) {
                udelay(1);
                if (i++ > 200000) {     // 200ms, there is for sure something wrong!!!
                        dprintk(1,
                                "%s: timeout at wait_end (last status: 0x%02x)\n",
                                ptr->name, ptr->status1);
                        break;
                }
        }
}

/* =========================================================================
   Local helper function:

   basic test of "connectivity", writes/reads to/from memory the SOF marker
   ========================================================================= */

static int
zr36050_basic_test (struct zr36050 *ptr)
{
        zr36050_write(ptr, ZR050_SOF_IDX, 0x00);
        zr36050_write(ptr, ZR050_SOF_IDX + 1, 0x00);
        if ((zr36050_read(ptr, ZR050_SOF_IDX) |
             zr36050_read(ptr, ZR050_SOF_IDX + 1)) != 0x0000) {
                dprintk(1,
                        KERN_ERR
                        "%s: attach failed, can't connect to jpeg processor!\n",
                        ptr->name);
                return -ENXIO;
        }
        zr36050_write(ptr, ZR050_SOF_IDX, 0xff);
        zr36050_write(ptr, ZR050_SOF_IDX + 1, 0xc0);
        if (((zr36050_read(ptr, ZR050_SOF_IDX) << 8) |
             zr36050_read(ptr, ZR050_SOF_IDX + 1)) != 0xffc0) {
                dprintk(1,
                        KERN_ERR
                        "%s: attach failed, can't connect to jpeg processor!\n",
                        ptr->name);
                return -ENXIO;
        }

        zr36050_wait_end(ptr);
        if ((ptr->status1 & 0x4) == 0) {
                dprintk(1,
                        KERN_ERR
                        "%s: attach failed, jpeg processor failed (end flag)!\n",
                        ptr->name);
                return -EBUSY;
        }

        return 0;               /* looks good! */
}

/* =========================================================================
   Local helper function:

   simple loop for pushing the init datasets
   ========================================================================= */

static int
zr36050_pushit (struct zr36050 *ptr,
                u16             startreg,
                u16             len,
                const char     *data)
{
        int i = 0;

        dprintk(4, "%s: write data block to 0x%04x (len=%d)\n", ptr->name,
                startreg, len);
        while (i < len) {
                zr36050_write(ptr, startreg++, data[i++]);
        }

        return i;
}

/* =========================================================================
   Basic datasets:

   jpeg baseline setup data (you find it on lots places in internet, or just
   extract it from any regular .jpg image...)

   Could be variable, but until it's not needed it they are just fixed to save
   memory. Otherwise expand zr36050 structure with arrays, push the values to
   it and initialize from there, as e.g. the linux zr36057/60 driver does it.
   ========================================================================= */

static const char zr36050_dqt[0x86] = {
        0xff, 0xdb,             //Marker: DQT
        0x00, 0x84,             //Length: 2*65+2
        0x00,                   //Pq,Tq first table
        0x10, 0x0b, 0x0c, 0x0e, 0x0c, 0x0a, 0x10, 0x0e,
        0x0d, 0x0e, 0x12, 0x11, 0x10, 0x13, 0x18, 0x28,
        0x1a, 0x18, 0x16, 0x16, 0x18, 0x31, 0x23, 0x25,
        0x1d, 0x28, 0x3a, 0x33, 0x3d, 0x3c, 0x39, 0x33,
        0x38, 0x37, 0x40, 0x48, 0x5c, 0x4e, 0x40, 0x44,
        0x57, 0x45, 0x37, 0x38, 0x50, 0x6d, 0x51, 0x57,
        0x5f, 0x62, 0x67, 0x68, 0x67, 0x3e, 0x4d, 0x71,
        0x79, 0x70, 0x64, 0x78, 0x5c, 0x65, 0x67, 0x63,
        0x01,                   //Pq,Tq second table
        0x11, 0x12, 0x12, 0x18, 0x15, 0x18, 0x2f, 0x1a,
        0x1a, 0x2f, 0x63, 0x42, 0x38, 0x42, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63,
        0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63, 0x63
};

static const char zr36050_dht[0x1a4] = {
        0xff, 0xc4,             //Marker: DHT
        0x01, 0xa2,             //Length: 2*AC, 2*DC
        0x00,                   //DC first table
        0x00, 0x01, 0x05, 0x01, 0x01, 0x01, 0x01, 0x01,
        0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
        0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B,
        0x01,                   //DC second table
        0x00, 0x03, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
        0x01, 0x01, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
        0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B,
        0x10,                   //AC first table
        0x00, 0x02, 0x01, 0x03, 0x03, 0x02, 0x04, 0x03,
        0x05, 0x05, 0x04, 0x04, 0x00, 0x00,
        0x01, 0x7D, 0x01, 0x02, 0x03, 0x00, 0x04, 0x11,
        0x05, 0x12, 0x21, 0x31, 0x41, 0x06, 0x13, 0x51, 0x61,
        0x07, 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xA1,
        0x08, 0x23, 0x42, 0xB1, 0xC1, 0x15, 0x52, 0xD1, 0xF0, 0x24,
        0x33, 0x62, 0x72, 0x82, 0x09, 0x0A, 0x16, 0x17,
        0x18, 0x19, 0x1A, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2A, 0x34,
        0x35, 0x36, 0x37, 0x38, 0x39, 0x3A, 0x43, 0x44,
        0x45, 0x46, 0x47, 0x48, 0x49, 0x4A, 0x53, 0x54, 0x55, 0x56,
        0x57, 0x58, 0x59, 0x5A, 0x63, 0x64, 0x65, 0x66,
        0x67, 0x68, 0x69, 0x6A, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,
        0x79, 0x7A, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88,
        0x89, 0x8A, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99,
        0x9A, 0xA2, 0xA3, 0xA4, 0xA5, 0xA6, 0xA7, 0xA8,
        0xA9, 0xAA, 0xB2, 0xB3, 0xB4, 0xB5, 0xB6, 0xB7, 0xB8, 0xB9,
        0xBA, 0xC2, 0xC3, 0xC4, 0xC5, 0xC6, 0xC7, 0xC8,
        0xC9, 0xCA, 0xD2, 0xD3, 0xD4, 0xD5, 0xD6, 0xD7, 0xD8, 0xD9,
        0xDA, 0xE1, 0xE2, 0xE3, 0xE4, 0xE5, 0xE6, 0xE7,
        0xE8, 0xE9, 0xEA, 0xF1, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7,
        0xF8, 0xF9, 0xFA,
        0x11,                   //AC second table
        0x00, 0x02, 0x01, 0x02, 0x04, 0x04, 0x03, 0x04,
        0x07, 0x05, 0x04, 0x04, 0x00, 0x01,
        0x02, 0x77, 0x00, 0x01, 0x02, 0x03, 0x11, 0x04,
        0x05, 0x21, 0x31, 0x06, 0x12, 0x41, 0x51, 0x07, 0x61, 0x71,
        0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91,
        0xA1, 0xB1, 0xC1, 0x09, 0x23, 0x33, 0x52, 0xF0, 0x15, 0x62,
        0x72, 0xD1, 0x0A, 0x16, 0x24, 0x34, 0xE1, 0x25,
        0xF1, 0x17, 0x18, 0x19, 0x1A, 0x26, 0x27, 0x28, 0x29, 0x2A,
        0x35, 0x36, 0x37, 0x38, 0x39, 0x3A, 0x43, 0x44,
        0x45, 0x46, 0x47, 0x48, 0x49, 0x4A, 0x53, 0x54, 0x55, 0x56,
        0x57, 0x58, 0x59, 0x5A, 0x63, 0x64, 0x65, 0x66,
        0x67, 0x68, 0x69, 0x6A, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,
        0x79, 0x7A, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,
        0x88, 0x89, 0x8A, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98,
        0x99, 0x9A, 0xA2, 0xA3, 0xA4, 0xA5, 0xA6, 0xA7,
        0xA8, 0xA9, 0xAA, 0xB2, 0xB3, 0xB4, 0xB5, 0xB6, 0xB7, 0xB8,
        0xB9, 0xBA, 0xC2, 0xC3, 0xC4, 0xC5, 0xC6, 0xC7,
        0xC8, 0xC9, 0xCA, 0xD2, 0xD3, 0xD4, 0xD5, 0xD6, 0xD7, 0xD8,
        0xD9, 0xDA, 0xE2, 0xE3, 0xE4, 0xE5, 0xE6, 0xE7,
        0xE8, 0xE9, 0xEA, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7, 0xF8,
        0xF9, 0xFA
};

/* jpeg baseline setup, this is just fixed in this driver (YUV pictures) */
#define NO_OF_COMPONENTS          0x3   //Y,U,V
#define BASELINE_PRECISION        0x8   //MCU size (?)
static const char zr36050_tq[8] = { 0, 1, 1, 0, 0, 0, 0, 0 };   //table idx's QT
static const char zr36050_td[8] = { 0, 1, 1, 0, 0, 0, 0, 0 };   //table idx's DC
static const char zr36050_ta[8] = { 0, 1, 1, 0, 0, 0, 0, 0 };   //table idx's AC

/* horizontal 422 decimation setup (maybe we support 411 or so later, too) */
static const char zr36050_decimation_h[8] = { 2, 1, 1, 0, 0, 0, 0, 0 };
static const char zr36050_decimation_v[8] = { 1, 1, 1, 0, 0, 0, 0, 0 };

/* =========================================================================
   Local helper functions:

   calculation and setup of parameter-dependent JPEG baseline segments
   (needed for compression only)
   ========================================================================= */

/* ------------------------------------------------------------------------- */

/* SOF (start of frame) segment depends on width, height and sampling ratio
                         of each color component */

static int
zr36050_set_sof (struct zr36050 *ptr)
{
        char sof_data[34];      // max. size of register set
        int i;

        dprintk(3, "%s: write SOF (%dx%d, %d components)\n", ptr->name,
                ptr->width, ptr->height, NO_OF_COMPONENTS);
        sof_data[0] = 0xff;
        sof_data[1] = 0xc0;
        sof_data[2] = 0x00;
        sof_data[3] = (3 * NO_OF_COMPONENTS) + 8;
        sof_data[4] = BASELINE_PRECISION;       // only '8' possible with zr36050
        sof_data[5] = (ptr->height) >> 8;
        sof_data[6] = (ptr->height) & 0xff;
        sof_data[7] = (ptr->width) >> 8;
        sof_data[8] = (ptr->width) & 0xff;
        sof_data[9] = NO_OF_COMPONENTS;
        for (i = 0; i < NO_OF_COMPONENTS; i++) {
                sof_data[10 + (i * 3)] = i;     // index identifier
                sof_data[11 + (i * 3)] = (ptr->h_samp_ratio[i] << 4) | (ptr->v_samp_ratio[i]);  // sampling ratios
                sof_data[12 + (i * 3)] = zr36050_tq[i]; // Q table selection
        }
        return zr36050_pushit(ptr, ZR050_SOF_IDX,
                              (3 * NO_OF_COMPONENTS) + 10, sof_data);
}

/* ------------------------------------------------------------------------- */

/* SOS (start of scan) segment depends on the used scan components
                        of each color component */

static int
zr36050_set_sos (struct zr36050 *ptr)
{
        char sos_data[16];      // max. size of register set
        int i;

        dprintk(3, "%s: write SOS\n", ptr->name);
        sos_data[0] = 0xff;
        sos_data[1] = 0xda;
        sos_data[2] = 0x00;
        sos_data[3] = 2 + 1 + (2 * NO_OF_COMPONENTS) + 3;
        sos_data[4] = NO_OF_COMPONENTS;
        for (i = 0; i < NO_OF_COMPONENTS; i++) {
                sos_data[5 + (i * 2)] = i;      // index
                sos_data[6 + (i * 2)] = (zr36050_td[i] << 4) | zr36050_ta[i];   // AC/DC tbl.sel.
        }
        sos_data[2 + 1 + (2 * NO_OF_COMPONENTS) + 2] = 00;      // scan start
        sos_data[2 + 1 + (2 * NO_OF_COMPONENTS) + 3] = 0x3F;
        sos_data[2 + 1 + (2 * NO_OF_COMPONENTS) + 4] = 00;
        return zr36050_pushit(ptr, ZR050_SOS1_IDX,
                              4 + 1 + (2 * NO_OF_COMPONENTS) + 3,
                              sos_data);
}

/* ------------------------------------------------------------------------- */

/* DRI (define restart interval) */

static int
zr36050_set_dri (struct zr36050 *ptr)
{
        char dri_data[6];       // max. size of register set

        dprintk(3, "%s: write DRI\n", ptr->name);
        dri_data[0] = 0xff;
        dri_data[1] = 0xdd;
        dri_data[2] = 0x00;
        dri_data[3] = 0x04;
        dri_data[4] = ptr->dri >> 8;
        dri_data[5] = ptr->dri & 0xff;
        return zr36050_pushit(ptr, ZR050_DRI_IDX, 6, dri_data);
}

/* =========================================================================
   Setup function:

   Setup compression/decompression of Zoran's JPEG processor
   ( see also zoran 36050 manual )

   ... sorry for the spaghetti code ...
   ========================================================================= */
static void
zr36050_init (struct zr36050 *ptr)
{
        int sum = 0;
        long bitcnt, tmp;

        if (ptr->mode == CODEC_DO_COMPRESSION) {
                dprintk(2, "%s: COMPRESSION SETUP\n", ptr->name);

                /* 050 communicates with 057 in master mode */
                zr36050_write(ptr, ZR050_HARDWARE, ZR050_HW_MSTR);

                /* encoding table preload for compression */
                zr36050_write(ptr, ZR050_MODE,
                              ZR050_MO_COMP | ZR050_MO_TLM);
                zr36050_write(ptr, ZR050_OPTIONS, 0);

                /* disable all IRQs */
                zr36050_write(ptr, ZR050_INT_REQ_0, 0);
                zr36050_write(ptr, ZR050_INT_REQ_1, 3); // low 2 bits always 1

                /* volume control settings */
                /*zr36050_write(ptr, ZR050_MBCV, ptr->max_block_vol);*/
                zr36050_write(ptr, ZR050_SF_HI, ptr->scalefact >> 8);
                zr36050_write(ptr, ZR050_SF_LO, ptr->scalefact & 0xff);

                zr36050_write(ptr, ZR050_AF_HI, 0xff);
                zr36050_write(ptr, ZR050_AF_M, 0xff);
                zr36050_write(ptr, ZR050_AF_LO, 0xff);

                /* setup the variable jpeg tables */
                sum += zr36050_set_sof(ptr);
                sum += zr36050_set_sos(ptr);
                sum += zr36050_set_dri(ptr);

                /* setup the fixed jpeg tables - maybe variable, though -
                 * (see table init section above) */
                dprintk(3, "%s: write DQT, DHT, APP\n", ptr->name);
                sum += zr36050_pushit(ptr, ZR050_DQT_IDX,
                                      sizeof(zr36050_dqt), zr36050_dqt);
                sum += zr36050_pushit(ptr, ZR050_DHT_IDX,
                                      sizeof(zr36050_dht), zr36050_dht);
                zr36050_write(ptr, ZR050_APP_IDX, 0xff);
                zr36050_write(ptr, ZR050_APP_IDX + 1, 0xe0 + ptr->app.appn);
                zr36050_write(ptr, ZR050_APP_IDX + 2, 0x00);
                zr36050_write(ptr, ZR050_APP_IDX + 3, ptr->app.len + 2);
                sum += zr36050_pushit(ptr, ZR050_APP_IDX + 4, 60,
                                      ptr->app.data) + 4;
                zr36050_write(ptr, ZR050_COM_IDX, 0xff);
                zr36050_write(ptr, ZR050_COM_IDX + 1, 0xfe);
                zr36050_write(ptr, ZR050_COM_IDX + 2, 0x00);
                zr36050_write(ptr, ZR050_COM_IDX + 3, ptr->com.len + 2);
                sum += zr36050_pushit(ptr, ZR050_COM_IDX + 4, 60,
                                      ptr->com.data) + 4;

                /* do the internal huffman table preload */
                zr36050_write(ptr, ZR050_MARKERS_EN, ZR050_ME_DHTI);

                zr36050_write(ptr, ZR050_GO, 1);        // launch codec
                zr36050_wait_end(ptr);
                dprintk(2, "%s: Status after table preload: 0x%02x\n",
                        ptr->name, ptr->status1);

                if ((ptr->status1 & 0x4) == 0) {
                        dprintk(1, KERN_ERR "%s: init aborted!\n",
                                ptr->name);
                        return; // something is wrong, its timed out!!!!
                }

                /* setup misc. data for compression (target code sizes) */

                /* size of compressed code to reach without header data */
                sum = ptr->real_code_vol - sum;
                bitcnt = sum << 3;      /* need the size in bits */

                tmp = bitcnt >> 16;
                dprintk(3,
                        "%s: code: csize=%d, tot=%d, bit=%ld, highbits=%ld\n",
                        ptr->name, sum, ptr->real_code_vol, bitcnt, tmp);
                zr36050_write(ptr, ZR050_TCV_NET_HI, tmp >> 8);
                zr36050_write(ptr, ZR050_TCV_NET_MH, tmp & 0xff);
                tmp = bitcnt & 0xffff;
                zr36050_write(ptr, ZR050_TCV_NET_ML, tmp >> 8);
                zr36050_write(ptr, ZR050_TCV_NET_LO, tmp & 0xff);

                bitcnt -= bitcnt >> 7;  // bits without stuffing
                bitcnt -= ((bitcnt * 5) >> 6);  // bits without eob

                tmp = bitcnt >> 16;
                dprintk(3, "%s: code: nettobit=%ld, highnettobits=%ld\n",
                        ptr->name, bitcnt, tmp);
                zr36050_write(ptr, ZR050_TCV_DATA_HI, tmp >> 8);
                zr36050_write(ptr, ZR050_TCV_DATA_MH, tmp & 0xff);
                tmp = bitcnt & 0xffff;
                zr36050_write(ptr, ZR050_TCV_DATA_ML, tmp >> 8);
                zr36050_write(ptr, ZR050_TCV_DATA_LO, tmp & 0xff);

                /* compression setup with or without bitrate control */
                zr36050_write(ptr, ZR050_MODE,
                              ZR050_MO_COMP | ZR050_MO_PASS2 |
                              (ptr->bitrate_ctrl ? ZR050_MO_BRC : 0));

                /* this headers seem to deliver "valid AVI" jpeg frames */
                zr36050_write(ptr, ZR050_MARKERS_EN,
                              ZR050_ME_DQT | ZR050_ME_DHT |
                              ((ptr->app.len > 0) ? ZR050_ME_APP : 0) |
                              ((ptr->com.len > 0) ? ZR050_ME_COM : 0));
        } else {
                dprintk(2, "%s: EXPANSION SETUP\n", ptr->name);

                /* 050 communicates with 055 in master mode */
                zr36050_write(ptr, ZR050_HARDWARE,
                              ZR050_HW_MSTR | ZR050_HW_CFIS_2_CLK);

                /* encoding table preload */
                zr36050_write(ptr, ZR050_MODE, ZR050_MO_TLM);

                /* disable all IRQs */
                zr36050_write(ptr, ZR050_INT_REQ_0, 0);
                zr36050_write(ptr, ZR050_INT_REQ_1, 3); // low 2 bits always 1

                dprintk(3, "%s: write DHT\n", ptr->name);
                zr36050_pushit(ptr, ZR050_DHT_IDX, sizeof(zr36050_dht),
                               zr36050_dht);

                /* do the internal huffman table preload */
                zr36050_write(ptr, ZR050_MARKERS_EN, ZR050_ME_DHTI);

                zr36050_write(ptr, ZR050_GO, 1);        // launch codec
                zr36050_wait_end(ptr);
                dprintk(2, "%s: Status after table preload: 0x%02x\n",
                        ptr->name, ptr->status1);

                if ((ptr->status1 & 0x4) == 0) {
                        dprintk(1, KERN_ERR "%s: init aborted!\n",
                                ptr->name);
                        return; // something is wrong, its timed out!!!!
                }

                /* setup misc. data for expansion */
                zr36050_write(ptr, ZR050_MODE, 0);
                zr36050_write(ptr, ZR050_MARKERS_EN, 0);
        }

        /* adr on selected, to allow GO from master */
        zr36050_read(ptr, 0);
}

/* =========================================================================
   CODEC API FUNCTIONS

   this functions are accessed by the master via the API structure
   ========================================================================= */

/* set compression/expansion mode and launches codec -
   this should be the last call from the master before starting processing */
static int
zr36050_set_mode (struct videocodec *codec,
                  int                mode)
{
        struct zr36050 *ptr = (struct zr36050 *) codec->data;

        dprintk(2, "%s: set_mode %d call\n", ptr->name, mode);

        if ((mode != CODEC_DO_EXPANSION) && (mode != CODEC_DO_COMPRESSION))
                return -EINVAL;

        ptr->mode = mode;
        zr36050_init(ptr);

        return 0;
}

/* set picture size (norm is ignored as the codec doesn't know about it) */
static int
zr36050_set_video (struct videocodec   *codec,
                   struct tvnorm       *norm,
                   struct vfe_settings *cap,
                   struct vfe_polarity *pol)
{
        struct zr36050 *ptr = (struct zr36050 *) codec->data;
        int size;

        dprintk(2, "%s: set_video %d.%d, %d/%d-%dx%d (0x%x) q%d call\n",
                ptr->name, norm->HStart, norm->VStart,
                cap->x, cap->y, cap->width, cap->height,
                cap->decimation, cap->quality);
        /* if () return -EINVAL;
         * trust the master driver that it knows what it does - so
         * we allow invalid startx/y and norm for now ... */
        ptr->width = cap->width / (cap->decimation & 0xff);
        ptr->height = cap->height / ((cap->decimation >> 8) & 0xff);

        /* (KM) JPEG quality */
        size = ptr->width * ptr->height;
        size *= 16; /* size in bits */
        /* apply quality setting */
        size = size * cap->quality / 200;

        /* Minimum: 1kb */
        if (size < 8192)
                size = 8192;
        /* Maximum: 7/8 of code buffer */
        if (size > ptr->total_code_vol * 7)
                size = ptr->total_code_vol * 7;

        ptr->real_code_vol = size >> 3; /* in bytes */

        /* Set max_block_vol here (previously in zr36050_init, moved
         * here for consistency with zr36060 code */
        zr36050_write(ptr, ZR050_MBCV, ptr->max_block_vol);

        return 0;
}

/* additional control functions */
static int
zr36050_control (struct videocodec *codec,
                 int                type,
                 int                size,
                 void              *data)
{
        struct zr36050 *ptr = (struct zr36050 *) codec->data;
        int *ival = (int *) data;

        dprintk(2, "%s: control %d call with %d byte\n", ptr->name, type,
                size);

        switch (type) {
        case CODEC_G_STATUS:    /* get last status */
                if (size != sizeof(int))
                        return -EFAULT;
                zr36050_read_status1(ptr);
                *ival = ptr->status1;
                break;

        case CODEC_G_CODEC_MODE:
                if (size != sizeof(int))
                        return -EFAULT;
                *ival = CODEC_MODE_BJPG;
                break;

        case CODEC_S_CODEC_MODE:
                if (size != sizeof(int))
                        return -EFAULT;
                if (*ival != CODEC_MODE_BJPG)
                        return -EINVAL;
                /* not needed, do nothing */
                return 0;

        case CODEC_G_VFE:
        case CODEC_S_VFE:
                /* not needed, do nothing */
                return 0;

        case CODEC_S_MMAP:
                /* not available, give an error */
                return -ENXIO;

        case CODEC_G_JPEG_TDS_BYTE:     /* get target volume in byte */
                if (size != sizeof(int))
                        return -EFAULT;
                *ival = ptr->total_code_vol;
                break;

        case CODEC_S_JPEG_TDS_BYTE:     /* get target volume in byte */
                if (size != sizeof(int))
                        return -EFAULT;
                ptr->total_code_vol = *ival;
                /* (Kieran Morrissey)
                 * code copied from zr36060.c to ensure proper bitrate */
                ptr->real_code_vol = (ptr->total_code_vol * 6) >> 3;
                break;

        case CODEC_G_JPEG_SCALE:        /* get scaling factor */
                if (size != sizeof(int))
                        return -EFAULT;
                *ival = zr36050_read_scalefactor(ptr);
                break;

        case CODEC_S_JPEG_SCALE:        /* set scaling factor */
                if (size != sizeof(int))
                        return -EFAULT;
                ptr->scalefact = *ival;
                break;

        case CODEC_G_JPEG_APP_DATA: {   /* get appn marker data */
                struct jpeg_app_marker *app = data;

                if (size != sizeof(struct jpeg_app_marker))
                        return -EFAULT;

                *app = ptr->app;
                break;
        }

        case CODEC_S_JPEG_APP_DATA: {    /* set appn marker data */
                struct jpeg_app_marker *app = data;

                if (size != sizeof(struct jpeg_app_marker))
                        return -EFAULT;

                ptr->app = *app;
                break;
        }

        case CODEC_G_JPEG_COM_DATA: {   /* get comment marker data */
                struct jpeg_com_marker *com = data;

                if (size != sizeof(struct jpeg_com_marker))
                        return -EFAULT;

                *com = ptr->com;
                break;
        }

        case CODEC_S_JPEG_COM_DATA: {   /* set comment marker data */
                struct jpeg_com_marker *com = data;

                if (size != sizeof(struct jpeg_com_marker))
                        return -EFAULT;

                ptr->com = *com;
                break;
        }

        default:
                return -EINVAL;
        }

        return size;
}

/* =========================================================================
   Exit and unregister function:

   Deinitializes Zoran's JPEG processor
   ========================================================================= */

static int
zr36050_unset (struct videocodec *codec)
{
        struct zr36050 *ptr = codec->data;

        if (ptr) {
                /* do wee need some codec deinit here, too ???? */

                dprintk(1, "%s: finished codec #%d\n", ptr->name,
                        ptr->num);
                kfree(ptr);
                codec->data = NULL;

                zr36050_codecs--;
                return 0;
        }

        return -EFAULT;
}

/* =========================================================================
   Setup and registry function:

   Initializes Zoran's JPEG processor

   Also sets pixel size, average code size, mode (compr./decompr.)
   (the given size is determined by the processor with the video interface)
   ========================================================================= */

static int
zr36050_setup (struct videocodec *codec)
{
        struct zr36050 *ptr;
        int res;

        dprintk(2, "zr36050: initializing MJPEG subsystem #%d.\n",
                zr36050_codecs);

        if (zr36050_codecs == MAX_CODECS) {
                dprintk(1,
                        KERN_ERR "zr36050: Can't attach more codecs!\n");
                return -ENOSPC;
        }
        //mem structure init
        codec->data = ptr = kzalloc(sizeof(struct zr36050), GFP_KERNEL);
        if (NULL == ptr) {
                dprintk(1, KERN_ERR "zr36050: Can't get enough memory!\n");
                return -ENOMEM;
        }

        snprintf(ptr->name, sizeof(ptr->name), "zr36050[%d]",
                 zr36050_codecs);
        ptr->num = zr36050_codecs++;
        ptr->codec = codec;

        //testing
        res = zr36050_basic_test(ptr);
        if (res < 0) {
                zr36050_unset(codec);
                return res;
        }
        //final setup
        memcpy(ptr->h_samp_ratio, zr36050_decimation_h, 8);
        memcpy(ptr->v_samp_ratio, zr36050_decimation_v, 8);

        ptr->bitrate_ctrl = 0;  /* 0 or 1 - fixed file size flag
                                 * (what is the difference?) */
        ptr->mode = CODEC_DO_COMPRESSION;
        ptr->width = 384;
        ptr->height = 288;
        ptr->total_code_vol = 16000;
        ptr->max_block_vol = 240;
        ptr->scalefact = 0x100;
        ptr->dri = 1;

        /* no app/com marker by default */
        ptr->app.appn = 0;
        ptr->app.len = 0;
        ptr->com.len = 0;

        zr36050_init(ptr);

        dprintk(1, KERN_INFO "%s: codec attached and running\n",
                ptr->name);

        return 0;
}

static const struct videocodec zr36050_codec = {
        .owner = THIS_MODULE,
        .name = "zr36050",
        .magic = 0L,            // magic not used
        .flags =
            CODEC_FLAG_JPEG | CODEC_FLAG_HARDWARE | CODEC_FLAG_ENCODER |
            CODEC_FLAG_DECODER,
        .type = CODEC_TYPE_ZR36050,
        .setup = zr36050_setup, // functionality
        .unset = zr36050_unset,
        .set_mode = zr36050_set_mode,
        .set_video = zr36050_set_video,
        .control = zr36050_control,
        // others are not used
};

/* =========================================================================
   HOOK IN DRIVER AS KERNEL MODULE
   ========================================================================= */

static int __init
zr36050_init_module (void)
{
        //dprintk(1, "ZR36050 driver %s\n",ZR050_VERSION);
        zr36050_codecs = 0;
        return videocodec_register(&zr36050_codec);
}

static void __exit
zr36050_cleanup_module (void)
{
        if (zr36050_codecs) {
                dprintk(1,
                        "zr36050: something's wrong - %d codecs left somehow.\n",
                        zr36050_codecs);
        }
        videocodec_unregister(&zr36050_codec);
}

module_init(zr36050_init_module);
module_exit(zr36050_cleanup_module);

MODULE_AUTHOR("Wolfgang Scherr <scherr@net4you.at>");
MODULE_DESCRIPTION("Driver module for ZR36050 jpeg processors "
                   ZR050_VERSION);
MODULE_LICENSE("GPL");