/*
   * svghelper.c - helper functions for outputting svg
   *
   * (C) Copyright 2009 Intel Corporation
   *
   * Authors:
   *     Arjan van de Ven <arjan@linux.intel.com>
   *
   * This program is free software; you can redistribute it and/or
   * modify it under the terms of the GNU General Public License
   * as published by the Free Software Foundation; version 2
   * of the License.
   */

  #include <inttypes.h>

  #include <stdio.h>
  #include <stdlib.h>
  #include <unistd.h>
  #include <string.h>
  
  #include "svghelper.h"
  
  static u64 first_time, last_time;
  static u64 turbo_frequency, max_freq;
  
  
  #define SLOT_MULT 30.0
  #define SLOT_HEIGHT 25.0

  
  int svg_page_width = 1000;

  #define MIN_TEXT_SIZE 0.01

  static u64 total_height;
  static FILE *svgfile;
  
  static double cpu2slot(int cpu)
  {
  	return 2 * cpu + 1;
  }
  
  static double cpu2y(int cpu)
  {
  	return cpu2slot(cpu) * SLOT_MULT;
  }

  static double time2pixels(u64 __time)

  {
  	double X;

  	X = 1.0 * svg_page_width * (__time - first_time) / (last_time - first_time);

  	return X;
  }

  /*
   * Round text sizes so that the svg viewer only needs a discrete
   * number of renderings of the font
   */
  static double round_text_size(double size)
  {
  	int loop = 100;
  	double target = 10.0;
  
  	if (size >= 10.0)
  		return size;
  	while (loop--) {
  		if (size >= target)
  			return target;
  		target = target / 2.0;
  	}
  	return size;
  }

  void open_svg(const char *filename, int cpus, int rows, u64 start, u64 end)

  {

  	int new_width;

  
  	svgfile = fopen(filename, "w");
  	if (!svgfile) {
  		fprintf(stderr, "Cannot open %s for output
  ", filename);
  		return;
  	}

  	first_time = start;
  	first_time = first_time / 100000000 * 100000000;
  	last_time = end;
  
  	/*
  	 * if the recording is short, we default to a width of 1000, but
  	 * for longer recordings we want at least 200 units of width per second
  	 */
  	new_width = (last_time - first_time) / 5000000;
  
  	if (new_width > svg_page_width)
  		svg_page_width = new_width;

  	total_height = (1 + rows + cpu2slot(cpus)) * SLOT_MULT;
  	fprintf(svgfile, "<?xml version=\"1.0\" standalone=\"no\"?> 
  ");

  	fprintf(svgfile, "<svg width=\"%i\" height=\"%" PRIu64 "\" version=\"1.1\" xmlns=\"http://www.w3.org/2000/svg\">
  ", svg_page_width, total_height);

  
  	fprintf(svgfile, "<defs>
    <style type=\"text/css\">
      <![CDATA[
  ");
  
  	fprintf(svgfile, "      rect          { stroke-width: 1; }
  ");
  	fprintf(svgfile, "      rect.process  { fill:rgb(180,180,180); fill-opacity:0.9; stroke-width:1;   stroke:rgb(  0,  0,  0); } 
  ");
  	fprintf(svgfile, "      rect.process2 { fill:rgb(180,180,180); fill-opacity:0.9; stroke-width:0;   stroke:rgb(  0,  0,  0); } 
  ");
  	fprintf(svgfile, "      rect.sample   { fill:rgb(  0,  0,255); fill-opacity:0.8; stroke-width:0;   stroke:rgb(  0,  0,  0); } 
  ");
  	fprintf(svgfile, "      rect.blocked  { fill:rgb(255,  0,  0); fill-opacity:0.5; stroke-width:0;   stroke:rgb(  0,  0,  0); } 
  ");

  	fprintf(svgfile, "      rect.waiting  { fill:rgb(224,214,  0); fill-opacity:0.8; stroke-width:0;   stroke:rgb(  0,  0,  0); } 
  ");

  	fprintf(svgfile, "      rect.WAITING  { fill:rgb(255,214, 48); fill-opacity:0.6; stroke-width:0;   stroke:rgb(  0,  0,  0); } 
  ");

  	fprintf(svgfile, "      rect.cpu      { fill:rgb(192,192,192); fill-opacity:0.2; stroke-width:0.5; stroke:rgb(128,128,128); } 
  ");
  	fprintf(svgfile, "      rect.pstate   { fill:rgb(128,128,128); fill-opacity:0.8; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c1       { fill:rgb(255,214,214); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c2       { fill:rgb(255,172,172); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c3       { fill:rgb(255,130,130); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c4       { fill:rgb(255, 88, 88); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c5       { fill:rgb(255, 44, 44); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      rect.c6       { fill:rgb(255,  0,  0); fill-opacity:0.5; stroke-width:0; } 
  ");
  	fprintf(svgfile, "      line.pstate   { stroke:rgb(255,255,  0); stroke-opacity:0.8; stroke-width:2; } 
  ");
  
  	fprintf(svgfile, "    ]]>
     </style>
  </defs>
  ");
  }
  
  void svg_box(int Yslot, u64 start, u64 end, const char *type)
  {
  	if (!svgfile)
  		return;
  
  	fprintf(svgfile, "<rect x=\"%4.8f\" width=\"%4.8f\" y=\"%4.1f\" height=\"%4.1f\" class=\"%s\"/>
  ",
  		time2pixels(start), time2pixels(end)-time2pixels(start), Yslot * SLOT_MULT, SLOT_HEIGHT, type);
  }

  void svg_sample(int Yslot, int cpu, u64 start, u64 end)

  {
  	double text_size;
  	if (!svgfile)
  		return;

  	fprintf(svgfile, "<rect x=\"%4.8f\" width=\"%4.8f\" y=\"%4.1f\" height=\"%4.1f\" class=\"sample\"/>
  ",
  		time2pixels(start), time2pixels(end)-time2pixels(start), Yslot * SLOT_MULT, SLOT_HEIGHT);

  
  	text_size = (time2pixels(end)-time2pixels(start));
  	if (cpu > 9)
  		text_size = text_size/2;
  	if (text_size > 1.25)
  		text_size = 1.25;

  	text_size = round_text_size(text_size);

  	if (text_size > MIN_TEXT_SIZE)

  		fprintf(svgfile, "<text x=\"%1.8f\" y=\"%1.8f\" font-size=\"%1.8fpt\">%i</text>
  ",

  			time2pixels(start), Yslot *  SLOT_MULT + SLOT_HEIGHT - 1, text_size,  cpu + 1);
  
  }

  static char *time_to_string(u64 duration)
  {
  	static char text[80];
  
  	text[0] = 0;
  
  	if (duration < 1000) /* less than 1 usec */
  		return text;
  
  	if (duration < 1000 * 1000) { /* less than 1 msec */
  		sprintf(text, "%4.1f us", duration / 1000.0);
  		return text;
  	}
  	sprintf(text, "%4.1f ms", duration / 1000.0 / 1000);
  
  	return text;
  }
  
  void svg_waiting(int Yslot, u64 start, u64 end)
  {
  	char *text;
  	const char *style;
  	double font_size;
  
  	if (!svgfile)
  		return;
  
  	style = "waiting";
  
  	if (end-start > 10 * 1000000) /* 10 msec */
  		style = "WAITING";
  
  	text = time_to_string(end-start);

  	font_size = 1.0 * (time2pixels(end)-time2pixels(start));

  	if (font_size > 3)
  		font_size = 3;

  	font_size = round_text_size(font_size);

  	fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\">
  ", time2pixels(start), Yslot * SLOT_MULT);
  	fprintf(svgfile, "<rect x=\"0\" width=\"%4.8f\" y=\"0\" height=\"%4.1f\" class=\"%s\"/>
  ",
  		time2pixels(end)-time2pixels(start), SLOT_HEIGHT, style);

  	if (font_size > MIN_TEXT_SIZE)

  		fprintf(svgfile, "<text transform=\"rotate(90)\" font-size=\"%1.8fpt\"> %s</text>
  ",
  			font_size, text);
  	fprintf(svgfile, "</g>
  ");

  }

  static char *cpu_model(void)
  {
  	static char cpu_m[255];
  	char buf[256];
  	FILE *file;
  
  	cpu_m[0] = 0;
  	/* CPU type */
  	file = fopen("/proc/cpuinfo", "r");
  	if (file) {
  		while (fgets(buf, 255, file)) {
  			if (strstr(buf, "model name")) {
  				strncpy(cpu_m, &buf[13], 255);
  				break;
  			}
  		}
  		fclose(file);
  	}

  
  	/* CPU type */
  	file = fopen("/sys/devices/system/cpu/cpu0/cpufreq/scaling_available_frequencies", "r");
  	if (file) {
  		while (fgets(buf, 255, file)) {
  			unsigned int freq;
  			freq = strtoull(buf, NULL, 10);
  			if (freq > max_freq)
  				max_freq = freq;
  		}
  		fclose(file);
  	}

  	return cpu_m;
  }
  
  void svg_cpu_box(int cpu, u64 __max_freq, u64 __turbo_freq)
  {
  	char cpu_string[80];
  	if (!svgfile)
  		return;
  
  	max_freq = __max_freq;
  	turbo_frequency = __turbo_freq;
  
  	fprintf(svgfile, "<rect x=\"%4.8f\" width=\"%4.8f\" y=\"%4.1f\" height=\"%4.1f\" class=\"cpu\"/>
  ",
  		time2pixels(first_time),
  		time2pixels(last_time)-time2pixels(first_time),
  		cpu2y(cpu), SLOT_MULT+SLOT_HEIGHT);
  
  	sprintf(cpu_string, "CPU %i", (int)cpu+1);

  	fprintf(svgfile, "<text x=\"%4.8f\" y=\"%4.8f\">%s</text>
  ",

  		10+time2pixels(first_time), cpu2y(cpu) + SLOT_HEIGHT/2, cpu_string);

  	fprintf(svgfile, "<text transform=\"translate(%4.8f,%4.8f)\" font-size=\"1.25pt\">%s</text>
  ",

  		10+time2pixels(first_time), cpu2y(cpu) + SLOT_MULT + SLOT_HEIGHT - 4, cpu_model());
  }
  
  void svg_process(int cpu, u64 start, u64 end, const char *type, const char *name)
  {
  	double width;
  
  	if (!svgfile)
  		return;

  
  	fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\">
  ", time2pixels(start), cpu2y(cpu));
  	fprintf(svgfile, "<rect x=\"0\" width=\"%4.8f\" y=\"0\" height=\"%4.1f\" class=\"%s\"/>
  ",
  		time2pixels(end)-time2pixels(start), SLOT_MULT+SLOT_HEIGHT, type);

  	width = time2pixels(end)-time2pixels(start);
  	if (width > 6)
  		width = 6;

  	width = round_text_size(width);

  	if (width > MIN_TEXT_SIZE)

  		fprintf(svgfile, "<text transform=\"rotate(90)\" font-size=\"%3.8fpt\">%s</text>
  ",
  			width, name);
  
  	fprintf(svgfile, "</g>
  ");

  }
  
  void svg_cstate(int cpu, u64 start, u64 end, int type)
  {
  	double width;
  	char style[128];
  
  	if (!svgfile)
  		return;
  
  
  	if (type > 6)
  		type = 6;
  	sprintf(style, "c%i", type);
  
  	fprintf(svgfile, "<rect class=\"%s\" x=\"%4.8f\" width=\"%4.8f\" y=\"%4.1f\" height=\"%4.1f\"/>
  ",
  		style,
  		time2pixels(start), time2pixels(end)-time2pixels(start),
  		cpu2y(cpu), SLOT_MULT+SLOT_HEIGHT);

  	width = (time2pixels(end)-time2pixels(start))/2.0;

  	if (width > 6)
  		width = 6;

  	width = round_text_size(width);

  	if (width > MIN_TEXT_SIZE)

  		fprintf(svgfile, "<text x=\"%4.8f\" y=\"%4.8f\" font-size=\"%3.8fpt\">C%i</text>
  ",
  			time2pixels(start), cpu2y(cpu)+width, width, type);

  }
  
  static char *HzToHuman(unsigned long hz)
  {
  	static char buffer[1024];
  	unsigned long long Hz;
  
  	memset(buffer, 0, 1024);
  
  	Hz = hz;
  
  	/* default: just put the Number in */
  	sprintf(buffer, "%9lli", Hz);
  
  	if (Hz > 1000)
  		sprintf(buffer, " %6lli Mhz", (Hz+500)/1000);
  
  	if (Hz > 1500000)
  		sprintf(buffer, " %6.2f Ghz", (Hz+5000.0)/1000000);
  
  	if (Hz == turbo_frequency)
  		sprintf(buffer, "Turbo");
  
  	return buffer;
  }
  
  void svg_pstate(int cpu, u64 start, u64 end, u64 freq)
  {
  	double height = 0;
  
  	if (!svgfile)
  		return;
  
  	if (max_freq)
  		height = freq * 1.0 / max_freq * (SLOT_HEIGHT + SLOT_MULT);
  	height = 1 + cpu2y(cpu) + SLOT_MULT + SLOT_HEIGHT - height;
  	fprintf(svgfile, "<line x1=\"%4.8f\" x2=\"%4.8f\" y1=\"%4.1f\" y2=\"%4.1f\" class=\"pstate\"/>
  ",
  		time2pixels(start), time2pixels(end), height, height);

  	fprintf(svgfile, "<text x=\"%4.8f\" y=\"%4.8f\" font-size=\"0.25pt\">%s</text>
  ",

  		time2pixels(start), height+0.9, HzToHuman(freq));
  
  }

  void svg_partial_wakeline(u64 start, int row1, char *desc1, int row2, char *desc2)

  {
  	double height;
  
  	if (!svgfile)
  		return;
  
  
  	if (row1 < row2) {

  		if (row1) {

  			fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  				time2pixels(start), row1 * SLOT_MULT + SLOT_HEIGHT,  time2pixels(start), row1 * SLOT_MULT + SLOT_HEIGHT + SLOT_MULT/32);

  			if (desc2)

  				fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\"><text transform=\"rotate(90)\" font-size=\"0.02pt\">%s &gt;</text></g>
  ",

  					time2pixels(start), row1 * SLOT_MULT + SLOT_HEIGHT + SLOT_HEIGHT/48, desc2);
  		}
  		if (row2) {

  			fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  				time2pixels(start), row2 * SLOT_MULT - SLOT_MULT/32,  time2pixels(start), row2 * SLOT_MULT);

  			if (desc1)

  				fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\"><text transform=\"rotate(90)\" font-size=\"0.02pt\">%s &gt;</text></g>
  ",

  					time2pixels(start), row2 * SLOT_MULT - SLOT_MULT/32, desc1);
  		}

  	} else {

  		if (row2) {

  			fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  				time2pixels(start), row2 * SLOT_MULT + SLOT_HEIGHT,  time2pixels(start), row2 * SLOT_MULT + SLOT_HEIGHT + SLOT_MULT/32);

  			if (desc1)

  				fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\"><text transform=\"rotate(90)\" font-size=\"0.02pt\">%s &lt;</text></g>
  ",

  					time2pixels(start), row2 * SLOT_MULT + SLOT_HEIGHT + SLOT_MULT/48, desc1);
  		}
  		if (row1) {

  			fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  				time2pixels(start), row1 * SLOT_MULT - SLOT_MULT/32,  time2pixels(start), row1 * SLOT_MULT);

  			if (desc2)

  				fprintf(svgfile, "<g transform=\"translate(%4.8f,%4.8f)\"><text transform=\"rotate(90)\" font-size=\"0.02pt\">%s &lt;</text></g>
  ",

  					time2pixels(start), row1 * SLOT_MULT - SLOT_HEIGHT/32, desc2);
  		}

  	}
  	height = row1 * SLOT_MULT;
  	if (row2 > row1)
  		height += SLOT_HEIGHT;
  	if (row1)
  		fprintf(svgfile, "<circle  cx=\"%4.8f\" cy=\"%4.2f\" r = \"0.01\"  style=\"fill:rgb(32,255,32)\"/>
  ",
  			time2pixels(start), height);
  }
  
  void svg_wakeline(u64 start, int row1, int row2)
  {
  	double height;
  
  	if (!svgfile)
  		return;
  
  
  	if (row1 < row2)
  		fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  			time2pixels(start), row1 * SLOT_MULT + SLOT_HEIGHT,  time2pixels(start), row2 * SLOT_MULT);
  	else
  		fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%4.2f\" style=\"stroke:rgb(32,255,32);stroke-width:0.009\"/>
  ",
  			time2pixels(start), row2 * SLOT_MULT + SLOT_HEIGHT,  time2pixels(start), row1 * SLOT_MULT);
  
  	height = row1 * SLOT_MULT;
  	if (row2 > row1)
  		height += SLOT_HEIGHT;
  	fprintf(svgfile, "<circle  cx=\"%4.8f\" cy=\"%4.2f\" r = \"0.01\"  style=\"fill:rgb(32,255,32)\"/>
  ",
  			time2pixels(start), height);
  }
  
  void svg_interrupt(u64 start, int row)
  {
  	if (!svgfile)
  		return;
  
  	fprintf(svgfile, "<circle  cx=\"%4.8f\" cy=\"%4.2f\" r = \"0.01\"  style=\"fill:rgb(255,128,128)\"/>
  ",
  			time2pixels(start), row * SLOT_MULT);
  	fprintf(svgfile, "<circle  cx=\"%4.8f\" cy=\"%4.2f\" r = \"0.01\"  style=\"fill:rgb(255,128,128)\"/>
  ",
  			time2pixels(start), row * SLOT_MULT + SLOT_HEIGHT);
  }
  
  void svg_text(int Yslot, u64 start, const char *text)
  {
  	if (!svgfile)
  		return;

  	fprintf(svgfile, "<text x=\"%4.8f\" y=\"%4.8f\">%s</text>
  ",

  		time2pixels(start), Yslot * SLOT_MULT+SLOT_HEIGHT/2, text);
  }
  
  static void svg_legenda_box(int X, const char *text, const char *style)
  {
  	double boxsize;
  	boxsize = SLOT_HEIGHT / 2;
  
  	fprintf(svgfile, "<rect x=\"%i\" width=\"%4.8f\" y=\"0\" height=\"%4.1f\" class=\"%s\"/>
  ",
  		X, boxsize, boxsize, style);

  	fprintf(svgfile, "<text transform=\"translate(%4.8f, %4.8f)\" font-size=\"%4.8fpt\">%s</text>
  ",

  		X + boxsize + 5, boxsize, 0.8 * boxsize, text);
  }
  
  void svg_legenda(void)
  {
  	if (!svgfile)
  		return;
  
  	svg_legenda_box(0,	"Running", "sample");

  	svg_legenda_box(100,	"Idle","c1");
  	svg_legenda_box(200,	"Deeper Idle", "c3");
  	svg_legenda_box(350,	"Deepest Idle", "c6");

  	svg_legenda_box(550,	"Sleeping", "process2");
  	svg_legenda_box(650,	"Waiting for cpu", "waiting");
  	svg_legenda_box(800,	"Blocked on IO", "blocked");
  }

  void svg_time_grid(void)

  {
  	u64 i;

  	if (!svgfile)
  		return;
  
  	i = first_time;
  	while (i < last_time) {
  		int color = 220;
  		double thickness = 0.075;
  		if ((i % 100000000) == 0) {
  			thickness = 0.5;
  			color = 192;
  		}
  		if ((i % 1000000000) == 0) {
  			thickness = 2.0;
  			color = 128;
  		}

  		fprintf(svgfile, "<line x1=\"%4.8f\" y1=\"%4.2f\" x2=\"%4.8f\" y2=\"%" PRIu64 "\" style=\"stroke:rgb(%i,%i,%i);stroke-width:%1.3f\"/>
  ",

  			time2pixels(i), SLOT_MULT/2, time2pixels(i), total_height, color, color, color, thickness);
  
  		i += 10000000;
  	}
  }
  
  void svg_close(void)
  {
  	if (svgfile) {
  		fprintf(svgfile, "</svg>
  ");
  		fclose(svgfile);
  		svgfile = NULL;
  	}
  }