#AutoIt3Wrapper_Icon=temperature-2-multi-size.ico
#RequireAdmin
#NoTrayIcon
#include <Temperature.au3>
#include <FFLabels.au3>
#include <GUIConstantsEx.au3>
#include <StaticConstants.au3>
#include <WindowsConstants.au3>

Global $winLogger = "C:\WINDOWS\system32\eventcreate.exe" ; application to use for event logging
Global $MessageSource = "HP TempMon"
Global $width = 185
Global $height = 15
Global $FontSize = 8
Global $FontFamily = 'Microsoft Sans Serif'
Global $normClr = 0xFF000000
Global $warnClr = 0xFF0000
Global $WMInameSpace = "HP_BIOSNumericSensor"
;~ Global $WMIReadSensor = "Chassis Thermal Index" ; HP 8300
Global $WMIReadSensor = "System Ambient Temperature" ; HP Z400
Global $warnTmp = 46.1111111111 ; degrees C
;~ Global $warnTmp = 24 ; degrees C for testing
Global $ovrtmpTime = 5 ; in minutes
Global $hGUI = GUICreate("Board Temperature", $width, $height, -1, 0, BitOR($WS_SYSMENU, $WS_POPUP), BitOR($WS_EX_TOPMOST, $WS_EX_WINDOWEDGE, $WS_EX_TOOLWINDOW))
Global $lb1 = _GUICtrlFFLabel_Create($hGUI, "", -1, -1, $width, $height, 8, -1, 0, 1, $normClr)
GUISetState(@SW_SHOW)

_TempSenseLoop()

Func _TempSenseLoop()
    While 1
        Sleep(250)
        $nMsg = GUIGetMsg()
        Switch $nMsg
            Case $GUI_EVENT_CLOSE
                GUIDelete($hGUI)
                Return
            Case Else
                $s_ambTempC = _doQueryHPNumSens(5, $WMInameSpace, $WMIReadSensor, "100") ; + 100
                Select
                    Case $s_ambTempC = ""
                        _GUICtrlFFLabel_SetTextColor($lb1, $normClr)
                        _GUICtrlFFLabel_SetData($lb1, "Unable to query temperature")
                    Case $s_ambTempC < $warnTmp
                        _GUICtrlFFLabel_SetTextColor($lb1, $normClr)
                        _GUICtrlFFLabel_SetData($lb1, "Board temperature: " & $s_ambTempC & "C" & "/" & Round(_CelsiusToFahrenheit($s_ambTempC), 0) & "F")
                    Case Else
                        _GUICtrlFFLabel_SetTextColor($lb1, $warnClr)
                        _GUICtrlFFLabel_SetData($lb1, "Board temperature: " & $s_ambTempC & "C" & "/" & Round(_CelsiusToFahrenheit($s_ambTempC), 0) & "F")
                        _doLog("System ambient temperature has exceeded " & $warnTmp & "C.", "WARN", $MessageSource)
                        _AlarmMonitor($ovrtmpTime, $warnTmp)
                EndSelect
        EndSwitch
    WEnd
EndFunc   ;==>_TempSenseLoop

Func _AlarmMonitor($min, $tmp)
    Local $Shutdown = 1
    Local $s_ambTemp
    Local $freq = 3500
    Local $shrt = 200
    Local $long = 500
    Local $timer = TimerInit()
    Local $wait = 1000 * ($min * 60)
    While TimerDiff($timer) < $wait
        $s_ambTemp = _doQueryHPNumSens(5, $WMInameSpace, $WMIReadSensor, "100") ; + 100
        If $s_ambTemp >= $tmp Then
            _GUICtrlFFLabel_SetData($lb1, "WARNING! OVERTEMP!")
            Beep($freq, $shrt)
            Beep($freq, $shrt)
            Beep($freq, $shrt)
            Sleep(250)
            Beep($freq, $long)
            Beep($freq, $long)
            Beep($freq, $long)
            Sleep(250)
            Beep($freq, $shrt)
            Beep($freq, $shrt)
            Beep($freq, $shrt)
            _GUICtrlFFLabel_SetData($lb1, "Board temperature: " & $s_ambTemp & "C" & "/" & _CelsiusToFahrenheit($s_ambTemp) & "F")
            Sleep(2000)
        ElseIf $s_ambTemp < $tmp Then
            $Shutdown = 0
            ExitLoop
        EndIf
    WEnd
    If $Shutdown = 1 Then
        _doLog("The system is shutting down due to overtemperature conditions. The temperature at the time of shutdown was " & $s_ambTemp & "C" & "/" & _CelsiusToFahrenheit($s_ambTemp) & "F", "ERRR", $MessageSource)
        Shutdown(24)
    ElseIf $Shutdown = 0 Then
        _doLog("System ambient temperature has been reduced below " & $warnTmp & "C.", "INFO", $MessageSource)
    EndIf
EndFunc   ;==>_AlarmMonitor

Func _doQueryHPNumSens($min, $class, $name, $sleep) ; $sleep = milliseconds
    Local $s_tempReading = ""
    Local $s_HPBiosWMIService = "winmgmts:\\" & @ComputerName & "\Root\HP\InstrumentedBIOS"
    Local $objHPBiosWMIService = ObjGet($s_HPBiosWMIService)
    Local $s_biosQuery = $objHPBiosWMIService.ExecQuery("select * from " & $class & " where Name='" & $name & "'")

    $timer = TimerInit() ; we may need to wait for the HP WMI extensions to enumerate in WMI, in WinPE this takes some minutes.
    $wait = 1000 * ($min * 60)

    If Not IsObj($s_biosQuery) Then ; we do need to wait, put up a splash screen
        $s_WMISplash = 1
        SplashTextOn("WMI", "Probing WMI (up to 10 minutes)...", 300, 50, -1, -1, 1)
        Sleep(1000)
        While TimerDiff($timer) < $wait
            Sleep(500)
            $objHPBiosWMIService = ObjGet($s_HPBiosWMIService)
            If IsObj($objHPBiosWMIService) Then
                Sleep(500)
                $colProdName = $s_biosQuery
                If IsObj($colProdName) Then
                    For $oItem In $colProdName
                        $s_tempReading = $oItem.CurrentReading
                    Next
                EndIf
            EndIf
        WEnd
    ElseIf IsObj($s_biosQuery) Then
        ConsoleWrite($s_biosQuery & @CRLF)
        For $oItem In $s_biosQuery
            $s_tempReading = $oItem.CurrentReading
        Next
    Else
        $s_tempReading = 0
    EndIf
    Sleep($sleep)
    Return $s_tempReading
EndFunc   ;==>_doQueryHPNumSens

Func _doLog($message, $type, $msgsrc)
    Select
        Case $type = "INFO"
            $logTyp = "INFORMATION"
        Case $type = "WARN"
            $logTyp = "WARNING"
        Case $type = "ERRR"
            $logTyp = "ERROR"
    EndSelect

    If FileExists($winLogger) Then
        RunWait(@ComSpec & " /c " & $winLogger & " /L Application /T " & $logTyp & " /SO " & Chr(34) & $msgsrc & Chr(34) & " /ID 1000 /D " & Chr(34) & $message & Chr(34), "", @SW_HIDE)
    EndIf
EndFunc   ;==>_doLog