Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Add more precision to the output for colorize.pl
[simgrid.git] / tools / MSG_visualization / colorize.pl
index 39da01f..9fdd3df 100755 (executable)
@@ -30,27 +30,95 @@ my (@coltab) = (
     $col_background . $col_blue,   $col_background . $col_magenta,
 );
 
+my %pid;
+
+# Get options
+#
+while (($_ = $ARGV[0]) =~ /^-/) {
+    shift;
+    if (/-location/i) {
+        $opt_print_location = 1;  shift;
+    } elsif (/-h(elp)?$|-u(sage)?$/i) {
+        print<<EOH
+colorize.pl - Log colorizer for SimGrid
+Syntax:
+
+    colorize.pl [options] <file>
+
+    where <file> is a text file of values or '-' for STDIN
+
+Options: () denote short version
+
+    -location          Print the location in the code of the message.
+EOH
+;
+       exit;
+    }
+}
+
+sub pidcolor {
+    my $pid = shift;
+
+    unless (defined($pid{$pid})) {
+       # first time we see this pid. Affect it a color
+       $pid{$pid}=(scalar keys %pid) % (scalar @coltab);
+    }
+    return $coltab[$pid{$pid}];
+}
+
+sub print_line {
+    my($host,$procname,$pid,$date,$location,$xbt_channel,$message)=@_;
+
+    print $col_norm;
+    printf "[% 10.6f]",$date;
+
+    print pidcolor($pid);
+
+    if(defined($location)) {
+       printf "[%10s:%-10s %s ]",$host,$procname,$location;
+    } else {
+       printf "[%10s:%-10s]",$host,$procname;
+    }
+    print " $message";
+    print $col_norm."\n";
+}
+
+# Read the messages and do the job
 while (<>) {
     $orgline = $thisline = $_;
 
-    if ( $thisline =~ /^\[(\w+):(\w+):\((\d+\)) ([0-9\.]*)\] ([^\[]*) \[([^\[]*)\] (.*)$/ ) {
+    # Typical line  [Gatien:slave:(9) 11.243148] msg/gos.c:137: [msg_gos/DEBUG] Action terminated
+    if ($thisline =~ /^\[(.+):([^:]+):\((\d+)\) ([\d\.]*)\] ([^\[]*) \[([^\[]*)\] (.*)$/) {
        $host=$1;
        $procname=$2;
        $pid=$3;
        $date=$4;
-       $location=$5;
+       if($opt_print_location) {
+           $location=$5;
+           $location =~ s/:$//;
+       } else {
+           $location = undef;
+       }
        $xbt_channel=$6;
        $message=$7;
 
-       print $col_norm;
-       printf "[% 10.3f]",$date;
-       print $coltab[($pid-1) % scalar(@coltab)];
-       printf "[%10s:%-10s]",$host,$procname;
-       print " $message";
-       print $col_norm."\n";
-        next;
+       print_line($host,$procname,$pid,$date,$location,$xbt_channel,$message);
+    # Typical line  [Boivin:slave:(2) 9.269357] [pmm/INFO] ROW: step(2)<>myrow(0). Receive data from TeX
+    } elsif ($thisline =~ /^\[([^:]+):([^:]+):\((\d+)\) ([\d\.]*)\] \[([^\[]*)\] (.*)$/) {
+       $host=$1;
+       $procname=$2;
+       $pid=$3;
+       $date=$4;
+       $xbt_channel=$5;
+       $message=$6;
+       print_line($host,$procname,$pid,$date,undef,$xbt_channel,$message);
+    } elsif ( $thisline =~ /^==(\d+)== (.*)$/) {
+       # take care of valgrind outputs
+       print pidcolor($1)."$2\n";
+
+    } else {
+       print $col_default. $orgline;
     }
-    print $col_default. $orgline;
 }
 
 print $col_norm;