1 files changed, 13 insertions, 13 deletions
diff --git a/pybot/drivers/hotplug/volt.py b/pybot/drivers/hotplug/volt.py
index db7e1c1..10866a6 100644
--- a/pybot/drivers/hotplug/volt.py
+++ b/pybot/drivers/hotplug/volt.py
@@ -1,19 +1,19 @@
 
+import math
+
 RD_VERSION = 0x00
 GET_VALUE = 0x01
 
-f1 = {
-    'name': 'getVersion',
-    'call': RD_VERSION,
-    'params': 0,
-    'read': 3
-}
+VCC = 65536
+
+def getVersion(dev):
+    dev.send([RD_VERSION])
+    raw = dev.read(3)
+    return raw[1] + raw[2] * 256
 
-f2 = {
-    'name': 'getValue',
-    'call': GET_VALUE,
-    'params': 0,
-    'read': 3
-}
+def getValue(dev):
+    dev.send([GET_VALUE])
+    raw = dev.read(3)
+    volt = (raw[1] + raw[2] * 256) * 5.0 / VCC
+    return math.floor(volt * 1000.0) / 1000
 
-FUNCTIONS = [f1, f2]