// SPDX-License-Identifier: GPL-2.0+
/*
 * comedi/drivers/amplc_dio200_common.c
 *
 * Common support code for "amplc_dio200" and "amplc_dio200_pci".
 *
 * Copyright (C) 2005-2013 MEV Ltd. <https://www.mev.co.uk/>
 *
 * COMEDI - Linux Control and Measurement Device Interface
 * Copyright (C) 1998,2000 David A. Schleef <ds@schleef.org>
 */

#include <linux/module.h>
#include <linux/interrupt.h>
#include <linux/comedi/comedidev.h>
#include <linux/comedi/comedi_8255.h> /* only for register defines */
#include <linux/comedi/comedi_8254.h>

#include "amplc_dio200.h"

/* 200 series registers */
#define DIO200_IO_SIZE  0x20
#define DIO200_PCIE_IO_SIZE 0x4000
#define DIO200_CLK_SCE(x) (0x18 + (x)) /* Group X/Y/Z clock sel reg */
#define DIO200_GAT_SCE(x) (0x1b + (x)) /* Group X/Y/Z gate sel reg */
#define DIO200_INT_SCE  0x1e /* Interrupt enable/status register */
/* Extra registers for new PCIe boards */
#define DIO200_ENHANCE  0x20 /* 1 to enable enhanced features */
#define DIO200_VERSION  0x24 /* Hardware version register */
#define DIO200_TS_CONFIG 0x600 /* Timestamp timer config register */
#define DIO200_TS_COUNT  0x602 /* Timestamp timer count register */

/*
 * Functions for constructing value for DIO_200_?CLK_SCE and
 * DIO_200_?GAT_SCE registers:
 *
 * 'which' is: 0 for CTR-X1, CTR-Y1, CTR-Z1; 1 for CTR-X2, CTR-Y2 or CTR-Z2.
 * 'chan' is the channel: 0, 1 or 2.
 * 'source' is the signal source: 0 to 7, or 0 to 31 for "enhanced" boards.
 */
static unsigned char clk_gat_sce(unsigned int which, unsigned int chan,
     unsigned int source)
{
 return (which << 5) | (chan << 3) |
        ((source & 030) << 3) | (source & 007);
}

/*
 * Periods of the internal clock sources in nanoseconds.
 */
static const unsigned int clock_period[32] = {
 [1] = 100,  /* 10 MHz */
 [2] = 1000,  /* 1 MHz */
 [3] = 10000,  /* 100 kHz */
 [4] = 100000,  /* 10 kHz */
 [5] = 1000000,  /* 1 kHz */
 [11] = 50,  /* 20 MHz (enhanced boards) */
 /* clock sources 12 and later reserved for enhanced boards */
};

/*
 * Timestamp timer configuration register (for new PCIe boards).
 */
#define TS_CONFIG_RESET  0x100 /* Reset counter to zero. */
#define TS_CONFIG_CLK_SRC_MASK 0x0FF /* Clock source. */
#define TS_CONFIG_MAX_CLK_SRC 2 /* Maximum clock source value. */

/*
 * Periods of the timestamp timer clock sources in nanoseconds.
 */
static const unsigned int ts_clock_period[TS_CONFIG_MAX_CLK_SRC + 1] = {
 1,   /* 1 nanosecond (but with 20 ns granularity). */
 1000,   /* 1 microsecond. */
 1000000,  /* 1 millisecond. */
};

struct dio200_subdev_8255 {
 unsigned int ofs;  /* DIO base offset */
};

struct dio200_subdev_intr {
 spinlock_t spinlock; /* protects the 'active' flag */
 unsigned int ofs;
 unsigned int valid_isns;
 unsigned int enabled_isns;
 unsigned int active:1;
};

#ifdef CONFIG_HAS_IOPORT

static unsigned char dio200___read8(struct comedi_device *dev,
        unsigned int offset)
{
 if (dev->mmio)
  return readb(dev->mmio + offset);
 return inb(dev->iobase + offset);
}

static void dio200___write8(struct comedi_device *dev,
       unsigned int offset, unsigned char val)
{
 if (dev->mmio)
  writeb(val, dev->mmio + offset);
 else
  outb(val, dev->iobase + offset);
}

static unsigned int dio200___read32(struct comedi_device *dev,
        unsigned int offset)
{
 if (dev->mmio)
  return readl(dev->mmio + offset);
 return inl(dev->iobase + offset);
}

static void dio200___write32(struct comedi_device *dev,
        unsigned int offset, unsigned int val)
{
 if (dev->mmio)
  writel(val, dev->mmio + offset);
 else
  outl(val, dev->iobase + offset);
}

#else /* CONFIG_HAS_IOPORT */

static unsigned char dio200___read8(struct comedi_device *dev,
        unsigned int offset)
{
 return readb(dev->mmio + offset);
}

static void dio200___write8(struct comedi_device *dev,
       unsigned int offset, unsigned char val)
{
 writeb(val, dev->mmio + offset);
}

static unsigned int dio200___read32(struct comedi_device *dev,
        unsigned int offset)
{
 return readl(dev->mmio + offset);
}

static void dio200___write32(struct comedi_device *dev,
        unsigned int offset, unsigned int val)
{
 writel(val, dev->mmio + offset);
}

#endif /* CONFIG_HAS_IOPORT */

static unsigned char dio200_read8(struct comedi_device *dev,
      unsigned int offset)
{
 const struct dio200_board *board = dev->board_ptr;

 if (board->is_pcie)
  offset <<= 3;

 return dio200___read8(dev, offset);
}

static void dio200_write8(struct comedi_device *dev,
     unsigned int offset, unsigned char val)
{
 const struct dio200_board *board = dev->board_ptr;

 if (board->is_pcie)
  offset <<= 3;

 dio200___write8(dev, offset, val);
}

static unsigned int dio200_read32(struct comedi_device *dev,
      unsigned int offset)
{
 const struct dio200_board *board = dev->board_ptr;

 if (board->is_pcie)
  offset <<= 3;

 return dio200___read32(dev, offset);
}

static void dio200_write32(struct comedi_device *dev,
      unsigned int offset, unsigned int val)
{
 const struct dio200_board *board = dev->board_ptr;

 if (board->is_pcie)
  offset <<= 3;

 dio200___write32(dev, offset, val);
}

static unsigned int dio200_subdev_8254_offset(struct comedi_device *dev,
           struct comedi_subdevice *s)
{
 const struct dio200_board *board = dev->board_ptr;
 struct comedi_8254 *i8254 = s->private;
 unsigned int offset;

 /* get the offset that was passed to comedi_8254_*_init() */
 if (dev->mmio)
  offset = (void __iomem *)i8254->context - dev->mmio;
 else
  offset = i8254->context - dev->iobase;

 /* remove the shift that was added for PCIe boards */
 if (board->is_pcie)
  offset >>= 3;

 /* this offset now works for the dio200_{read,write} helpers */
 return offset;
}

static int dio200_subdev_intr_insn_bits(struct comedi_device *dev,
     struct comedi_subdevice *s,
     struct comedi_insn *insn,
     unsigned int *data)
{
 const struct dio200_board *board = dev->board_ptr;
 struct dio200_subdev_intr *subpriv = s->private;

 if (board->has_int_sce) {
  /* Just read the interrupt status register.  */
  data[1] = dio200_read8(dev, subpriv->ofs) & subpriv->valid_isns;
 } else {
  /* No interrupt status register. */
  data[0] = 0;
 }

 return insn->n;
}

static void dio200_stop_intr(struct comedi_device *dev,
        struct comedi_subdevice *s)
{
 const struct dio200_board *board = dev->board_ptr;
 struct dio200_subdev_intr *subpriv = s->private;

 subpriv->active = false;
 subpriv->enabled_isns = 0;
 if (board->has_int_sce)
  dio200_write8(dev, subpriv->ofs, 0);
}

static void dio200_start_intr(struct comedi_device *dev,
         struct comedi_subdevice *s)
{
 const struct dio200_board *board = dev->board_ptr;
 struct dio200_subdev_intr *subpriv = s->private;
 struct comedi_cmd *cmd = &s->async->cmd;
 unsigned int n;
 unsigned int isn_bits;

 /* Determine interrupt sources to enable. */
 isn_bits = 0;
 if (cmd->chanlist) {
  for (n = 0; n < cmd->chanlist_len; n++)
   isn_bits |= (1U << CR_CHAN(cmd->chanlist[n]));
 }
 isn_bits &= subpriv->valid_isns;
 /* Enable interrupt sources. */
 subpriv->enabled_isns = isn_bits;
 if (board->has_int_sce)
  dio200_write8(dev, subpriv->ofs, isn_bits);
}

static int dio200_inttrig_start_intr(struct comedi_device *dev,
         struct comedi_subdevice *s,
         unsigned int trig_num)
{
 struct dio200_subdev_intr *subpriv = s->private;
 struct comedi_cmd *cmd = &s->async->cmd;
 unsigned long flags;

 if (trig_num != cmd->start_arg)
  return -EINVAL;

 spin_lock_irqsave(&subpriv->spinlock, flags);
 s->async->inttrig = NULL;
 if (subpriv->active)
  dio200_start_intr(dev, s);

 spin_unlock_irqrestore(&subpriv->spinlock, flags);

 return 1;
}

static void dio200_read_scan_intr(struct comedi_device *dev,
      struct comedi_subdevice *s,
      unsigned int triggered)
{
 struct comedi_cmd *cmd = &s->async->cmd;
 unsigned short val;
 unsigned int n, ch;

 val = 0;
 for (n = 0; n < cmd->chanlist_len; n++) {
  ch = CR_CHAN(cmd->chanlist[n]);
  if (triggered & (1U << ch))
   val |= (1U << n);
 }

 comedi_buf_write_samples(s, &val, 1);

 if (cmd->stop_src == TRIG_COUNT &&
     s->async->scans_done >= cmd->stop_arg)
  s->async->events |= COMEDI_CB_EOA;
}

static int dio200_handle_read_intr(struct comedi_device *dev,
       struct comedi_subdevice *s)
{
 const struct dio200_board *board = dev->board_ptr;
 struct dio200_subdev_intr *subpriv = s->private;
 unsigned int triggered;
 unsigned int intstat;
 unsigned int cur_enabled;
 unsigned long flags;

 triggered = 0;

 spin_lock_irqsave(&subpriv->spinlock, flags);
 if (board->has_int_sce) {
  /*
 * Collect interrupt sources that have triggered and disable
 * them temporarily.  Loop around until no extra interrupt
 * sources have triggered, at which point, the valid part of
 * the interrupt status register will read zero, clearing the
 * cause of the interrupt.
 *
 * Mask off interrupt sources already seen to avoid infinite
 * loop in case of misconfiguration.
 */
  cur_enabled = subpriv->enabled_isns;
  while ((intstat = (dio200_read8(dev, subpriv->ofs) &
       subpriv->valid_isns & ~triggered)) != 0) {
   triggered |= intstat;
   cur_enabled &= ~triggered;
   dio200_write8(dev, subpriv->ofs, cur_enabled);
  }
 } else {
  /*
 * No interrupt status register.  Assume the single interrupt
 * source has triggered.
 */
  triggered = subpriv->enabled_isns;
 }

 if (triggered) {
  /*
 * Some interrupt sources have triggered and have been
 * temporarily disabled to clear the cause of the interrupt.
 *
 * Reenable them NOW to minimize the time they are disabled.
 */
  cur_enabled = subpriv->enabled_isns;
  if (board->has_int_sce)
   dio200_write8(dev, subpriv->ofs, cur_enabled);

  if (subpriv->active) {
   /*
 * The command is still active.
 *
 * Ignore interrupt sources that the command isn't
 * interested in (just in case there's a race
 * condition).
 */
   if (triggered & subpriv->enabled_isns) {
    /* Collect scan data. */
    dio200_read_scan_intr(dev, s, triggered);
   }
  }
 }
 spin_unlock_irqrestore(&subpriv->spinlock, flags);

 comedi_handle_events(dev, s);

 return (triggered != 0);
}

static int dio200_subdev_intr_cancel(struct comedi_device *dev,
         struct comedi_subdevice *s)
{
 struct dio200_subdev_intr *subpriv = s->private;
 unsigned long flags;

 spin_lock_irqsave(&subpriv->spinlock, flags);
 if (subpriv->active)
  dio200_stop_intr(dev, s);

 spin_unlock_irqrestore(&subpriv->spinlock, flags);

 return 0;
}

static int dio200_subdev_intr_cmdtest(struct comedi_device *dev,
          struct comedi_subdevice *s,
          struct comedi_cmd *cmd)
{
 int err = 0;

 /* Step 1 : check if triggers are trivially valid */

 err |= comedi_check_trigger_src(&cmd->start_src, TRIG_NOW | TRIG_INT);
 err |= comedi_check_trigger_src(&cmd->scan_begin_src, TRIG_EXT);
 err |= comedi_check_trigger_src(&cmd->convert_src, TRIG_NOW);
 err |= comedi_check_trigger_src(&cmd->scan_end_src, TRIG_COUNT);
 err |= comedi_check_trigger_src(&cmd->stop_src, TRIG_COUNT | TRIG_NONE);

 if (err)
  return 1;

 /* Step 2a : make sure trigger sources are unique */

 err |= comedi_check_trigger_is_unique(cmd->start_src);
 err |= comedi_check_trigger_is_unique(cmd->stop_src);

 /* Step 2b : and mutually compatible */

 if (err)
  return 2;

 /* Step 3: check if arguments are trivially valid */

 err |= comedi_check_trigger_arg_is(&cmd->start_arg, 0);
 err |= comedi_check_trigger_arg_is(&cmd->scan_begin_arg, 0);
 err |= comedi_check_trigger_arg_is(&cmd->convert_arg, 0);
 err |= comedi_check_trigger_arg_is(&cmd->scan_end_arg,
        cmd->chanlist_len);

 if (cmd->stop_src == TRIG_COUNT)
  err |= comedi_check_trigger_arg_min(&cmd->stop_arg, 1);
 else /* TRIG_NONE */
  err |= comedi_check_trigger_arg_is(&cmd->stop_arg, 0);

 if (err)
  return 3;

 /* step 4: fix up any arguments */

 /* if (err) return 4; */

 return 0;
}

static int dio200_subdev_intr_cmd(struct comedi_device *dev,
      struct comedi_subdevice *s)
{
 struct comedi_cmd *cmd = &s->async->cmd;
 struct dio200_subdev_intr *subpriv = s->private;
 unsigned long flags;

 spin_lock_irqsave(&subpriv->spinlock, flags);

 subpriv->active = true;

 if (cmd->start_src == TRIG_INT)
  s->async->inttrig = dio200_inttrig_start_intr;
 else /* TRIG_NOW */
  dio200_start_intr(dev, s);

 spin_unlock_irqrestore(&subpriv->spinlock, flags);

 return 0;
}

static int dio200_subdev_intr_init(struct comedi_device *dev,
       struct comedi_subdevice *s,
       unsigned int offset,
       unsigned int valid_isns)
{
 const struct dio200_board *board = dev->board_ptr;
 struct dio200_subdev_intr *subpriv;

 subpriv = comedi_alloc_spriv(s, sizeof(*subpriv));
 if (!subpriv)
  return -ENOMEM;

 subpriv->ofs = offset;
 subpriv->valid_isns = valid_isns;
 spin_lock_init(&subpriv->spinlock);

 if (board->has_int_sce)
  /* Disable interrupt sources. */
  dio200_write8(dev, subpriv->ofs, 0);

 s->type = COMEDI_SUBD_DI;
 s->subdev_flags = SDF_READABLE | SDF_CMD_READ | SDF_PACKED;
 if (board->has_int_sce) {
  s->n_chan = DIO200_MAX_ISNS;
  s->len_chanlist = DIO200_MAX_ISNS;
 } else {
  /* No interrupt source register.  Support single channel. */
  s->n_chan = 1;
  s->len_chanlist = 1;
 }
 s->range_table = &range_digital;
 s->maxdata = 1;
 s->insn_bits = dio200_subdev_intr_insn_bits;
 s->do_cmdtest = dio200_subdev_intr_cmdtest;
 s->do_cmd = dio200_subdev_intr_cmd;
 s->cancel = dio200_subdev_intr_cancel;

 return 0;
}

static irqreturn_t dio200_interrupt(int irq, void *d)
{
 struct comedi_device *dev = d;
 struct comedi_subdevice *s = dev->read_subdev;
 int handled;

 if (!dev->attached)
  return IRQ_NONE;

 handled = dio200_handle_read_intr(dev, s);

 return IRQ_RETVAL(handled);
}

static void dio200_subdev_8254_set_gate_src(struct comedi_device *dev,
         struct comedi_subdevice *s,
         unsigned int chan,
         unsigned int src)
{
 unsigned int offset = dio200_subdev_8254_offset(dev, s);

 dio200_write8(dev, DIO200_GAT_SCE(offset >> 3),
        clk_gat_sce((offset >> 2) & 1, chan, src));
}

static void dio200_subdev_8254_set_clock_src(struct comedi_device *dev,
          struct comedi_subdevice *s,
          unsigned int chan,
          unsigned int src)
{
 unsigned int offset = dio200_subdev_8254_offset(dev, s);

 dio200_write8(dev, DIO200_CLK_SCE(offset >> 3),
        clk_gat_sce((offset >> 2) & 1, chan, src));
}

static int dio200_subdev_8254_config(struct comedi_device *dev,
         struct comedi_subdevice *s,
         struct comedi_insn *insn,
         unsigned int *data)
{
 const struct dio200_board *board = dev->board_ptr;
 struct comedi_8254 *i8254 = s->private;
 unsigned int chan = CR_CHAN(insn->chanspec);
 unsigned int max_src = board->is_pcie ? 31 : 7;
 unsigned int src;

 if (!board->has_clk_gat_sce)
  return -EINVAL;

 switch (data[0]) {
 case INSN_CONFIG_SET_GATE_SRC:
  src = data[2];
  if (src > max_src)
   return -EINVAL;

  dio200_subdev_8254_set_gate_src(dev, s, chan, src);
  i8254->gate_src[chan] = src;
  break;
 case INSN_CONFIG_GET_GATE_SRC:
  data[2] = i8254->gate_src[chan];
  break;
 case INSN_CONFIG_SET_CLOCK_SRC:
  src = data[1];
  if (src > max_src)
   return -EINVAL;

  dio200_subdev_8254_set_clock_src(dev, s, chan, src);
  i8254->clock_src[chan] = src;
  break;
 case INSN_CONFIG_GET_CLOCK_SRC:
  data[1] = i8254->clock_src[chan];
  data[2] = clock_period[i8254->clock_src[chan]];
  break;
 default:
  return -EINVAL;
 }

 return insn->n;
}

static int dio200_subdev_8254_init(struct comedi_device *dev,
       struct comedi_subdevice *s,
       unsigned int offset)
{
 const struct dio200_board *board = dev->board_ptr;
 struct comedi_8254 *i8254;
 unsigned int regshift;
 int chan;

 /*
 * PCIe boards need the offset shifted in order to get the
 * correct base address of the timer.
 */
 if (board->is_pcie) {
  offset <<= 3;
  regshift = 3;
 } else {
  regshift = 0;
 }

 if (dev->mmio) {
  i8254 = comedi_8254_mm_alloc(dev->mmio + offset,
          0, I8254_IO8, regshift);
 } else {
  i8254 = comedi_8254_io_alloc(dev->iobase + offset,
          0, I8254_IO8, regshift);
 }
 if (IS_ERR(i8254))
  return PTR_ERR(i8254);

 comedi_8254_subdevice_init(s, i8254);

 i8254->insn_config = dio200_subdev_8254_config;

 /*
 * There could be multiple timers so this driver does not
 * use dev->pacer to save the i8254 pointer. Instead,
 * comedi_8254_subdevice_init() saved the i8254 pointer in
 * s->private.  Mark the subdevice as having private data
 * to be automatically freed when the device is detached.
 */
 comedi_set_spriv_auto_free(s);

 /* Initialize channels. */
 if (board->has_clk_gat_sce) {
  for (chan = 0; chan < 3; chan++) {
   /* Gate source 0 is VCC (logic 1). */
   dio200_subdev_8254_set_gate_src(dev, s, chan, 0);
   /* Clock source 0 is the dedicated clock input. */
   dio200_subdev_8254_set_clock_src(dev, s, chan, 0);
  }
 }

 return 0;
}

static void dio200_subdev_8255_set_dir(struct comedi_device *dev,
           struct comedi_subdevice *s)
{
 struct dio200_subdev_8255 *subpriv = s->private;
 int config;

 config = I8255_CTRL_CW;
 /* 1 in io_bits indicates output, 1 in config indicates input */
 if (!(s->io_bits & 0x0000ff))
  config |= I8255_CTRL_A_IO;
 if (!(s->io_bits & 0x00ff00))
  config |= I8255_CTRL_B_IO;
 if (!(s->io_bits & 0x0f0000))
  config |= I8255_CTRL_C_LO_IO;
 if (!(s->io_bits & 0xf00000))
  config |= I8255_CTRL_C_HI_IO;
 dio200_write8(dev, subpriv->ofs + I8255_CTRL_REG, config);
}

static int dio200_subdev_8255_bits(struct comedi_device *dev,
       struct comedi_subdevice *s,
       struct comedi_insn *insn,
       unsigned int *data)
{
 struct dio200_subdev_8255 *subpriv = s->private;
 unsigned int mask;
 unsigned int val;

 mask = comedi_dio_update_state(s, data);
 if (mask) {
  if (mask & 0xff) {
   dio200_write8(dev, subpriv->ofs + I8255_DATA_A_REG,
          s->state & 0xff);
  }
  if (mask & 0xff00) {
   dio200_write8(dev, subpriv->ofs + I8255_DATA_B_REG,
          (s->state >> 8) & 0xff);
  }
  if (mask & 0xff0000) {
   dio200_write8(dev, subpriv->ofs + I8255_DATA_C_REG,
          (s->state >> 16) & 0xff);
  }
 }

 val = dio200_read8(dev, subpriv->ofs + I8255_DATA_A_REG);
 val |= dio200_read8(dev, subpriv->ofs + I8255_DATA_B_REG) << 8;
 val |= dio200_read8(dev, subpriv->ofs + I8255_DATA_C_REG) << 16;

 data[1] = val;

 return insn->n;
}

static int dio200_subdev_8255_config(struct comedi_device *dev,
         struct comedi_subdevice *s,
         struct comedi_insn *insn,
         unsigned int *data)
{
 unsigned int chan = CR_CHAN(insn->chanspec);
 unsigned int mask;
 int ret;

 if (chan < 8)
  mask = 0x0000ff;
 else if (chan < 16)
  mask = 0x00ff00;
 else if (chan < 20)
  mask = 0x0f0000;
 else
  mask = 0xf00000;

 ret = comedi_dio_insn_config(dev, s, insn, data, mask);
 if (ret)
  return ret;

 dio200_subdev_8255_set_dir(dev, s);

 return insn->n;
}

static int dio200_subdev_8255_init(struct comedi_device *dev,
       struct comedi_subdevice *s,
       unsigned int offset)
{
 struct dio200_subdev_8255 *subpriv;

 subpriv = comedi_alloc_spriv(s, sizeof(*subpriv));
 if (!subpriv)
  return -ENOMEM;

 subpriv->ofs = offset;

 s->type = COMEDI_SUBD_DIO;
 s->subdev_flags = SDF_READABLE | SDF_WRITABLE;
 s->n_chan = 24;
 s->range_table = &range_digital;
 s->maxdata = 1;
 s->insn_bits = dio200_subdev_8255_bits;
 s->insn_config = dio200_subdev_8255_config;
 dio200_subdev_8255_set_dir(dev, s);
 return 0;
}

static int dio200_subdev_timer_read(struct comedi_device *dev,
        struct comedi_subdevice *s,
        struct comedi_insn *insn,
        unsigned int *data)
{
 unsigned int n;

 for (n = 0; n < insn->n; n++)
  data[n] = dio200_read32(dev, DIO200_TS_COUNT);
 return n;
}

static void dio200_subdev_timer_reset(struct comedi_device *dev,
          struct comedi_subdevice *s)
{
 unsigned int clock;

 clock = dio200_read32(dev, DIO200_TS_CONFIG) & TS_CONFIG_CLK_SRC_MASK;
 dio200_write32(dev, DIO200_TS_CONFIG, clock | TS_CONFIG_RESET);
 dio200_write32(dev, DIO200_TS_CONFIG, clock);
}

static void dio200_subdev_timer_get_clock_src(struct comedi_device *dev,
           struct comedi_subdevice *s,
           unsigned int *src,
           unsigned int *period)
{
 unsigned int clk;

 clk = dio200_read32(dev, DIO200_TS_CONFIG) & TS_CONFIG_CLK_SRC_MASK;
 *src = clk;
 *period = (clk < ARRAY_SIZE(ts_clock_period)) ?
    ts_clock_period[clk] : 0;
}

static int dio200_subdev_timer_set_clock_src(struct comedi_device *dev,
          struct comedi_subdevice *s,
          unsigned int src)
{
 if (src > TS_CONFIG_MAX_CLK_SRC)
  return -EINVAL;
 dio200_write32(dev, DIO200_TS_CONFIG, src);
 return 0;
}

static int dio200_subdev_timer_config(struct comedi_device *dev,
          struct comedi_subdevice *s,
          struct comedi_insn *insn,
          unsigned int *data)
{
 int ret = 0;

 switch (data[0]) {
 case INSN_CONFIG_RESET:
  dio200_subdev_timer_reset(dev, s);
  break;
 case INSN_CONFIG_SET_CLOCK_SRC:
  ret = dio200_subdev_timer_set_clock_src(dev, s, data[1]);
  if (ret < 0)
   ret = -EINVAL;
  break;
 case INSN_CONFIG_GET_CLOCK_SRC:
  dio200_subdev_timer_get_clock_src(dev, s, &data[1], &data['color: green'>2]);
  break;
 default:
  ret = -EINVAL;
  break;
 }
 return ret < 0 ? ret : insn->n;
}

void amplc_dio200_set_enhance(struct comedi_device *dev, unsigned char val)
{
 dio200_write8(dev, DIO200_ENHANCE, val);
}
EXPORT_SYMBOL_GPL(amplc_dio200_set_enhance);

int amplc_dio200_common_attach(struct comedi_device *dev, unsigned int irq,
          unsigned long req_irq_flags)
{
 const struct dio200_board *board = dev->board_ptr;
 struct comedi_subdevice *s;
 unsigned int n;
 int ret;

 if (!IS_ENABLED(CONFIG_HAS_IOPORT) && !dev->mmio) {
  dev_err(dev->class_dev,
   "error! need I/O port support\n");
  return -ENXIO;
 }

 ret = comedi_alloc_subdevices(dev, board->n_subdevs);
 if (ret)
  return ret;

 for (n = 0; n < dev->n_subdevices; n++) {
  s = &dev->subdevices[n];
  switch (board->sdtype[n]) {
  case sd_8254:
   /* counter subdevice (8254) */
   ret = dio200_subdev_8254_init(dev, s,
            board->sdinfo[n]);
   if (ret < 0)
    return ret;
   break;
  case sd_8255:
   /* digital i/o subdevice (8255) */
   ret = dio200_subdev_8255_init(dev, s,
            board->sdinfo[n]);
   if (ret < 0)
    return ret;
   break;
  case sd_intr:
   /* 'INTERRUPT' subdevice */
   if (irq && !dev->read_subdev) {
    ret = dio200_subdev_intr_init(dev, s,
             DIO200_INT_SCE,
             board->sdinfo[n]);
    if (ret < 0)
     return ret;
    dev->read_subdev = s;
   } else {
    s->type = COMEDI_SUBD_UNUSED;
   }
   break;
  case sd_timer:
   s->type  = COMEDI_SUBD_TIMER;
   s->subdev_flags = SDF_READABLE | SDF_LSAMPL;
   s->n_chan = 1;
   s->maxdata = 0xffffffff;
   s->insn_read = dio200_subdev_timer_read;
   s->insn_config = dio200_subdev_timer_config;
   break;
  default:
   s->type = COMEDI_SUBD_UNUSED;
   break;
  }
 }

 if (irq && dev->read_subdev) {
  if (request_irq(irq, dio200_interrupt, req_irq_flags,
    dev->board_name, dev) >= 0) {
   dev->irq = irq;
  } else {
   dev_warn(dev->class_dev,
     "warning! irq %u unavailable!\n", irq);
  }
 }

 return 0;
}
EXPORT_SYMBOL_GPL(amplc_dio200_common_attach);

static int __init amplc_dio200_common_init(void)
{
 return 0;
}
module_init(amplc_dio200_common_init);

static void __exit amplc_dio200_common_exit(void)
{
}
module_exit(amplc_dio200_common_exit);

MODULE_AUTHOR("Comedi https://www.comedi.org");
MODULE_DESCRIPTION("Comedi helper for amplc_dio200 and amplc_dio200_pci");
MODULE_LICENSE("GPL");