From d709d4e2bb7f795087ade83129200967408b084b Mon Sep 17 00:00:00 2001
From: Alex <52292902+alexrudd2@users.noreply.github.com>
Date: Wed, 26 Jul 2023 10:39:55 -0500
Subject: [PATCH] Add functionality for 'heating fluid' temp measurement (#32)

---
 ika/driver.py | 5 +++++
 ika/mock.py   | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/ika/driver.py b/ika/driver.py
index 44fa2e0..bd28175 100755
--- a/ika/driver.py
+++ b/ika/driver.py
@@ -152,6 +152,7 @@ class HotplateProtocol:
     READ_ACTUAL_PROCESS_TEMP = "IN_PV_1"
     READ_ACTUAL_SURFACE_TEMP = "IN_PV_2"
     READ_ACTUAL_SPEED = "IN_PV_4"
+    READ_ACTUAL_FLUID_TEMP = "IN_PV_7"  # for double temp probe "PT1000"
     READ_VISCOSITY_TREND_VALUE = "IN_PV_5"
     READ_PROCESS_TEMP_SETPOINT = "IN_SP_1"
     READ_SURFACE_TEMP_SETPOINT = "IN_SP_2"
@@ -204,6 +205,7 @@ async def get(self, include_surface_control=False):
         speed_sp = await self.query(self.READ_SPEED_SETPOINT)
         process_temp = await self.query(self.READ_ACTUAL_PROCESS_TEMP)
         process_temp_sp = await self.query(self.READ_PROCESS_TEMP_SETPOINT)
+        fluid_temp = await self.query(self.READ_ACTUAL_FLUID_TEMP)
         shaker_status = await self.query(self.READ_SHAKER_STATUS)
         process_heater_status = await self.query(self.READ_PROCESS_HEATER_STATUS)
         surface_data = {
@@ -226,6 +228,9 @@ async def get(self, include_surface_control=False):
                 'active': process_heater_status,
             },
             'surface_temp': surface_data,
+            'fluid_temp': {
+                'actual': fluid_temp,
+            }
         }
         return response
 
diff --git a/ika/mock.py b/ika/mock.py
index c75c565..6ec17e9 100644
--- a/ika/mock.py
+++ b/ika/mock.py
@@ -107,6 +107,9 @@ def __init__(self, *args, **kwargs):
                 "setpoint": 300,
                 "active": False,
             },
+            "fluid_temp": {
+                "actual": 100,
+            }
         }
 
     async def query(self, command):
@@ -127,6 +130,8 @@ async def query(self, command):
                 return round(uniform(15, 100), 2)
             elif command == self.READ_ACTUAL_SURFACE_TEMP:
                 return round(uniform(80, 120), 2)
+            elif command == self.READ_ACTUAL_FLUID_TEMP:
+                return round(uniform(20, 110), 2)
             elif command == self.READ_SURFACE_TEMP_SETPOINT:
                 return self.state["surface_temp"]["setpoint"]
             elif command == self.READ_PROCESS_TEMP_SETPOINT: