bin/weathermon-iio

   1 #!/usr/bin/lua
   2
   3 require "uci"
   4 cur = uci.cursor()
   5
   6 lfs = require "lfs"
   7 json = require "json"
   8 socket = require "socket"
   9
  10 require "wm_util"
  11
  12 io.stdout:setvbuf('no')
  13
  14 function dump(o)
  15    if type(o) == 'table' then
  16       local s = '{ '
  17       for k,v in pairs(o) do
  18          if type(k) ~= 'number' then k = '"'..k..'"' end
  19          s = s .. '['..k..'] = ' .. dump(v) .. ','
  20       end
  21       return s .. '} '
  22    else
  23       return tostring(o)
  24    end
  25 end
  26
  27 function get_device_list(config_name)
  28
  29   local devices
  30   devices = {}
  31
  32   cur.foreach(config_name, "device", function(s)
  33     devices[#devices+1] = s[".name"]
  34   end)
  35
  36   return devices
  37
  38 end
  39
  40 function get_device(config_name,device_name)
  41
  42   local device
  43   device = {}
  44
  45   cur.foreach(config_name, "device", function(s)
  46     if s[".name"] == device_name then
  47       device = s
  48     end
  49   end)
  50
  51   return device
  52
  53 end
  54
  55 function find_device(name,subsystem)
  56
  57   local search_base
  58
  59   if subsystem == "iio" then
  60     search_base = "/sys/bus/iio/devices"
  61     for file in lfs.dir(search_base) do
  62       if get_file_content(search_base.."/"..file.."/name") == name then
  63         return search_base.."/"..file
  64       end
  65     end
  66   elseif subsystem == "hwmon" then
  67     search_base = "/sys/class/hwmon"
  68     for file in lfs.dir(search_base) do
  69       if get_file_content(search_base.."/"..file.."/device/name") == name then
  70         return search_base.."/"..file.."/device"
  71       end
  72     end
  73   end
  74
  75   return nil
  76
  77 end
  78
  79 function get_parameter(record)
  80   return tonumber(get_file_content(record["path"])) * record["scale"] + record["correction"]
  81 end
  82
  83 function string.fromhex(str)
  84   return (str:gsub('..', function (cc)
  85     return string.char(tonumber(cc, 16))
  86   end))
  87 end
  88
  89 function string.tohex(str)
  90   return (str:gsub('.', function (c)
  91     return string.format('%02X', string.byte(c))
  92   end))
  93 end
  94
  95 function get_mhz(record)
  96   local p = record["rs232"]
  97   p:read(9,100)
  98   p:write(string.fromhex("ff0186000000000079"))
  99   local e, s = p:read(9,1000)
 100   if (e == 0) and (s:len() == 9) and (s:byte(1) == 255) then
 101     local crc = 0
 102     for i = 2, 8 do
 103       crc = crc + s:byte(i)
 104       if (crc>=256) then
 105         crc = crc - 256
 106       end
 107     end
 108     crc = 255 - crc
 109     crc = crc + 1;
 110     if crc == s:byte(9) then
 111       return s:byte(3)*256+s:byte(4)
 112     end
 113   end
 114
 115   return nil
 116
 117 end
 118
 119 function init_serial_device(device,subsystem,parameters)
 120
 121   rs232 = require("luars232")
 122
 123   pcall(function ()
 124
 125     local e, port = rs232.open(device["port"])
 126
 127     if subsystem == "mhz" then
 128
 129       assert(port:set_baud_rate(rs232.RS232_BAUD_9600) == rs232.RS232_ERR_NOERROR)
 130       assert(port:set_data_bits(rs232.RS232_DATA_8) == rs232.RS232_ERR_NOERROR)
 131       assert(port:set_parity(rs232.RS232_PARITY_NONE) == rs232.RS232_ERR_NOERROR)
 132       assert(port:set_stop_bits(rs232.RS232_STOP_1) == rs232.RS232_ERR_NOERROR)
 133       assert(port:set_flow_control(rs232.RS232_FLOW_OFF)  == rs232.RS232_ERR_NOERROR)
 134
 135       getparameter = {}
 136
 137       getparameter["rs232"] = port
 138       getparameter["function"] = get_mhz
 139       getparameter["name"] = "CO2_PPM"
 140       getparameter["sensor"] = "MHZ19"
 141       parameters[#parameters+1] = getparameter
 142
 143     end
 144
 145   end)
 146
 147 end
 148
 149 function init_i2c_device(device,subsystem,parameters)
 150
 151   if not i2c_bus then
 152     i2c_bus = 0
 153   end
 154
 155   pcall(function ()
 156     local f = io.open("/sys/class/i2c-dev/i2c-"..i2c_bus.."/device/new_device","w")
 157     io.output(f)
 158     io.write(device["name"].." "..device["address"])
 159     io.close(f)
 160   end)
 161
 162   device_path=find_device(device["name"],subsystem)
 163
 164   if device_path and device["set_param"] then
 165     for key,record in pairs(device["set_param"]) do
 166       setparam = split(record,":")
 167       setpath = device_path.."/"..setparam[1]
 168       pcall(function ()
 169         local f = io.open(setpath,"w")
 170         io.output(f)
 171         io.write(setparam[2])
 172         io.close(f)
 173       end)
 174     end
 175   end
 176
 177   if device_path and device["parameter"] then
 178
 179     for key,record in pairs(device["parameter"]) do
 180
 181       getparam = split(record,":")
 182       getparameter = {}
 183       getparameter["path"] = device_path.."/"..getparam[1]
 184       getparameter["name"] = getparam[2]
 185       getscale = getparam[3]
 186       getcorrection = getparam[4]
 187       if not getscale then
 188         getscale = 1
 189       end
 190       if not getcorrection then
 191         getcorrection = 0
 192       end
 193       getparameter["scale"] = tonumber(getscale)
 194       getparameter["sensor"] = device["name"]:upper()
 195       getparameter["correction"] = tonumber(getcorrection)
 196       getparameter["function"] = get_parameter
 197       parameters[#parameters+1] = getparameter
 198
 199     end
 200
 201   end
 202
 203 end
 204
 205 function init_device(device,parameters)
 206
 207   if device["module"] then
 208     os.execute("modprobe "..device["module"])
 209   end
 210
 211   if device["type"] then
 212
 213     devtype = split(device["type"],":")
 214     bus = devtype[1]
 215     subsystem = devtype[2]
 216
 217     if (bus == "i2c") then
 218
 219       init_i2c_device(device,subsystem,parameters)
 220
 221     elseif (bus == "serial") then
 222
 223       init_serial_device(device,subsystem,parameters)
 224
 225     end
 226
 227   end
 228
 229 end
 230
 231 function init(config_name)
 232
 233   local parameters= {}
 234
 235   i2c_bus = uci.get(config_name,"hardware","i2c_bus")
 236
 237   local devices = get_device_list(config_name)
 238
 239   for key,devname in pairs(devices) do
 240
 241     device = get_device(config_name,devname)
 242
 243     if device then
 244       init_device(device,parameters)
 245     end
 246
 247   end
 248
 249   return parameters
 250
 251 end
 252
 253 function get_parameters(parameters)
 254   local results = {}
 255   for key,record in pairs(parameters) do
 256
 257     if not results[record["sensor"]] then
 258       results[record["sensor"]] = {}
 259     end
 260     pcall(function ()
 261         results[record["sensor"]][record["name"]] = record["function"](record)
 262       end)
 263   end
 264
 265   return results
 266 end
 267
 268 config_name = arg[1]
 269 if not config_name then
 270   config_name = "weathermon"
 271 end
 272
 273 web_id = get_devid(config_name)
 274
 275 parameters = init(config_name)
 276
 277 local delay = uci.get(config_name,"process","delay")
 278
 279 local working_dir = uci.get(config_name,"process","working_dir")
 280 if working_dir then
 281   lfs.mkdir(working_dir)
 282 end
 283
 284 if not delay then
 285   delay = 60
 286 end
 287
 288 while true do
 289     values = get_parameters(parameters)
 290     records = {}
 291     records[web_id] = {}
 292     for key,record in pairs(values) do
 293       records[web_id][key] = record
 294       records[web_id]["timestamp"] = os.date("%Y-%m-%dT%H:%M:%S")
 295     end
 296     for key,value in pairs(values) do
 297       value["device"] = key
 298       print(json.encode(value))
 299     end
 300   socket.sleep(delay)
 301 end