/*      $NetBSD: ss_mustek.c,v 1.43 2016/11/20 15:37:19 mlelstv Exp $   */

/*
* Copyright (c) 1995 Joachim Koenig-Baltes.  All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
*    must display the following acknowledgement:
*      This product includes software developed by Joachim Koenig-Baltes.
* 4. The name of the author may not be used to endorse or promote products
*    derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

/*
* special driver for MUSTEK flatbed scanners MFS 06000CX and MFS 12000CX
* these scanners come with their own scsi card, containing an NCR53C400
* SCSI controller chip. I'm in the progress of writing a driver for this
* card to work under NetBSD-current. I've hooked it up to a Seagate ST01
* hostadapter in the meantime, giving 350KB/sec for higher resolutions!
*
* I tried to connect it to my Adaptec 1542B, but with no success. It seems,
* it does not like synchronous negotiation between Hostadapter and other
* targets, but I could not turn this off for the 1542B.
*
* There is also an other reason why you would not like to connect it to your
* favourite SCSI host adapter: The Mustek DOES NOT DISCONNECT. It will block
* other traffic from the bus while a transfer is active.
*/

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: ss_mustek.c,v 1.43 2016/11/20 15:37:19 mlelstv Exp $");

#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/systm.h>
#include <sys/fcntl.h>
#include <sys/errno.h>
#include <sys/ioctl.h>
#include <sys/malloc.h>
#include <sys/buf.h>
#include <sys/bufq.h>
#include <sys/proc.h>
#include <sys/device.h>
#include <sys/conf.h>           /* for cdevsw */
#include <sys/scanio.h>

#include <dev/scsipi/scsipi_all.h>
#include <dev/scsipi/scsi_all.h>
#include <dev/scsipi/scsi_scanner.h>
#include <dev/scsipi/scsipiconf.h>
#include <dev/scsipi/scsipi_base.h>
#include <dev/scsipi/ssvar.h>
#include <dev/scsipi/ss_mustek.h>

#define MUSTEK_RETRIES 4

static int      mustek_get_params(struct ss_softc *);
static int      mustek_set_params(struct ss_softc *, struct scan_io *);
static int      mustek_trigger_scanner(struct ss_softc *);
static void     mustek_minphys(struct ss_softc *, struct buf *);
static int      mustek_read(struct ss_softc *, struct buf *);
static int      mustek_rewind_scanner(struct ss_softc *);

/* only used internally */
static int      mustek_get_status(struct ss_softc *, int, int);
static void     mustek_compute_sizes(struct ss_softc *);

/* structure for the special handlers */
static struct ss_special mustek_special = {
       mustek_set_params,
       mustek_trigger_scanner,
       mustek_get_params,
       mustek_minphys,
       mustek_read,
       mustek_rewind_scanner,
       NULL,                   /* no adf support right now */
       NULL                    /* no adf support right now */
};

/* mustek_attach: attach special functions to ss */
void
mustek_attach(struct ss_softc *ss, struct scsipibus_attach_args *sa)
{
#ifdef SCSIPI_DEBUG
       struct scsipi_periph *periph = sa->sa_periph;
#endif

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_attach: start\n"));
       ss->sio.scan_scanner_type = 0;

       printf("\n%s: ", device_xname(ss->sc_dev));

       /* first, check the model which determines resolutions */
       if (!memcmp(sa->sa_inqbuf.product, "MFS-06000CX", 11)) {
               ss->sio.scan_scanner_type = MUSTEK_06000CX;
               printf("Mustek 6000CX Flatbed 3-pass color scanner, "
                   "3 - 600 dpi\n");
       }
       if (!memcmp(sa->sa_inqbuf.product, "MFS-12000CX", 11)) {
               ss->sio.scan_scanner_type = MUSTEK_12000CX;
               printf("Mustek 12000CX Flatbed 3-pass color scanner, "
                   "6 - 1200 dpi\n");
       }

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_attach: scanner_type = %d\n",
           ss->sio.scan_scanner_type));

       /* install special handlers */
       ss->special = &mustek_special;

       /*
        * populate the scanio struct with legal values
        * the default should come from user space
        */
       ss->sio.scan_width              = 1200;
       ss->sio.scan_height             = 1200;
       ss->sio.scan_x_resolution       = 99;
       ss->sio.scan_y_resolution       = 99;
       ss->sio.scan_x_origin           = 0;
       ss->sio.scan_y_origin           = 0;
       ss->sio.scan_brightness         = 100;
       ss->sio.scan_contrast           = 100;
       ss->sio.scan_quality            = 100;
       ss->sio.scan_image_mode         = SIM_GRAYSCALE;

       mustek_compute_sizes(ss);
}

static int
mustek_get_params (struct ss_softc *ss)
{
       return 0;
}

/*
* check the parameters if the mustek is capable of fulfilling it
* but don't send the command to the scanner in case the user wants
* to change parameters by more than one call
*/
static int
mustek_set_params(struct ss_softc *ss, struct scan_io *sio)
{
       int error;

       /* if the scanner is triggered, then rewind it */
       if (ss->flags & SSF_TRIGGERED) {
               error = mustek_rewind_scanner(ss);
               if (error)
                       return error;
       }

       /* size constraints: 8.5" horizontally and 14" vertically */
#ifdef MUSTEK_INCH_SPEC
       /* sizes must be a multiple of 1/8" */
       sio->scan_x_origin -= sio->scan_x_origin % 150;
       sio->scan_y_origin -= sio->scan_y_origin % 150;
       sio->scan_width -= sio->scan_width % 150;
       sio->scan_height -= sio->scan_height % 150;
#endif
       if (sio->scan_width == 0 ||
           sio->scan_x_origin + sio->scan_width > 10200 ||
           sio->scan_height == 0 ||
           sio->scan_y_origin + sio->scan_height > 16800)
               return EINVAL;

       /*
        * for now, only realize the values for the MUSTEK_06000CX
        * in the future, values for the MUSTEK_12000CX will be implemented
        */

       /*
        * resolution (dpi) must be <= 300 and a multiple of 3 or
        * between 300 and 600 and a multiple of 30
        */
       sio->scan_x_resolution -= sio->scan_x_resolution <= 300 ?
           sio->scan_x_resolution % 3 : sio->scan_x_resolution % 30;
       sio->scan_y_resolution -= sio->scan_y_resolution <= 300 ?
           sio->scan_y_resolution % 3 : sio->scan_y_resolution % 30;
       if (sio->scan_x_resolution < 3 || sio->scan_x_resolution > 600 ||
           sio->scan_x_resolution != sio->scan_y_resolution)
               return EINVAL;

       /* assume brightness values are between 64 and 136 in steps of 3 */
       sio->scan_brightness -= (sio->scan_brightness - 64) % 3;
       if (sio->scan_brightness < 64 || sio->scan_brightness > 136)
               return EINVAL;

       /* contrast values must be between 16 and 184 in steps of 7 */
       sio->scan_contrast -= (sio->scan_contrast - 16) % 7;
       if (sio->scan_contrast < 16 || sio->scan_contrast > 184)
               return EINVAL;

       /*
        * velocity: between 0 (fast) and 4 (slow) which will be mapped
        * to 100% = 4, 80% = 3, 60% = 2, 40% = 1, 20% = 0
        * must be a multiple of 20
        */
       sio->scan_quality -= sio->scan_quality % 20;
       if (sio->scan_quality < 20 || sio->scan_quality > 100)
               return EINVAL;

       switch (sio->scan_image_mode) {
       case SIM_BINARY_MONOCHROME:
       case SIM_DITHERED_MONOCHROME:
       case SIM_GRAYSCALE:
       case SIM_RED:
       case SIM_GREEN:
       case SIM_BLUE:
               break;
       default:
               return EINVAL;
       }

       /* change ss_softc to the new values, but save ro-variables */
       sio->scan_scanner_type = ss->sio.scan_scanner_type;
       memcpy(&ss->sio, sio, sizeof(struct scan_io));

       mustek_compute_sizes(ss);

       return 0;
}

/*
* trim the requested transfer to a multiple of the line size
* this is called only from ssread() which guarantees, scanner is triggered
* In the future, it will trim the transfer to not read to much at a time
* because the mustek cannot disconnect. It will be calculated by the
* resolution, the velocity and the number of bytes per line.
*/
static void
mustek_minphys(struct ss_softc *ss, struct buf *bp)
{
#ifdef SCSIPI_DEBUG
       struct scsipi_periph *periph = ss->sc_periph;
#endif

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_minphys: before: %d\n",
           bp->b_bcount));
       bp->b_bcount -= bp->b_bcount %
           ((ss->sio.scan_pixels_per_line * ss->sio.scan_bits_per_pixel) / 8);
       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_minphys: after:  %d\n",
           bp->b_bcount));
}

/*
* trigger the scanner to start a scan operation
* this includes sending the mode- and window-data, starting the scanner
* and getting the image size info
*/
static int
mustek_trigger_scanner(struct ss_softc *ss)
{
       struct mustek_mode_select_cmd mode_cmd;
       struct mustek_mode_select_data mode_data;
       struct mustek_set_window_cmd window_cmd;
       struct mustek_set_window_data window_data;
       struct mustek_start_scan_cmd start_scan_cmd;
       struct scsipi_periph *periph = ss->sc_periph;
       int pixel_tlx, pixel_tly, pixel_brx, pixel_bry, paperlength;
       int error;

       mustek_compute_sizes(ss);

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_trigger_scanner\n"));

       /* set the window params and send the scsi command */
       memset(&window_cmd, 0, sizeof(window_cmd));
       window_cmd.opcode = MUSTEK_SET_WINDOW;
       window_cmd.length = sizeof(window_data);

       memset(&window_data, 0, sizeof(window_data));
       window_data.frame.header = MUSTEK_LINEART_BACKGROUND | MUSTEK_UNIT_SPEC;
#ifdef MUSTEK_INCH_SPEC
       /* the positional values are all 1 byte because 256 / 8 = 32" */
       pixel_tlx = ss->sio.scan_x_origin / 150;
       pixel_tly = ss->sio.scan_y_origin / 150;
       pixel_brx = pixel_tlx + ss->sio.scan_width / 150;
       pixel_bry = pixel_tly + ss->sio.scan_height / 150;
#else
       pixel_tlx = (ss->sio.scan_x_origin * ss->sio.scan_x_resolution) / 1200;
       pixel_tly = (ss->sio.scan_y_origin * ss->sio.scan_y_resolution) / 1200;
       pixel_brx = pixel_tlx +
           (ss->sio.scan_width * ss->sio.scan_x_resolution) / 1200;
       pixel_bry = pixel_tly +
           (ss->sio.scan_height * ss->sio.scan_y_resolution) / 1200;
#endif
       _lto2l(pixel_tlx, window_data.frame.tl_x);
       _lto2l(pixel_tly, window_data.frame.tl_y);
       _lto2l(pixel_brx, window_data.frame.br_x);
       _lto2l(pixel_bry, window_data.frame.br_y);

#if MUSTEK_WINDOWS >= 1
       window_data.window1 = window_data.frame;
       window_data.window1.header = MUSTEK_WINDOW_MASK | MUSTEK_UNIT_SPEC;
#endif

       /* send the set window command to the scanner */
       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_set_parms: set_window\n"));
       error = scsipi_command(periph, (void *)&window_cmd, sizeof(window_cmd),
           (void *)&window_data, sizeof(window_data),
           MUSTEK_RETRIES, 5000, NULL, XS_CTL_DATA_OUT);
       if (error)
               return error;

       /*
        * do what it takes to actualize the mode
        */
       memset(&mode_cmd, 0, sizeof(mode_cmd));
       mode_cmd.opcode = MUSTEK_MODE_SELECT;
       _lto2b(sizeof(mode_data), mode_cmd.length);

       memset(&mode_data, 0, sizeof(mode_data));
       mode_data.mode =
           MUSTEK_MODE_MASK | MUSTEK_HT_PATTERN_BUILTIN | MUSTEK_UNIT_SPEC;
       if (ss->sio.scan_x_resolution <= 300)
               mode_data.resolution = ss->sio.scan_x_resolution / 3;
       else
               /*
                * the resolution values is computed by modulo 100, but not
                * for 600dpi, where the value is 100 (a bit tricky, but ...)
                */
               mode_data.resolution =
                   ((ss->sio.scan_x_resolution - 1) % 100) + 1;

       mode_data.brightness = (ss->sio.scan_brightness - 64) / 3;
       mode_data.contrast = (ss->sio.scan_contrast - 16) / 7;
       mode_data.grain = 0;
       mode_data.velocity = ss->sio.scan_quality / 20 - 1;
#ifdef MUSTEK_INCH_SPEC
       paperlength = 14 * 8;   /* 14" */
#else
       paperlength = 14 * ss->sio.scan_y_resolution;   /* 14" */
#endif
       _lto2l(paperlength, mode_data.paperlength);

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_trigger_scanner: mode_select\n"));
       /* send the command to the scanner */
       error = scsipi_command(periph, (void *)&mode_cmd, sizeof(mode_cmd),
           (void *)&mode_data, sizeof(mode_data),
           MUSTEK_RETRIES, 5000, NULL, XS_CTL_DATA_OUT);
       if (error)
               return error;

       /*
        * now construct and send the start command
        */
       memset(&start_scan_cmd, 0, sizeof(start_scan_cmd));
       start_scan_cmd.opcode = MUSTEK_START_STOP;
       start_scan_cmd.mode = MUSTEK_SCAN_START;
       if (ss->sio.scan_x_resolution <= 300)
               start_scan_cmd.mode |= MUSTEK_RES_STEP_1;
       else
               start_scan_cmd.mode |= MUSTEK_RES_STEP_10;
       switch (ss->sio.scan_image_mode) {
       case SIM_BINARY_MONOCHROME:
       case SIM_DITHERED_MONOCHROME:
               start_scan_cmd.mode |= MUSTEK_BIT_MODE | MUSTEK_GRAY_FILTER;
               break;
       case SIM_GRAYSCALE:
               start_scan_cmd.mode |= MUSTEK_GRAY_MODE | MUSTEK_GRAY_FILTER;
               break;
       case SIM_RED:
               start_scan_cmd.mode |= MUSTEK_GRAY_MODE | MUSTEK_RED_FILTER;
               break;
       case SIM_GREEN:
               start_scan_cmd.mode |= MUSTEK_GRAY_MODE | MUSTEK_GREEN_FILTER;
               break;
       case SIM_BLUE:
               start_scan_cmd.mode |= MUSTEK_GRAY_MODE | MUSTEK_BLUE_FILTER;
               break;
       }

       /* send the command to the scanner */
       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_trigger_scanner: start_scan\n"));
       error = scsipi_command(periph,
           (void *)&start_scan_cmd, sizeof(start_scan_cmd), NULL, 0,
           MUSTEK_RETRIES, 5000, NULL, 0);
       if (error)
               return error;

       /*
        * now check if scanner ready this time with update of size info
        * we wait here so that if the user issues a read directly afterwards,
        * the scanner will respond directly (otherwise we had to sleep with
        * a buffer locked in memory)
        */
       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_trigger_scanner: get_status\n"));
       error = mustek_get_status(ss, 60, 1);
       if (error)
               return error;

       return 0;
}

/* stop a scan operation in progress */
static int
mustek_rewind_scanner(struct ss_softc *ss)
{
       struct mustek_start_scan_cmd cmd;
       struct scsipi_periph *periph = ss->sc_periph;
       int error;

       if (ss->sio.scan_window_size != 0) {
               /*
                * only if not all data has been read, the scanner has to be
                * stopped
                */
               memset(&cmd, 0, sizeof(cmd));
               cmd.opcode = MUSTEK_START_STOP;
               cmd.mode = MUSTEK_SCAN_STOP;

               /* send the command to the scanner */
               SC_DEBUG(periph, SCSIPI_DB1,
                   ("mustek_rewind_scanner: stop_scan\n"));
               error = scsipi_command(periph,
                   (void *)&cmd, sizeof(cmd),
                   NULL, 0,
                   MUSTEK_RETRIES, 5000, NULL, 0);
               if (error)
                       return error;
       }

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_rewind_scanner: end\n"));

       return 0;
}

/* read the requested number of bytes/lines from the scanner */
static int
mustek_read(struct ss_softc *ss, struct buf *bp)
{
       struct mustek_read_cmd cmd;
       struct scsipi_xfer *xs;
       struct scsipi_periph *periph = ss->sc_periph;
       u_long lines_to_read;
       int error __diagused;

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_read: start\n"));

       memset(&cmd, 0, sizeof(cmd));
       cmd.opcode = MUSTEK_READ;

       /* instead of the bytes, the mustek wants the number of lines */
       lines_to_read = bp->b_bcount /
           ((ss->sio.scan_pixels_per_line * ss->sio.scan_bits_per_pixel) / 8);
       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_read: read %ld lines\n",
           lines_to_read));
       _lto3b(lines_to_read, cmd.length);

       /* go ask the adapter to do all this for us */
       xs = scsipi_make_xs_locked(periph,
           (struct scsipi_generic *) &cmd, sizeof(cmd),
           (u_char *) bp->b_data, bp->b_bcount,
           MUSTEK_RETRIES, 10000, bp,
           XS_CTL_NOSLEEP | XS_CTL_ASYNC | XS_CTL_DATA_IN);
       if (xs == NULL) {
               /*
                * out of memory. Keep this buffer in the queue, and
                * retry later.
                */
               callout_reset(&ss->sc_callout, hz / 2, ssrestart,
                   periph);
               return 0;
       }
#ifdef DIAGNOSTIC
       if (bufq_get(ss->buf_queue) != bp)
               panic("ssstart(): dequeued wrong buf");
#else
       bufq_get(ss->buf_queue);
#endif
       error = scsipi_execute_xs(xs);
       /* with a scsipi_xfer preallocated, scsipi_command can't fail */
       KASSERT(error == 0);
       ss->sio.scan_lines -= lines_to_read;
#if 0
       if (ss->sio.scan_lines < 0)
               ss->sio.scan_lines = 0;
#endif
       ss->sio.scan_window_size -= bp->b_bcount;
#if 0
       if (ss->sio.scan_window_size < 0)
               ss->sio.scan_window_size = 0;
#endif
       return 0;
}

/*
* check if the scanner is ready to take commands
*   wait timeout seconds and try only every second
*   if update, then update picture size info
*
*   returns EBUSY if scanner not ready
*/
static int
mustek_get_status(struct ss_softc *ss, int timeout, int update)
{
       struct mustek_get_status_cmd cmd;
       struct mustek_get_status_data data;
       struct scsipi_periph *periph = ss->sc_periph;
       int error, lines, bytes_per_line;

       memset(&cmd, 0, sizeof(cmd));
       cmd.opcode = MUSTEK_GET_STATUS;
       cmd.length = sizeof(data);

       while (1) {
               SC_DEBUG(periph, SCSIPI_DB1, ("mustek_get_status: stat_cmd\n"));
               error = scsipi_command(periph, (void *)&cmd, sizeof(cmd),
                   (void *)&data, sizeof(data),
                   MUSTEK_RETRIES, 5000, NULL, XS_CTL_DATA_IN);
               if (error)
                       return error;
               if ((data.ready_busy == MUSTEK_READY) ||
                   (timeout-- <= 0))
                       break;
               /* please wait a second */
               kpause("mtkrdy", false, hz, NULL);
       }

       if (update) {
               bytes_per_line = _2ltol(data.bytes_per_line);
               lines = _3ltol(data.lines);
               if (lines != ss->sio.scan_lines) {
                       printf("mustek: lines actual(%d) != computed(%ld)\n",
                           lines, ss->sio.scan_lines);
                       return EIO;
               }
               if (bytes_per_line * lines != ss->sio.scan_window_size) {
                       printf("mustek: win-size actual(%d) != computed(%ld)\n",
                           bytes_per_line * lines, ss->sio.scan_window_size);
                   return EIO;
               }

               SC_DEBUG(periph, SCSIPI_DB1,
                   ("mustek_get_size: bpl=%ld, lines=%ld\n",
                   (ss->sio.scan_pixels_per_line * ss->sio.scan_bits_per_pixel)
                   / 8, ss->sio.scan_lines));
               SC_DEBUG(periph, SCSIPI_DB1, ("window size = %ld\n",
                   ss->sio.scan_window_size));
       }

       SC_DEBUG(periph, SCSIPI_DB1, ("mustek_get_status: end\n"));
       if (data.ready_busy == MUSTEK_READY)
               return 0;
       else
               return EBUSY;
}

/*
* mustek_compute_sizes: compute window_size and lines for the picture
*   this function is called from different places in the code
*/
static void
mustek_compute_sizes(struct ss_softc *ss)
{

       switch (ss->sio.scan_image_mode) {
       case SIM_BINARY_MONOCHROME:
       case SIM_DITHERED_MONOCHROME:
               ss->sio.scan_bits_per_pixel = 1;
               break;
       case SIM_GRAYSCALE:
       case SIM_RED:
       case SIM_GREEN:
       case SIM_BLUE:
               ss->sio.scan_bits_per_pixel = 8;
               break;
       }

       /*
        * horizontal number of bytes is always a multiple of 2,
        * in 8-bit mode at least
        */
       ss->sio.scan_pixels_per_line =
           (ss->sio.scan_width * ss->sio.scan_x_resolution) / 1200;
       if (ss->sio.scan_bits_per_pixel == 1)
               /* make it a multiple of 16, and thus of 2 bytes */
               ss->sio.scan_pixels_per_line =
                   (ss->sio.scan_pixels_per_line + 15) & 0xfffffff0;
       else
               ss->sio.scan_pixels_per_line =
                   (ss->sio.scan_pixels_per_line + 1) & 0xfffffffe;

       ss->sio.scan_lines =
           (ss->sio.scan_height * ss->sio.scan_y_resolution) / 1200;
       ss->sio.scan_window_size = ss->sio.scan_lines *
           ((ss->sio.scan_pixels_per_line * ss->sio.scan_bits_per_pixel) / 8);
}