TCPMv2: Use a common function to check if the battery is capable

Move the battery checks to a common function, which is used to enable
Try.SRC and enable PD comm for CONFIG_SYSTEM_LOCKED.

BRANCH=None
BUG=b:169453974
TEST=On a board with battery which takes longer time to enable its
discharge FET, cut off the battery and booted the board. Checked
on startup the Try.SRC was disabled and the PD comm was disabled.
When the battery discharge FET became active, PD comm was enabled.

Change-Id: I236d5f8a462ece056b4adedd0b6e6ef1e402966c
Signed-off-by: Wai-Hong Tam <waihong@google.com>
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/ec/+/2461942
Reviewed-by: Aseda Aboagye <aaboagye@chromium.org>

1 files changed, 11 insertions, 2 deletions

diff --git a/include/usb_pd.h b/include/usb_pd.h
index 46222ea14a..0fea201e9e 100644
--- a/include/usb_pd.h
+++ b/include/usb_pd.h
@@ -995,8 +995,17 @@ enum pd_data_role pd_get_data_role(int port);
  */
 enum pd_power_role pd_get_power_role(int port);
 
-/*
- * Return true if PD is capable of trying as source else false
+/**
+ * Check if the battery is capable of powering the system
+ *
+ * @return true if capable of, else false.
+ */
+bool pd_is_battery_capable(void);
+
+/**
+ * Check if PD is capable of trying as source
+ *
+ * @return true if capable of, else false.
  */
 bool pd_is_try_source_capable(void);