Solution to task 2

1 files changed, 69 insertions, 0 deletions

diff --git a/challenge-248/jo-37/perl/ch-2.pl b/challenge-248/jo-37/perl/ch-2.pl
new file mode 100755
index 0000000000..ee94250365
--- /dev/null
+++ b/challenge-248/jo-37/perl/ch-2.pl
@@ -0,0 +1,69 @@
+#!/usr/bin/perl -s
+
+use v5.24;
+use Test2::V0 '!float';
+use PDL;
+use PDL::NiceSlice;
+
+our ($tests, $examples);
+
+run_tests() if $tests || $examples;	# does not return
+
+die <<EOS unless @ARGV;
+usage: $0 [-examples] [-tests] [--] [M]
+
+-examples
+    run the examples from the challenge
+ 
+-tests
+    run some tests
+
+M
+    a matrix in any format as accepted by the PDL string constructor,
+    e.g. '1,0,0;0,1,0;0,0,1'
+
+EOS
+
+
+### Input and Output
+
+say submatrix_sum("@ARGV");
+
+
+### Implementation
+
+sub submatrix_sum {
+	my $m = pdl @_;
+
+    $m->range(ndcoords($m(1:,1:)), 2)->reorder(2, 3, 0, 1)
+        ->clump(2)->sumover;
+}
+
+
+### Examples and tests
+
+sub run_tests {
+    SKIP: {
+        skip "examples" unless $examples;
+
+        is submatrix_sum(
+            [1,  2,  3,  4],
+            [5,  6,  7,  8],
+            [9, 10, 11, 12])->unpdl,
+            [[14, 18, 22], [30, 34, 38]], 'example 1';
+
+        is submatrix_sum(
+            [1, 0, 0, 0],
+            [0, 1, 0, 0],
+            [0, 0, 1, 0],
+            [0, 0, 0, 1])->unpdl,
+            [[2, 1, 0], [1, 2, 1], [0, 1, 2]], 'example 2';
+    }
+
+    SKIP: {
+        skip "tests" unless $tests;
+	}
+
+    done_testing;
+    exit;
+}