/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * Copyright (c) 2008 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 "arn_core.h"
#include "arn_hw.h"
#include "arn_reg.h"
#include "arn_phy.h"

/* ARGSUSED */
void
ath9k_hw_write_regs(struct ath_hal *ah, uint32_t modesIndex, uint32_t freqIndex,
		    int regWrites)
{
	struct ath_hal_5416 *ahp = AH5416(ah);

	/* LINTED: E_CONSTANT_CONDITION */
	REG_WRITE_ARRAY(&ahp->ah_iniBB_RfGain, freqIndex, regWrites);
}

boolean_t
ath9k_hw_set_channel(struct ath_hal *ah, struct ath9k_channel *chan)
{
	uint32_t channelSel = 0;
	uint32_t bModeSynth = 0;
	uint32_t aModeRefSel = 0;
	uint32_t reg32 = 0;
	uint16_t freq;
	struct chan_centers centers;

	ath9k_hw_get_channel_centers(ah, chan, &centers);
	freq = centers.synth_center;

	if (freq < 4800) {
		uint32_t txctl;

		if (((freq - 2192) % 5) == 0) {
			channelSel = ((freq - 672) * 2 - 3040) / 10;
			bModeSynth = 0;
		} else if (((freq - 2224) % 5) == 0) {
			channelSel = ((freq - 704) * 2 - 3040) / 10;
			bModeSynth = 1;
		} else {
			arn_problem("%s: invalid channel %u MHz\n",
			    __func__, freq);
			return (B_FALSE);
		}

		channelSel = (channelSel << 2) & 0xff;
		channelSel = ath9k_hw_reverse_bits(channelSel, 8);

		txctl = REG_READ(ah, AR_PHY_CCK_TX_CTRL);
		if (freq == 2484) {

			REG_WRITE(ah, AR_PHY_CCK_TX_CTRL,
			    txctl | AR_PHY_CCK_TX_CTRL_JAPAN);
		} else {
			REG_WRITE(ah, AR_PHY_CCK_TX_CTRL,
			    txctl & ~AR_PHY_CCK_TX_CTRL_JAPAN);
		}

	} else if ((freq % 20) == 0 && freq >= 5120) {
		channelSel =
		    ath9k_hw_reverse_bits(((freq - 4800) / 20 << 2), 8);
		aModeRefSel = ath9k_hw_reverse_bits(1, 2);
	} else if ((freq % 10) == 0) {
		channelSel =
		    ath9k_hw_reverse_bits(((freq - 4800) / 10 << 1), 8);
		if (AR_SREV_9100(ah) || AR_SREV_9160_10_OR_LATER(ah))
			aModeRefSel = ath9k_hw_reverse_bits(2, 2);
		else
			aModeRefSel = ath9k_hw_reverse_bits(1, 2);
	} else if ((freq % 5) == 0) {
		channelSel = ath9k_hw_reverse_bits((freq - 4800) / 5, 8);
		aModeRefSel = ath9k_hw_reverse_bits(1, 2);
	} else {
		arn_problem("%s: invalid channel %u MHz\n", __func__, freq);
		return (B_FALSE);
	}

	reg32 =
	    (channelSel << 8) | (aModeRefSel << 2) | (bModeSynth << 1) |
	    (1 << 5) | 0x1;

	REG_WRITE(ah, AR_PHY(0x37), reg32);

	ah->ah_curchan = chan;

	AH5416(ah)->ah_curchanRadIndex = -1;

	return (B_TRUE);
}

boolean_t
ath9k_hw_ar9280_set_channel(struct ath_hal *ah,
    struct ath9k_channel *chan)
{
	uint16_t bMode, fracMode, aModeRefSel = 0;
	uint32_t freq, ndiv, channelSel = 0, channelFrac = 0, reg32 = 0;
	struct chan_centers centers;
	uint32_t refDivA = 24;

	ath9k_hw_get_channel_centers(ah, chan, &centers);
	freq = centers.synth_center;

	reg32 = REG_READ(ah, AR_PHY_SYNTH_CONTROL);
	reg32 &= 0xc0000000;

	if (freq < 4800) {
		uint32_t txctl;

		bMode = 1;
		fracMode = 1;
		aModeRefSel = 0;
		channelSel = (freq * 0x10000) / 15;

		txctl = REG_READ(ah, AR_PHY_CCK_TX_CTRL);
		if (freq == 2484) {

			REG_WRITE(ah, AR_PHY_CCK_TX_CTRL,
			    txctl | AR_PHY_CCK_TX_CTRL_JAPAN);
		} else {
			REG_WRITE(ah, AR_PHY_CCK_TX_CTRL,
			    txctl & ~AR_PHY_CCK_TX_CTRL_JAPAN);
		}
	} else {
		bMode = 0;
		fracMode = 0;

		if ((freq % 20) == 0) {
			aModeRefSel = 3;
		} else if ((freq % 10) == 0) {
			aModeRefSel = 2;
		} else {
			aModeRefSel = 0;

			fracMode = 1;
			refDivA = 1;
			channelSel = (freq * 0x8000) / 15;

			REG_RMW_FIELD(ah, AR_AN_SYNTH9,
			    AR_AN_SYNTH9_REFDIVA, refDivA);
		}
		if (!fracMode) {
			ndiv = (freq * (refDivA >> aModeRefSel)) / 60;
			channelSel = ndiv & 0x1ff;
			channelFrac = (ndiv & 0xfffffe00) * 2;
			channelSel = (channelSel << 17) | channelFrac;
		}
	}

	reg32 = reg32 |
	    (bMode << 29) |
	    (fracMode << 28) | (aModeRefSel << 26) | (channelSel);

	REG_WRITE(ah, AR_PHY_SYNTH_CONTROL, reg32);

	ah->ah_curchan = chan;

	AH5416(ah)->ah_curchanRadIndex = -1;

	return (B_TRUE);
}

static void
ath9k_phy_modify_rx_buffer(uint32_t *rfBuf, uint32_t reg32,
    uint32_t numBits, uint32_t firstBit, uint32_t column)
{
	uint32_t tmp32, mask, arrayEntry, lastBit;
	int32_t bitPosition, bitsLeft;

	tmp32 = ath9k_hw_reverse_bits(reg32, numBits);
	arrayEntry = (firstBit - 1) / 8;
	bitPosition = (firstBit - 1) % 8;
	bitsLeft = numBits;
	while (bitsLeft > 0) {
		lastBit = (bitPosition + bitsLeft > 8) ?
		    8 : bitPosition + bitsLeft;
		mask = (((1 << lastBit) - 1) ^ ((1 << bitPosition) - 1)) <<
		    (column * 8);
		rfBuf[arrayEntry] &= ~mask;
		rfBuf[arrayEntry] |= ((tmp32 << bitPosition) <<
		    (column * 8)) & mask;
		bitsLeft -= 8 - bitPosition;
		tmp32 = tmp32 >> (8 - bitPosition);
		bitPosition = 0;
		arrayEntry++;
	}
}

boolean_t
ath9k_hw_set_rf_regs(struct ath_hal *ah, struct ath9k_channel *chan,
    uint16_t modesIndex)
{
	struct ath_hal_5416 *ahp = AH5416(ah);

	uint32_t eepMinorRev;
	uint32_t ob5GHz = 0, db5GHz = 0;
	uint32_t ob2GHz = 0, db2GHz = 0;
	/* LINTED E_FUNC_SET_NOT_USED */
	int regWrites = 0;

	if (AR_SREV_9280_10_OR_LATER(ah))
		return (B_TRUE);

	eepMinorRev = ath9k_hw_get_eeprom(ah, EEP_MINOR_REV);

	RF_BANK_SETUP(ahp->ah_analogBank0Data, &ahp->ah_iniBank0, 1);

	RF_BANK_SETUP(ahp->ah_analogBank1Data, &ahp->ah_iniBank1, 1);

	RF_BANK_SETUP(ahp->ah_analogBank2Data, &ahp->ah_iniBank2, 1);

	RF_BANK_SETUP(ahp->ah_analogBank3Data, &ahp->ah_iniBank3,
	    modesIndex);
	{
		int i;
		for (i = 0; i < ahp->ah_iniBank6TPC.ia_rows; i++) {
			ahp->ah_analogBank6Data[i] =
			    INI_RA(&ahp->ah_iniBank6TPC, i, modesIndex);
		}
	}

	if (eepMinorRev >= 2) {
		if (IS_CHAN_2GHZ(chan)) {
			ob2GHz = ath9k_hw_get_eeprom(ah, EEP_OB_2);
			db2GHz = ath9k_hw_get_eeprom(ah, EEP_DB_2);
			ath9k_phy_modify_rx_buffer(ahp->ah_analogBank6Data,
			    ob2GHz, 3, 197, 0);
			ath9k_phy_modify_rx_buffer(ahp->ah_analogBank6Data,
			    db2GHz, 3, 194, 0);
		} else {
			ob5GHz = ath9k_hw_get_eeprom(ah, EEP_OB_5);
			db5GHz = ath9k_hw_get_eeprom(ah, EEP_DB_5);
			ath9k_phy_modify_rx_buffer(ahp->ah_analogBank6Data,
			    ob5GHz, 3, 203, 0);
			ath9k_phy_modify_rx_buffer(ahp->ah_analogBank6Data,
			    db5GHz, 3, 200, 0);
		}
	}

	RF_BANK_SETUP(ahp->ah_analogBank7Data, &ahp->ah_iniBank7, 1);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank0, ahp->ah_analogBank0Data,
	    regWrites);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank1, ahp->ah_analogBank1Data,
	    regWrites);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank2, ahp->ah_analogBank2Data,
	    regWrites);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank3, ahp->ah_analogBank3Data,
	    regWrites);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank6TPC, ahp->ah_analogBank6Data,
	    regWrites);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank7, ahp->ah_analogBank7Data,
	    regWrites);

	return (B_TRUE);
}

void
ath9k_hw_rfdetach(struct ath_hal *ah)
{
	struct ath_hal_5416 *ahp = AH5416(ah);

	if (ahp->ah_analogBank0Data != NULL) {
		kmem_free(ahp->ah_analogBank0Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank0.ia_rows));
		ahp->ah_analogBank0Data = NULL;
	}
	if (ahp->ah_analogBank1Data != NULL) {
		kmem_free(ahp->ah_analogBank1Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank1.ia_rows));
		ahp->ah_analogBank1Data = NULL;
	}
	if (ahp->ah_analogBank2Data != NULL) {
		kmem_free(ahp->ah_analogBank2Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank2.ia_rows));
		ahp->ah_analogBank2Data = NULL;
	}
	if (ahp->ah_analogBank3Data != NULL) {
		kmem_free(ahp->ah_analogBank3Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank3.ia_rows));
		ahp->ah_analogBank3Data = NULL;
	}
	if (ahp->ah_analogBank6Data != NULL) {
		kmem_free(ahp->ah_analogBank6Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank6.ia_rows));
		ahp->ah_analogBank6Data = NULL;
	}
	if (ahp->ah_analogBank6TPCData != NULL) {
		kmem_free(ahp->ah_analogBank6TPCData,
		    (sizeof (uint32_t) * ahp->ah_iniBank6TPC.ia_rows));
		ahp->ah_analogBank6TPCData = NULL;
	}
	if (ahp->ah_analogBank7Data != NULL) {
		kmem_free(ahp->ah_analogBank7Data,
		    (sizeof (uint32_t) * ahp->ah_iniBank7.ia_rows));
		ahp->ah_analogBank7Data = NULL;
	}
	if (ahp->ah_addac5416_21 != NULL) {
		kmem_free(ahp->ah_addac5416_21,
		    (sizeof (uint32_t) * ahp->ah_iniAddac.ia_rows *
		    ahp->ah_iniAddac.ia_columns));
		ahp->ah_addac5416_21 = NULL;
	}
	if (ahp->ah_bank6Temp != NULL) {
		kmem_free(ahp->ah_bank6Temp,
		    (sizeof (uint32_t) * ahp->ah_iniBank6.ia_rows));
		ahp->ah_bank6Temp = NULL;
	}
}

boolean_t
ath9k_hw_init_rf(struct ath_hal *ah, int *status)
{
	struct ath_hal_5416 *ahp = AH5416(ah);

	if (!AR_SREV_9280_10_OR_LATER(ah)) {

		ahp->ah_analogBank0Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank0.ia_rows), KM_SLEEP);
		ahp->ah_analogBank1Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank1.ia_rows), KM_SLEEP);
		ahp->ah_analogBank2Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank2.ia_rows), KM_SLEEP);
		ahp->ah_analogBank3Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank3.ia_rows), KM_SLEEP);
		ahp->ah_analogBank6Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank6.ia_rows), KM_SLEEP);
		ahp->ah_analogBank6TPCData =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank6TPC.ia_rows), KM_SLEEP);
		ahp->ah_analogBank7Data =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank7.ia_rows), KM_SLEEP);

		if (ahp->ah_analogBank0Data == NULL ||
		    ahp->ah_analogBank1Data == NULL ||
		    ahp->ah_analogBank2Data == NULL ||
		    ahp->ah_analogBank3Data == NULL ||
		    ahp->ah_analogBank6Data == NULL ||
		    ahp->ah_analogBank6TPCData == NULL ||
		    ahp->ah_analogBank7Data == NULL) {
			ARN_DBG((ARN_DBG_FATAL, "arn: ath9k_hw_init_rf(): "
			    "cannot allocate RF banks\n"));
			*status = ENOMEM;
			return (B_FALSE);
		}

		ahp->ah_addac5416_21 =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniAddac.ia_rows *
		    ahp->ah_iniAddac.ia_columns), KM_SLEEP);
		if (ahp->ah_addac5416_21 == NULL) {
			ARN_DBG((ARN_DBG_FATAL, "arn: ath9k_hw_init_rf(): "
			    "cannot allocate ah_addac5416_21\n"));
			*status = ENOMEM;
			return (B_FALSE);
		}

		ahp->ah_bank6Temp =
		    kmem_zalloc((sizeof (uint32_t) *
		    ahp->ah_iniBank6.ia_rows), KM_SLEEP);
		if (ahp->ah_bank6Temp == NULL) {
			ARN_DBG((ARN_DBG_FATAL, "arn: ath9k_hw_init_rf(): "
			    "cannot allocate ah_bank6Temp\n"));
			*status = ENOMEM;
			return (B_FALSE);
		}
	}

	return (B_TRUE);
}

/* ARGSUSED */
void
ath9k_hw_decrease_chain_power(struct ath_hal *ah, struct ath9k_channel *chan)
{
	/* LINTED E_FUNC_SET_NOT_USED */
	int i, regWrites = 0;
	struct ath_hal_5416 *ahp = AH5416(ah);
	uint32_t bank6SelMask;
	uint32_t *bank6Temp = ahp->ah_bank6Temp;

	switch (ahp->ah_diversityControl) {
	case ATH9K_ANT_FIXED_A:
		bank6SelMask =
		    (ahp-> ah_antennaSwitchSwap & ANTSWAP_AB) ? REDUCE_CHAIN_0 :
		    REDUCE_CHAIN_1;
		break;
	case ATH9K_ANT_FIXED_B:
		bank6SelMask =
		    (ahp-> ah_antennaSwitchSwap & ANTSWAP_AB) ? REDUCE_CHAIN_1 :
		    REDUCE_CHAIN_0;
		break;
	case ATH9K_ANT_VARIABLE:
	default:
		return;
	}

	for (i = 0; i < ahp->ah_iniBank6.ia_rows; i++)
		bank6Temp[i] = ahp->ah_analogBank6Data[i];

	REG_WRITE(ah, AR_PHY_BASE + 0xD8, bank6SelMask);

	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 189, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 190, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 191, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 192, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 193, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 222, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 245, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 246, 0);
	ath9k_phy_modify_rx_buffer(bank6Temp, 1, 1, 247, 0);

	REG_WRITE_RF_ARRAY(&ahp->ah_iniBank6, bank6Temp, regWrites);

	REG_WRITE(ah, AR_PHY_BASE + 0xD8, 0x00000053);
#ifdef ALTER_SWITCH
	REG_WRITE(ah, PHY_SWITCH_CHAIN_0,
	    (REG_READ(ah, PHY_SWITCH_CHAIN_0) & ~0x38)
	    | ((REG_READ(ah, PHY_SWITCH_CHAIN_0) >> 3) & 0x38));
#endif
}