text: /* Copyright 2020 The Chromium OS Authors. All rights reserved.
   * Use of this source code is governed by a BSD-style license that can be
   * found in the LICENSE file.
   */
  
  #include "chg_control.h"
  #include "gpio.h"
  #include "ioexpanders.h"
  #include "registers.h"
  #include "timer.h"
  
  #define CHG_P5V_POWER  0
  #define CHG_VBUS_POWER 1
  
  void chg_reset(void)
  {
  	/* Disconnect DUT Power */
  	chg_power_select(CHG_POWER_OFF);
  
  	/* Disconnect CHG CC1(Rd) and CC2(Rd) */
  	chg_attach_cc_rds(0);
  
  	msleep(100);
  
  	/* Connect CHG CC1(Rd) and CC2(Rd) to detect charger */
  	chg_attach_cc_rds(1);
  }
  
  void chg_power_select(enum chg_power_select_t type)
  {
  	switch (type) {
  	case CHG_POWER_OFF:
  		dut_chg_en(0);
  		vbus_dischrg_en(1);
  		break;
  	case CHG_POWER_PP5000:
  		vbus_dischrg_en(0);
  		host_or_chg_ctl(CHG_P5V_POWER);
  		dut_chg_en(1);
  		break;
  	case CHG_POWER_VBUS:
  		vbus_dischrg_en(0);
  		host_or_chg_ctl(CHG_VBUS_POWER);
  		dut_chg_en(1);
  		break;
  	}
  }
  
  void chg_attach_cc_rds(bool en)
  {
  	if (en) {
  		/*
  		 * Configure USB_CHG_CC1_MCU and USB_CHG_CC2_MCU as
  		 * ANALOG input
  		 */
  		STM32_GPIO_MODER(GPIO_A) = (STM32_GPIO_MODER(GPIO_A)
  			| (3 << (2*2)) | /* PA2 in ANALOG mode */
  			  (3 << (2*4))); /* PA4 in ANALOG mode */
  	} else {
  		/*
  		 * Configure USB_CHG_CC1_MCU and USB_CHG_CC2_MCU as GPIO and
  		 * drive high to trigger disconnect.
  		 * NOTE: The CC line has an external fixed Rd pull-down.
  		 * Driving the CC line High overrides the pull down and this
  		 * triggers a disconnection.
  		 */
  		/* Set level high */
  		gpio_set_level(GPIO_USB_CHG_CC1_MCU, 1);
  		gpio_set_level(GPIO_USB_CHG_CC2_MCU, 1);
  
  		/* Disable Analog mode and Enable GPO */
  		STM32_GPIO_MODER(GPIO_A) = (STM32_GPIO_MODER(GPIO_A)
  			& ~(3 << (2*2) | /* PA2 disable ADC */
  			    3 << (2*4))) /* PA4 disable ADC */
  			|  (1 << (2*2) | /* Set as GPO */
  			    1 << (2*4)); /* Set as GPO */
  	}
  }
  