energymeter.py 58 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909
  1. #!/usr/bin/env python3
  2. """
  3. EnergyMeter: A Python module for interfacing with several energy meters.
  4. """
  5. __author__ = 'Stan Janssen'
  6. __email__ = 'stanjanssen@finetuned.nl'
  7. __url__ = 'https://finetuned.nl/'
  8. __license__ = 'Apache License, Version 2.0'
  9. __version__ = '1.0.2'
  10. __status__ = 'Beta'
  11. import random
  12. import socket
  13. import struct
  14. import asyncio
  15. import time
  16. import minimalmodbus
  17. from collections import Iterable
  18. class ABBMeter:
  19. """Meter class that uses a minimalmodbus Instrument to query an ABB meter."""
  20. def __init__(self, port, baudrate=38400, slaveaddress=1, type=None):
  21. """ Initialize the ABBMeter object.
  22. Arguments:
  23. * port: a serial port (string)
  24. * baudrate: the baudrate to use (integer)
  25. * slaveaddress: the address of the modbus device (integer)
  26. * Specification of the type. Used to limit the registers to a specific set.
  27. Returns:
  28. * An ABBMeter object
  29. """
  30. minimalmodbus.BAUDRATE = baudrate
  31. minimalmodbus.TIMEOUT = 0.5
  32. self.instrument = minimalmodbus.Instrument(port, slaveaddress)
  33. if type in ABBMeter.REGSETS:
  34. self.registers = [register for register in ABBMeter.REGS if register['name'] in ABBMeter.REGSETS[type]]
  35. else:
  36. self.registers = ABBMeter.REGS
  37. def read(self, regnames=None):
  38. """ Read one, many or all registers from the device
  39. Args:
  40. * registers (str or list). If None, read all. If string, read
  41. single register. If list, read all registers from list.
  42. Returns:
  43. * If single register, it returns a single value. If all or list,
  44. return a dict with the keys and values.
  45. Raises:
  46. * KeyError, TypeError, IOError
  47. """
  48. if regnames is None:
  49. return self._batch_read(self.registers)
  50. if type(regnames) is list:
  51. registers = [register for register in self.registers if register['name'] in regnames]
  52. if len(registers) < len(regnames):
  53. regs_not_available = [regname for regname in regnames if regname not in \
  54. [register['name'] for register in self.registers]]
  55. print("Warning: the following registers are not available on this device: " +
  56. ", ".join(regs_not_available))
  57. print("The available registers are: %s" +
  58. ", ".join(register['name'] for register in self.registers))
  59. if len(registers) == 0:
  60. return {}
  61. registers.sort(key=lambda reg: reg['start'])
  62. return self._batch_read(registers)
  63. elif type(regnames) is str:
  64. registers = [register for register in self.registers if register['name'] == regnames]
  65. if len(registers) == 0:
  66. return "Register not found on device."
  67. return self._read_single(registers[0])
  68. else:
  69. raise TypeError
  70. def _read_single(self, register):
  71. """
  72. Read a single register and return the value.
  73. """
  74. if register['length'] is 1:
  75. return self.instrument.read_register(registeraddress=register['start'],
  76. numberOfDecimals=register['decimals'],
  77. signed=register['signed'])
  78. if register['length'] is 2:
  79. value = self.instrument.read_long(registeraddress=register['start'],
  80. signed=register['signed'])
  81. return value / 10 ** register['decimals']
  82. if register['length'] is 4:
  83. value = self.instrument.read_registers(registeraddress=register['start'],
  84. numberOfRegisters=register['length'])
  85. return self._convert_value(values=value,
  86. signed=register['signed'],
  87. numberOfDecimals=register['decimals'])
  88. def _read_multiple(self, registers):
  89. """
  90. Read multiple registers from the slave device and return their values as a dict.
  91. Arguments:
  92. * A list of registers (complete structs)
  93. Returns:
  94. * A dict containing all keys and values
  95. """
  96. first_reg = min([register['start'] for register in registers])
  97. num_regs = max([register['start'] + register['length'] for register in registers]) - first_reg
  98. values = self.instrument.read_registers(registeraddress=first_reg,
  99. numberOfRegisters=num_regs)
  100. return self._interpret_result(values, registers)
  101. def _batch_read(self, registers):
  102. """
  103. Read multiple registers in batches, limiting each batch to at most 125 registers.
  104. Arguments:
  105. * A list of registers (complete structs)
  106. Returns:
  107. * A dict containing all keys and values
  108. """
  109. # Count up to at most 128 registers:
  110. start_reg = registers[0]['start']
  111. batch = []
  112. results = {}
  113. for register in registers:
  114. if register['start'] + register['length'] - start_reg <= 125:
  115. batch.append(register)
  116. else:
  117. results.update(self._read_multiple(batch))
  118. batch = []
  119. batch.append(register)
  120. start_reg = register['start']
  121. results.update(self._read_multiple(batch))
  122. return results
  123. def _interpret_result(self, data, registers):
  124. """
  125. Pull the returned string apart and package the data back to its
  126. intended form.
  127. Arguments:
  128. * data: list of register values returned from the device
  129. * registers: the original requested set of registers
  130. Returns:
  131. * A dict containing the register names and resulting values
  132. """
  133. first_reg = min([register['start'] for register in registers])
  134. results = {}
  135. for register in registers:
  136. regname = register['name']
  137. start = register['start'] - first_reg
  138. end = start + register['length']
  139. values = data[start:end]
  140. results[regname] = self._convert_value(values=values,
  141. signed=register['signed'],
  142. numberOfDecimals=register['decimals'])
  143. return results
  144. def _convert_value(self, values, signed=False, numberOfDecimals=0):
  145. """
  146. Convert a list of returned integers to the intended value.
  147. Arguments:
  148. * bytestring: a list of integers that together represent the value
  149. * signed: whether the value is a signed value
  150. * decimals: number of decimals the return value should contain
  151. """
  152. numberOfRegisters = len(values)
  153. formatcode_i = '>'
  154. formatcode_o = '>'
  155. if numberOfRegisters == 1:
  156. formatcode_i += "H"
  157. if signed:
  158. formatcode_o += "h"
  159. else:
  160. formatcode_o += "H"
  161. if numberOfRegisters == 2:
  162. formatcode_i += 'HH'
  163. if signed:
  164. formatcode_o += "l"
  165. else:
  166. formatcode_o += "L"
  167. if numberOfRegisters == 4:
  168. formatcode_i += "HHHH"
  169. if signed:
  170. formatcode_o += "q"
  171. else:
  172. formatcode_o += "Q"
  173. bytestring = struct.pack(formatcode_i, *values)
  174. value = struct.unpack(formatcode_o, bytestring)[0]
  175. if value in ABBMeter.NULLS:
  176. return None
  177. else:
  178. return float(value) / 10 ** numberOfDecimals
  179. # Register map of the ABB A and B series energy meters.
  180. REGS = [{'name': 'active_import', 'start': 20480, 'length': 4, 'signed': True, 'decimals': 2},
  181. {'name': 'active_export', 'start': 20484, 'length': 4, 'signed': True, 'decimals': 2},
  182. {'name': 'active_net', 'start': 20488, 'length': 4, 'signed': True, 'decimals': 2},
  183. {'name': 'reactive_import', 'start': 20492, 'length': 4, 'signed': True, 'decimals': 2},
  184. {'name': 'reactive_export', 'start': 20496, 'length': 4, 'signed': True, 'decimals': 2},
  185. {'name': 'reactive_net', 'start': 20500, 'length': 4, 'signed': True, 'decimals': 2},
  186. {'name': 'apparent_import', 'start': 20504, 'length': 4, 'signed': True, 'decimals': 2},
  187. {'name': 'apparent_export', 'start': 20508, 'length': 4, 'signed': True, 'decimals': 2},
  188. {'name': 'apparent_net', 'start': 20512, 'length': 4, 'signed': True, 'decimals': 2},
  189. {'name': 'active_import_co2', 'start': 20516, 'length': 4, 'signed': True, 'decimals': 2},
  190. {'name': 'active_import_currency', 'start': 20532, 'length': 4, 'signed': True, 'decimals': 2},
  191. {'name': 'active_import_tariff_1', 'start': 20848, 'length': 4, 'signed': False, 'decimals': 2},
  192. {'name': 'active_import_tariff_2', 'start': 20852, 'length': 4, 'signed': False, 'decimals': 2},
  193. {'name': 'active_import_tariff_3', 'start': 20856, 'length': 4, 'signed': False, 'decimals': 2},
  194. {'name': 'active_import_tariff_4', 'start': 20860, 'length': 4, 'signed': False, 'decimals': 2},
  195. {'name': 'active_export_tariff_1', 'start': 20880, 'length': 4, 'signed': False, 'decimals': 2},
  196. {'name': 'active_export_tariff_2', 'start': 20884, 'length': 4, 'signed': False, 'decimals': 2},
  197. {'name': 'active_export_tariff_3', 'start': 20888, 'length': 4, 'signed': False, 'decimals': 2},
  198. {'name': 'active_export_tariff_4', 'start': 20892, 'length': 4, 'signed': False, 'decimals': 2},
  199. {'name': 'reactive_import_tariff_1', 'start': 20912, 'length': 4, 'signed': False, 'decimals': 2},
  200. {'name': 'reactive_import_tariff_2', 'start': 20916, 'length': 4, 'signed': False, 'decimals': 2},
  201. {'name': 'reactive_import_tariff_3', 'start': 20920, 'length': 4, 'signed': False, 'decimals': 2},
  202. {'name': 'reactive_import_tariff_4', 'start': 20924, 'length': 4, 'signed': False, 'decimals': 2},
  203. {'name': 'reactive_export_tariff_1', 'start': 20944, 'length': 4, 'signed': False, 'decimals': 2},
  204. {'name': 'reactive_export_tariff_2', 'start': 20948, 'length': 4, 'signed': False, 'decimals': 2},
  205. {'name': 'reactive_export_tariff_3', 'start': 20952, 'length': 4, 'signed': False, 'decimals': 2},
  206. {'name': 'reactive_export_tariff_4', 'start': 20956, 'length': 4, 'signed': False, 'decimals': 2},
  207. {'name': 'active_import_l1', 'start': 21600, 'length': 4, 'signed': False, 'decimals': 2},
  208. {'name': 'active_import_l2', 'start': 21604, 'length': 4, 'signed': False, 'decimals': 2},
  209. {'name': 'active_import_l3', 'start': 21608, 'length': 4, 'signed': False, 'decimals': 2},
  210. {'name': 'active_export_l1', 'start': 21612, 'length': 4, 'signed': False, 'decimals': 2},
  211. {'name': 'active_export_l2', 'start': 21616, 'length': 4, 'signed': False, 'decimals': 2},
  212. {'name': 'active_export_l3', 'start': 21620, 'length': 4, 'signed': False, 'decimals': 2},
  213. {'name': 'active_net_l1', 'start': 21624, 'length': 4, 'signed': True, 'decimals': 2},
  214. {'name': 'active_net_l2', 'start': 21628, 'length': 4, 'signed': True, 'decimals': 2},
  215. {'name': 'active_net_l3', 'start': 21632, 'length': 4, 'signed': True, 'decimals': 2},
  216. {'name': 'reactive_import_l1', 'start': 21636, 'length': 4, 'signed': False, 'decimals': 2},
  217. {'name': 'reactive_import_l2', 'start': 21640, 'length': 4, 'signed': False, 'decimals': 2},
  218. {'name': 'reactive_import_l3', 'start': 21644, 'length': 4, 'signed': False, 'decimals': 2},
  219. {'name': 'reactive_export_l1', 'start': 21648, 'length': 4, 'signed': False, 'decimals': 2},
  220. {'name': 'reactive_export_l2', 'start': 21652, 'length': 4, 'signed': False, 'decimals': 2},
  221. {'name': 'reactive_export_l3', 'start': 21656, 'length': 4, 'signed': False, 'decimals': 2},
  222. {'name': 'reactive_net_l1', 'start': 21660, 'length': 4, 'signed': True, 'decimals': 2},
  223. {'name': 'reactive_net_l2', 'start': 21664, 'length': 4, 'signed': True, 'decimals': 2},
  224. {'name': 'reactive_net_l3', 'start': 21668, 'length': 4, 'signed': True, 'decimals': 2},
  225. {'name': 'apparent_import_l1', 'start': 21672, 'length': 4, 'signed': False, 'decimals': 2},
  226. {'name': 'apparent_import_l2', 'start': 21676, 'length': 4, 'signed': False, 'decimals': 2},
  227. {'name': 'apparent_import_l3', 'start': 21680, 'length': 4, 'signed': False, 'decimals': 2},
  228. {'name': 'apparent_export_l1', 'start': 21684, 'length': 4, 'signed': False, 'decimals': 2},
  229. {'name': 'apparent_export_l2', 'start': 21688, 'length': 4, 'signed': False, 'decimals': 2},
  230. {'name': 'apparent_export_l3', 'start': 21692, 'length': 4, 'signed': False, 'decimals': 2},
  231. {'name': 'apparent_net_l1', 'start': 21696, 'length': 4, 'signed': True, 'decimals': 2},
  232. {'name': 'apparent_net_l2', 'start': 21700, 'length': 4, 'signed': True, 'decimals': 2},
  233. {'name': 'apparent_net_l3', 'start': 21704, 'length': 4, 'signed': True, 'decimals': 2},
  234. {'name': 'resettable_active_import', 'start': 21804, 'length': 4, 'signed': False, 'decimals': 2},
  235. {'name': 'resettable_active_export', 'start': 21808, 'length': 4, 'signed': False, 'decimals': 2},
  236. {'name': 'resettable_reactive_import', 'start': 21812, 'length': 4, 'signed': False, 'decimals': 2},
  237. {'name': 'resettable_reactive_export', 'start': 21816, 'length': 4, 'signed': False, 'decimals': 2},
  238. {'name': 'voltage_l1_n', 'start': 23296, 'length': 2, 'signed': False, 'decimals': 1},
  239. {'name': 'voltage_l2_n', 'start': 23298, 'length': 2, 'signed': False, 'decimals': 1},
  240. {'name': 'voltage_l3_n', 'start': 23300, 'length': 2, 'signed': False, 'decimals': 1},
  241. {'name': 'voltage_l1_l2', 'start': 23302, 'length': 2, 'signed': False, 'decimals': 1},
  242. {'name': 'voltage_l3_l2', 'start': 23304, 'length': 2, 'signed': False, 'decimals': 1},
  243. {'name': 'voltage_l1_l3', 'start': 23306, 'length': 2, 'signed': False, 'decimals': 1},
  244. {'name': 'current_l1', 'start': 23308, 'length': 2, 'signed': False, 'decimals': 2},
  245. {'name': 'current_l2', 'start': 23310, 'length': 2, 'signed': False, 'decimals': 2},
  246. {'name': 'current_l3', 'start': 23312, 'length': 2, 'signed': False, 'decimals': 2},
  247. {'name': 'current_n', 'start': 23314, 'length': 2, 'signed': False, 'decimals': 2},
  248. {'name': 'active_power_total', 'start': 23316, 'length': 2, 'signed': True, 'decimals': 2},
  249. {'name': 'active_power_l1', 'start': 23318, 'length': 2, 'signed': True, 'decimals': 2},
  250. {'name': 'active_power_l2', 'start': 23320, 'length': 2, 'signed': True, 'decimals': 2},
  251. {'name': 'active_power_l3', 'start': 23322, 'length': 2, 'signed': True, 'decimals': 2},
  252. {'name': 'reactive_power_total', 'start': 23324, 'length': 2, 'signed': True, 'decimals': 2},
  253. {'name': 'reactive_power_l1', 'start': 23326, 'length': 2, 'signed': True, 'decimals': 2},
  254. {'name': 'reactive_power_l2', 'start': 23328, 'length': 2, 'signed': True, 'decimals': 2},
  255. {'name': 'reactive_power_l3', 'start': 23330, 'length': 2, 'signed': True, 'decimals': 2},
  256. {'name': 'apparent_power_total', 'start': 23332, 'length': 2, 'signed': True, 'decimals': 2},
  257. {'name': 'apparent_power_l1', 'start': 23334, 'length': 2, 'signed': True, 'decimals': 2},
  258. {'name': 'apparent_power_l2', 'start': 23336, 'length': 2, 'signed': True, 'decimals': 2},
  259. {'name': 'apparent_power_l3', 'start': 23338, 'length': 2, 'signed': True, 'decimals': 2},
  260. {'name': 'frequency', 'start': 23340, 'length': 1, 'signed': False, 'decimals': 2},
  261. {'name': 'phase_angle_power_total', 'start': 23341, 'length': 1, 'signed': True, 'decimals': 1},
  262. {'name': 'phase_angle_power_l1', 'start': 23342, 'length': 1, 'signed': True, 'decimals': 1},
  263. {'name': 'phase_angle_power_l2', 'start': 23343, 'length': 1, 'signed': True, 'decimals': 1},
  264. {'name': 'phase_angle_power_l3', 'start': 23344, 'length': 1, 'signed': True, 'decimals': 1},
  265. {'name': 'phase_angle_voltage_l1', 'start': 23345, 'length': 1, 'signed': True, 'decimals': 1},
  266. {'name': 'phase_angle_voltage_l2', 'start': 23346, 'length': 1, 'signed': True, 'decimals': 1},
  267. {'name': 'phase_angle_voltage_l3', 'start': 23347, 'length': 1, 'signed': True, 'decimals': 1},
  268. {'name': 'phase_angle_current_l1', 'start': 23351, 'length': 1, 'signed': True, 'decimals': 1},
  269. {'name': 'phase_angle_current_l2', 'start': 23352, 'length': 1, 'signed': True, 'decimals': 1},
  270. {'name': 'phase_angle_current_l3', 'start': 23353, 'length': 1, 'signed': True, 'decimals': 1},
  271. {'name': 'power_factor_total', 'start': 23354, 'length': 1, 'signed': True, 'decimals': 3},
  272. {'name': 'power_factor_l1', 'start': 23355, 'length': 1, 'signed': True, 'decimals': 3},
  273. {'name': 'power_factor_l2', 'start': 23356, 'length': 1, 'signed': True, 'decimals': 3},
  274. {'name': 'power_factor_l3', 'start': 23357, 'length': 1, 'signed': True, 'decimals': 3},
  275. {'name': 'current_quadrant_total', 'start': 23358, 'length': 1, 'signed': False, 'decimals': 0},
  276. {'name': 'current_quadrant_l1', 'start': 23359, 'length': 1, 'signed': False, 'decimals': 0},
  277. {'name': 'current_quadrant_l2', 'start': 23360, 'length': 1, 'signed': False, 'decimals': 0},
  278. {'name': 'current_quadrant_l3', 'start': 23361, 'length': 1, 'signed': False, 'decimals': 0},
  279. {'name': 'voltage_harmonics_l1_n_thd', 'start': 23808, 'length': 2, 'signed': False, 'decimals': 1},
  280. {'name': 'voltage_harmonics_l1_n_2nd', 'start': 23810, 'length': 2, 'signed': False, 'decimals': 1},
  281. {'name': 'voltage_harmonics_l1_n_3rd', 'start': 23812, 'length': 2, 'signed': False, 'decimals': 1},
  282. {'name': 'voltage_harmonics_l1_n_4th', 'start': 23814, 'length': 2, 'signed': False, 'decimals': 1},
  283. {'name': 'voltage_harmonics_l1_n_5th', 'start': 23816, 'length': 2, 'signed': False, 'decimals': 1},
  284. {'name': 'voltage_harmonics_l1_n_6th', 'start': 23818, 'length': 2, 'signed': False, 'decimals': 1},
  285. {'name': 'voltage_harmonics_l1_n_7th', 'start': 23820, 'length': 2, 'signed': False, 'decimals': 1},
  286. {'name': 'voltage_harmonics_l1_n_8th', 'start': 23822, 'length': 2, 'signed': False, 'decimals': 1},
  287. {'name': 'voltage_harmonics_l1_n_9th', 'start': 23824, 'length': 2, 'signed': False, 'decimals': 1},
  288. {'name': 'voltage_harmonics_l1_n_10th', 'start': 23826, 'length': 2, 'signed': False, 'decimals': 1},
  289. {'name': 'voltage_harmonics_l1_n_11th', 'start': 23828, 'length': 2, 'signed': False, 'decimals': 1},
  290. {'name': 'voltage_harmonics_l1_n_12th', 'start': 23830, 'length': 2, 'signed': False, 'decimals': 1},
  291. {'name': 'voltage_harmonics_l1_n_13th', 'start': 23832, 'length': 2, 'signed': False, 'decimals': 1},
  292. {'name': 'voltage_harmonics_l1_n_14th', 'start': 23834, 'length': 2, 'signed': False, 'decimals': 1},
  293. {'name': 'voltage_harmonics_l1_n_15th', 'start': 23836, 'length': 2, 'signed': False, 'decimals': 1},
  294. {'name': 'voltage_harmonics_l1_n_16th', 'start': 23838, 'length': 2, 'signed': False, 'decimals': 1},
  295. {'name': 'voltage_harmonics_l2_n_thd', 'start': 23936, 'length': 2, 'signed': False, 'decimals': 1},
  296. {'name': 'voltage_harmonics_l2_n_2nd', 'start': 23938, 'length': 2, 'signed': False, 'decimals': 1},
  297. {'name': 'voltage_harmonics_l2_n_3rd', 'start': 23940, 'length': 2, 'signed': False, 'decimals': 1},
  298. {'name': 'voltage_harmonics_l2_n_4th', 'start': 23942, 'length': 2, 'signed': False, 'decimals': 1},
  299. {'name': 'voltage_harmonics_l2_n_5th', 'start': 23944, 'length': 2, 'signed': False, 'decimals': 1},
  300. {'name': 'voltage_harmonics_l2_n_6th', 'start': 23946, 'length': 2, 'signed': False, 'decimals': 1},
  301. {'name': 'voltage_harmonics_l2_n_7th', 'start': 23948, 'length': 2, 'signed': False, 'decimals': 1},
  302. {'name': 'voltage_harmonics_l2_n_8th', 'start': 23950, 'length': 2, 'signed': False, 'decimals': 1},
  303. {'name': 'voltage_harmonics_l2_n_9th', 'start': 23952, 'length': 2, 'signed': False, 'decimals': 1},
  304. {'name': 'voltage_harmonics_l2_n_10th', 'start': 23954, 'length': 2, 'signed': False, 'decimals': 1},
  305. {'name': 'voltage_harmonics_l2_n_11th', 'start': 23956, 'length': 2, 'signed': False, 'decimals': 1},
  306. {'name': 'voltage_harmonics_l2_n_12th', 'start': 23958, 'length': 2, 'signed': False, 'decimals': 1},
  307. {'name': 'voltage_harmonics_l2_n_13th', 'start': 23960, 'length': 2, 'signed': False, 'decimals': 1},
  308. {'name': 'voltage_harmonics_l2_n_14th', 'start': 23962, 'length': 2, 'signed': False, 'decimals': 1},
  309. {'name': 'voltage_harmonics_l2_n_15th', 'start': 23964, 'length': 2, 'signed': False, 'decimals': 1},
  310. {'name': 'voltage_harmonics_l2_n_16th', 'start': 23966, 'length': 2, 'signed': False, 'decimals': 1},
  311. {'name': 'voltage_harmonics_l3_n_thd', 'start': 24064, 'length': 2, 'signed': False, 'decimals': 1},
  312. {'name': 'voltage_harmonics_l3_n_2nd', 'start': 24066, 'length': 2, 'signed': False, 'decimals': 1},
  313. {'name': 'voltage_harmonics_l3_n_3rd', 'start': 24068, 'length': 2, 'signed': False, 'decimals': 1},
  314. {'name': 'voltage_harmonics_l3_n_4th', 'start': 24070, 'length': 2, 'signed': False, 'decimals': 1},
  315. {'name': 'voltage_harmonics_l3_n_5th', 'start': 24072, 'length': 2, 'signed': False, 'decimals': 1},
  316. {'name': 'voltage_harmonics_l3_n_6th', 'start': 24074, 'length': 2, 'signed': False, 'decimals': 1},
  317. {'name': 'voltage_harmonics_l3_n_7th', 'start': 24076, 'length': 2, 'signed': False, 'decimals': 1},
  318. {'name': 'voltage_harmonics_l3_n_8th', 'start': 24078, 'length': 2, 'signed': False, 'decimals': 1},
  319. {'name': 'voltage_harmonics_l3_n_9th', 'start': 24080, 'length': 2, 'signed': False, 'decimals': 1},
  320. {'name': 'voltage_harmonics_l3_n_10th', 'start': 24082, 'length': 2, 'signed': False, 'decimals': 1},
  321. {'name': 'voltage_harmonics_l3_n_11th', 'start': 24084, 'length': 2, 'signed': False, 'decimals': 1},
  322. {'name': 'voltage_harmonics_l3_n_12th', 'start': 24086, 'length': 2, 'signed': False, 'decimals': 1},
  323. {'name': 'voltage_harmonics_l3_n_13th', 'start': 24088, 'length': 2, 'signed': False, 'decimals': 1},
  324. {'name': 'voltage_harmonics_l3_n_14th', 'start': 24090, 'length': 2, 'signed': False, 'decimals': 1},
  325. {'name': 'voltage_harmonics_l3_n_15th', 'start': 24092, 'length': 2, 'signed': False, 'decimals': 1},
  326. {'name': 'voltage_harmonics_l3_n_16th', 'start': 24094, 'length': 2, 'signed': False, 'decimals': 1},
  327. {'name': 'voltage_harmonics_l1_l2_thd', 'start': 24192, 'length': 2, 'signed': False, 'decimals': 1},
  328. {'name': 'voltage_harmonics_l1_l2_2nd', 'start': 24194, 'length': 2, 'signed': False, 'decimals': 1},
  329. {'name': 'voltage_harmonics_l1_l2_3rd', 'start': 24196, 'length': 2, 'signed': False, 'decimals': 1},
  330. {'name': 'voltage_harmonics_l1_l2_4th', 'start': 24198, 'length': 2, 'signed': False, 'decimals': 1},
  331. {'name': 'voltage_harmonics_l1_l2_5th', 'start': 24200, 'length': 2, 'signed': False, 'decimals': 1},
  332. {'name': 'voltage_harmonics_l1_l2_6th', 'start': 24202, 'length': 2, 'signed': False, 'decimals': 1},
  333. {'name': 'voltage_harmonics_l1_l2_7th', 'start': 24204, 'length': 2, 'signed': False, 'decimals': 1},
  334. {'name': 'voltage_harmonics_l1_l2_8th', 'start': 24206, 'length': 2, 'signed': False, 'decimals': 1},
  335. {'name': 'voltage_harmonics_l1_l2_9th', 'start': 24208, 'length': 2, 'signed': False, 'decimals': 1},
  336. {'name': 'voltage_harmonics_l1_l2_10th', 'start': 24210, 'length': 2, 'signed': False, 'decimals': 1},
  337. {'name': 'voltage_harmonics_l1_l2_11th', 'start': 24212, 'length': 2, 'signed': False, 'decimals': 1},
  338. {'name': 'voltage_harmonics_l1_l2_12th', 'start': 24214, 'length': 2, 'signed': False, 'decimals': 1},
  339. {'name': 'voltage_harmonics_l1_l2_13th', 'start': 24216, 'length': 2, 'signed': False, 'decimals': 1},
  340. {'name': 'voltage_harmonics_l1_l2_14th', 'start': 24218, 'length': 2, 'signed': False, 'decimals': 1},
  341. {'name': 'voltage_harmonics_l1_l2_15th', 'start': 24220, 'length': 2, 'signed': False, 'decimals': 1},
  342. {'name': 'voltage_harmonics_l1_l2_16th', 'start': 24222, 'length': 2, 'signed': False, 'decimals': 1},
  343. {'name': 'voltage_harmonics_l3_l2_thd', 'start': 24320, 'length': 2, 'signed': False, 'decimals': 1},
  344. {'name': 'voltage_harmonics_l3_l2_2nd', 'start': 24322, 'length': 2, 'signed': False, 'decimals': 1},
  345. {'name': 'voltage_harmonics_l3_l2_3rd', 'start': 24324, 'length': 2, 'signed': False, 'decimals': 1},
  346. {'name': 'voltage_harmonics_l3_l2_4th', 'start': 24326, 'length': 2, 'signed': False, 'decimals': 1},
  347. {'name': 'voltage_harmonics_l3_l2_5th', 'start': 24328, 'length': 2, 'signed': False, 'decimals': 1},
  348. {'name': 'voltage_harmonics_l3_l2_6th', 'start': 24330, 'length': 2, 'signed': False, 'decimals': 1},
  349. {'name': 'voltage_harmonics_l3_l2_7th', 'start': 24332, 'length': 2, 'signed': False, 'decimals': 1},
  350. {'name': 'voltage_harmonics_l3_l2_8th', 'start': 24334, 'length': 2, 'signed': False, 'decimals': 1},
  351. {'name': 'voltage_harmonics_l3_l2_9th', 'start': 24336, 'length': 2, 'signed': False, 'decimals': 1},
  352. {'name': 'voltage_harmonics_l3_l2_10th', 'start': 24338, 'length': 2, 'signed': False, 'decimals': 1},
  353. {'name': 'voltage_harmonics_l3_l2_11th', 'start': 24340, 'length': 2, 'signed': False, 'decimals': 1},
  354. {'name': 'voltage_harmonics_l3_l2_12th', 'start': 24342, 'length': 2, 'signed': False, 'decimals': 1},
  355. {'name': 'voltage_harmonics_l3_l2_13th', 'start': 24344, 'length': 2, 'signed': False, 'decimals': 1},
  356. {'name': 'voltage_harmonics_l3_l2_14th', 'start': 24346, 'length': 2, 'signed': False, 'decimals': 1},
  357. {'name': 'voltage_harmonics_l3_l2_15th', 'start': 24348, 'length': 2, 'signed': False, 'decimals': 1},
  358. {'name': 'voltage_harmonics_l3_l2_16th', 'start': 24350, 'length': 2, 'signed': False, 'decimals': 1},
  359. {'name': 'voltage_harmonics_l1_l3_thd', 'start': 24448, 'length': 2, 'signed': False, 'decimals': 1},
  360. {'name': 'voltage_harmonics_l1_l3_2nd', 'start': 24450, 'length': 2, 'signed': False, 'decimals': 1},
  361. {'name': 'voltage_harmonics_l1_l3_3rd', 'start': 24452, 'length': 2, 'signed': False, 'decimals': 1},
  362. {'name': 'voltage_harmonics_l1_l3_4th', 'start': 24454, 'length': 2, 'signed': False, 'decimals': 1},
  363. {'name': 'voltage_harmonics_l1_l3_5th', 'start': 24456, 'length': 2, 'signed': False, 'decimals': 1},
  364. {'name': 'voltage_harmonics_l1_l3_6th', 'start': 24458, 'length': 2, 'signed': False, 'decimals': 1},
  365. {'name': 'voltage_harmonics_l1_l3_7th', 'start': 24460, 'length': 2, 'signed': False, 'decimals': 1},
  366. {'name': 'voltage_harmonics_l1_l3_8th', 'start': 24462, 'length': 2, 'signed': False, 'decimals': 1},
  367. {'name': 'voltage_harmonics_l1_l3_9th', 'start': 24464, 'length': 2, 'signed': False, 'decimals': 1},
  368. {'name': 'voltage_harmonics_l1_l3_10th', 'start': 24466, 'length': 2, 'signed': False, 'decimals': 1},
  369. {'name': 'voltage_harmonics_l1_l3_11th', 'start': 24468, 'length': 2, 'signed': False, 'decimals': 1},
  370. {'name': 'voltage_harmonics_l1_l3_12th', 'start': 24470, 'length': 2, 'signed': False, 'decimals': 1},
  371. {'name': 'voltage_harmonics_l1_l3_13th', 'start': 24472, 'length': 2, 'signed': False, 'decimals': 1},
  372. {'name': 'voltage_harmonics_l1_l3_14th', 'start': 24474, 'length': 2, 'signed': False, 'decimals': 1},
  373. {'name': 'voltage_harmonics_l1_l3_15th', 'start': 24476, 'length': 2, 'signed': False, 'decimals': 1},
  374. {'name': 'voltage_harmonics_l1_l3_16th', 'start': 24478, 'length': 2, 'signed': False, 'decimals': 1},
  375. {'name': 'current_harmonics_l1_thd', 'start': 24576, 'length': 2, 'signed': False, 'decimals': 1},
  376. {'name': 'current_harmonics_l1_2nd', 'start': 24578, 'length': 2, 'signed': False, 'decimals': 1},
  377. {'name': 'current_harmonics_l1_3rd', 'start': 24580, 'length': 2, 'signed': False, 'decimals': 1},
  378. {'name': 'current_harmonics_l1_4th', 'start': 24582, 'length': 2, 'signed': False, 'decimals': 1},
  379. {'name': 'current_harmonics_l1_5th', 'start': 24584, 'length': 2, 'signed': False, 'decimals': 1},
  380. {'name': 'current_harmonics_l1_6th', 'start': 24586, 'length': 2, 'signed': False, 'decimals': 1},
  381. {'name': 'current_harmonics_l1_7th', 'start': 24588, 'length': 2, 'signed': False, 'decimals': 1},
  382. {'name': 'current_harmonics_l1_8th', 'start': 24590, 'length': 2, 'signed': False, 'decimals': 1},
  383. {'name': 'current_harmonics_l1_9th', 'start': 24592, 'length': 2, 'signed': False, 'decimals': 1},
  384. {'name': 'current_harmonics_l1_10th', 'start': 24594, 'length': 2, 'signed': False, 'decimals': 1},
  385. {'name': 'current_harmonics_l1_11th', 'start': 24596, 'length': 2, 'signed': False, 'decimals': 1},
  386. {'name': 'current_harmonics_l1_12th', 'start': 24598, 'length': 2, 'signed': False, 'decimals': 1},
  387. {'name': 'current_harmonics_l1_13th', 'start': 24600, 'length': 2, 'signed': False, 'decimals': 1},
  388. {'name': 'current_harmonics_l1_14th', 'start': 24602, 'length': 2, 'signed': False, 'decimals': 1},
  389. {'name': 'current_harmonics_l1_15th', 'start': 24604, 'length': 2, 'signed': False, 'decimals': 1},
  390. {'name': 'current_harmonics_l1_16th', 'start': 24606, 'length': 2, 'signed': False, 'decimals': 1},
  391. {'name': 'current_harmonics_l2_thd', 'start': 24704, 'length': 2, 'signed': False, 'decimals': 1},
  392. {'name': 'current_harmonics_l2_2nd', 'start': 24706, 'length': 2, 'signed': False, 'decimals': 1},
  393. {'name': 'current_harmonics_l2_3rd', 'start': 24708, 'length': 2, 'signed': False, 'decimals': 1},
  394. {'name': 'current_harmonics_l2_4th', 'start': 24710, 'length': 2, 'signed': False, 'decimals': 1},
  395. {'name': 'current_harmonics_l2_5th', 'start': 24712, 'length': 2, 'signed': False, 'decimals': 1},
  396. {'name': 'current_harmonics_l2_6th', 'start': 24714, 'length': 2, 'signed': False, 'decimals': 1},
  397. {'name': 'current_harmonics_l2_7th', 'start': 24716, 'length': 2, 'signed': False, 'decimals': 1},
  398. {'name': 'current_harmonics_l2_8th', 'start': 24718, 'length': 2, 'signed': False, 'decimals': 1},
  399. {'name': 'current_harmonics_l2_9th', 'start': 24720, 'length': 2, 'signed': False, 'decimals': 1},
  400. {'name': 'current_harmonics_l2_10th', 'start': 24722, 'length': 2, 'signed': False, 'decimals': 1},
  401. {'name': 'current_harmonics_l2_11th', 'start': 24724, 'length': 2, 'signed': False, 'decimals': 1},
  402. {'name': 'current_harmonics_l2_12th', 'start': 24726, 'length': 2, 'signed': False, 'decimals': 1},
  403. {'name': 'current_harmonics_l2_13th', 'start': 24728, 'length': 2, 'signed': False, 'decimals': 1},
  404. {'name': 'current_harmonics_l2_14th', 'start': 24730, 'length': 2, 'signed': False, 'decimals': 1},
  405. {'name': 'current_harmonics_l2_15th', 'start': 24732, 'length': 2, 'signed': False, 'decimals': 1},
  406. {'name': 'current_harmonics_l2_16th', 'start': 24734, 'length': 2, 'signed': False, 'decimals': 1},
  407. {'name': 'current_harmonics_l3_thd', 'start': 24832, 'length': 2, 'signed': False, 'decimals': 1},
  408. {'name': 'current_harmonics_l3_2nd', 'start': 24834, 'length': 2, 'signed': False, 'decimals': 1},
  409. {'name': 'current_harmonics_l3_3rd', 'start': 24836, 'length': 2, 'signed': False, 'decimals': 1},
  410. {'name': 'current_harmonics_l3_4th', 'start': 24838, 'length': 2, 'signed': False, 'decimals': 1},
  411. {'name': 'current_harmonics_l3_5th', 'start': 24840, 'length': 2, 'signed': False, 'decimals': 1},
  412. {'name': 'current_harmonics_l3_6th', 'start': 24842, 'length': 2, 'signed': False, 'decimals': 1},
  413. {'name': 'current_harmonics_l3_7th', 'start': 24844, 'length': 2, 'signed': False, 'decimals': 1},
  414. {'name': 'current_harmonics_l3_8th', 'start': 24846, 'length': 2, 'signed': False, 'decimals': 1},
  415. {'name': 'current_harmonics_l3_9th', 'start': 24848, 'length': 2, 'signed': False, 'decimals': 1},
  416. {'name': 'current_harmonics_l3_10th', 'start': 24850, 'length': 2, 'signed': False, 'decimals': 1},
  417. {'name': 'current_harmonics_l3_11th', 'start': 24852, 'length': 2, 'signed': False, 'decimals': 1},
  418. {'name': 'current_harmonics_l3_12th', 'start': 24854, 'length': 2, 'signed': False, 'decimals': 1},
  419. {'name': 'current_harmonics_l3_13th', 'start': 24856, 'length': 2, 'signed': False, 'decimals': 1},
  420. {'name': 'current_harmonics_l3_14th', 'start': 24858, 'length': 2, 'signed': False, 'decimals': 1},
  421. {'name': 'current_harmonics_l3_15th', 'start': 24860, 'length': 2, 'signed': False, 'decimals': 1},
  422. {'name': 'current_harmonics_l3_16th', 'start': 24862, 'length': 2, 'signed': False, 'decimals': 1},
  423. {'name': 'current_harmonics_n_thd', 'start': 24960, 'length': 2, 'signed': False, 'decimals': 1},
  424. {'name': 'current_harmonics_n_2nd', 'start': 24962, 'length': 2, 'signed': False, 'decimals': 1},
  425. {'name': 'current_harmonics_n_3rd', 'start': 24964, 'length': 2, 'signed': False, 'decimals': 1},
  426. {'name': 'current_harmonics_n_4th', 'start': 24966, 'length': 2, 'signed': False, 'decimals': 1},
  427. {'name': 'current_harmonics_n_5th', 'start': 24968, 'length': 2, 'signed': False, 'decimals': 1},
  428. {'name': 'current_harmonics_n_6th', 'start': 24970, 'length': 2, 'signed': False, 'decimals': 1},
  429. {'name': 'current_harmonics_n_7th', 'start': 24972, 'length': 2, 'signed': False, 'decimals': 1},
  430. {'name': 'current_harmonics_n_8th', 'start': 24974, 'length': 2, 'signed': False, 'decimals': 1},
  431. {'name': 'current_harmonics_n_9th', 'start': 24976, 'length': 2, 'signed': False, 'decimals': 1},
  432. {'name': 'current_harmonics_n_10th', 'start': 24978, 'length': 2, 'signed': False, 'decimals': 1},
  433. {'name': 'current_harmonics_n_11th', 'start': 24980, 'length': 2, 'signed': False, 'decimals': 1},
  434. {'name': 'current_harmonics_n_12th', 'start': 24982, 'length': 2, 'signed': False, 'decimals': 1},
  435. {'name': 'current_harmonics_n_13th', 'start': 24984, 'length': 2, 'signed': False, 'decimals': 1},
  436. {'name': 'current_harmonics_n_14th', 'start': 24986, 'length': 2, 'signed': False, 'decimals': 1},
  437. {'name': 'current_harmonics_n_15th', 'start': 24988, 'length': 2, 'signed': False, 'decimals': 1},
  438. {'name': 'current_harmonics_n_16th', 'start': 24990, 'length': 2, 'signed': False, 'decimals': 1}]
  439. REGSETS = {
  440. "A43": ["active_import", "active_export", "active_net",
  441. "reactive_import", "reactive_export", "reactive_net",
  442. "apparent_import", "apparent_export", "apparent_net",
  443. "active_import_co2", "active_import_currency",
  444. "active_import_l1", "active_import_l2", "active_import_l3",
  445. "active_export_l1", "active_export_l2", "active_export_l3",
  446. "active_net_l1", "active_net_l2", "active_net_l3",
  447. "reactive_import_l1", "reactive_import_l2",
  448. "reactive_import_l3", "reactive_export_l1",
  449. "reactive_export_l2", "reactive_export_l3",
  450. "reactive_net_l1", "reactive_net_l2", "reactive_net_l3",
  451. "apparent_import_l1", "apparent_import_l2",
  452. "apparent_import_l3", "apparent_export_l1",
  453. "apparent_export_l2", "apparent_export_l3",
  454. "apparent_net_l1", "apparent_net_l2", "apparent_net_l3",
  455. "voltage_l1_n", "voltage_l2_n", "voltage_l3_n",
  456. "voltage_l1_l2", "voltage_l3_l2", "voltage_l1_l3",
  457. "current_l1", "current_l2", "current_l3",
  458. "active_power_total", "active_power_l1", "active_power_l2",
  459. "active_power_l3", "reactive_power_total",
  460. "reactive_power_l1", "reactive_power_l2",
  461. "reactive_power_l3", "apparent_power_total",
  462. "apparent_power_l1", "apparent_power_l2",
  463. "apparent_power_l3", "frequency", "phase_angle_power_total",
  464. "power_factor_total", "power_factor_l1", "power_factor_l2",
  465. "power_factor_l3", "current_quadrant_total",
  466. "current_quadrant_l1", "current_quadrant_l2",
  467. "current_quadrant_l3"],
  468. "B23": ["active_import", "active_export", "active_net",
  469. "reactive_import", "reactive_export", "reactive_net",
  470. "apparent_import", "apparent_export", "apparent_net",
  471. "active_import_co2", "active_import_currency",
  472. "active_import_l1", "active_import_l2", "active_import_l3",
  473. "active_export_l1", "active_export_l2", "active_export_l3",
  474. "active_net_l1", "active_net_l2", "active_net_l3",
  475. "reactive_import_l1", "reactive_import_l2",
  476. "reactive_import_l3", "reactive_export_l1",
  477. "reactive_export_l2", "reactive_export_l3",
  478. "reactive_net_l1", "reactive_net_l2", "reactive_net_l3",
  479. "apparent_import_l1", "apparent_import_l2",
  480. "apparent_import_l3", "apparent_export_l1",
  481. "apparent_export_l2", "apparent_export_l3",
  482. "apparent_net_l1", "apparent_net_l2", "apparent_net_l3",
  483. "voltage_l1_n", "voltage_l2_n", "voltage_l3_n",
  484. "voltage_l1_l2", "voltage_l3_l2", "voltage_l1_l3",
  485. "current_l1", "current_l2", "current_l3",
  486. "active_power_total", "active_power_l1", "active_power_l2",
  487. "active_power_l3", "reactive_power_total",
  488. "reactive_power_l1", "reactive_power_l2",
  489. "reactive_power_l3", "apparent_power_total",
  490. "apparent_power_l1", "apparent_power_l2",
  491. "apparent_power_l3", "frequency", "phase_angle_power_total",
  492. "power_factor_total", "power_factor_l1", "power_factor_l2",
  493. "power_factor_l3", "current_quadrant_total",
  494. "current_quadrant_l1", "current_quadrant_l2",
  495. "current_quadrant_l3"],
  496. "B21": ["active_import", "active_export", "active_net",
  497. "reactive_import", "reactive_export", "reactive_net",
  498. "apparent_import", "apparent_export", "apparent_net",
  499. "active_import_co2", "active_import_currency",
  500. "voltage_l1_n", "current_l1", "active_power_total",
  501. "reactive_power_total", "apparent_power_total", "frequency",
  502. "phase_angle_power_total", "power_factor_total",
  503. "current_quadrant_total"]
  504. }
  505. NULLS = [pow(2, n) - 1 for n in (64, 63, 32, 31, 16, 15)]
  506. class ModbusTCPMeter:
  507. """
  508. Implementation for a Modbus TCP Energy Meter.
  509. """
  510. def __init__(self, port, tcp_port=502, slaveaddress=126, type=None, baudrate=None):
  511. self.port = port
  512. self.tcp_port = tcp_port
  513. self.device_id = slaveaddress
  514. def read(self, regnames=None):
  515. if regnames is None:
  516. registers = self.REGS
  517. return self._read_multiple(registers)
  518. if type(regnames) is str:
  519. registers = [register for register in self.REGS if register['name'] == regnames]
  520. return self._read_single(registers[0])
  521. if type(regnames) is list:
  522. registers = [register for register in self.REGS if register['name'] in regnames]
  523. return self._read_multiple(registers)
  524. def _read_single(self, register):
  525. message = self._modbus_message(start_reg=register['start'], num_regs=register['length'])
  526. data = self._perform_request(message)
  527. return self._convert_value(data, signed=register['signed'], decimals=register['decimals'])
  528. def _read_multiple(self, registers):
  529. registers.sort(key=lambda reg: reg['start'])
  530. results = {}
  531. for reg_range in self._split_ranges(registers):
  532. first_reg = min([register['start'] for register in reg_range])
  533. num_regs = max([register['start'] + register['length'] for register in reg_range]) - first_reg
  534. message = self._modbus_message(start_reg=first_reg, num_regs=num_regs)
  535. data = self._perform_request(message)
  536. results.update(self._interpret_result(data, reg_range))
  537. return results
  538. def _split_ranges(self, registers):
  539. """
  540. Generator that splits the registers list into continuous parts.
  541. """
  542. reg_list = []
  543. prev_end = registers[0]['start'] - 1
  544. for r in registers:
  545. if r['start'] - prev_end > 1:
  546. yield reg_list
  547. reg_list = []
  548. reg_list.append(r)
  549. prev_end = r['start'] + r['length']
  550. yield reg_list
  551. def _modbus_message(self, start_reg, num_regs):
  552. transaction_id = random.randint(1, 2**16 - 1)
  553. return struct.pack(">HHHBBHH", transaction_id,
  554. self.PROTOCOL_CODE,
  555. 6,
  556. self.device_id,
  557. self.FUNCTION_CODE,
  558. start_reg - self.REG_OFFSET,
  559. num_regs)
  560. def _perform_request(self, message):
  561. if self.device is None:
  562. self._connect()
  563. self.device.send(message)
  564. data = bytes()
  565. expect_bytes = 9 + 2 * struct.unpack(">H", message[-2:])[0]
  566. attempt = 1
  567. while len(data) is not expect_bytes:
  568. time.sleep(0.05)
  569. data += self.device.recv(2048)
  570. if attempt >= 10:
  571. return 2 * struct.unpack(">H", message[-2:])[0] * [0]
  572. attempt += 1
  573. return data[9:]
  574. def _interpret_result(self, data, registers):
  575. """
  576. Pull the returned string apart and package the data back to its
  577. intended form.
  578. Arguments:
  579. * data: list of register values returned from the device
  580. * registers: the original requested set of registers
  581. Returns:
  582. * A dict containing the register names and resulting values
  583. """
  584. first_reg = min([register['start'] for register in registers])
  585. results = {}
  586. for register in registers:
  587. regname = register['name']
  588. start = (register['start'] - first_reg) * 2
  589. end = start + register['length'] * 2
  590. values = data[start:end]
  591. results[regname] = self._convert_value(values=values,
  592. signed=register['signed'],
  593. decimals=register['decimals'],
  594. isFloat=register['isFloat'])
  595. if regname == "power_factor_total" and results[regname] == 0:
  596. results[regname] = 1 # The SMA will send out a 0 when the power factor is 100%
  597. return results
  598. def _convert_value(self, values, signed=False, decimals=0, isFloat=False):
  599. """
  600. Convert a list of returned integers to the intended value.
  601. Arguments:
  602. * bytestring: a list of integers that together represent the value
  603. * signed: whether the value is a signed value
  604. * decimals: number of decimals the return value should contain
  605. * isFloat: whether the valie is a float
  606. """
  607. numberOfBytes = len(values)
  608. formatcode_o = '>'
  609. if isFloat:
  610. formatcode_o += 'f'
  611. elif numberOfBytes == 1:
  612. if signed:
  613. formatcode_o += "b"
  614. else:
  615. formatcode_o += "B"
  616. elif numberOfBytes == 2:
  617. if signed:
  618. formatcode_o += "h"
  619. else:
  620. formatcode_o += "H"
  621. elif numberOfBytes == 4:
  622. if signed:
  623. formatcode_o += "l"
  624. else:
  625. formatcode_o += "L"
  626. value = struct.unpack(formatcode_o, bytes(values))[0]
  627. if value in self.NULLS:
  628. return None
  629. else:
  630. return float(value) / 10 ** decimals
  631. def _connect(self):
  632. self.device = socket.create_connection(address=(self.port, self.tcp_port))
  633. class SMAMeter(ModbusTCPMeter):
  634. def __init__(self, *args, **kwargs):
  635. super().__init__(*args, **kwargs)
  636. REGS = [
  637. {'name': 'current_ac', 'start': 40188, 'length': 1, 'signed': False, 'decimals': 1},
  638. {'name': 'current_l1', 'start': 40189, 'length': 1, 'signed': False, 'decimals': 1},
  639. {'name': 'current_l2', 'start': 40190, 'length': 1, 'signed': False, 'decimals': 1},
  640. {'name': 'current_l3', 'start': 40191, 'length': 1, 'signed': False, 'decimals': 1},
  641. {'name': 'voltage_l1_l2', 'start': 40193, 'length': 1, 'signed': False, 'decimals': 1},
  642. {'name': 'voltage_l2_l3', 'start': 40194, 'length': 1, 'signed': False, 'decimals': 1},
  643. {'name': 'voltage_l3_l1', 'start': 40195, 'length': 1, 'signed': False, 'decimals': 1},
  644. {'name': 'voltage_l1_n', 'start': 40196, 'length': 1, 'signed': False, 'decimals': 1},
  645. {'name': 'voltage_l2_n', 'start': 40197, 'length': 1, 'signed': False, 'decimals': 1},
  646. {'name': 'voltage_l3_n', 'start': 40198, 'length': 1, 'signed': False, 'decimals': 1},
  647. {'name': 'active_power_total', 'start': 40200, 'length': 1, 'signed': True, 'decimals': -1},
  648. {'name': 'frequency', 'start': 40202, 'length': 1, 'signed': False, 'decimals': 2},
  649. {'name': 'apparent_power_total', 'start': 40204, 'length': 1, 'signed': True, 'decimals': -1},
  650. {'name': 'reactive_power_total', 'start': 40206, 'length': 1, 'signed': True, 'decimals': -1},
  651. {'name': 'power_factor_total', 'start': 40208, 'length': 1, 'signed': True, 'decimals': 3},
  652. {'name': 'active_export', 'start': 40210, 'length': 2, 'signed': False, 'decimals': 3},
  653. {'name': 'dc_power', 'start': 40217, 'length': 1, 'signed': False, 'decimals': -1},
  654. {'name': 'temperature_internal', 'start': 40219, 'length': 1, 'signed': False, 'decimals': 0},
  655. {'name': 'temperature_other', 'start': 40222, 'length': 1, 'signed': False, 'decimals': 0},
  656. {'name': 'operating_status', 'start': 40224, 'length': 1, 'signed': False, 'decimals': 0}
  657. ]
  658. REG_OFFSET = 1
  659. PROTOCOL_CODE = 0
  660. FUNCTION_CODE = 3
  661. NULLS = [2**16 - 1, 2**15 - 1, 2**15, -2**15]
  662. class MulticubeMeter(ModbusTCPMeter):
  663. """
  664. Implementation for a Multicube energy meter over Modbus TCP.
  665. """
  666. def __init__(self, port, tcp_port=1502, slaveaddress=1, type=None, baudrate=None, auto_scale=True):
  667. super().__init__(port, tcp_port, slaveaddress, type, baudrate)
  668. self.device = socket.create_connection(address=(port, tcp_port))
  669. self.device_id = slaveaddress
  670. self.REGS = [
  671. {'name': 'energy_scale', 'start': 512, 'length': 2, 'decimals': 0, 'signed': True},
  672. {'name': 'active_net', 'start': 514, 'length': 2, 'decimals': 0, 'signed': True},
  673. {'name': 'apparent_net', 'start': 516, 'length': 2, 'decimals': 0, 'signed': True},
  674. {'name': 'reactive_net', 'start': 518, 'length': 2, 'decimals': 0, 'signed': True},
  675. {'name': 'active_power_total', 'start': 2816, 'length': 1, 'decimals': 0, 'signed': True},
  676. {'name': 'apparent_power_total', 'start': 2817, 'length': 1, 'decimals': 0, 'signed': True},
  677. {'name': 'reactive_power_total', 'start': 2818, 'length': 1, 'decimals': 0, 'signed': True},
  678. {'name': 'power_factor_total', 'start': 2819, 'length': 1, 'decimals': 3, 'signed': True},
  679. {'name': 'frequency', 'start': 2820, 'length': 1, 'decimals': 1, 'signed': True},
  680. {'name': 'voltage_l1_n', 'start': 2821, 'length': 1, 'decimals': 0, 'signed': True},
  681. {'name': 'current_l1', 'start': 2822, 'length': 1, 'decimals': 0, 'signed': True},
  682. {'name': 'active_power_l1', 'start': 2823, 'length': 1, 'decimals': 0, 'signed': True},
  683. {'name': 'voltage_l2_n', 'start': 2824, 'length': 1, 'decimals': 0, 'signed': True},
  684. {'name': 'current_l2', 'start': 2825, 'length': 1, 'decimals': 0, 'signed': True},
  685. {'name': 'active_power_l2', 'start': 2826, 'length': 1, 'decimals': 0, 'signed': True},
  686. {'name': 'voltage_l3_n', 'start': 2827, 'length': 1, 'decimals': 0, 'signed': True},
  687. {'name': 'current_l3', 'start': 2828, 'length': 1, 'decimals': 0, 'signed': True},
  688. {'name': 'active_power_l3', 'start': 2829, 'length': 1, 'decimals': 0, 'signed': True},
  689. {'name': 'power_factor_l1', 'start': 2830, 'length': 1, 'decimals': 0, 'signed': True},
  690. {'name': 'power_factor_l2', 'start': 2831, 'length': 1, 'decimals': 0, 'signed': True},
  691. {'name': 'power_factor_l3', 'start': 2832, 'length': 1, 'decimals': 0, 'signed': True},
  692. {'name': 'voltage_l1_l2', 'start': 2833, 'length': 1, 'decimals': 0, 'signed': True},
  693. {'name': 'voltage_l2_l3', 'start': 2834, 'length': 1, 'decimals': 0, 'signed': True},
  694. {'name': 'voltage_l3_l1', 'start': 2835, 'length': 1, 'decimals': 0, 'signed': True},
  695. {'name': 'current_n', 'start': 2836, 'length': 1, 'decimals': 0, 'signed': True},
  696. {'name': 'amps_scale', 'start': 2837, 'length': 1, 'decimals': 0, 'signed': True},
  697. {'name': 'phase_volts_scale', 'start': 2838, 'length': 1, 'decimals': 0, 'signed': True},
  698. {'name': 'line_volts_scale', 'start': 2839, 'length': 1, 'decimals': 0, 'signed': True},
  699. {'name': 'power_scale', 'start': 2840, 'length': 1, 'decimals': 0, 'signed': True},
  700. {'name': 'apparent_power_l1', 'start': 3072, 'length': 1, 'decimals': 0, 'signed': True},
  701. {'name': 'apparent_power_l2', 'start': 3073, 'length': 1, 'decimals': 0, 'signed': True},
  702. {'name': 'apparent_power_l3', 'start': 3074, 'length': 1, 'decimals': 0, 'signed': True},
  703. {'name': 'reactive_power_l1', 'start': 3075, 'length': 1, 'decimals': 0, 'signed': True},
  704. {'name': 'reactive_power_l2', 'start': 3076, 'length': 1, 'decimals': 0, 'signed': True},
  705. {'name': 'reactive_power_l3', 'start': 3077, 'length': 1, 'decimals': 0, 'signed': True},
  706. {'name': 'peak_current_l1', 'start': 3078, 'length': 1, 'decimals': 0, 'signed': True},
  707. {'name': 'peak_current_l2', 'start': 3079, 'length': 1, 'decimals': 0, 'signed': True},
  708. {'name': 'peak_current_l3', 'start': 3080, 'length': 1, 'decimals': 0, 'signed': True},
  709. {'name': 'current_l1_thd', 'start': 3081, 'length': 1, 'decimals': 3, 'signed': True},
  710. {'name': 'current_l2_thd', 'start': 3082, 'length': 1, 'decimals': 3, 'signed': True},
  711. {'name': 'current_l3_thd', 'start': 3083, 'length': 1, 'decimals': 3, 'signed': True}
  712. ]
  713. if auto_scale:
  714. self.set_scaling()
  715. def set_scaling(self):
  716. """
  717. Call this function before reading anything to set up the correct scaling for this meter.
  718. """
  719. decimals_mapping = {1: 2, 2: 1, 3: 0, 4: -1, 5: -2, 6: -3, 7: -4}
  720. a_registers = ['current_l1', 'current_l2', 'current_l3', 'current_n']
  721. scale = int(self.read('amps_scale'))
  722. for r in self.REGS:
  723. if r['name'] in a_registers:
  724. r['decimals'] = decimals_mapping[scale]
  725. pv_registers = ['voltage_l1_n', 'voltage_l2_n', 'voltage_l3_n']
  726. scale = int(self.read('phase_volts_scale'))
  727. for r in self.REGS:
  728. if r['name'] in pv_registers:
  729. r['decimals'] = decimals_mapping[scale]
  730. lv_registers = ['voltage_l1_l2', 'voltage_l2_l3', 'voltage_l3_l1']
  731. scale = int(self.read('line_volts_scale'))
  732. for r in self.REGS:
  733. if r['name'] in lv_registers:
  734. r['decimals'] = decimals_mapping[scale]
  735. p_registers = ['active_power_total',
  736. 'reactive_power_total',
  737. 'apparent_power_total',
  738. 'active_power_l1',
  739. 'active_power_l2',
  740. 'active_power_l3',
  741. 'apparent_power_l1',
  742. 'apparent_power_l2',
  743. 'apparent_power_l3',
  744. 'reactive_power_l1',
  745. 'reactive_power_l2',
  746. 'reactive_power_l3']
  747. scale = int(self.read('power_scale'))
  748. for r in self.REGS:
  749. if r['name'] in p_registers:
  750. r['decimals'] = decimals_mapping[scale]
  751. e_registers = ['active_net', 'apparent_net', 'reactive_net']
  752. decimals_mapping = {3: 3, 4: 2, 5: 1, 6: 0, 7: -1}
  753. scale = int(self.read('energy_scale'))
  754. for r in self.REGS:
  755. if r['name'] in e_registers:
  756. r['decimals'] = decimals_mapping[scale]
  757. REG_OFFSET = 0
  758. PROTOCOL_CODE = 0
  759. FUNCTION_CODE = 3
  760. NULLS = [2**16 - 1, 2**15 - 1, 2**15, -2**15]
  761. class AsyncModbusTCPMeter(ModbusTCPMeter):
  762. def __init__(self, *args, **kwargs):
  763. super().__init__(*args, **kwargs)
  764. self.reader = self.writer = None
  765. async def _connect(self):
  766. self.reader, self.writer = await asyncio.open_connection(host=self.port, port=self.tcp_port)
  767. async def read(self, regnames=None):
  768. if regnames is None:
  769. registers = self.REGS
  770. return await self._read_multiple(registers)
  771. if isinstance(regnames, str):
  772. registers = [register for register in self.REGS if register['name'] == regnames]
  773. if len(registers) == 0:
  774. return "Register not found on device."
  775. return await self._read_single(registers[0])
  776. if isinstance(regnames, Iterable):
  777. registers = [register for register in self.REGS if register['name'] in regnames]
  778. return await self._read_multiple(registers)
  779. async def _read_single(self, register):
  780. num_regs = register['length']
  781. message = self._modbus_message(register['start'], num_regs)
  782. if self.writer is None:
  783. await self._connect()
  784. self.writer.write(message)
  785. await self.writer.drain()
  786. data = await self.reader.readexactly(9 + 2 * num_regs)
  787. return self._convert_value(data[9:], signed=register['signed'], decimals=register['decimals'], isFloat=register['isFloat']|False)
  788. async def _read_multiple(self, registers):
  789. registers.sort(key=lambda reg: reg['start'])
  790. results = {}
  791. for reg_range in self._split_ranges(registers):
  792. # Prepare the request
  793. first_reg = min([register['start'] for register in reg_range])
  794. num_regs = max([register['start'] + register['length'] for register in reg_range]) - first_reg
  795. if self.writer is None:
  796. await self._connect()
  797. self.writer.write(self._modbus_message(first_reg, num_regs))
  798. await self.writer.drain()
  799. # Receive the response
  800. data = await self.reader.readexactly(9 + 2 * num_regs)
  801. results.update(self._interpret_result(data[9:], reg_range))
  802. return results
  803. class AsyncABBTCPMeter(AsyncModbusTCPMeter):
  804. def __init__(self, *args, **kwargs):
  805. super().__init__(*args, **kwargs)
  806. if type in ABBMeter.REGSETS:
  807. self.registers = [register for register in ABBMeter.REGS if register['name'] in ABBMeter.REGSETS[type]]
  808. else:
  809. self.registers = ABBMeter.REGS
  810. REGS = ABBMeter.REGS
  811. REGSETS = ABBMeter.REGSETS
  812. NULLS = ABBMeter.NULLS
  813. PROTOCOL_CODE = 0
  814. FUNCTION_CODE = 3
  815. REG_OFFSET = 0