/*
 * wmtempmon: a sensor monitor for WindowMaker. this little app is mainly based
 * on wmsensormon and other simple dockapps.
 *
 * version = 0.1
 * date: Fri Oct 31 20:41:14 CET 2008 @861 /Internet Time/
 * author: Roman 'gryf' Dobosz
 * requirements: configured lm_sensors and sensor program, nvidia-settings, cut,
 *               grep.
 * licence: gpl
 */

#include <sys/param.h>
#include <sys/types.h>
#include "standards.h"
#include <X11/Xlib.h>
#include <X11/xpm.h>
#include <X11/extensions/shape.h>
#include <string.h>
#include "../wmgeneral/wmgeneral.h"
#include "../wmgeneral/misc.h"
#include "wmtempmon_master2.xpm"
#include "wmtempmon_mask.xbm"
         
#define MAXSTRLEN 8
#define TEMP 40
#define TEMP_OVER 47
#define GPU_T 70
#define GPU_T_OVER 85

void display_values(int, int, int);
int get_temp(int core_number);
int get_offset(int temp, int cpu);

int main(int argc, char **argv){
    int temp1=0, temp2=0, temp3=0;
	/* offset is one of 0 (normal), 7 (alert), 14 (warning) */
    int offset1=0, offset2=0, offset3=0;
	int counter = 0;

    openXwindow(argc, argv, wmtempmon_master2_xpm, wmtempmon_mask_bits, wmtempmon_mask_width, wmtempmon_mask_height);

	while(TRUE){
		if(counter==0){
			temp1 = get_temp(0);
			offset1 = get_offset(temp1, 1);
			temp2 = get_temp(1);
			offset2 = get_offset(temp2, 1);
			temp3 = get_temp(2);
			offset3 = get_offset(temp3, 0);
			counter = 200;
		}
		
		// core 1
		copyXPMArea(0, 87 + offset1, 23, 7, 4, 7); // LCD: "CPU"
		copyXPMArea(69, 87 + offset1, 5, 7, 22, 7); // LCD: number of cpu
		copyXPMArea(66, 65 + offset1, 9, 7, 51, 7); // LCD: "°C"
		display_values(temp1, 0, offset1);

		// core 2
		copyXPMArea(0, 87 + offset2, 23, 7, 4, 21);
		copyXPMArea(75, 87 + offset2, 5, 7, 22, 21);
		copyXPMArea(66, 65 + offset2, 9, 7, 51, 21);
		display_values(temp2, 14, offset2);

		// gpu
		copyXPMArea(23, 87 + offset3, 23, 7, 4, 35);
		copyXPMArea(66, 65 + offset3, 9, 7, 51, 35);
		display_values(temp3, 28, offset3);
		RedrawWindow();
		counter--;
		usleep(5000);
	}
}

int get_offset(int temp, int cpu){
	int alt, wrn;
	if(cpu == 1){
		wrn = TEMP;
		alt = TEMP_OVER;
	}else{
		wrn = GPU_T;
		alt = GPU_T_OVER;
	}
	if(temp >= alt){
		return 7;  // Alert
	}else if(temp >= wrn){
		return 14; // Warning
	}else{
		return 0;  // Normal
	}
}
	

void display_values(int temp, int offset, int offset2){
    char text[5], num1, num2, num3, num4;

    sprintf(text, "%03d", temp);
    num1 = (text[0] - '0');
    num2 = (text[1] - '0');
    num3 = (text[2] - '0');
    num4 = (text[3] - '0');
    if(num1)
		copyXPMArea(5 * num1, 65 + offset2, 5, 7, 31, 7 + offset);
    else
		copyXPMArea(60, 65 + offset2, 5, 7, 31, 7 + offset);
    copyXPMArea(5 * num2, 65 + offset2, 5, 7, 38, 7 + offset);
    copyXPMArea(5 * num3, 65 + offset2, 5, 7, 45, 7 + offset);
    copyXPMArea(5 * num4, 65 + offset2, 5, 7, 51, 7 + offset);
}

int get_temp(int core_number){
	// Core temperature. argument is core number. core no.2 is GPU
	FILE *file;
	int core=0;
	char cmd[] = "                                                                                           ";
	if(core_number==2){
		sprintf(cmd, "echo `nvidia-settings -q 'GPUCoreTemp'|grep Attribute|cut -d ':' -f 3|cut -d '.' -f 1`");
	}else{
		sprintf(cmd, "echo `sensors |grep 'Core %d'|cut -d ':' -f 2|cut -d '.' -f 1`", core_number);
	}
    file = popen(cmd, "r");
    while (! feof(file)) {
        char line[MAXSTRLEN + 1];
        bzero(line, MAXSTRLEN + 1);
        fgets(line, MAXSTRLEN, file);
        if(line[0] != 0){
			sscanf(line, "%d", &core);
        }
    }
	pclose(file);

	return core;
}