# vim: ft=ruby

class GPU2 < Worker
	def mainloop()
		IO.popen("nvidia-smi --query-gpu=temperature.gpu --format=csv,noheader -l 3") do |pipe|
			pipe.each do |line|
            line.chomp!

            t = line.to_i

            if t > @my_config['temp_crit'].to_i
              fg_colour = "crit_colour"
            elsif t > @my_config['temp_warn'].to_i
              fg_colour = "warn_colour"
            else
              fg_colour = "ok_colour"
            end

            rstr = "%%{FC{%s}}%s%%{F-}" % [ fg_colour, t ]
            self.write(rstr)
			end
		end
	end
end

Modules.add("gpu2", "GPU2")