/*      $NetBSD: wdc_buddha.c,v 1.11 2023/12/20 00:40:42 thorpej Exp $  */

/*-
 * Copyright (c) 2000 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Michael L. Hitch and Ignatios Souvatzis.
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``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 FOUNDATION OR CONTRIBUTORS
 * 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.
 */

#include <sys/types.h>
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/device.h>
#include <sys/bus.h>

#include <machine/cpu.h>
#include <machine/intr.h>
#include <machine/bswap.h>

#include <amiga/amiga/cia.h>
#include <amiga/amiga/custom.h>
#include <amiga/amiga/device.h>
#include <amiga/dev/zbusvar.h>

#include <dev/ata/atavar.h>
#include <dev/ic/wdcvar.h>

#define BUDDHA_MAX_CHANNELS 3

struct wdc_buddha_softc {
        struct wdc_softc sc_wdcdev;
        struct ata_channel *wdc_chanarray[BUDDHA_MAX_CHANNELS];
        struct ata_channel channels[BUDDHA_MAX_CHANNELS];
        struct bus_space_tag sc_iot;
        struct isr sc_isr;
        volatile char *ba;
};

int     wdc_buddha_probe(device_t, cfdata_t, void *);
void    wdc_buddha_attach(device_t, device_t, void *);
int     wdc_buddha_intr(void *);

CFATTACH_DECL_NEW(wdc_buddha, sizeof(struct wdc_buddha_softc),
    wdc_buddha_probe, wdc_buddha_attach, NULL, NULL);

int
wdc_buddha_probe(device_t parent, cfdata_t cfp, void *aux)
{
        struct zbus_args *zap;

        zap = aux;

        if (zap->manid != 4626)
                return 0;

        if ((zap->prodid != 0) &&       /* Buddha */
            (zap->prodid != 42))        /* Catweasel */
                return 0;

        return 1;
}

void
wdc_buddha_attach(device_t parent, device_t self, void *aux)
{
        struct wdc_buddha_softc *sc;
        struct zbus_args *zap;
        int nchannels;
        int ch;

        sc = device_private(self);
        sc->sc_wdcdev.sc_atac.atac_dev = self;
        zap = aux;

        sc->ba = zap->va;

        sc->sc_iot.base = (bus_addr_t)sc->ba;
        sc->sc_iot.absm = &amiga_bus_stride_4swap;

        nchannels = 2;
        if (zap->prodid == 42) {
                aprint_normal(": Catweasel Z2\n");
                nchannels = 3;
        } else if (zap->serno == 0) 
                aprint_normal(": Buddha\n");
        else
                aprint_normal(": Buddha Flash\n");

        /* XXX pio mode setting not implemented yet. */
        sc->sc_wdcdev.sc_atac.atac_cap = ATAC_CAP_DATA16;
        sc->sc_wdcdev.sc_atac.atac_pio_cap = 0;
        sc->sc_wdcdev.sc_atac.atac_channels = sc->wdc_chanarray;
        sc->sc_wdcdev.sc_atac.atac_nchannels = nchannels;
        sc->sc_wdcdev.wdc_maxdrives = 2;

        wdc_allocate_regs(&sc->sc_wdcdev);

        for (ch = 0; ch < nchannels; ch++) {
                struct ata_channel *cp;
                struct wdc_regs *wdr;
                int i;

                cp = &sc->channels[ch];
                sc->wdc_chanarray[ch] = cp;

                cp->ch_channel = ch;
                cp->ch_atac = &sc->sc_wdcdev.sc_atac;

                /*
                 * XXX According to the Buddha docs, we should use a method
                 * array that adds 0x40 to the address for byte accesses, to
                 * get the slow timing for command accesses, and the 0x00 
                 * offset for the word (fast) accesses. This will be
                 * reconsidered when implementing setting the timing.
                 *
                 * XXX We also could consider to abuse the 32bit capability, or
                 * 32bit accesses to the words (which will read in two words)
                 * for better performance.
                 *              -is
                 */
                wdr = CHAN_TO_WDC_REGS(cp);

                wdr->cmd_iot = &sc->sc_iot;
                if (bus_space_map(wdr->cmd_iot, 0x210+ch*0x80, 8, 0,
                    &wdr->cmd_baseioh)) {
                        aprint_error_dev(self, "couldn't map cmd registers\n");
                        return;
                }

                wdr->ctl_iot = &sc->sc_iot;
                if (bus_space_map(wdr->ctl_iot, 0x250+ch*0x80, 2, 0,
                    &wdr->ctl_ioh)) {
                        bus_space_unmap(wdr->cmd_iot, wdr->cmd_baseioh, 8);
                        aprint_error_dev(self, "couldn't map ctl registers\n");
                        return;
                }

                for (i = 0; i < WDC_NREG; i++) {
                        if (bus_space_subregion(wdr->cmd_iot, wdr->cmd_baseioh,
                            i, i == 0 ? 4 : 1, &wdr->cmd_iohs[i]) != 0) {
                                aprint_error_dev(self,
                                    "couldn't subregion cmd regs\n");
                                return;
                        }
                }

                wdc_init_shadow_regs(wdr);
                wdcattach(cp);
        }

        sc->sc_isr.isr_intr = wdc_buddha_intr;
        sc->sc_isr.isr_arg = sc;
        sc->sc_isr.isr_ipl = 2;
        add_isr (&sc->sc_isr);
        sc->ba[0xfc0] = 0;      /* enable interrupts */
}

int
wdc_buddha_intr(void *arg)
{
        struct wdc_buddha_softc *sc = (struct wdc_buddha_softc *)arg;
        int nchannels, i, ret;
        volatile char *p;

        ret = 0;
        nchannels = sc->sc_wdcdev.sc_atac.atac_nchannels;
        p = sc->ba;

        for (i=0; i<nchannels; i++) {
                if (p[0xf00 + 0x40*i] & 0x80) {
                        wdcintr(&sc->channels[i]);
                        ret = 1;
                }
        }

        return ret;
}