#!/usr/bin/env perl
# time NUM_THREADS=3 perl pcount.pl big

use v5.36;
use autodie;
use MCE;

exit 1 if not @ARGV;

my $mul_pattern = 'mul\(\d{1,3},\d{1,3}\)';
my $filename = shift;
my $count = 0;

sub reduce_count ($worker_count) {
    $count += $worker_count;
}

my $mce = MCE->new(
    max_workers => $ENV{NUM_THREADS} // MCE::Util::get_ncpu(),
    chunk_size => 65536*16,
    use_slurpio => 1,
    gather => \&reduce_count,
    user_func => sub {
        my ($mce, $slurp_ref, $chunk_id) = @_;
        my $count = () = $$slurp_ref =~ m/$mul_pattern/g;
        $mce->gather($count);
    }
)->spawn;

$mce->process({ input_data => $filename });
$mce->shutdown;

print "Found $count matches.\n";