Add support for grep and while. Inline expect_list into expect_any so that the pp...

[perl/modules/Sub-Nary.git] / lib / Sub / Nary.pm

diff --git a/lib/Sub/Nary.pm b/lib/Sub/Nary.pm
index 4e244782a0e8650e0f5ca61a422f39b9a853cb6a..058048249bb723f9a23481098addd389b342615e 100644 (file)
--- a/lib/Sub/Nary.pm
+++ b/lib/Sub/Nary.pm
@@ -194,6 +194,21 @@ sub combine {
  }} map { (ref) ? $_ : { $_ => 1 } } grep defined, @_;
 }
 
+sub power {
+ my ($p, $n, $c) = @_;
+ my $s = sum values %$p;
+ return { 0 => $s * $c } unless $n;
+ if ($n eq 'list') {
+  my $z = delete $p->{0};
+  return { 'list' => $c } unless $z;
+  return { 0      => $c } if $z == 1;
+  return { 0 => $c * $z, list => $c * ($s - $z) };
+ }
+ my $r = combine map { { %$p } } 1 .. $n;
+ $r->{$_} *= $c for keys %$r;
+ return $r;
+}
+
 sub add {
  reduce {
   $a->{$_} += $b->{$_} for keys %$b;
@@ -208,7 +223,8 @@ $ops{$_} = 0      for qw/stub nextstate/;
 $ops{$_} = 1      for qw/padsv/;
 $ops{$_} = 'list' for qw/padav/;
 $ops{$_} = 'list' for qw/padhv rv2hv/;
-$ops{$_} = 'list' for qw/padany match entereval readline/;
+$ops{$_} = 'list' for qw/padany/;
+$ops{$_} = 'list' for qw/match entereval readline/;
 
 $ops{each}      = { 0 => 0.5, 2 => 0.5 };
 $ops{stat}      = { 0 => 0.5, 13 => 0.5 };
@@ -252,7 +268,7 @@ sub enter {
 sub expect_return {
  my ($self, $op) = @_;
 
- return ($self->expect_list($op))[0] => 1 if name($op) eq 'return';
+ return ($self->expect_kids($op))[0] => 1 if name($op) eq 'return';
 
  if ($op->flags & OPf_KIDS) {
   for ($op = $op->first; not null $op; $op = $op->sibling) {
@@ -261,48 +277,24 @@ sub expect_return {
   }
  }
 
- return;
+ return 0;
 }
 
-sub expect_list {
+sub expect_any {
  my ($self, $op) = @_;
 
  my $n = name($op);
+ return ($self->expect_kids($op))[0] => 1 if $n eq 'return';
+
  my $meth = $self->can('pp_' . $n);
  return $self->$meth($op) if $meth;
+
  if (exists $ops{$n}) {
   my $r = $ops{$n};
-  $r = { %$r } if ref $r eq 'HASH';
+  $r = { %$r } if ref $r;
   return $r => 0;
  }
 
- if ($op->flags & OPf_KIDS) {
-  my @res = (0);
-  my ($p, $r);
-  for ($op = $op->first; not null $op; $op = $op->sibling) {
-   my $n = name($op);
-   next if $n eq 'pushmark';
-   if ($n eq 'nextstate'
-       and not null(($op = $op->sibling)->sibling)) {
-    ($p, $r) = $self->expect_return($op);
-    return $p => 1 if $r;
-   } else {
-    ($p, $r) = $self->expect_any($op);
-    return $p => 1 if $r;
-    push @res, $p;
-   }
-  }
-  return (combine @res) => 0;
- }
-
- return;
-}
-
-sub expect_any {
- my ($self, $op) = @_;
-
- return ($self->expect_list($op))[0] => 1 if name($op) eq 'return';
-
  if (class($op) eq 'LOGOP' and not null $op->first) {
   my @res;
   my ($p, $r);
@@ -326,7 +318,31 @@ sub expect_any {
   return (add @res) => 0;
  }
 
- return $self->expect_list($op);
+ return $self->expect_kids($op);
+}
+
+sub expect_kids {
+ my ($self, $op) = @_;
+
+ return 0 unless $op->flags & OPf_KIDS;
+
+ my @res = (0);
+ my ($p, $r);
+ for ($op = $op->first; not null $op; $op = $op->sibling) {
+  my $n = name($op);
+  next if $n eq 'pushmark';
+  if ($n eq 'nextstate'
+      and not null(($op = $op->sibling)->sibling)) {
+   ($p, $r) = $self->expect_return($op);
+   return $p => 1 if $r;
+  } else {
+   ($p, $r) = $self->expect_any($op);
+   return $p => 1 if $r;
+   push @res, $p;
+  }
+ }
+
+ return (combine @res) => 0;
 }
 
 # Stolen from B::Deparse
@@ -451,7 +467,7 @@ sub pp_aassign {
  $op = $op->first;
 
  # Can't assign to return
- my ($p, $r) = $self->expect_list($op->sibling);
+ my ($p, $r) = $self->expect_any($op->sibling);
  return $p => 0 if not exists $p->{list};
 
  $self->expect_any($op);
@@ -485,6 +501,41 @@ sub pp_flip {
  return 'list'
 }
 
+sub pp_grepwhile {
+ my ($self, $op) = @_;
+
+ $op = $op->first;
+ return 'list' if name($op) ne 'grepstart';
+
+ $op = $op->first->sibling;
+ my ($p, $r) = $self->expect_any($op);
+ return $p => 1 if $r;
+
+ $op = $op->sibling;
+ ($p, $r) = $self->expect_any($op);
+ return $p => 1 if $r;
+
+ return 'list';
+}
+
+sub pp_mapwhile {
+ my ($self, $op) = @_;
+
+ $op = $op->first;
+ return 'list' if name($op) ne 'mapstart';
+
+ $op = $op->first->sibling;
+ my ($p1, $r) = $self->expect_any($op);
+ return $p1 => 1 if $r;
+
+ $op = $op->sibling;
+ (my $p2, $r) = $self->expect_any($op);
+ return $p2 => 1 if $r;
+ $p2 = { $p2 => 1 } unless ref $p2;
+
+ return add map { power $p1, $_, $p2->{$_} } keys %$p2;
+}
+
 =head1 EXPORT
 
 An object-oriented module shouldn't export any function, and so does this one.