/*
 * Copyright (c) 2017 The DragonFly Project.  All rights reserved.
 *
 * This code is derived from software contributed to The DragonFly Project
 * by Matthew Dillon <dillon@backplane.com>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "kcollect.h"

void
start_gnuplot(int ac __unused, char **av __unused, const char *plotfile)
{
	OutFP = popen("gnuplot", "w");
	if (OutFP == NULL) {
		fprintf(stderr, "can't find gnuplot\n");
		exit(1);
	}

	/*
	 * If plotfile is specified allow .jpg or .JPG or .png or .PNG
	 */
	if (plotfile) {
		const char *ext;

		if ((ext = strrchr(plotfile, '.')) == NULL) {
			ext = "";
		} else {
			++ext;
		}
		if (strcmp(ext, "jpg") == 0 ||
		    strcmp(ext, "JPG") == 0) {
			fprintf(OutFP, "set terminal jpeg size %d,%d\n",
				OutputWidth, OutputHeight);
		} else if (strcmp(ext, "png") == 0 ||
			   strcmp(ext, "PNG") == 0) {
			fprintf(OutFP, "set terminal png size %d,%d\n",
				OutputWidth, OutputHeight);
		} else {
			fprintf(stderr, "plotfile must be .jpg or .png\n");
			exit(1);
		}
		fprintf(OutFP, "set output \"%s\"\n", plotfile);
	} else {
		fprintf(OutFP, "set terminal x11 persist size %d,%d\n",
			OutputWidth, OutputHeight);
	}
}

void
dump_gnuplot(kcollect_t *ary, size_t count)
{
	int plot1[] = { KCOLLECT_MEMFRE, KCOLLECT_MEMCAC,
			KCOLLECT_MEMINA, KCOLLECT_MEMACT,
			KCOLLECT_MEMWIR, KCOLLECT_LOAD };
	int plot2[] = { KCOLLECT_IDLEPCT, KCOLLECT_INTRPCT,
			KCOLLECT_SYSTPCT, KCOLLECT_USERPCT,
			KCOLLECT_SWAPPCT,
			KCOLLECT_VMFAULT, -1/*KCOLLECT_SYSCALLS*/,
			KCOLLECT_NLOOKUP };
	const char *id1[] = {
			"free", "cache",
			"inact", "active",
			"wired", "load" };
	const char *id2[] = {
			"idle", "intr", "system", "user",
			"swap",
			"faults", "syscalls", "nlookups" };
	struct tm *tmv;
	char buf[64];
	uint64_t value;
	time_t t;
	double dv;
	double smoothed_dv;
	int i;
	int j;
	int jj;
	int k;

	/*
	 * NOTE: be sure to reset any fields adjusted by the second plot,
	 *	 in case we are streaming plots with -f.
	 */
	fprintf(OutFP, "set xdata time\n");
	fprintf(OutFP, "set timefmt \"%%d-%%b-%%Y %%H:%%M:%%S\"\n");
	fprintf(OutFP, "set style fill solid 1.0\n");
	fprintf(OutFP, "set multiplot layout 2,1\n");
	fprintf(OutFP, "set key outside\n");
	fprintf(OutFP, "set lmargin 10\n");
	fprintf(OutFP, "set rmargin 25\n");
	fprintf(OutFP, "set xtics rotate\n");
	fprintf(OutFP, "set format x '%%H:%%M'\n");

	fprintf(OutFP, "set ylabel \"GB\"\n");

	fprintf(OutFP, "set yrange [0:%d]\n",
		(int)((KCOLLECT_GETSCALE(ary[0].data[KCOLLECT_MEMFRE]) +
		       999999) / 1000000000));
	fprintf(OutFP, "set autoscale y2\n");
	fprintf(OutFP, "set y2label \"Load\"\n");
	fprintf(OutFP, "set ytics nomirror\n");
	fprintf(OutFP, "set y2tics nomirror\n");

	fprintf(OutFP,
		"plot "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 axes x1y2 title \"%s\" with lines lw 1\n",
		id1[0], id1[1], id1[2], id1[3], id1[4], id1[5]);

	for (jj = 0; jj < (int)(sizeof(plot1) / sizeof(plot1[0])); ++jj) {
		j = plot1[jj];

		smoothed_dv = 0.0;
		for (i = count - 1; i >= 2; --i) {
			/*
			 * Timestamp
			 */
			t = ary[i].realtime.tv_sec;
			if (t < 1000)
				continue;
			if (UseGMT)
				tmv = gmtime(&t);
			else
				tmv = localtime(&t);
			strftime(buf, sizeof(buf), "%d-%b-%Y %H:%M:%S", tmv);
			value = ary[i].data[j];
			if (jj <= 4) {
				for (k = jj + 1; k <= 4; ++k)
					value += ary[i].data[plot1[k]];
				dv = (double)value / 1e9;
			} else {
				dv = (double)value / 100.0;
			}
			if (SmoothOpt) {
				if (i == (int)(count - 1)) {
					smoothed_dv = dv;
				} else if (smoothed_dv < dv) {
					smoothed_dv =
					 (smoothed_dv * 5.0 + 5 * dv) /
					 10.0;
				} else {
					smoothed_dv =
					 (smoothed_dv * 9.0 + 1 * dv) /
					 10.0;
				}
				dv = smoothed_dv;
			}
			fprintf(OutFP, "%s %6.2f\n", buf, dv);
		}
		fprintf(OutFP, "e\n");
	}

	int ncpu = 4;
	size_t ncpu_size = sizeof(ncpu);
	sysctlbyname("hw.ncpu", &ncpu, &ncpu_size, NULL, 0);

	fprintf(OutFP, "set ylabel \"Cpu Utilization\"\n");
	fprintf(OutFP, "set y2label \"MOps/sec (smoothed)\"\n");
	fprintf(OutFP, "set ytics nomirror\n");
	fprintf(OutFP, "set y2tics nomirror\n");
	fprintf(OutFP, "set yrange [0:105]\n");
	fprintf(OutFP, "set y2range [0:%d.0]\n",
		(93750 * ncpu + 999999) / 1000000);

	fprintf(OutFP,
		"plot "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with boxes lw 1, "
		"\"-\" using 1:3 title \"%s\" with lines lw 1, "
		"\"-\" using 1:3 axes x1y2 title \"%s\" with lines lw 1, "
		"\"-\" using 1:3 axes x1y2 title \"%s\" with lines lw 1, "
		"\"-\" using 1:3 axes x1y2 title \"%s\" with lines lw 1\n",
		id2[0], id2[1], id2[2], id2[3], id2[4], id2[5], id2[6], id2[7]);

	for (jj = 0; jj < (int)(sizeof(plot2) / sizeof(plot2[0])); ++jj) {
		j = plot2[jj];

		/*
		 * Skip if disabled
		 */
		if (j < 0) {
			fprintf(OutFP, "e\n");
			continue;
		}

		/*
		 * Dump points
		 */
		smoothed_dv = 0.0;
		for (i = count - 1; i >= 2; --i) {
			/*
			 * Timestamp
			 */
			t = ary[i].realtime.tv_sec;
			if (t < 1000)
				continue;
			if (UseGMT)
				tmv = gmtime(&t);
			else
				tmv = localtime(&t);
			strftime(buf, sizeof(buf), "%d-%b-%Y %H:%M:%S", tmv);
			value = ary[i].data[j];

			if (jj <= 3) {
				/*
				 * intr/sys/user/idle percentages
				 */
				for (k = jj + 1; k <= 3; ++k)
					value += ary[i].data[plot2[k]];
				dv = (double)value / 100.0;
				if (SmoothOpt) {
					if (i == (int)(count - 1)) {
						smoothed_dv = dv;
					} else if (smoothed_dv < dv) {
						smoothed_dv =
						 (smoothed_dv * 5.0 + 5 * dv) /
					         10.0;
					} else {
						smoothed_dv =
						 (smoothed_dv * 9.0 + 1 * dv) /
					         10.0;
					}
					dv = smoothed_dv;
				}
			} else {
				if (jj >= 5) {
					/* fault counters */
					dv = (double)value / KCOLLECT_INTERVAL;
					dv = dv / 1e6;
				} else {
					/* swap percentage (line graph) */
					dv = (double)value / 100.0;
				}
				if (i == (int)(count - 1)) {
					smoothed_dv = dv;
				} else if (smoothed_dv < dv) {
					smoothed_dv =
					 (smoothed_dv * 5.0 + 5 * dv) /
					 10.0;
				} else {
					smoothed_dv = (smoothed_dv * 9.0 + dv) /
						      10.0;
				}
				dv = smoothed_dv;
			}
			fprintf(OutFP, "%s %6.2f\n", buf, dv);
		}
		fprintf(OutFP, "e\n");
	}
	fflush(OutFP);
}