/*
 * Copyright (c) 2008-2011 Atheros Communications Inc.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include "hw.h"
#include "hw-ops.h"
#include "ar9002_phy.h"

#define AR9285_CLCAL_REDO_THRESH    1
/* AGC & I/Q calibrations time limit, ms */
#define AR9002_CAL_MAX_TIME             30000

enum ar9002_cal_types {
        ADC_GAIN_CAL = BIT(0),
        ADC_DC_CAL = BIT(1),
        IQ_MISMATCH_CAL = BIT(2),
};

static bool ar9002_hw_is_cal_supported(struct ath_hw *ah,
                                struct ath9k_channel *chan,
                                enum ar9002_cal_types cal_type)
{
        bool supported = false;
        switch (ah->supp_cals & cal_type) {
        case IQ_MISMATCH_CAL:
                supported = true;
                break;
        case ADC_GAIN_CAL:
        case ADC_DC_CAL:
                /* Run even/odd ADCs calibrations for HT40 channels only */
                if (IS_CHAN_HT40(chan))
                        supported = true;
                break;
        }
        return supported;
}

static void ar9002_hw_setup_calibration(struct ath_hw *ah,
                                        struct ath9k_cal_list *currCal)
{
        struct ath_common *common = ath9k_hw_common(ah);

        REG_RMW_FIELD(ah, AR_PHY_TIMING_CTRL4(0),
                      AR_PHY_TIMING_CTRL4_IQCAL_LOG_COUNT_MAX,
                      currCal->calData->calCountMax);

        switch (currCal->calData->calType) {
        case IQ_MISMATCH_CAL:
                REG_WRITE(ah, AR_PHY_CALMODE, AR_PHY_CALMODE_IQ);
                ath_dbg(common, CALIBRATE,
                        "starting IQ Mismatch Calibration\n");
                break;
        case ADC_GAIN_CAL:
                REG_WRITE(ah, AR_PHY_CALMODE, AR_PHY_CALMODE_ADC_GAIN);
                ath_dbg(common, CALIBRATE, "starting ADC Gain Calibration\n");
                break;
        case ADC_DC_CAL:
                REG_WRITE(ah, AR_PHY_CALMODE, AR_PHY_CALMODE_ADC_DC_PER);
                ath_dbg(common, CALIBRATE, "starting ADC DC Calibration\n");
                break;
        }

        REG_SET_BIT(ah, AR_PHY_TIMING_CTRL4(0),
                    AR_PHY_TIMING_CTRL4_DO_CAL);
}

static bool ar9002_hw_per_calibration(struct ath_hw *ah,
                                      struct ath9k_channel *ichan,
                                      u8 rxchainmask,
                                      struct ath9k_cal_list *currCal)
{
        struct ath9k_hw_cal_data *caldata = ah->caldata;
        bool iscaldone = false;

        if (currCal->calState == CAL_RUNNING) {
                if (!(REG_READ(ah, AR_PHY_TIMING_CTRL4(0)) &
                      AR_PHY_TIMING_CTRL4_DO_CAL)) {

                        currCal->calData->calCollect(ah);
                        ah->cal_samples++;

                        if (ah->cal_samples >=
                            currCal->calData->calNumSamples) {
                                int i, numChains = 0;
                                for (i = 0; i < AR5416_MAX_CHAINS; i++) {
                                        if (rxchainmask & (1 << i))
                                                numChains++;
                                }

                                currCal->calData->calPostProc(ah, numChains);
                                caldata->CalValid |= currCal->calData->calType;
                                currCal->calState = CAL_DONE;
                                iscaldone = true;
                        } else {
                                ar9002_hw_setup_calibration(ah, currCal);
                        }
                } else if (time_after(jiffies, ah->cal_start_time +
                                      msecs_to_jiffies(AR9002_CAL_MAX_TIME))) {
                        REG_CLR_BIT(ah, AR_PHY_TIMING_CTRL4(0),
                                    AR_PHY_TIMING_CTRL4_DO_CAL);
                        ath_dbg(ath9k_hw_common(ah), CALIBRATE,
                                "calibration timeout\n");
                        currCal->calState = CAL_WAITING;        /* Try later */
                        iscaldone = true;
                }
        } else if (!(caldata->CalValid & currCal->calData->calType)) {
                ath9k_hw_reset_calibration(ah, currCal);
        }

        return iscaldone;
}

static void ar9002_hw_iqcal_collect(struct ath_hw *ah)
{
        int i;

        for (i = 0; i < AR5416_MAX_CHAINS; i++) {
                ah->totalPowerMeasI[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_0(i));
                ah->totalPowerMeasQ[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_1(i));
                ah->totalIqCorrMeas[i] +=
                        (int32_t) REG_READ(ah, AR_PHY_CAL_MEAS_2(i));
                ath_dbg(ath9k_hw_common(ah), CALIBRATE,
                        "%d: Chn %d pmi=0x%08x;pmq=0x%08x;iqcm=0x%08x;\n",
                        ah->cal_samples, i, ah->totalPowerMeasI[i],
                        ah->totalPowerMeasQ[i],
                        ah->totalIqCorrMeas[i]);
        }
}

static void ar9002_hw_adc_gaincal_collect(struct ath_hw *ah)
{
        int i;

        for (i = 0; i < AR5416_MAX_CHAINS; i++) {
                ah->totalAdcIOddPhase[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_0(i));
                ah->totalAdcIEvenPhase[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_1(i));
                ah->totalAdcQOddPhase[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_2(i));
                ah->totalAdcQEvenPhase[i] +=
                        REG_READ(ah, AR_PHY_CAL_MEAS_3(i));

                ath_dbg(ath9k_hw_common(ah), CALIBRATE,
                        "%d: Chn %d oddi=0x%08x; eveni=0x%08x; oddq=0x%08x; evenq=0x%08x;\n",
                        ah->cal_samples, i,
                        ah->totalAdcIOddPhase[i],
                        ah->totalAdcIEvenPhase[i],
                        ah->totalAdcQOddPhase[i],
                        ah->totalAdcQEvenPhase[i]);
        }
}

static void ar9002_hw_adc_dccal_collect(struct ath_hw *ah)
{
        int i;

        for (i = 0; i < AR5416_MAX_CHAINS; i++) {
                ah->totalAdcDcOffsetIOddPhase[i] +=
                        (int32_t) REG_READ(ah, AR_PHY_CAL_MEAS_0(i));
                ah->totalAdcDcOffsetIEvenPhase[i] +=
                        (int32_t) REG_READ(ah, AR_PHY_CAL_MEAS_1(i));
                ah->totalAdcDcOffsetQOddPhase[i] +=
                        (int32_t) REG_READ(ah, AR_PHY_CAL_MEAS_2(i));
                ah->totalAdcDcOffsetQEvenPhase[i] +=
                        (int32_t) REG_READ(ah, AR_PHY_CAL_MEAS_3(i));

                ath_dbg(ath9k_hw_common(ah), CALIBRATE,
                        "%d: Chn %d oddi=0x%08x; eveni=0x%08x; oddq=0x%08x; evenq=0x%08x;\n",
                        ah->cal_samples, i,
                        ah->totalAdcDcOffsetIOddPhase[i],
                        ah->totalAdcDcOffsetIEvenPhase[i],
                        ah->totalAdcDcOffsetQOddPhase[i],
                        ah->totalAdcDcOffsetQEvenPhase[i]);
        }
}

static void ar9002_hw_iqcalibrate(struct ath_hw *ah, u8 numChains)
{
        struct ath_common *common = ath9k_hw_common(ah);
        u32 powerMeasQ, powerMeasI, iqCorrMeas;
        u32 qCoffDenom, iCoffDenom;
        int32_t qCoff, iCoff;
        int iqCorrNeg, i;

        for (i = 0; i < numChains; i++) {
                powerMeasI = ah->totalPowerMeasI[i];
                powerMeasQ = ah->totalPowerMeasQ[i];
                iqCorrMeas = ah->totalIqCorrMeas[i];

                ath_dbg(common, CALIBRATE,
                        "Starting IQ Cal and Correction for Chain %d\n",
                        i);

                ath_dbg(common, CALIBRATE,
                        "Original: Chn %d iq_corr_meas = 0x%08x\n",
                        i, ah->totalIqCorrMeas[i]);

                iqCorrNeg = 0;

                if (iqCorrMeas > 0x80000000) {
                        iqCorrMeas = (0xffffffff - iqCorrMeas) + 1;
                        iqCorrNeg = 1;
                }

                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_i = 0x%08x\n",
                        i, powerMeasI);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_q = 0x%08x\n",
                        i, powerMeasQ);
                ath_dbg(common, CALIBRATE, "iqCorrNeg is 0x%08x\n", iqCorrNeg);

                iCoffDenom = (powerMeasI / 2 + powerMeasQ / 2) / 128;
                qCoffDenom = powerMeasQ / 64;

                if ((powerMeasQ != 0) && (iCoffDenom != 0) &&
                    (qCoffDenom != 0)) {
                        iCoff = iqCorrMeas / iCoffDenom;
                        qCoff = powerMeasI / qCoffDenom - 64;
                        ath_dbg(common, CALIBRATE, "Chn %d iCoff = 0x%08x\n",
                                i, iCoff);
                        ath_dbg(common, CALIBRATE, "Chn %d qCoff = 0x%08x\n",
                                i, qCoff);

                        iCoff = iCoff & 0x3f;
                        ath_dbg(common, CALIBRATE,
                                "New: Chn %d iCoff = 0x%08x\n", i, iCoff);
                        if (iqCorrNeg == 0x0)
                                iCoff = 0x40 - iCoff;

                        if (qCoff > 15)
                                qCoff = 15;
                        else if (qCoff <= -16)
                                qCoff = -16;

                        ath_dbg(common, CALIBRATE,
                                "Chn %d : iCoff = 0x%x  qCoff = 0x%x\n",
                                i, iCoff, qCoff);

                        REG_RMW_FIELD(ah, AR_PHY_TIMING_CTRL4(i),
                                      AR_PHY_TIMING_CTRL4_IQCORR_Q_I_COFF,
                                      iCoff);
                        REG_RMW_FIELD(ah, AR_PHY_TIMING_CTRL4(i),
                                      AR_PHY_TIMING_CTRL4_IQCORR_Q_Q_COFF,
                                      qCoff);
                        ath_dbg(common, CALIBRATE,
                                "IQ Cal and Correction done for Chain %d\n",
                                i);
                }
        }

        REG_SET_BIT(ah, AR_PHY_TIMING_CTRL4(0),
                    AR_PHY_TIMING_CTRL4_IQCORR_ENABLE);
}

static void ar9002_hw_adc_gaincal_calibrate(struct ath_hw *ah, u8 numChains)
{
        struct ath_common *common = ath9k_hw_common(ah);
        u32 iOddMeasOffset, iEvenMeasOffset, qOddMeasOffset, qEvenMeasOffset;
        u32 qGainMismatch, iGainMismatch, val, i;

        for (i = 0; i < numChains; i++) {
                iOddMeasOffset = ah->totalAdcIOddPhase[i];
                iEvenMeasOffset = ah->totalAdcIEvenPhase[i];
                qOddMeasOffset = ah->totalAdcQOddPhase[i];
                qEvenMeasOffset = ah->totalAdcQEvenPhase[i];

                ath_dbg(common, CALIBRATE,
                        "Starting ADC Gain Cal for Chain %d\n", i);

                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_odd_i = 0x%08x\n",
                        i, iOddMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_even_i = 0x%08x\n",
                        i, iEvenMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_odd_q = 0x%08x\n",
                        i, qOddMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_even_q = 0x%08x\n",
                        i, qEvenMeasOffset);

                if (iOddMeasOffset != 0 && qEvenMeasOffset != 0) {
                        iGainMismatch =
                                ((iEvenMeasOffset * 32) /
                                 iOddMeasOffset) & 0x3f;
                        qGainMismatch =
                                ((qOddMeasOffset * 32) /
                                 qEvenMeasOffset) & 0x3f;

                        ath_dbg(common, CALIBRATE,
                                "Chn %d gain_mismatch_i = 0x%08x\n",
                                i, iGainMismatch);
                        ath_dbg(common, CALIBRATE,
                                "Chn %d gain_mismatch_q = 0x%08x\n",
                                i, qGainMismatch);

                        val = REG_READ(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(i));
                        val &= 0xfffff000;
                        val |= (qGainMismatch) | (iGainMismatch << 6);
                        REG_WRITE(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(i), val);

                        ath_dbg(common, CALIBRATE,
                                "ADC Gain Cal done for Chain %d\n", i);
                }
        }

        REG_WRITE(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(0),
                  REG_READ(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(0)) |
                  AR_PHY_NEW_ADC_GAIN_CORR_ENABLE);
}

static void ar9002_hw_adc_dccal_calibrate(struct ath_hw *ah, u8 numChains)
{
        struct ath_common *common = ath9k_hw_common(ah);
        u32 iOddMeasOffset, iEvenMeasOffset, val, i;
        int32_t qOddMeasOffset, qEvenMeasOffset, qDcMismatch, iDcMismatch;
        const struct ath9k_percal_data *calData =
                ah->cal_list_curr->calData;
        u32 numSamples =
                (1 << (calData->calCountMax + 5)) * calData->calNumSamples;

        for (i = 0; i < numChains; i++) {
                iOddMeasOffset = ah->totalAdcDcOffsetIOddPhase[i];
                iEvenMeasOffset = ah->totalAdcDcOffsetIEvenPhase[i];
                qOddMeasOffset = ah->totalAdcDcOffsetQOddPhase[i];
                qEvenMeasOffset = ah->totalAdcDcOffsetQEvenPhase[i];

                ath_dbg(common, CALIBRATE,
                        "Starting ADC DC Offset Cal for Chain %d\n", i);

                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_odd_i = %d\n",
                        i, iOddMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_even_i = %d\n",
                        i, iEvenMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_odd_q = %d\n",
                        i, qOddMeasOffset);
                ath_dbg(common, CALIBRATE, "Chn %d pwr_meas_even_q = %d\n",
                        i, qEvenMeasOffset);

                iDcMismatch = (((iEvenMeasOffset - iOddMeasOffset) * 2) /
                               numSamples) & 0x1ff;
                qDcMismatch = (((qOddMeasOffset - qEvenMeasOffset) * 2) /
                               numSamples) & 0x1ff;

                ath_dbg(common, CALIBRATE,
                        "Chn %d dc_offset_mismatch_i = 0x%08x\n",
                        i, iDcMismatch);
                ath_dbg(common, CALIBRATE,
                        "Chn %d dc_offset_mismatch_q = 0x%08x\n",
                        i, qDcMismatch);

                val = REG_READ(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(i));
                val &= 0xc0000fff;
                val |= (qDcMismatch << 12) | (iDcMismatch << 21);
                REG_WRITE(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(i), val);

                ath_dbg(common, CALIBRATE,
                        "ADC DC Offset Cal done for Chain %d\n", i);
        }

        REG_WRITE(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(0),
                  REG_READ(ah, AR_PHY_NEW_ADC_DC_GAIN_CORR(0)) |
                  AR_PHY_NEW_ADC_DC_OFFSET_CORR_ENABLE);
}

static void ar9287_hw_olc_temp_compensation(struct ath_hw *ah)
{
        u32 rddata;
        int32_t delta, currPDADC, slope;

        rddata = REG_READ(ah, AR_PHY_TX_PWRCTRL4);
        currPDADC = MS(rddata, AR_PHY_TX_PWRCTRL_PD_AVG_OUT);

        if (ah->initPDADC == 0 || currPDADC == 0) {
                /*
                 * Zero value indicates that no frames have been transmitted
                 * yet, can't do temperature compensation until frames are
                 * transmitted.
                 */
                return;
        } else {
                slope = ah->eep_ops->get_eeprom(ah, EEP_TEMPSENSE_SLOPE);

                if (slope == 0) { /* to avoid divide by zero case */
                        delta = 0;
                } else {
                        delta = ((currPDADC - ah->initPDADC)*4) / slope;
                }
                REG_RMW_FIELD(ah, AR_PHY_CH0_TX_PWRCTRL11,
                              AR_PHY_TX_PWRCTRL_OLPC_TEMP_COMP, delta);
                REG_RMW_FIELD(ah, AR_PHY_CH1_TX_PWRCTRL11,
                              AR_PHY_TX_PWRCTRL_OLPC_TEMP_COMP, delta);
        }
}

static void ar9280_hw_olc_temp_compensation(struct ath_hw *ah)
{
        u32 rddata, i;
        int delta, currPDADC, regval;

        rddata = REG_READ(ah, AR_PHY_TX_PWRCTRL4);
        currPDADC = MS(rddata, AR_PHY_TX_PWRCTRL_PD_AVG_OUT);

        if (ah->initPDADC == 0 || currPDADC == 0)
                return;

        if (ah->eep_ops->get_eeprom(ah, EEP_DAC_HPWR_5G))
                delta = (currPDADC - ah->initPDADC + 4) / 8;
        else
                delta = (currPDADC - ah->initPDADC + 5) / 10;

        if (delta != ah->PDADCdelta) {
                ah->PDADCdelta = delta;
                for (i = 1; i < AR9280_TX_GAIN_TABLE_SIZE; i++) {
                        regval = ah->originalGain[i] - delta;
                        if (regval < 0)
                                regval = 0;

                        REG_RMW_FIELD(ah,
                                      AR_PHY_TX_GAIN_TBL1 + i * 4,
                                      AR_PHY_TX_GAIN, regval);
                }
        }
}

static void ar9271_hw_pa_cal(struct ath_hw *ah, bool is_reset)
{
        u32 regVal;
        unsigned int i;
        u32 regList[][2] = {
                { AR9285_AN_TOP3, 0 },
                { AR9285_AN_RXTXBB1, 0 },
                { AR9285_AN_RF2G1, 0 },
                { AR9285_AN_RF2G2, 0 },
                { AR9285_AN_TOP2, 0 },
                { AR9285_AN_RF2G8, 0 },
                { AR9285_AN_RF2G7, 0 },
                { AR9285_AN_RF2G3, 0 },
        };

        REG_READ_ARRAY(ah, regList, ARRAY_SIZE(regList));

        ENABLE_REG_RMW_BUFFER(ah);
        /* 7834, b1=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G6, 1 << 0);
        /* 9808, b27=1 */
        REG_SET_BIT(ah, 0x9808, 1 << 27);
        /* 786c,b23,1, pwddac=1 */
        REG_SET_BIT(ah, AR9285_AN_TOP3, AR9285_AN_TOP3_PWDDAC);
        /* 7854, b5,1, pdrxtxbb=1 */
        REG_SET_BIT(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDRXTXBB1);
        /* 7854, b7,1, pdv2i=1 */
        REG_SET_BIT(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDV2I);
        /* 7854, b8,1, pddacinterface=1 */
        REG_SET_BIT(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDDACIF);
        /* 7824,b12,0, offcal=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G2, AR9285_AN_RF2G2_OFFCAL);
        /* 7838, b1,0, pwddb=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G7, AR9285_AN_RF2G7_PWDDB);
        /* 7820,b11,0, enpacal=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_ENPACAL);
        /* 7820,b25,1, pdpadrv1=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPADRV1);
        /* 7820,b24,0, pdpadrv2=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPADRV2);
        /* 7820,b23,0, pdpaout=0 */
        REG_CLR_BIT(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPAOUT);
        /* 783c,b14-16,7, padrvgn2tab_0=7 */
        REG_RMW_FIELD(ah, AR9285_AN_RF2G8, AR9285_AN_RF2G8_PADRVGN2TAB0, 7);
        /*
         * 7838,b29-31,0, padrvgn1tab_0=0
         * does not matter since we turn it off
         */
        REG_RMW_FIELD(ah, AR9285_AN_RF2G7, AR9285_AN_RF2G7_PADRVGN2TAB0, 0);
        /* 7828, b0-11, ccom=fff */
        REG_RMW_FIELD(ah, AR9285_AN_RF2G3, AR9271_AN_RF2G3_CCOMP, 0xfff);
        REG_RMW_BUFFER_FLUSH(ah);

        /* Set:
         * localmode=1,bmode=1,bmoderxtx=1,synthon=1,
         * txon=1,paon=1,oscon=1,synthon_force=1
         */
        REG_WRITE(ah, AR9285_AN_TOP2, 0xca0358a0);
        udelay(30);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G6, AR9271_AN_RF2G6_OFFS, 0);

        /* find off_6_1; */
        for (i = 6; i > 0; i--) {
                regVal = REG_READ(ah, AR9285_AN_RF2G6);
                regVal |= (1 << (20 + i));
                REG_WRITE(ah, AR9285_AN_RF2G6, regVal);
                udelay(1);
                /* regVal = REG_READ(ah, 0x7834); */
                regVal &= (~(0x1 << (20 + i)));
                regVal |= (MS(REG_READ(ah, AR9285_AN_RF2G9),
                              AR9285_AN_RXTXBB1_SPARE9)
                            << (20 + i));
                REG_WRITE(ah, AR9285_AN_RF2G6, regVal);
        }

        regVal = (regVal >> 20) & 0x7f;

        /* Update PA cal info */
        if ((!is_reset) && (ah->pacal_info.prev_offset == regVal)) {
                if (ah->pacal_info.max_skipcount < MAX_PACAL_SKIPCOUNT)
                        ah->pacal_info.max_skipcount =
                                2 * ah->pacal_info.max_skipcount;
                ah->pacal_info.skipcount = ah->pacal_info.max_skipcount;
        } else {
                ah->pacal_info.max_skipcount = 1;
                ah->pacal_info.skipcount = 0;
                ah->pacal_info.prev_offset = regVal;
        }


        ENABLE_REG_RMW_BUFFER(ah);
        /* 7834, b1=1 */
        REG_SET_BIT(ah, AR9285_AN_RF2G6, 1 << 0);
        /* 9808, b27=0 */
        REG_CLR_BIT(ah, 0x9808, 1 << 27);
        REG_RMW_BUFFER_FLUSH(ah);

        ENABLE_REGWRITE_BUFFER(ah);
        for (i = 0; i < ARRAY_SIZE(regList); i++)
                REG_WRITE(ah, regList[i][0], regList[i][1]);

        REGWRITE_BUFFER_FLUSH(ah);
}

static inline void ar9285_hw_pa_cal(struct ath_hw *ah, bool is_reset)
{
        struct ath_common *common = ath9k_hw_common(ah);
        u32 regVal;
        int i, offset, offs_6_1, offs_0;
        u32 ccomp_org, reg_field;
        u32 regList[][2] = {
                { 0x786c, 0 },
                { 0x7854, 0 },
                { 0x7820, 0 },
                { 0x7824, 0 },
                { 0x7868, 0 },
                { 0x783c, 0 },
                { 0x7838, 0 },
        };

        ath_dbg(common, CALIBRATE, "Running PA Calibration\n");

        /* PA CAL is not needed for high power solution */
        if (ah->eep_ops->get_eeprom(ah, EEP_TXGAIN_TYPE) ==
            AR5416_EEP_TXGAIN_HIGH_POWER)
                return;

        for (i = 0; i < ARRAY_SIZE(regList); i++)
                regList[i][1] = REG_READ(ah, regList[i][0]);

        regVal = REG_READ(ah, 0x7834);
        regVal &= (~(0x1));
        REG_WRITE(ah, 0x7834, regVal);
        regVal = REG_READ(ah, 0x9808);
        regVal |= (0x1 << 27);
        REG_WRITE(ah, 0x9808, regVal);

        REG_RMW_FIELD(ah, AR9285_AN_TOP3, AR9285_AN_TOP3_PWDDAC, 1);
        REG_RMW_FIELD(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDRXTXBB1, 1);
        REG_RMW_FIELD(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDV2I, 1);
        REG_RMW_FIELD(ah, AR9285_AN_RXTXBB1, AR9285_AN_RXTXBB1_PDDACIF, 1);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G2, AR9285_AN_RF2G2_OFFCAL, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G7, AR9285_AN_RF2G7_PWDDB, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_ENPACAL, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPADRV1, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPADRV2, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G1, AR9285_AN_RF2G1_PDPAOUT, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G8, AR9285_AN_RF2G8_PADRVGN2TAB0, 7);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G7, AR9285_AN_RF2G7_PADRVGN2TAB0, 0);
        ccomp_org = MS(REG_READ(ah, AR9285_AN_RF2G6), AR9285_AN_RF2G6_CCOMP);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G6, AR9285_AN_RF2G6_CCOMP, 0xf);

        REG_WRITE(ah, AR9285_AN_TOP2, 0xca0358a0);
        udelay(30);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G6, AR9285_AN_RF2G6_OFFS, 0);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G3, AR9285_AN_RF2G3_PDVCCOMP, 0);

        for (i = 6; i > 0; i--) {
                regVal = REG_READ(ah, 0x7834);
                regVal |= (1 << (19 + i));
                REG_WRITE(ah, 0x7834, regVal);
                udelay(1);
                regVal = REG_READ(ah, 0x7834);
                regVal &= (~(0x1 << (19 + i)));
                reg_field = MS(REG_READ(ah, 0x7840), AR9285_AN_RXTXBB1_SPARE9);
                regVal |= (reg_field << (19 + i));
                REG_WRITE(ah, 0x7834, regVal);
        }

        REG_RMW_FIELD(ah, AR9285_AN_RF2G3, AR9285_AN_RF2G3_PDVCCOMP, 1);
        udelay(1);
        reg_field = MS(REG_READ(ah, AR9285_AN_RF2G9), AR9285_AN_RXTXBB1_SPARE9);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G3, AR9285_AN_RF2G3_PDVCCOMP, reg_field);
        offs_6_1 = MS(REG_READ(ah, AR9285_AN_RF2G6), AR9285_AN_RF2G6_OFFS);
        offs_0   = MS(REG_READ(ah, AR9285_AN_RF2G3), AR9285_AN_RF2G3_PDVCCOMP);

        offset = (offs_6_1<<1) | offs_0;
        offset = offset - 0;
        offs_6_1 = offset>>1;
        offs_0 = offset & 1;

        if ((!is_reset) && (ah->pacal_info.prev_offset == offset)) {
                if (ah->pacal_info.max_skipcount < MAX_PACAL_SKIPCOUNT)
                        ah->pacal_info.max_skipcount =
                                2 * ah->pacal_info.max_skipcount;
                ah->pacal_info.skipcount = ah->pacal_info.max_skipcount;
        } else {
                ah->pacal_info.max_skipcount = 1;
                ah->pacal_info.skipcount = 0;
                ah->pacal_info.prev_offset = offset;
        }

        REG_RMW_FIELD(ah, AR9285_AN_RF2G6, AR9285_AN_RF2G6_OFFS, offs_6_1);
        REG_RMW_FIELD(ah, AR9285_AN_RF2G3, AR9285_AN_RF2G3_PDVCCOMP, offs_0);

        regVal = REG_READ(ah, 0x7834);
        regVal |= 0x1;
        REG_WRITE(ah, 0x7834, regVal);
        regVal = REG_READ(ah, 0x9808);
        regVal &= (~(0x1 << 27));
        REG_WRITE(ah, 0x9808, regVal);

        for (i = 0; i < ARRAY_SIZE(regList); i++)
                REG_WRITE(ah, regList[i][0], regList[i][1]);

        REG_RMW_FIELD(ah, AR9285_AN_RF2G6, AR9285_AN_RF2G6_CCOMP, ccomp_org);
}

static void ar9002_hw_pa_cal(struct ath_hw *ah, bool is_reset)
{
        if (AR_SREV_9271(ah)) {
                if (is_reset || !ah->pacal_info.skipcount)
                        ar9271_hw_pa_cal(ah, is_reset);
                else
                        ah->pacal_info.skipcount--;
        } else if (AR_SREV_9285_12_OR_LATER(ah)) {
                if (is_reset || !ah->pacal_info.skipcount)
                        ar9285_hw_pa_cal(ah, is_reset);
                else
                        ah->pacal_info.skipcount--;
        }
}

static void ar9002_hw_olc_temp_compensation(struct ath_hw *ah)
{
        if (OLC_FOR_AR9287_10_LATER(ah))
                ar9287_hw_olc_temp_compensation(ah);
        else if (OLC_FOR_AR9280_20_LATER(ah))
                ar9280_hw_olc_temp_compensation(ah);
}

static int ar9002_hw_calibrate(struct ath_hw *ah, struct ath9k_channel *chan,
                               u8 rxchainmask, bool longcal)
{
        struct ath9k_cal_list *currCal = ah->cal_list_curr;
        bool nfcal, nfcal_pending = false, percal_pending;
        int ret;

        nfcal = !!(REG_READ(ah, AR_PHY_AGC_CONTROL(ah)) & AR_PHY_AGC_CONTROL_NF);
        if (ah->caldata) {
                nfcal_pending = test_bit(NFCAL_PENDING, &ah->caldata->cal_flags);
                if (longcal)            /* Remember to not miss */
                        set_bit(LONGCAL_PENDING, &ah->caldata->cal_flags);
                else if (test_bit(LONGCAL_PENDING, &ah->caldata->cal_flags))
                        longcal = true; /* Respin a previous one */
        }

        percal_pending = (currCal &&
                          (currCal->calState == CAL_RUNNING ||
                           currCal->calState == CAL_WAITING));

        if (percal_pending && !nfcal) {
                if (!ar9002_hw_per_calibration(ah, chan, rxchainmask, currCal))
                        return 0;

                /* Looking for next waiting calibration if any */
                for (currCal = currCal->calNext; currCal != ah->cal_list_curr;
                     currCal = currCal->calNext) {
                        if (currCal->calState == CAL_WAITING)
                                break;
                }
                if (currCal->calState == CAL_WAITING) {
                        percal_pending = true;
                        ah->cal_list_curr = currCal;
                } else {
                        percal_pending = false;
                        ah->cal_list_curr = ah->cal_list;
                }
        }

        /* Do not start a next calibration if the longcal is in action */
        if (percal_pending && !nfcal && !longcal) {
                ath9k_hw_reset_calibration(ah, currCal);

                return 0;
        }

        /* Do NF cal only at longer intervals */
        if (longcal || nfcal_pending) {
                /*
                 * Get the value from the previous NF cal and update
                 * history buffer.
                 */
                if (ath9k_hw_getnf(ah, chan)) {
                        /*
                         * Load the NF from history buffer of the current
                         * channel.
                         * NF is slow time-variant, so it is OK to use a
                         * historical value.
                         */
                        ret = ath9k_hw_loadnf(ah, ah->curchan);
                        if (ret < 0)
                                return ret;
                }

                if (longcal) {
                        if (ah->caldata)
                                clear_bit(LONGCAL_PENDING,
                                          &ah->caldata->cal_flags);
                        ath9k_hw_start_nfcal(ah, false);
                        /* Do periodic PAOffset Cal */
                        ar9002_hw_pa_cal(ah, false);
                        ar9002_hw_olc_temp_compensation(ah);
                }
        }

        return !percal_pending;
}

/* Carrier leakage Calibration fix */
static bool ar9285_hw_cl_cal(struct ath_hw *ah, struct ath9k_channel *chan)
{
        struct ath_common *common = ath9k_hw_common(ah);

        REG_SET_BIT(ah, AR_PHY_CL_CAL_CTL, AR_PHY_CL_CAL_ENABLE);
        if (IS_CHAN_HT20(chan)) {
                REG_SET_BIT(ah, AR_PHY_CL_CAL_CTL, AR_PHY_PARALLEL_CAL_ENABLE);
                REG_SET_BIT(ah, AR_PHY_TURBO, AR_PHY_FC_DYN2040_EN);
                REG_CLR_BIT(ah, AR_PHY_AGC_CONTROL(ah),
                            AR_PHY_AGC_CONTROL_FLTR_CAL);
                REG_CLR_BIT(ah, AR_PHY_TPCRG1, AR_PHY_TPCRG1_PD_CAL_ENABLE);
                REG_SET_BIT(ah, AR_PHY_AGC_CONTROL(ah), AR_PHY_AGC_CONTROL_CAL);
                if (!ath9k_hw_wait(ah, AR_PHY_AGC_CONTROL(ah),
                                  AR_PHY_AGC_CONTROL_CAL, 0, AH_WAIT_TIMEOUT)) {
                        ath_dbg(common, CALIBRATE,
                                "offset calibration failed to complete in %d ms; noisy environment?\n",
                                AH_WAIT_TIMEOUT / 1000);
                        return false;
                }
                REG_CLR_BIT(ah, AR_PHY_TURBO, AR_PHY_FC_DYN2040_EN);
                REG_CLR_BIT(ah, AR_PHY_CL_CAL_CTL, AR_PHY_PARALLEL_CAL_ENABLE);
                REG_CLR_BIT(ah, AR_PHY_CL_CAL_CTL, AR_PHY_CL_CAL_ENABLE);
        }
        REG_CLR_BIT(ah, AR_PHY_ADC_CTL, AR_PHY_ADC_CTL_OFF_PWDADC);
        REG_SET_BIT(ah, AR_PHY_AGC_CONTROL(ah), AR_PHY_AGC_CONTROL_FLTR_CAL);
        REG_SET_BIT(ah, AR_PHY_TPCRG1, AR_PHY_TPCRG1_PD_CAL_ENABLE);
        REG_SET_BIT(ah, AR_PHY_AGC_CONTROL(ah), AR_PHY_AGC_CONTROL_CAL);
        if (!ath9k_hw_wait(ah, AR_PHY_AGC_CONTROL(ah), AR_PHY_AGC_CONTROL_CAL,
                          0, AH_WAIT_TIMEOUT)) {
                ath_dbg(common, CALIBRATE,
                        "offset calibration failed to complete in %d ms; noisy environment?\n",
                        AH_WAIT_TIMEOUT / 1000);
                return false;
        }

        REG_SET_BIT(ah, AR_PHY_ADC_CTL, AR_PHY_ADC_CTL_OFF_PWDADC);
        REG_CLR_BIT(ah, AR_PHY_CL_CAL_CTL, AR_PHY_CL_CAL_ENABLE);
        REG_CLR_BIT(ah, AR_PHY_AGC_CONTROL(ah), AR_PHY_AGC_CONTROL_FLTR_CAL);

        return true;
}

static bool ar9285_hw_clc(struct ath_hw *ah, struct ath9k_channel *chan)
{
        int i;
        u_int32_t txgain_max;
        u_int32_t clc_gain, gain_mask = 0, clc_num = 0;
        u_int32_t reg_clc_I0, reg_clc_Q0;
        u_int32_t i0_num = 0;
        u_int32_t q0_num = 0;
        u_int32_t total_num = 0;
        u_int32_t reg_rf2g5_org;
        bool retv = true;

        if (!(ar9285_hw_cl_cal(ah, chan)))
                return false;

        txgain_max = MS(REG_READ(ah, AR_PHY_TX_PWRCTRL7),
                        AR_PHY_TX_PWRCTRL_TX_GAIN_TAB_MAX);

        for (i = 0; i < (txgain_max+1); i++) {
                clc_gain = (REG_READ(ah, (AR_PHY_TX_GAIN_TBL1+(i<<2))) &
                           AR_PHY_TX_GAIN_CLC) >> AR_PHY_TX_GAIN_CLC_S;
                if (!(gain_mask & (1 << clc_gain))) {
                        gain_mask |= (1 << clc_gain);
                        clc_num++;
                }
        }

        for (i = 0; i < clc_num; i++) {
                reg_clc_I0 = (REG_READ(ah, (AR_PHY_CLC_TBL1 + (i << 2)))
                              & AR_PHY_CLC_I0) >> AR_PHY_CLC_I0_S;
                reg_clc_Q0 = (REG_READ(ah, (AR_PHY_CLC_TBL1 + (i << 2)))
                              & AR_PHY_CLC_Q0) >> AR_PHY_CLC_Q0_S;
                if (reg_clc_I0 == 0)
                        i0_num++;

                if (reg_clc_Q0 == 0)
                        q0_num++;
        }
        total_num = i0_num + q0_num;
        if (total_num > AR9285_CLCAL_REDO_THRESH) {
                reg_rf2g5_org = REG_READ(ah, AR9285_RF2G5);
                if (AR_SREV_9285E_20(ah)) {
                        REG_WRITE(ah, AR9285_RF2G5,
                                  (reg_rf2g5_org & AR9285_RF2G5_IC50TX) |
                                  AR9285_RF2G5_IC50TX_XE_SET);
                } else {
                        REG_WRITE(ah, AR9285_RF2G5,
                                  (reg_rf2g5_org & AR9285_RF2G5_IC50TX) |
                                  AR9285_RF2G5_IC50TX_SET);
                }
                retv = ar9285_hw_cl_cal(ah, chan);
                REG_WRITE(ah, AR9285_RF2G5, reg_rf2g5_org);
        }
        return retv;
}

static bool ar9002_hw_init_cal(struct ath_hw *ah, struct ath9k_channel *chan)
{
        struct ath_common *common = ath9k_hw_common(ah);

        if (AR_SREV_9271(ah)) {
                if (!ar9285_hw_cl_cal(ah, chan))
                        return false;
        } else if (AR_SREV_9285(ah) && AR_SREV_9285_12_OR_LATER(ah)) {
                if (!ar9285_hw_clc(ah, chan))
                        return false;
        } else {
                if (AR_SREV_9280_20_OR_LATER(ah)) {
                        if (!AR_SREV_9287_11_OR_LATER(ah))
                                REG_CLR_BIT(ah, AR_PHY_ADC_CTL,
                                            AR_PHY_ADC_CTL_OFF_PWDADC);
                        REG_SET_BIT(ah, AR_PHY_AGC_CONTROL(ah),
                                    AR_PHY_AGC_CONTROL_FLTR_CAL);
                }

                /* Calibrate the AGC */
                REG_WRITE(ah, AR_PHY_AGC_CONTROL(ah),
                          REG_READ(ah, AR_PHY_AGC_CONTROL(ah)) |
                          AR_PHY_AGC_CONTROL_CAL);

                /* Poll for offset calibration complete */
                if (!ath9k_hw_wait(ah, AR_PHY_AGC_CONTROL(ah),
                                   AR_PHY_AGC_CONTROL_CAL,
                                   0, AH_WAIT_TIMEOUT)) {
                        ath_dbg(common, CALIBRATE,
                                "offset calibration failed to complete in %d ms; noisy environment?\n",
                                AH_WAIT_TIMEOUT / 1000);
                        return false;
                }

                if (AR_SREV_9280_20_OR_LATER(ah)) {
                        if (!AR_SREV_9287_11_OR_LATER(ah))
                                REG_SET_BIT(ah, AR_PHY_ADC_CTL,
                                            AR_PHY_ADC_CTL_OFF_PWDADC);
                        REG_CLR_BIT(ah, AR_PHY_AGC_CONTROL(ah),
                                    AR_PHY_AGC_CONTROL_FLTR_CAL);
                }
        }

        /* Do PA Calibration */
        ar9002_hw_pa_cal(ah, true);
        ath9k_hw_loadnf(ah, chan);
        ath9k_hw_start_nfcal(ah, true);

        ah->cal_list = ah->cal_list_last = ah->cal_list_curr = NULL;

        /* Enable IQ, ADC Gain and ADC DC offset CALs */
        if (AR_SREV_9100(ah) || AR_SREV_9160_10_OR_LATER(ah)) {
                ah->supp_cals = IQ_MISMATCH_CAL;

                if (AR_SREV_9160_10_OR_LATER(ah))
                        ah->supp_cals |= ADC_GAIN_CAL | ADC_DC_CAL;

                if (AR_SREV_9287(ah))
                        ah->supp_cals &= ~ADC_GAIN_CAL;

                if (ar9002_hw_is_cal_supported(ah, chan, ADC_GAIN_CAL)) {
                        INIT_CAL(&ah->adcgain_caldata);
                        INSERT_CAL(ah, &ah->adcgain_caldata);
                        ath_dbg(common, CALIBRATE,
                                        "enabling ADC Gain Calibration\n");
                }

                if (ar9002_hw_is_cal_supported(ah, chan, ADC_DC_CAL)) {
                        INIT_CAL(&ah->adcdc_caldata);
                        INSERT_CAL(ah, &ah->adcdc_caldata);
                        ath_dbg(common, CALIBRATE,
                                        "enabling ADC DC Calibration\n");
                }

                if (ar9002_hw_is_cal_supported(ah, chan, IQ_MISMATCH_CAL)) {
                        INIT_CAL(&ah->iq_caldata);
                        INSERT_CAL(ah, &ah->iq_caldata);
                        ath_dbg(common, CALIBRATE, "enabling IQ Calibration\n");
                }

                ah->cal_list_curr = ah->cal_list;

                if (ah->cal_list_curr)
                        ath9k_hw_reset_calibration(ah, ah->cal_list_curr);
        }

        if (ah->caldata)
                ah->caldata->CalValid = 0;

        return true;
}

static const struct ath9k_percal_data iq_cal_multi_sample = {
        IQ_MISMATCH_CAL,
        MAX_CAL_SAMPLES,
        PER_MIN_LOG_COUNT,
        ar9002_hw_iqcal_collect,
        ar9002_hw_iqcalibrate
};
static const struct ath9k_percal_data iq_cal_single_sample = {
        IQ_MISMATCH_CAL,
        MIN_CAL_SAMPLES,
        PER_MAX_LOG_COUNT,
        ar9002_hw_iqcal_collect,
        ar9002_hw_iqcalibrate
};
static const struct ath9k_percal_data adc_gain_cal_multi_sample = {
        ADC_GAIN_CAL,
        MAX_CAL_SAMPLES,
        PER_MIN_LOG_COUNT,
        ar9002_hw_adc_gaincal_collect,
        ar9002_hw_adc_gaincal_calibrate
};
static const struct ath9k_percal_data adc_gain_cal_single_sample = {
        ADC_GAIN_CAL,
        MIN_CAL_SAMPLES,
        PER_MAX_LOG_COUNT,
        ar9002_hw_adc_gaincal_collect,
        ar9002_hw_adc_gaincal_calibrate
};
static const struct ath9k_percal_data adc_dc_cal_multi_sample = {
        ADC_DC_CAL,
        MAX_CAL_SAMPLES,
        PER_MIN_LOG_COUNT,
        ar9002_hw_adc_dccal_collect,
        ar9002_hw_adc_dccal_calibrate
};
static const struct ath9k_percal_data adc_dc_cal_single_sample = {
        ADC_DC_CAL,
        MIN_CAL_SAMPLES,
        PER_MAX_LOG_COUNT,
        ar9002_hw_adc_dccal_collect,
        ar9002_hw_adc_dccal_calibrate
};

static void ar9002_hw_init_cal_settings(struct ath_hw *ah)
{
        if (AR_SREV_9100(ah)) {
                ah->iq_caldata.calData = &iq_cal_multi_sample;
                ah->supp_cals = IQ_MISMATCH_CAL;
                return;
        }

        if (AR_SREV_9160_10_OR_LATER(ah)) {
                if (AR_SREV_9280_20_OR_LATER(ah)) {
                        ah->iq_caldata.calData = &iq_cal_single_sample;
                        ah->adcgain_caldata.calData =
                                &adc_gain_cal_single_sample;
                        ah->adcdc_caldata.calData =
                                &adc_dc_cal_single_sample;
                } else {
                        ah->iq_caldata.calData = &iq_cal_multi_sample;
                        ah->adcgain_caldata.calData =
                                &adc_gain_cal_multi_sample;
                        ah->adcdc_caldata.calData =
                                &adc_dc_cal_multi_sample;
                }
                ah->supp_cals = ADC_GAIN_CAL | ADC_DC_CAL | IQ_MISMATCH_CAL;

                if (AR_SREV_9287(ah))
                        ah->supp_cals &= ~ADC_GAIN_CAL;
        }
}

void ar9002_hw_attach_calib_ops(struct ath_hw *ah)
{
        struct ath_hw_private_ops *priv_ops = ath9k_hw_private_ops(ah);
        struct ath_hw_ops *ops = ath9k_hw_ops(ah);

        priv_ops->init_cal_settings = ar9002_hw_init_cal_settings;
        priv_ops->init_cal = ar9002_hw_init_cal;
        priv_ops->setup_calibration = ar9002_hw_setup_calibration;

        ops->calibrate = ar9002_hw_calibrate;
}