/*
 * wmtempnv: a sensor monitor for WindowMaker. this little app is mainly based
 * on wmsensormon and other simple dockapps.
 *
 * version = 0.4
 *
 * requirements: configured lm_sensors, sensor program and
 *               nvidia-settings package
 * 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 "../wmgeneral/misc.h"
#include "wmtempnv_master2.xpm"
#include "wmtempnv_mask.xbm"

#include <NVCtrl/NVCtrl.h>
#include <NVCtrl/NVCtrlLib.h>

#define MAXSTRLEN 8
#define TEMP 40
#define TEMP_OVER 47
#define GPU_T 70
#define GPU_T_OVER 85
#define MAXFNAME 50

void display_values(int, int, int);
int get_temp(int core_number, Display*);
int get_offset(int temp, int cpu);
void display_help(char* progname);
int get_gpu_temp(char* path, Display *disp);
void read_file_into(char *filepath, int *output);
Display *display;

int main(int argc, char **argv){
    short got_path=0;
    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;
    char* path = "";
    display = XOpenDisplay(NULL);

    if (argc > 1) {
        if (argc > 2) {
            display_help(argv[0]);
            exit(2);
        }
        
        if (argc == 2 &&
                (strcmp(argv[1], "--help") == 0 ||
                 strcmp(argv[1], "-h") == 0)) {
            display_help(argv[0]);
            exit(0);
        }

        got_path = 1;
        path = argv[1];
    }

    openXwindow(argc, argv, wmtempnv_master2_xpm, wmtempnv_mask_bits,
            wmtempnv_mask_width, wmtempnv_mask_height);

    while(TRUE){
        if (counter < 1){
            counter = 5;
            temp1 = get_temp(0, display);
            offset1 = get_offset(temp1, 1);
            temp2 = get_temp(1, display);
            offset2 = get_offset(temp2, 1);
            temp3 = get_gpu_temp(path, display);
            offset3 = get_offset(temp3, 0);
        }

        // 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(100000);
    }
}

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, Display *disp){
    // Core temperature. argument is core number.
    FILE *file;
    char filename[MAXFNAME];
    int core_temp = 0;
	Bool res;

    snprintf(filename, MAXFNAME,
            "/sys/bus/platform/devices/coretemp.0/temp%d_input",
            core_number + 2);
    read_file_into(filename, &core_temp);
    return core_temp;
}

int get_gpu_temp(char* path, Display *disp){
    // return GPU temperature. Argument is path in sysfs or empty string.
    FILE *file;
    char filename[MAXFNAME];
    int gpu_temp = 0;
	Bool res;

    if (strcmp(path, "") == 0){
		res = XNVCTRLQueryTargetAttribute(disp,
				NV_CTRL_TARGET_TYPE_GPU, 0, 0,
				NV_CTRL_GPU_CORE_TEMPERATURE, &gpu_temp);

		if (res == False) gpu_temp = 0;
    }else{
        read_file_into(path, &gpu_temp);
    }
    return gpu_temp;
}

void read_file_into(char *filepath, int *output) {
    // Read an integer from the provided filepath and write it in the address
    FILE *fp;
    if((fp = fopen(filepath, "r")) != NULL){
        if(fscanf(fp, "%d", output) != EOF){
            *output = *output / 1000;
            fclose(fp);
        }
    }
}

void display_help(char* progname){
    printf("Dockapp for monitoring CPU and Nvidia GPU temperatures.\n");
    printf("Usage:\n\t%s [full path for temp in sysfs]\n\n", progname);
    printf("As an optional parameter you can provide `temp_input' ");
    printf("full path from sysfs,\ncorresponding to your gfx card ");
    printf("to read temperature from. Otherwise\nfunctionality of ");
    printf("nv-control will be used (package nvidia-settings).\n");
}