labjack/temp_graph.c at master · josephtoles/labjack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include "temp_graph.h"

//constants
const char* TEMP_ROOT_FILE_NAME = "temp_graph.C";
const int MARKER_STYLES[4] = {20, 20, 20, 20};
const int MARKER_COLORS[4] = {2, 3, 4, 28};

//private functions
int create_temp_script(double x[], double* y[4], int n, int update_delay);
void temp_int_handler(int);
void temp_cleanup();
void add_graph(FILE* f, int index, double x[], double y[], int n);

//variables
int pID;

//The graph will be displayed for a time equal to the update_delay
int flash_temp_animation(double x[], double* y[4], int n, int update_delay)
{
    signal(SIGINT, temp_int_handler);
    int pID = fork();
    if(pID==0)
    {
        create_temp_script(x,y,n, update_delay); //modify to allow multiple arrays
        exit(0);
    }
    return 0;
}

//Constant strings for writing root file
const char TEMP_BEGINNING[] =
    "#include<unistd.h>\n"
    "void temp_graph() {\n"
    "TCanvas *c1 = new TCanvas(\"c1\", \"Temperature\", 10, 40, 700, 450);\n"
    //"c1->SetFillColor(42);\n"
    "c1->SetGrid();\n"
    "TMultiGraph *mg = new TMultiGraph();\n";

    /* root file graphs are inserted here */


const char TEMP_MIDDLE[] =
    //"mg->Draw(\"ap\");\n"
    "mg->Draw(\"ALP\");\n"
    "c1->Update();\n";

    /* root file sleep command goes here */

const char TEMP_END[] =
    "exit();\n"
    "}\n";

void add_graph(FILE* f, int index, double x[], double y[], int n)
{
    fprintf(f, "const Int_t n%d = %d;\n", index, n);
    fprintf(f, "Double_t x%d[] = {", index);
    for(int i=0; i<n; ++i)
    {
        fprintf(f, "%f", x[i]);
        if(i != n-1)
            fprintf(f, ",");
    }
    fprintf(f, "};\n");
    fprintf(f, "Double_t y%d[] = {", index);
    for(int i=0; i<n; ++i)
    {
        fprintf(f, "%f", y[i]);
        if(i != n-1)
            fprintf(f, ",");
    }
    fprintf(f, "};\n");
    fprintf(f, "TGraph *gr%d = new TGraph(n%d,x%d,y%d);\n", index, index, index, index);
    fprintf(f, "gr%d->SetMarkerColor(%d);\n", index, MARKER_COLORS[index]);
    fprintf(f, "gr%d->SetLineColor(%d);\n", index, MARKER_COLORS[index]);
    fprintf(f, "gr%d->SetMarkerStyle(%d);\n", index, MARKER_STYLES[index]);
    fprintf(f, "mg->Add(gr%d);\n", index);
}


int create_temp_script(double x[], double* y[4], int n, int update_delay)
{
    //open file
    FILE *f = fopen(TEMP_ROOT_FILE_NAME, "w");
    if (f == NULL)
    {
            printf("Error opening file!\n");
            return 1;
    }

    //beginning
    fprintf(f, TEMP_BEGINNING);

    //graphs
    for(int i=0; i<4; ++i)
        add_graph(f, i, x, y[i], n);

    //middle
    fprintf(f, TEMP_MIDDLE);

    //sleep
    fprintf(f, "sleep(%d);\n", update_delay);

    //end
    fprintf(f, TEMP_END);
    fclose(f);

    //execute
    char com[100] = "root -l ";
    strcat(com, TEMP_ROOT_FILE_NAME);
    strcat(com, " &> /dev/null");
    system(com);
    temp_cleanup();
    return 0;
}

void temp_int_handler(int sig)
{
    signal(sig, SIG_IGN);
    temp_cleanup();
    kill(pID, 1);
    exit(0);
}

void temp_cleanup()
{
    //uncomment following line after testing
    //unlink(TEMP_ROOT_FILE_NAME);
}