/*
 * (C) Copyright 2002
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

/*
 * I2C test
 *
 * For verifying the I2C bus, a full I2C bus scanning is performed.
 *
 * #ifdef CONFIG_SYS_POST_I2C_ADDRS
 *   The test is considered as passed if all the devices and only the devices
 *   in the list are found.
 *   #ifdef CONFIG_SYS_POST_I2C_IGNORES
 *     Ignore devices listed in CONFIG_SYS_POST_I2C_IGNORES.  These devices
 *     are optional or not vital to board functionality.
 *   #endif
 * #else [ ! CONFIG_SYS_POST_I2C_ADDRS ]
 *   The test is considered as passed if any I2C device is found.
 * #endif
 */

#include <common.h>
#include <post.h>
#include <i2c.h>

#if CONFIG_POST & CONFIG_SYS_POST_I2C

#ifdef CONFIG_POST_AML
#include <aml_i2c.h>

extern struct aml_i2c_device aml_i2c_devices;

//=============================================================================
static int i2c_xfer(unsigned int addr)
{
    unsigned char cmd = 0x0;
	
	struct i2c_msg msgs[] = {
	    {
	        .addr = addr,
	        .flags = 0,
	        .len = 1,
	        .buf = &cmd,
	    }
	};

	if(aml_i2c_xfer(msgs, 1) < 0) 
		return -1;

	return 0;
}

//=============================================================================
int i2c_post_test(int flags)
{
	int i, ret;
	struct i2c_board_info *ptests = aml_i2c_devices.aml_i2c_boards;
	struct i2c_board_info *ptest = NULL;
	int num = aml_i2c_devices.dev_num;
	
	ret = 0;	
	for(i=0; i<num; i++){	
		ptest = ptests+i;
				
		if(ptest->device_init)
			ptest->device_init();
				
		if(i2c_xfer(ptest->addr) < 0){					
			post_log("<%d>%s:%d: I2C[board: %s addr: 0x%x]: test fail.\n", SYSTEST_INFO_L2, __FUNCTION__, __LINE__, ptest->type, ptest->addr);			
			ret = -1;
		}
		else{				
			post_log("<%d>%s:%d: I2C[board: %s addr: 0x%x]: test pass.\n", SYSTEST_INFO_L2, __FUNCTION__, __LINE__, ptest->type, ptest->addr);
		}
		
		if(ptest->device_uninit)
			ptest->device_uninit();					
	}
	
	return ret;	
}

//=============================================================================
#else

static int i2c_ignore_device(unsigned int chip)
{
#ifdef CONFIG_SYS_POST_I2C_IGNORES
	const unsigned char i2c_ignore_list[] = CONFIG_SYS_POST_I2C_IGNORES;
	int i;

	for (i = 0; i < sizeof(i2c_ignore_list); i++)
		if (i2c_ignore_list[i] == chip)
			return 1;
#endif

	return 0;
}

int i2c_post_test (int flags)
{
	unsigned int i;
#ifndef CONFIG_SYS_POST_I2C_ADDRS
	/* Start at address 1, address 0 is the general call address */
	for (i = 1; i < 128; i++) {
		if (i2c_ignore_device(i))
			continue;
		if (i2c_probe (i) == 0)
			return 0;
	}

	/* No devices found */
	return -1;
#else
	unsigned int ret  = 0;
	int j;
	unsigned char i2c_addr_list[] = CONFIG_SYS_POST_I2C_ADDRS;

	/* Start at address 1, address 0 is the general call address */
	for (i = 1; i < 128; i++) {
		if (i2c_ignore_device(i))
			continue;
		if (i2c_probe(i) != 0)
			continue;

		for (j = 0; j < sizeof(i2c_addr_list); ++j) {
			if (i == i2c_addr_list[j]) {
				i2c_addr_list[j] = 0xff;
				break;
			}
		}

		if (j == sizeof(i2c_addr_list)) {
			ret = -1;
			post_log("I2C: addr %02x not expected\n", i);
		}
	}

	for (i = 0; i < sizeof(i2c_addr_list); ++i) {
		if (i2c_addr_list[i] == 0xff)
			continue;
		post_log("I2C: addr %02x did not respond\n", i2c_addr_list[i]);
		ret = -1;
	}

	return ret;
#endif
}

#endif /*CONFIG_POST_AML*/
#endif /* CONFIG_POST & CONFIG_SYS_POST_I2C */
