#! /usr/bin/perl
use CGI;
use GD;
use Image::Magick;
use DBI;
use lib '../include';
use nms;
my $cgi = CGI->new;

# Sekrit night-mode
my $night = defined($cgi->param('night'));

my $dbh = nms::db_connect();

GD::Image->trueColor(1);
my ($img, $text_img);

$img = GD::Image->new('bg07.png');
if ($night) {
	my ($width, $height) = ($img->width, $img->height); 

	$img = GD::Image->new($width, $height);
	$img->alphaBlending(0);
	$img->saveAlpha(1);
	my $blank = $img->colorAllocateAlpha(0, 0, 0, 127);
	$img->filledRectangle(0, 0, $img->width - 1, $img->height - 1, $blank);

	$text_img = GD::Image->new($width, $height);
	$text_img->alphaBlending(0);
	$text_img->saveAlpha(1);
	my $blank = $text_img->colorAllocateAlpha(0, 0, 0, 127);
	$text_img->filledRectangle(0, 0, $text_img->width - 1, $text_img->height - 1, $blank);
} else {
	$img = GD::Image->new('snmp-bg.png');
	$text_img = $img;
}

my $blk = $img->colorResolve(0, 0, 0);

for my $y (42..236) {
	my $i = 4.0 * ($y - 236.0) / (42.0 - 237.0);
	my $clr = get_color($i);
	
	$img->filledRectangle(12, $y, 33, $y+1, $clr);
	$text_img->filledRectangle(12, $y, 33, $y+1, $clr);
}

$text_img->rectangle(12,42,33,236,$blk);

my $tclr = $night ? $text_img->colorResolve(255, 255, 255) : $blk;
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 40, 47 + (236-42)*0.0/4.0, "100 Gbit/sec");
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 40, 47 + (236-42)*1.0/4.0, "10 Gbit/sec");
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 40, 47 + (236-42)*2.0/4.0, "1 Gbit/sec");
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 40, 47 + (236-42)*3.0/4.0, "100 Mbit/sec");
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 40, 47 + (236-42)*4.0/4.0, "10 Mbit/sec");
$text_img->stringFT($tclr, "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf", 10, 0, 1000, 620, "NMS (C) 2005-2006 Tech:Server");

my $q = $dbh->prepare('select * from switches natural join placements natural left join
( select switch,sum(bytes_in) as bytes_in,sum(bytes_out) as bytes_out from get_datarate() group
by switch ) t1 order by zorder');
$q->execute();
while (my $ref = $q->fetchrow_hashref()) {

	# for now:
	# 10Mbit/switch = green
	# 100Mbit/switch = yellow
	# 1Gbit/switch = red
	# 10Gbit/switch = white
	
	my $clr;

	if (defined($ref->{'bytes_in'})) {
		my $intensity = 0.0;
		my $traffic = 4.0 * $ref->{'bytes_in'} + $ref->{'bytes_out'};  # average and convert to bits (should be about the same in practice)

		my $max = 10_000_000_000.0;   # 10Gbit
		my $min =     10_000_000.0;   # 10Mbit
		if ($traffic >= $min) {
			$intensity = log($traffic / $min) / log(10);
			$intensity = 3.0 if ($intensity > 3.0);
		}
		$clr = get_color($intensity);
	} else {
		$clr = $img->colorResolve(0, 0, 255);
	}
	
	$ref->{'placement'} =~ /\((\d+),(\d+)\),\((\d+),(\d+)\)/;
	$img->filledRectangle($3,$4,$1,$2,$clr);
	$text_img->filledRectangle($3,$4,$1,$2,$clr);

	$img->rectangle($3,$4,$1,$2,$blk);
	$text_img->rectangle($3,$4,$1,$2,$blk);
        $text_img->stringUp(gdSmallFont,$3,$2-3,$ref->{'sysname'},$blk);
}
$dbh->disconnect;

print $cgi->header(-type=>'image/png');
if ($night) {
	my $magick = Image::Magick->new;
	$magick->BlobToImage($img->png);
	$magick->Blur(sigma=>10.0, channels=>'All');
	$magick->Gamma(gamma=>1.90);

	my $m2 = Image::Magick->new;
	$m2->Read('snmp-bg.png');
	$m2->Negate();
	$m2->Composite(image=>$magick, compose=>'Atop');

	my $m3 = Image::Magick->new;
	$m3->BlobToImage($text_img->png);
	$m2->Composite(image=>$m3, compose=>'Atop');
	
	$img = $m2->ImageToBlob();
	print $img;
} else {	
	print $img->png;
}

sub get_color {
	my $intensity = shift;
	my $gamma = 1.0/1.90;
	if ($intensity > 3.0) {
		return $img->colorResolve(255.0 * ((4.0 - $intensity) ** $gamma), 255.0 * ((4.0 - $intensity) ** $gamma), 255.0 * ((4.0 - $intensity) ** $gamma));
	} elsif ($intensity > 2.0) {
		return $img->colorResolve(255.0, 255.0 * (($intensity - 2.0) ** $gamma), 255.0 * (($intensity - 2.0) ** $gamma));
	} elsif ($intensity > 1.0) {
		return $img->colorResolve(255.0, 255.0 * ((2.0 - $intensity) ** $gamma), 0);
	} else {
		return $img->colorResolve(255.0 * ($intensity ** $gamma), 255, 0);
	}
}
