File: gnuplot.cpp

package info (click to toggle)
fityk 1.3.2-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 3,772 kB
  • sloc: cpp: 34,595; ansic: 4,676; python: 963; makefile: 384; sh: 119; xml: 91; java: 31; ruby: 27; perl: 25
file content (141 lines) | stat: -rw-r--r-- 3,898 bytes parent folder | download | duplicates (2)
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
129
130
131
132
133
134
135
136
137
138
139
140
141
// This file is part of fityk program. Copyright 2001-2013 Marcin Wojdyr
// Licence: GNU General Public License ver. 2+

// CLI-only file

#include "gnuplot.h"

#include <stdlib.h>
#include <sys/types.h>
#include <signal.h>
#include <errno.h>
#include <cmath>
#include <string>
#include <algorithm>
#ifndef _WIN32
#include <unistd.h>
#endif

#include "fityk/fityk.h"
#include "fityk/common.h" // is_finite()

#define GNUPLOT_PATH "gnuplot"

using namespace std;
using fityk::Point;
using fityk::is_finite;


GnuPlot::GnuPlot()
    : failed_(false), gnuplot_pipe_(NULL)
{
}

GnuPlot::~GnuPlot()
{
    if (gnuplot_pipe_)
        fclose(gnuplot_pipe_);
}

void GnuPlot::fork_and_make_pipe()
{
#ifndef _WIN32
    signal(SIGPIPE, SIG_IGN);
    int fd[2];
    int ret = pipe(fd);
    if (ret == -1) {
        perror("pipe");
        exit(1);
    }
    pid_t childpid = fork();
    if (childpid == -1) {
        perror("fork");
        exit(1);
    }

    if (childpid == 0) {
        // Child process closes up output side of pipe
        close(fd[1]);
        // and input side is stdin
        dup2(fd[0], 0);
        if (fd[0] > 2)
            close(fd[0]);
        //putenv("PAGER=");
        execlp(GNUPLOT_PATH, GNUPLOT_PATH, /*"-",*/ (char*) NULL);
        // if we are here, sth went wrong
        fprintf(stderr, "** Calling `" GNUPLOT_PATH
                        "' failed. Plotting disabled. **\n");
        abort(); // exit() messed with open files.
    } else {
        // Parent process closes up input side of pipe
        close(fd[0]);
        gnuplot_pipe_ = fdopen(fd[1], "w"); //fdopen() - POSIX, not ANSI
    }
#endif //!_WIN32
}

bool GnuPlot::test_gnuplot_pipe()
{
#ifdef _WIN32
    return false;
#else //!_WIN32
    errno = 0;
    fprintf(gnuplot_pipe_, " "); //pipe test
    fflush(gnuplot_pipe_);
    if (errno != 0) // errno == EPIPE if the pipe doesn't work
        failed_ = true;
    return errno == 0;
#endif //_WIN32
}

void GnuPlot::plot()
{
    // plot only the active dataset and model
    int dm_number = ftk->get_default_dataset();
    const vector<Point>& points = ftk->get_data(dm_number);
    double left_x = ftk->get_view_boundary('L');
    double right_x = ftk->get_view_boundary('R');
    vector<Point>::const_iterator begin
        = lower_bound(points.begin(), points.end(), Point(left_x,0));
    vector<Point>::const_iterator end
        = upper_bound(points.begin(), points.end(), Point(right_x,0));
    bool has_points = (begin != end);

    if (gnuplot_pipe_ == NULL && has_points)
        fork_and_make_pipe();

    if (gnuplot_pipe_ == NULL || failed_ || !test_gnuplot_pipe())
        return;

    // send "plot ..." through the pipe to gnuplot
    string plot_string = "plot " + ftk->get_info("view")
        + " \'-\' title \"data\", '-' title \"sum\" with line\n";
    fprintf(gnuplot_pipe_, "%s", plot_string.c_str());
    if (fflush(gnuplot_pipe_) != 0)
        fprintf(stderr, "Flushing pipe program-to-gnuplot failed.\n");

    // data
    if (has_points) {
        for (vector<Point>::const_iterator i = begin; i != end; ++i)
            if (i->is_active && is_finite(i->x) && is_finite(i->y))
                fprintf(gnuplot_pipe_, "%f  %f\n", double(i->x), double(i->y));
    } else
        // if there are no points, we send empty dataset to reset the plot
        fprintf(gnuplot_pipe_, "0.0  0.0\n");
    fprintf(gnuplot_pipe_, "e\n");//gnuplot needs 'e' at the end of data

    // model
    if (has_points) {
        for (vector<Point>::const_iterator i = begin; i != end; ++i)
            if (i->is_active && is_finite(i->x)) {
                double y = ftk->get_model_value(i->x, dm_number);
                if (is_finite(y))
                    fprintf(gnuplot_pipe_, "%f  %f\n", double(i->x), y);
            }
    } else
        fprintf(gnuplot_pipe_, "0.0  0.0\n");
    fprintf(gnuplot_pipe_, "e\n");

    fflush(gnuplot_pipe_);
}