Current sensor scripts

3 months ago · 9695cf862c
3 changed files with 260 additions and 0 deletions
--- a/code/modules/SDL_Pi_INA3221.py
+++ b/code/modules/SDL_Pi_INA3221.py
@ -0,0 +1,165 @@
+#!/usr/bin/env python
+
+# SDL_Pi_INA3221.py Python Driver Code
+# SwitchDoc Labs March 4, 2015	 
+# V 1.2
+
+
+#encoding: utf-8
+ 
+from datetime import datetime
+
+import smbus
+
+# constants
+
+#/*=========================================================================
+#    I2C ADDRESS/BITS
+#    -----------------------------------------------------------------------*/
+INA3221_ADDRESS =                         (0x40)    # 1000000 (A0+A1=GND)
+INA3221_READ    =                         (0x01)
+#/*=========================================================================*/
+
+#/*=========================================================================
+#    CONFIG REGISTER (R/W)
+#    -----------------------------------------------------------------------*/
+INA3221_REG_CONFIG            =          (0x00)
+#    /*---------------------------------------------------------------------*/
+INA3221_CONFIG_RESET          =          (0x8000)  # Reset Bit
+	
+INA3221_CONFIG_ENABLE_CHAN1   =          (0x4000)  # Enable Channel 1
+INA3221_CONFIG_ENABLE_CHAN2   =          (0x2000)  # Enable Channel 2
+INA3221_CONFIG_ENABLE_CHAN3   =          (0x1000)  # Enable Channel 3
+
+INA3221_CONFIG_AVG2     =                (0x0800)  # AVG Samples Bit 2 - See table 3 spec
+INA3221_CONFIG_AVG1     =                (0x0400)  # AVG Samples Bit 1 - See table 3 spec
+INA3221_CONFIG_AVG0     =                (0x0200)  # AVG Samples Bit 0 - See table 3 spec
+
+INA3221_CONFIG_VBUS_CT2 =                (0x0100)  # VBUS bit 2 Conversion time - See table 4 spec
+INA3221_CONFIG_VBUS_CT1 =                (0x0080)  # VBUS bit 1 Conversion time - See table 4 spec
+INA3221_CONFIG_VBUS_CT0 =                (0x0040)  # VBUS bit 0 Conversion time - See table 4 spec
+
+INA3221_CONFIG_VSH_CT2  =                (0x0020)  # Vshunt bit 2 Conversion time - See table 5 spec
+INA3221_CONFIG_VSH_CT1  =                (0x0010)  # Vshunt bit 1 Conversion time - See table 5 spec
+INA3221_CONFIG_VSH_CT0  =                (0x0008)  # Vshunt bit 0 Conversion time - See table 5 spec
+
+INA3221_CONFIG_MODE_2   =                (0x0004)  # Operating Mode bit 2 - See table 6 spec
+INA3221_CONFIG_MODE_1   = 	         (0x0002)  # Operating Mode bit 1 - See table 6 spec
+INA3221_CONFIG_MODE_0 	=                (0x0001)  # Operating Mode bit 0 - See table 6 spec
+
+#/*=========================================================================*/
+
+#/*=========================================================================
+#    SHUNT VOLTAGE REGISTER (R)
+#    -----------------------------------------------------------------------*/
+INA3221_REG_SHUNTVOLTAGE_1   =             (0x01)
+#/*=========================================================================*/
+
+#/*=========================================================================
+#    BUS VOLTAGE REGISTER (R)
+#    -----------------------------------------------------------------------*/
+INA3221_REG_BUSVOLTAGE_1     =             (0x02)
+#/*=========================================================================*/
+
+SHUNT_RESISTOR_VALUE         = (0.01)   # default shunt resistor value of 0.1 Ohm
+
+
+
+class SDL_Pi_INA3221():
+
+
+
+    ###########################
+    # INA3221 Code
+    ###########################
+    def __init__(self, twi=2, addr=INA3221_ADDRESS, shunt_resistor = SHUNT_RESISTOR_VALUE  ):
+        self._bus = smbus.SMBus(twi)
+        self._addr = addr
+        config = INA3221_CONFIG_ENABLE_CHAN1 |		\
+                    INA3221_CONFIG_ENABLE_CHAN2 |	\
+                    INA3221_CONFIG_ENABLE_CHAN3 |	\
+                    INA3221_CONFIG_AVG1 |		\
+                    INA3221_CONFIG_VBUS_CT2 |		\
+                    INA3221_CONFIG_VSH_CT2 |		\
+                    INA3221_CONFIG_MODE_2 |		\
+                    INA3221_CONFIG_MODE_1 |		\
+                    INA3221_CONFIG_MODE_0
+
+
+
+
+        self._write_register_little_endian(INA3221_REG_CONFIG, config)
+
+
+    def _write(self, register, data):
+        #print "addr =0x%x register = 0x%x data = 0x%x " % (self._addr, register, data)
+        self._bus.write_byte_data(self._addr, register, data)
+
+
+    def _read(self, data):
+
+        returndata = self._bus.read_byte_data(self._addr, data)
+        #print "addr = 0x%x data = 0x%x %i returndata = 0x%x " % (self._addr, data, data, returndata)
+        return returndata
+
+
+    def _read_register_little_endian(self, register): 
+	
+        result = self._bus.read_word_data(self._addr,register) & 0xFFFF
+        lowbyte = (result & 0xFF00)>>8 
+        highbyte = (result & 0x00FF) << 8
+        switchresult = lowbyte + highbyte 
+        #print "Read 16 bit Word addr =0x%x register = 0x%x switchresult = 0x%x " % (self._addr, register, switchresult)
+        return switchresult
+   
+   
+    def _write_register_little_endian(self, register, data): 
+
+        data = data & 0xFFFF
+        # reverse configure byte for little endian
+        lowbyte = data>>8
+        highbyte = (data & 0x00FF)<<8
+        switchdata = lowbyte + highbyte
+        self._bus.write_word_data(self._addr, register, switchdata)
+        #print "Write  16 bit Word addr =0x%x register = 0x%x data = 0x%x " % (self._addr, register, data)
+       
+
+
+    def _getBusVoltage_raw(self, channel):
+	#Gets the raw bus voltage (16-bit signed integer, so +-32767)
+
+        value = self._read_register_little_endian(INA3221_REG_BUSVOLTAGE_1+(channel -1) *2) 
+        if value > 32767:
+            value -= 65536
+        return value
+
+    def _getShuntVoltage_raw(self, channel):
+	#Gets the raw shunt voltage (16-bit signed integer, so +-32767)
+	
+        value = self._read_register_little_endian(INA3221_REG_SHUNTVOLTAGE_1+(channel -1) *2)
+        if value > 32767:
+            value -= 65536
+        return value
+
+    # public functions
+
+    def getBusVoltage_V(self, channel):
+	# Gets the Bus voltage in volts
+
+        value = self._getBusVoltage_raw(channel)
+        return value * 0.001
+
+
+    def getShuntVoltage_mV(self, channel):
+	# Gets the shunt voltage in mV (so +-168.3mV)
+
+        value = self._getShuntVoltage_raw(channel)
+        return value * 0.005
+
+    def getCurrent_mA(self, channel):
+    #Gets the current value in mA, taking into account the config settings and current LSB
+    	
+        valueDec = self.getShuntVoltage_mV(channel)/ SHUNT_RESISTOR_VALUE               
+        return valueDec;
+
+
--- a/code/modules/testSDL_Pi_INA3221.py
+++ b/code/modules/testSDL_Pi_INA3221.py
@ -0,0 +1,95 @@
+#!/usr/bin/env python
+#
+# Test SDL_Pi_INA3221
+# John C. Shovic, SwitchDoc Labs
+# 03/05/2015
+#
+#
+
+# imports
+
+import sys
+import time
+import datetime
+import random 
+import SDL_Pi_INA3221
+
+# Main Program
+
+print("")
+print("Test SDL_Pi_INA3221 Version 1.0 - SwitchDoc Labs")
+print("")
+print("Sample uses 0x40 and SunAirPlus board INA3221")
+print(" Will work with the INA3221 SwitchDoc Labs Breakout Board")
+print("Program Started at:"+ time.strftime("%Y-%m-%d %H:%M:%S"))
+print("")
+
+filename = time.strftime("%Y-%m-%d%H:%M:%SRTCTest") + ".txt"
+starttime = datetime.datetime.utcnow()
+
+ina3221 = SDL_Pi_INA3221.SDL_Pi_INA3221(addr=0x40)
+
+# the three channels of the INA3221 named for SunAirPlus Solar Power Controller channels (www.switchdoc.com)
+LIPO_BATTERY_CHANNEL = 1
+SOLAR_CELL_CHANNEL   = 2
+OUTPUT_CHANNEL       = 3
+
+
+while True:
+
+  	print("------------------------------")
+  	shuntvoltage1 = 0
+  	busvoltage1   = 0
+  	current_mA1   = 0
+  	loadvoltage1  = 0
+
+
+  	busvoltage1 = ina3221.getBusVoltage_V(LIPO_BATTERY_CHANNEL)
+  	shuntvoltage1 = ina3221.getShuntVoltage_mV(LIPO_BATTERY_CHANNEL)
+  	# minus is to get the "sense" right.   - means the battery is charging, + that it is discharging
+  	current_mA1 = ina3221.getCurrent_mA(LIPO_BATTERY_CHANNEL)  
+
+  	loadvoltage1 = busvoltage1 + (shuntvoltage1 / 1000)
+  
+  	print("LIPO_Battery Bus Voltage: %3.2f V " % busvoltage1)
+  	print("LIPO_Battery Shunt Voltage: %3.2f mV " % shuntvoltage1)
+  	print("LIPO_Battery Load Voltage:  %3.2f V" % loadvoltage1)
+  	print("LIPO_Battery Current 1:  %3.2f mA" % current_mA1)
+  	print
+
+  	shuntvoltage2 = 0
+  	busvoltage2 = 0
+  	current_mA2 = 0
+  	loadvoltage2 = 0
+
+  	busvoltage2 = ina3221.getBusVoltage_V(SOLAR_CELL_CHANNEL)
+  	shuntvoltage2 = ina3221.getShuntVoltage_mV(SOLAR_CELL_CHANNEL)
+  	current_mA2 = -ina3221.getCurrent_mA(SOLAR_CELL_CHANNEL)
+  	loadvoltage2 = busvoltage2 + (shuntvoltage2 / 1000)
+  
+  	print("Solar Cell Bus Voltage 2:  %3.2f V " % busvoltage2)
+  	print("Solar Cell Shunt Voltage 2: %3.2f mV " % shuntvoltage2)
+  	print("Solar Cell Load Voltage 2:  %3.2f V" % loadvoltage2)
+  	print("Solar Cell Current 2:  %3.2f mA" % current_mA2)
+  	print 
+
+  	shuntvoltage3 = 0
+  	busvoltage3 = 0
+  	current_mA3 = 0
+  	loadvoltage3 = 0
+
+  	busvoltage3 = ina3221.getBusVoltage_V(OUTPUT_CHANNEL)
+  	shuntvoltage3 = ina3221.getShuntVoltage_mV(OUTPUT_CHANNEL)
+  	current_mA3 = ina3221.getCurrent_mA(OUTPUT_CHANNEL)
+  	loadvoltage3 = busvoltage3 + (shuntvoltage3 / 1000)
+  
+  	print("Output Bus Voltage 3:  %3.2f V " % busvoltage3)
+  	print("Output Shunt Voltage 3: %3.2f mV " % shuntvoltage3)
+  	print("Output Load Voltage 3:  %3.2f V" % loadvoltage3)
+  	print("Output Current 3:  %3.2f mA" % current_mA3)
+  	print()
+        
+		
+
+        #
+  	time.sleep(2.0)
--- a/misc/GUI_icon.png
+++ b/misc/GUI_icon.png