impulse_data_multi.pl

#!/usr/bin/perl

#Extract the values of the impulse (F*dt) for each integrator for a set of logs for trajectories in a given range
#Data sets are given in descending order through the nested integrator:  [0][0],  [0][1], ... [1][0],  [1][1] ... [N][0]... [N][N]   for  [level][integrator]

sub compare_sort
{
    $a=~m/(\d+) (\d+)/;
    my $a1=$1; 
    my $a2=$2;

    $b=~m/(\d+) (\d+)/;
    my $b1=$1; 
    my $b2=$2;

    if($a1 < $b1){
	return -1;
    }elsif($a1 > $b1){
	return 1;
    }else{
	if($a2 < $b2){
	    return -1;
	}elsif($a2 > $b2){
	    return 1;
	}else{
	    return 0;
	}
    }
}

$ARGC = scalar(@ARGV);
if($ARGC < 2){
    print "Need start trajectory and at least one filename\n";
    exit 1;
}
$start_traj = $ARGV[0];

@logs = ();
for($i=1;$i<$ARGC;$i++){
    push(@logs, $ARGV[$i]);
}

%data = ();
%data_stage = ();
$ncomplete = 0;

$traj;
foreach $log (@logs){
    open(IN, $log);

    $active = 0;
    foreach $line (<IN>){
	if($active == 0){	 
	    if($line =~m/# Trajectory = (\d+)/){
		$traj = $1;
		#print "Found start of trajectory ${traj}\n";
		if($traj >= $start_traj){
		    #print "Trajectory is in range\n";
		    $active = 1;	    
		}
	    }
	}else{
	    if($line=~m/Total time for trajectory/){
		#print "Found end of trajectory ${traj}\n";
		$active = 0;
		
		#Found complete trajectory
		foreach $key (keys %data_stage){
		    $to = ${data{$key}};
		    $from = ${data_stage{$key}};
		    push(@{$to}, @{$from});
		}
		$ncomplete = $ncomplete +1;

		#}elsif($line=~m/\[(\d+)\]\[(\d)\] Force average:.*Max impulse: ([\d\.e\+\-]+)/){
	    }elsif($line=~m/\[(\d+)\]\[(\d+)\] Fdt max\s+: ([\d\.e\+\-]+)/){
		#print "$1 $2 $3\n";
		$key = "$1 $2";
		if( !( exists $data{$key}) ){
		    #print "Adding key $key\n";
		    $data{$key} = [];
		}
		#print "Push to key $key\n";
		push(@{$data_stage{$key}}, $3);
	    }
	}
    }
    close(IN);
}

@keys = keys %data;
@keys_sorted = sort compare_sort (@keys);


foreach $key (@keys_sorted){
    @vals = @{$data{$key}};
    #print "$key\n";
    foreach $v (@vals){
	print "$v\n";	
    }
    print "\n";
}

#print "$ncomplete complete trajectories\n";