X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=common%2Fcmd_dtt.c;h=f2e750f2f1642f0fa71a6be1edac10eb48578f75;hb=f3b4bc458de3285a04c5f60df6372ad5ca79fa60;hp=dc57744f8981b82e41c0eaa161ce6584089c0717;hpb=1a4596601fd395f3afb8f82f3f840c5e00bdd57a;p=oweals%2Fu-boot.git

diff --git a/common/cmd_dtt.c b/common/cmd_dtt.c
index dc57744f89..f2e750f2f1 100644
--- a/common/cmd_dtt.c
+++ b/common/cmd_dtt.c
@@ -2,7 +2,7 @@
  * (C) Copyright 2001
  * Erik Theisen, Wave 7 Optics, etheisen@mindspring.com
  *
- * SPDX-License-Identifier:	GPL-2.0+ 
+ * SPDX-License-Identifier:	GPL-2.0+
  */
 
 #include <common.h>
@@ -57,8 +57,13 @@ int dtt_i2c(void)
 	/* Force a compilation error, if there are more then 32 sensors */
 	BUILD_BUG_ON(sizeof(sensors) > 32);
 	/* switch to correct I2C bus */
+#ifdef CONFIG_SYS_I2C
+	old_bus = i2c_get_bus_num();
+	i2c_set_bus_num(CONFIG_SYS_DTT_BUS_NUM);
+#else
 	old_bus = I2C_GET_BUS();
 	I2C_SET_BUS(CONFIG_SYS_DTT_BUS_NUM);
+#endif
 
 	_initialize_dtt();
 
@@ -70,7 +75,11 @@ int dtt_i2c(void)
 		printf("DTT%d: %i C\n", i + 1, dtt_get_temp(sensors[i]));
 
 	/* switch back to original I2C bus */
+#ifdef CONFIG_SYS_I2C
+	i2c_set_bus_num(old_bus);
+#else
 	I2C_SET_BUS(old_bus);
+#endif
 #endif
 
 	return 0;