1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#!/usr/bin/perl6
use v6;
my $self = $*PROGRAM-NAME;
sub MAIN($file1, $file2)
{
my @m1;
readPlot(@m1, $file1);
my @m2;
readPlot(@m2, $file2);
my $mode = ($file1 ~~ /".pgfplots"$/);
my @msum = sumPlot(@m1, @m2, $mode);
writePlot(@msum, $mode);
}
sub sumPlot(@m1, @m2, $mode)
{
my Int $total = @m1.elems;
die "$self: Plots not of equal size $total and "~@m2.elems~"\n" if ($total != @m2.elems);
my @msum;
for 0..^$total -> Int $ind {
@msum[$ind] = sumThisRow(@m1[$ind], @m2[$ind], $mode);
}
return @msum;
}
sub sumThisRow(@row1, @row2, $mode)
{
my Int $total = @row1.elems;
die "$self: Row not of equal size $total and "~@row2.elems~"\n" if ($total != @row2.elems);
my @a;
my $n = ($mode) ?? 2 !! 1;
for 0..^$n -> Int $ind {
@a[$ind] = @row1[$ind];
}
for $n..^$total -> Int $ind {
@a[$ind] = @row1[$ind] + @row2[$ind];
}
return @a;
}
sub vectorEqual(@v1, @v2)
{
my $n = @v1.elems;
return 0 if ($n != @v2.elems);
for 0..^$n -> Int $ind {
next if (@v1[$ind] == @v2[$ind]);
die "$self: @v1[$ind] different from @v2[$ind] for $ind\n";
}
return 1;
}
sub writePlot(@m, $mode)
{
my Int $total = @m.elems;
my $prevOmega = 0;
for 0..^$total -> Int $ind {
my @thisrow = @m[$ind];
my $thisOmega = ($mode) ?? @thisrow[0][1] !! 0;
say "" if ($mode && $ind > 0 && $prevOmega != $thisOmega);
@thisrow.join(' ').say;
$prevOmega = $thisOmega;
}
}
sub readPlot(@m, $file)
{
my $input = open($file, :r);
my @thisrow;
my (Int $x, Int $y) = (0, 0);
my Int $rows;
my Int $ln = 0;
for $file.IO.lines -> $line {
++$ln;
next if ($line ~~ /^ '#' /);
my @temp = split(/\s/, $line);
next if (@temp.elems < 2);
@m[$y++] = @temp;
}
note "$self: Found $y rows in $file\n";
}
|
