bin/weather-lcd

   1 #!/usr/bin/lua
   2
   3 socket = require "socket"
   4 json = require "json"
   5 require "uci"
   6 cur = uci.cursor()
   7 require "wm_util"
   8
   9 config_name = arg[1]
  10 if not config_name then
  11   config_name = "weathermon"
  12 end
  13
  14 weather_file = uci.get(config_name,"process","dump_file")
  15 weather_db = uci.get(config_name,"process","logdb")
  16
  17 graph_duration = uci.get(config_name,"display","graph_duration")
  18
  19 if graph_duration and not(weather_db == "") then
  20   pcall(function ()
  21       local dbdriver = require "luasql.sqlite3"
  22       local env = assert(dbdriver.sqlite3())
  23       log_con = assert(env:connect(weather_db))
  24     end)
  25 end
  26
  27 function round(num, numDecimalPlaces)
  28   local mult = 10^(numDecimalPlaces or 0)
  29   return math.floor(num * mult + 0.5) / mult
  30 end
  31
  32 function trim(s)
  33   return (s:gsub("^%s*(.-)%s*$", "%1"))
  34 end
  35
  36 function lpad(s, l, c)
  37   local res = string.rep(c or ' ', l - #s) .. s
  38   return res, res ~= s
  39 end
  40
  41 function process_file()
  42   txt = get_file_content(weather_file)
  43   data = json.decode(txt)
  44   for k,v in pairs(data) do
  45     return v
  46   end
  47 end
  48
  49 function get_display_config()
  50
  51   width  = tonumber(uci.get(config_name,"display","width"))
  52   height = tonumber(uci.get(config_name,"display","height"))
  53   charwidth  = tonumber(uci.get(config_name,"display","charwidth"))
  54   charheight = tonumber(uci.get(config_name,"display","charheight"))
  55   title  = uci.get(config_name,"display","title")
  56
  57   pages = {}
  58   pagetitles = {}
  59   pagedurations = {}
  60   pagecolumns = {}
  61   pagecolwidth = {}
  62
  63   local page, def, line, col, pos
  64
  65   cur.foreach(config_name,"display", function(s)
  66       page = {}
  67       local columns = s["columns"]
  68       if not columns or columns == "" then
  69         columns = 1
  70       else
  71         columns = tonumber(columns)
  72       end
  73       local pagetitle = s["title"]
  74       if not pagetitle then
  75         pagetitle = title
  76       end
  77       local pageduration = s["duration"]
  78       if not (pagetitle=="") then
  79         line = 2
  80       else
  81         line = 1
  82       end
  83       col = 1
  84       pos = 0
  85       for k,v in pairs(s["parameter"]) do
  86         def = {}
  87         if not(v == "") then
  88           v = split(v,":")
  89           def["sensor"] = v[1]
  90           def["param"] = v[2]
  91           def["label"] = v[3]
  92           def["unit"] = v[4]
  93           def["scale"] = tonumber(v[5])
  94           def["round"] = tonumber(v[6])
  95           def["pos"] = tonumber(v[7])
  96           def["line"] = line
  97           def["col"] = col
  98           page[#page+1] = def
  99         end
 100         col = col + 1
 101         if col > columns then
 102           col = 1
 103           line = line + 1
 104         end
 105       end
 106       pages[#pages+1] = page
 107       pagetitles[#pages] = pagetitle
 108       pagedurations[#pages] = pageduration
 109       pagecolumns[#pages] = columns
 110       pagecolwidth[#pages] = math.floor(width/columns)
 111     end)
 112
 113 end
 114
 115 function connect_server()
 116   local ip = uci.get(config_name,"display","server")
 117   local port = uci.get(config_name,"display","port")
 118   conn = socket.connect(ip,port)
 119   return conn
 120 end
 121
 122 function write_command(conn,command)
 123   conn:send(command.."\n")
 124   local str=conn:receive()
 125   return str
 126 end
 127
 128 function setup_pages(conn)
 129   write_command(conn,"hello")
 130   write_command(conn,"client_set -name weathermon")
 131   for i,page in pairs(pages) do
 132     local pageid = "page"..trim(tostring(i))
 133     write_command(conn,"screen_add "..pageid)
 134     local pagetitle = pagetitles[i]
 135     write_command(conn,"screen_set "..pageid.." -cursor off")
 136     if not(pagetitle == "") then
 137       write_command(conn,"screen_set "..pageid.." -name "..pagetitle)
 138       write_command(conn,"widget_add "..pageid.." "..pageid..".title title")
 139       write_command(conn,"widget_set "..pageid.." "..pageid..".title \"".. pagetitle.."\"")
 140     end
 141     local pageduration = pagedurations[i]
 142     if pageduration and not(pageduration == "") then
 143       write_command(conn,"screen_set "..pageid.." -duration "..pageduration)
 144     end
 145     for j,def in pairs(page) do
 146       defid = def["sensor"].."."..def["param"]
 147       write_command(conn,"widget_add "..pageid.." "..defid.." string")
 148     end
 149   end
 150   if log_con then
 151     for i,page in pairs(pages) do
 152       for j,def in pairs(page) do
 153         pageid = def["sensor"].."."..def["param"]
 154         pagetitle = trim(def["label"])..", "..def["unit"]
 155         write_command(conn,"screen_add "..pageid)
 156         write_command(conn,"screen_set "..pageid.." -cursor off -name "..pagetitle.." -duration "..graph_duration)
 157         write_command(conn,"widget_add "..pageid.." "..pageid..".title title")
 158         if height>2 then
 159           write_command(conn,"widget_add "..pageid.." "..pageid..".max string")
 160           write_command(conn,"widget_add "..pageid.." "..pageid..".min string")
 161           for k = 3,height-1 do
 162             write_command(conn,"widget_add "..pageid.." "..pageid..".place"..trim(tostring(k)).." string")
 163           end
 164         end
 165         write_command(conn,"widget_set "..pageid.." "..pageid..".title \"".. pagetitle.."\"")
 166       end
 167     end
 168   end
 169 end
 170
 171 function process_vals(vals)
 172   for i,page in pairs(pages) do
 173     col_width = pagecolwidth[i]
 174     local pageid = "page"..trim(tostring(i))
 175     for j,def in pairs(page) do
 176       val = vals[def["sensor"]][def["param"]]
 177       if val then
 178         val = round(val * def["scale"], def["round"])
 179       end
 180       defid = def["sensor"].."."..def["param"]
 181       val = lpad(tostring(val),def["pos"]).." "..def["unit"]
 182       if not(def["label"] == "") then
 183         val = def["label"]..": "..val
 184       end
 185       write_command(conn,"widget_set "..pageid.." "..defid.." "..trim(tostring((def["col"]-1)*col_width+1)).." "..def["line"].." \""..val.."\"")
 186     end
 187   end
 188 end
 189
 190 function process_graphs()
 191   for i,page in pairs(pages) do
 192     for j,def in pairs(page) do
 193       local sensor = def["sensor"]
 194       local param  = def["param"]
 195       local pageid = sensor.."."..param
 196       local cur = assert (log_con:execute(string.format("select hour,avg(value) val from (select strftime('%%Y-%%m-%%d %%H',time_stamp)||':00' hour,value from log where sensor='%s' and param='%s') group by hour order by hour desc limit 24",sensor,param)))
 197       local row = cur:fetch ({}, "a")
 198       local vals = {}
 199       local maxval = -99999999
 200       local minval =  99999999
 201       local len
 202       for k=width,1,-1 do
 203         if row then
 204           val = row["val"] * def["scale"]
 205           row = cur:fetch ({}, "a")
 206         else
 207           val = nil
 208         end
 209         vals[k] = val
 210         if val and (val > maxval) then maxval = val; end
 211         if val and (val < minval) then minval = val; end
 212       end
 213       minval = math.floor(minval)
 214       maxval = math.ceil(maxval)
 215       if height>2 then
 216         local minvalstr = trim(tostring(minval))
 217         local maxvalstr = trim(tostring(maxval))
 218         len = math.max(string.len(minvalstr),string.len(maxvalstr))
 219         write_command(conn,"widget_set "..pageid.." "..pageid..".max "..trim(tostring(width-string.len(maxvalstr)+1)).." 2 "..maxvalstr)
 220         write_command(conn,"widget_set "..pageid.." "..pageid..".min "..trim(tostring(width-string.len(minvalstr)+1)).." "..height.." "..minvalstr)
 221         for k = 3,height-1 do
 222           write_command(conn,"widget_set "..pageid.." "..pageid..".place"..trim(tostring(k)).." "..trim(tostring(width-len+1)).." "..k.." \" "..string.rep("-",len-1).."\"")
 223         end
 224       else
 225         len = 0
 226       end
 227       local m = width
 228       for k=1,width do
 229         write_command(conn,"widget_del "..pageid.." "..pageid..".bar"..trim(tostring(k)))
 230       end
 231       for k = width-len,1,-1 do
 232         val = vals[m]
 233         if val then
 234           h = math.floor(0.5+(val-minval)/(maxval-minval)*(height-1)*charheight)
 235         else
 236           h = 0
 237         end
 238         write_command(conn,"widget_add "..pageid.." "..pageid..".bar"..trim(tostring(k)).." vbar")
 239         write_command(conn,"widget_set "..pageid.." "..pageid..".bar"..trim(tostring(k)).." "..k.." "..height.." "..h)
 240         m = m - 1
 241       end
 242     end
 243   end
 244 end
 245
 246 get_display_config()
 247
 248 conn = connect_server()
 249
 250 setup_pages(conn)
 251
 252 while true do
 253
 254   pcall( function ()
 255
 256       vals = process_file()
 257       process_vals(vals)
 258
 259       if log_con then
 260         process_graphs()
 261       end
 262
 263       os.execute("inotifywait -e MODIFY \""..weather_file.."\"")
 264
 265     end)
 266
 267     socket.sleep(3)
 268
 269 end