misc/plot_stuff

   1 #!/usr/bin/env python
   2
   3 import cairo;
   4 import os;
   5 import locale;
   6 import Numeric;
   7 import math;
   8
   9
  10
  11 class Plot:
  12         def __init__(self, width, height):
  13                 self.width=width;
  14                 self.height=height;
  15
  16                 self.margin_top=20;
  17                 self.margin_bottom=40;
  18                 self.margin_left=40;
  19                 self.margin_right=20;
  20
  21                 self.range_xmin=0.0;
  22                 self.range_xmax=1.0;
  23                 self.range_ymin=0.0;
  24                 self.range_ymax=1.0;
  25
  26                 self.symbol_size = 2;
  27                 self.tick_size = 5;
  28
  29                 self.with_lines = 1;
  30
  31                 self.surface = cairo.ImageSurface (cairo.FORMAT_ARGB32,
  32                                 self.width, self.height);
  33                 self.ctx = cairo.Context (self.surface);
  34
  35         def set_x_range (self, xmin, xmax):
  36                 if xmin == xmax:
  37                         print "Invalid x range";
  38                         xmin -= 1;
  39                         xmax += 1;
  40                 self.range_xmin=float(xmin);
  41                 self.range_xmax=float(xmax);
  42
  43         def set_y_range (self, ymin, ymax):
  44                 print "y range %g %g" % (ymin, ymax);
  45                 if ymin == ymax:
  46                         print "Invalid y range";
  47                         ymin -= 1;
  48                         ymax += 1;
  49                 self.range_ymin=float(ymin);
  50                 self.range_ymax=float(ymax);
  51
  52         def set_x_range_data (self, data):
  53                 xmin = min(data);
  54                 xmax = max(data);
  55                 self.set_x_range(xmin,xmax);
  56
  57         def set_y_range_data (self, data):
  58                 ymin = min(data);
  59                 ymax = max(data);
  60                 self.set_y_range(ymin,ymax);
  61
  62         def set_sane_ticks (self):
  63                 logrange = math.log10((self.range_xmax - self.range_xmin)/3);
  64                 tick = math.pow(10, math.floor(logrange));
  65                 self.tick_major_x = tick;
  66
  67                 logrange = math.log10((self.range_ymax - self.range_ymin)/3);
  68                 tick = math.pow(10, math.floor(logrange));
  69                 self.tick_major_y = tick;
  70
  71                 print "tick %g %g" % (self.tick_major_x, self.tick_major_y);
  72
  73         def draw_border (self):
  74                 self.ctx.move_to (self.margin_left, self.margin_top);
  75                 self.ctx.line_to (self.width - self.margin_right, self.margin_top);
  76                 self.ctx.line_to (self.width - self.margin_right, self.height - self.margin_bottom);
  77                 self.ctx.line_to (self.margin_left, self.height - self.margin_bottom);
  78                 self.ctx.close_path ();
  79
  80                 self.ctx.set_source_rgb (0, 0, 0);
  81                 self.ctx.stroke();
  82
  83         def marker (self, x, y):
  84                 self.ctx.move_to (x, y);
  85                 self.ctx.save ();
  86                 self.ctx.identity_matrix ();
  87                 self.ctx.rel_move_to (-self.symbol_size, 0);
  88                 self.ctx.rel_line_to (self.symbol_size*2, 0);
  89                 self.ctx.rel_move_to (-self.symbol_size, -self.symbol_size);
  90                 self.ctx.rel_line_to (0, self.symbol_size*2);
  91                 self.ctx.restore ();
  92
  93         def draw_plot (self, points):
  94                 self.ctx.save();
  95
  96                 self.ctx.translate (self.margin_left,
  97                                 self.height - self.margin_bottom);
  98                 self.ctx.scale ((self.width - self.margin_right - self.margin_left)/
  99                                 (self.range_xmax - self.range_xmin),
 100                                 -(self.height - self.margin_top - self.margin_bottom)/
 101                                 (self.range_ymax - self.range_ymin));
 102                 self.ctx.translate (-self.range_xmin, -self.range_ymin);
 103
 104                 if (self.with_lines):
 105                         self.ctx.move_to (points[0][0], points[0][1]);
 106                         for point in points[1:]:
 107                                 self.ctx.line_to (point[0], point[1]);
 108                 else:
 109                         for point in points:
 110                                 self.marker (point[0], point[1]);
 111
 112                 self.ctx.restore();
 113                 self.ctx.stroke();
 114
 115         def vtick (self, y):
 116                 self.ctx.move_to (self.range_xmin, y);
 117                 self.ctx.save ();
 118                 self.ctx.identity_matrix ();
 119                 self.ctx.rel_move_to (0, 0);
 120                 self.ctx.rel_line_to (-self.tick_size, 0);
 121                 self.ctx.restore ();
 122
 123         def htick (self, x):
 124                 self.ctx.move_to (x, self.range_ymin);
 125                 self.ctx.save ();
 126                 self.ctx.identity_matrix ();
 127                 self.ctx.rel_move_to (0, 0);
 128                 self.ctx.rel_line_to (0, self.tick_size);
 129                 self.ctx.restore ();
 130
 131         def vlabel (self, y):
 132                 self.ctx.move_to (self.range_xmin, y);
 133                 self.ctx.save ();
 134                 self.ctx.identity_matrix ();
 135                 self.ctx.rel_move_to (-self.tick_size * 6, 0);
 136                 self.ctx.show_text("%g" % y);
 137                 self.ctx.restore ();
 138
 139         def hlabel (self, x):
 140                 self.ctx.move_to (x, self.range_ymin);
 141                 self.ctx.save ();
 142                 self.ctx.identity_matrix ();
 143                 self.ctx.rel_move_to (0, self.tick_size * 4);
 144                 self.ctx.show_text("%g" % x);
 145                 self.ctx.restore ();
 146
 147         def draw_ticks (self):
 148                 self.ctx.save ();
 149                 self.ctx.translate (self.margin_left, self.height - self.margin_bottom);
 150                 self.x_scale = (self.width - self.margin_right - self.margin_left)/(self.range_xmax - self.range_xmin);
 151                 self.y_scale = (self.height - self.margin_top - self.margin_bottom)/(self.range_ymax - self.range_ymin);
 152                 print "scale %g %g" % (self.x_scale, self.y_scale);
 153                 self.ctx.scale (self.x_scale, -self.y_scale);
 154                 self.ctx.translate (-self.range_xmin, -self.range_ymin);
 155
 156                 x = self.range_ymin;
 157                 while (x <= self.range_ymax):
 158                         self.vtick(x);
 159                         x+=self.tick_major_y;
 160                 x = self.range_xmin;
 161                 while (x <= self.range_xmax):
 162                         self.htick(x);
 163                         x+=self.tick_major_x;
 164
 165                 x = self.range_ymin;
 166                 while (x <= self.range_ymax):
 167                         self.vlabel(x);
 168                         x+=self.tick_major_y;
 169                 x = self.range_xmin;
 170                 while (x <= self.range_xmax):
 171                         self.hlabel(x);
 172                         x+=self.tick_major_x;
 173
 174                 self.ctx.restore();
 175                 self.ctx.set_source_rgb (0, 0, 0);
 176                 self.ctx.stroke();
 177
 178 def readdata(filename):
 179         file = open(filename);
 180
 181         points = [];
 182         for line in file.readlines():
 183                 points.append(map(float,line.split()));
 184
 185         file.close();
 186         return points;
 187
 188
 189 def plot_mse():
 190         plot = Plot(800,600);
 191         plot.set_x_range_data(tpts[0]);
 192         m = max(Numeric.transpose(tpts[14]));
 193         #plot.set_x_range(0, m);
 194         plot.set_y_range(0, m);
 195
 196         plot.set_sane_ticks();
 197
 198         plot.ctx.set_source_rgb (1, 1, 1);
 199         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 200         plot.ctx.paint();
 201
 202         plot.draw_border ();
 203         plot.draw_ticks ();
 204
 205         plot.ctx.set_line_width(0.5);
 206
 207         plot.ctx.set_source_rgb (1, 0, 0);
 208         plot.draw_plot (Numeric.transpose(tpts[0:15:14]));
 209
 210         plot.ctx.set_source_rgb (0, 1, 0);
 211         plot.draw_plot (Numeric.transpose(tpts[0:16:15]));
 212
 213         #plot.with_lines = 0;
 214         #plot.draw_plot (Numeric.transpose(tpts[14:16:1]));
 215
 216         plot.surface.write_to_png('00000-mse.png');
 217
 218 def plot_buffer_level():
 219         plot = Plot(800,600);
 220         plot.set_x_range_data(tpts[0]);
 221         m = max(Numeric.transpose(tpts[11]));
 222         plot.set_y_range(0, m);
 223
 224         plot.set_sane_ticks();
 225
 226         plot.ctx.set_source_rgb (1, 1, 1);
 227         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 228         plot.ctx.paint();
 229
 230         plot.draw_border ();
 231         plot.draw_ticks ();
 232
 233         plot.ctx.set_line_width(0.5);
 234
 235         plot.ctx.set_source_rgb (1, 0, 0);
 236         plot.draw_plot (Numeric.transpose(tpts[0:12:11]));
 237
 238         plot.surface.write_to_png('00000-buffer-level.png');
 239
 240 def plot_bits_per_pic():
 241         plot = Plot(800,600);
 242         plot.set_x_range_data(tpts[0]);
 243
 244         data_x = Numeric.transpose(tpts[0]);
 245         data_y1 = Numeric.transpose(tpts[8]);
 246         data_y2 = Numeric.transpose(tpts[9]);
 247         data_y = Numeric.transpose(tpts[8]+tpts[9]);
 248         m = max(Numeric.transpose(data_y));
 249         plot.set_y_range(0, m);
 250
 251         plot.set_sane_ticks();
 252
 253         plot.ctx.set_source_rgb (1, 1, 1);
 254         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 255         plot.ctx.paint();
 256
 257         plot.draw_border ();
 258         plot.draw_ticks ();
 259
 260         plot.ctx.set_line_width(0.5);
 261
 262         plot.ctx.set_source_rgb (1, 0, 0);
 263         plot.with_lines = 0;
 264         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 265
 266         plot.ctx.set_source_rgb (0, 1, 0);
 267         plot.with_lines = 0;
 268         plot.draw_plot (Numeric.transpose([data_x,data_y1]));
 269
 270         plot.ctx.set_source_rgb (0, 0, 1);
 271         plot.with_lines = 0;
 272         plot.draw_plot (Numeric.transpose([data_x,data_y2]));
 273
 274         plot.ctx.set_source_rgb (0.5, 0.5, 0.5);
 275         plot.with_lines = 0;
 276         plot.draw_plot (Numeric.transpose([data_x,Numeric.transpose(tpts[3])]));
 277
 278         plot.ctx.set_source_rgb (0.5, 0.5, 0.5);
 279         plot.with_lines = 0;
 280         plot.draw_plot (Numeric.transpose([data_x,Numeric.transpose(tpts[4])]));
 281
 282         plot.surface.write_to_png('00000-bits-per-pic.png');
 283
 284 def plot_badblock_ratio():
 285         plot = Plot(800,600);
 286         data_x = Numeric.transpose(tpts[0]);
 287         data_y = Numeric.transpose(tpts[17]);
 288         mx = max(data_x);
 289         my = max(data_y);
 290         plot.set_x_range(0, mx);
 291         plot.set_y_range(0, my);
 292
 293         plot.set_sane_ticks();
 294
 295         plot.ctx.set_source_rgb (1, 1, 1);
 296         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 297         plot.ctx.paint();
 298
 299         plot.draw_border ();
 300         plot.draw_ticks ();
 301
 302         plot.ctx.set_line_width(0.5);
 303
 304         plot.ctx.set_source_rgb (1, 0, 0);
 305         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 306
 307         plot.surface.write_to_png('00000-badblock-ratio.png');
 308
 309 def plot_est_act_residual():
 310         plot = Plot(800,600);
 311         mx = max(Numeric.transpose(tpts[7]));
 312         my = max(Numeric.transpose(tpts[9]));
 313         m = max(mx,my);
 314         plot.set_x_range(0, m);
 315         plot.set_y_range(0, m);
 316
 317         plot.set_sane_ticks();
 318
 319         plot.ctx.set_source_rgb (1, 1, 1);
 320         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 321         plot.ctx.paint();
 322
 323         plot.draw_border ();
 324         plot.draw_ticks ();
 325
 326         plot.ctx.set_line_width(0.5);
 327
 328         plot.ctx.set_source_rgb (0, 1, 0);
 329         plot.with_lines = 1;
 330         plot.draw_plot ( [[0,0],[m,m]]);
 331
 332         plot.ctx.set_source_rgb (1, 0, 0);
 333         plot.with_lines = 0;
 334         plot.draw_plot (Numeric.transpose(tpts[7:10:2]));
 335
 336         plot.surface.write_to_png('00000-est-act-residual.png');
 337
 338 def plot_alloc_est_residual():
 339         plot = Plot(800,600);
 340         data_x = Numeric.transpose(tpts[4]);
 341         data_y = Numeric.transpose(tpts[7]);
 342         mx = max(data_x);
 343         my = max(data_y);
 344         m = max(mx,my);
 345         plot.set_x_range(0, m);
 346         plot.set_y_range(0, m);
 347
 348         plot.set_sane_ticks();
 349
 350         plot.ctx.set_source_rgb (1, 1, 1);
 351         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 352         plot.ctx.paint();
 353
 354         plot.draw_border ();
 355         plot.draw_ticks ();
 356
 357         plot.ctx.set_line_width(0.5);
 358
 359         plot.ctx.set_source_rgb (0, 1, 0);
 360         plot.with_lines = 1;
 361         plot.draw_plot ( [[0,0],[m,m]]);
 362
 363         plot.ctx.set_source_rgb (1, 0, 0);
 364         plot.with_lines = 0;
 365         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 366
 367         plot.surface.write_to_png('00000-alloc-est-residual.png');
 368
 369 def plot_est_act_mc():
 370         plot = Plot(800,600);
 371         mx = max(Numeric.transpose(tpts[6]));
 372         my = max(Numeric.transpose(tpts[8]));
 373         m = max(mx,my);
 374         plot.set_x_range(0, m);
 375         plot.set_y_range(0, m);
 376
 377         plot.set_sane_ticks();
 378
 379         plot.ctx.set_source_rgb (1, 1, 1);
 380         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 381         plot.ctx.paint();
 382
 383         plot.draw_border ();
 384         plot.draw_ticks ();
 385
 386         plot.ctx.set_line_width(0.5);
 387
 388         plot.ctx.set_source_rgb (0, 1, 0);
 389         plot.with_lines = 1;
 390         plot.draw_plot ( [[0,0],[m,m]]);
 391
 392         plot.ctx.set_source_rgb (1, 0, 0);
 393         plot.with_lines = 0;
 394         plot.draw_plot (Numeric.transpose(tpts[6:9:2]));
 395
 396         plot.surface.write_to_png('00000-est-act-mc.png');
 397
 398 def plot_est_mc_scs():
 399         plot = Plot(800,600);
 400         data_x = Numeric.transpose(tpts[6]);
 401         data_y = Numeric.transpose(tpts[10]);
 402         mx = max(data_x);
 403         my = max(data_y);
 404         plot.set_x_range(0, mx);
 405         plot.set_y_range(0, 0.1);
 406
 407         plot.set_sane_ticks();
 408
 409         plot.ctx.set_source_rgb (1, 1, 1);
 410         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 411         plot.ctx.paint();
 412
 413         plot.draw_border ();
 414         plot.draw_ticks ();
 415
 416         plot.ctx.set_line_width(0.5);
 417
 418         plot.ctx.set_source_rgb (1, 0, 0);
 419         plot.with_lines = 0;
 420         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 421
 422         plot.surface.write_to_png('00000-est-mc-scs.png');
 423
 424 def plot_scs_mse():
 425         plot = Plot(800,600);
 426         #mx = max(Numeric.transpose(tpts[10]));
 427         mx = 1;
 428         my = max(Numeric.transpose(tpts[14]));
 429         plot.set_x_range(0, mx);
 430         plot.set_y_range(0, my);
 431
 432         plot.set_sane_ticks();
 433
 434         plot.ctx.set_source_rgb (1, 1, 1);
 435         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 436         plot.ctx.paint();
 437
 438         plot.draw_border ();
 439         plot.draw_ticks ();
 440
 441         plot.ctx.set_line_width(0.5);
 442
 443         plot.ctx.set_source_rgb (1, 0, 0);
 444         plot.with_lines = 0;
 445         plot.draw_plot (Numeric.transpose(tpts[10:15:4]));
 446
 447         plot.surface.write_to_png('00000-scs-mse.png');
 448
 449 def plot_scs():
 450         plot = Plot(800,600);
 451         data_x = Numeric.transpose(tpts[0]);
 452         data_y = Numeric.transpose(tpts[14]);
 453         mx = max(data_x);
 454         my = max(data_y);
 455         plot.set_x_range(0, mx);
 456         plot.set_y_range(0, my);
 457
 458         plot.set_sane_ticks();
 459
 460         plot.ctx.set_source_rgb (1, 1, 1);
 461         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 462         plot.ctx.paint();
 463
 464         plot.draw_border ();
 465         plot.draw_ticks ();
 466
 467         plot.ctx.set_line_width(0.5);
 468
 469         plot.ctx.set_source_rgb (1, 0, 0);
 470         plot.with_lines = 0;
 471         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 472
 473         plot.surface.write_to_png('00000-scs.png');
 474
 475 def plot_badblock_vs_mse():
 476         plot = Plot(800,600);
 477         data_x = Numeric.transpose(tpts[17]);
 478         data_y = Numeric.transpose(tpts[14]);
 479         mx = max(data_x);
 480         my = max(data_y);
 481         plot.set_x_range(0, mx);
 482         plot.set_y_range(0, my);
 483
 484         plot.set_sane_ticks();
 485
 486         plot.ctx.set_source_rgb (1, 1, 1);
 487         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 488         plot.ctx.paint();
 489
 490         plot.draw_border ();
 491         plot.draw_ticks ();
 492
 493         plot.ctx.set_line_width(0.5);
 494
 495         plot.ctx.set_source_rgb (1, 0, 0);
 496         plot.with_lines = 0;
 497         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 498
 499         plot.surface.write_to_png('00000-badblock-vs-mse.png');
 500
 501 def plot_mc_error_vs_mse():
 502         plot = Plot(800,600);
 503         data_x = Numeric.transpose(tpts[13]);
 504         data_y = Numeric.transpose(tpts[14]);
 505         #data_x = Numeric.transpose(tpts[0]);
 506         #data_y = Numeric.transpose(tpts[13])/Numeric.transpose(tpts[14]);
 507         mx = max(data_x);
 508         my = max(data_y);
 509         plot.set_x_range_data(data_x);
 510         plot.set_y_range(0, my);
 511
 512         plot.set_sane_ticks();
 513
 514         plot.ctx.set_source_rgb (1, 1, 1);
 515         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 516         plot.ctx.paint();
 517
 518         plot.draw_border ();
 519         plot.draw_ticks ();
 520
 521         plot.ctx.set_line_width(0.5);
 522
 523         plot.ctx.set_source_rgb (0, 1, 0);
 524         plot.with_lines = 1;
 525         plot.draw_plot ( [[0,0],[mx,mx]]);
 526
 527         plot.ctx.set_source_rgb (1, 0, 0);
 528         plot.with_lines = 0;
 529         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 530
 531         plot.surface.write_to_png('00000-mc-error-vs-mse.png');
 532
 533 def plot_hist_slope():
 534         plot = Plot(800,600);
 535         data_x = Numeric.transpose(tpts[0]);
 536         data_y = Numeric.transpose(tpts[18]);
 537         plot.set_x_range_data(data_x);
 538         plot.set_y_range_data(data_y);
 539
 540         plot.set_sane_ticks();
 541
 542         plot.ctx.set_source_rgb (1, 1, 1);
 543         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 544         plot.ctx.paint();
 545
 546         plot.draw_border ();
 547         plot.draw_ticks ();
 548
 549         plot.ctx.set_line_width(0.5);
 550
 551         plot.ctx.set_source_rgb (1, 0, 0);
 552         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 553
 554         plot.surface.write_to_png('00000-hist-slope.png');
 555
 556 def plot_hist_slope_vs_mse():
 557         plot = Plot(800,600);
 558         data_x = Numeric.transpose(tpts[18]);
 559         data_y = Numeric.transpose(tpts[14]);
 560         plot.set_x_range_data(data_x);
 561         plot.set_y_range_data(data_y);
 562
 563         plot.set_sane_ticks();
 564
 565         plot.ctx.set_source_rgb (1, 1, 1);
 566         plot.ctx.set_operator (cairo.OPERATOR_SOURCE);
 567         plot.ctx.paint();
 568
 569         plot.draw_border ();
 570         plot.draw_ticks ();
 571
 572         plot.ctx.set_line_width(0.5);
 573
 574         plot.with_lines = 0;
 575         plot.ctx.set_source_rgb (1, 0, 0);
 576         plot.draw_plot (Numeric.transpose([data_x,data_y]));
 577
 578         plot.surface.write_to_png('00000-hist-slope-vs-mse.png');
 579
 580
 581 pts = readdata("schro_dump.picture");
 582 #pts = pts[100:200];
 583 tpts = Numeric.transpose(pts);
 584
 585 plot_mse();
 586 plot_buffer_level();
 587 plot_alloc_est_residual();
 588 plot_est_act_residual();
 589 plot_est_act_mc();
 590 plot_bits_per_pic();
 591 plot_scs_mse();
 592 plot_scs();
 593 plot_est_mc_scs();
 594 plot_badblock_ratio();
 595 plot_badblock_vs_mse();
 596 plot_mc_error_vs_mse();
 597 plot_hist_slope();
 598 plot_hist_slope_vs_mse();
 599
 600