New feature: trigger frequency and snapshot count.

1 files changed, 56 insertions, 11 deletions

diff --git a/pw.c b/pw.c
index 6dd5d1e..260a07d 100644
--- a/pw.c
+++ b/pw.c
@@ -79,6 +79,7 @@ enum status_flags {
   stat_lino =     0x0100, // render line numbers
   stat_bkgnd =    0x0200, // running in the background
   stat_hlite =    0x0400, // running in the background
+  stat_oneshot =  0x0800, // running in the background
 };
 
 typedef struct pwstate {
@@ -89,6 +90,7 @@ typedef struct pwstate {
   int hist;
   int columns;
   unsigned stat;
+  int sncount, tcount;
   char *curcmd, *savedcmd;
   char cmdbuf[cmdsize];
 } pwstate;
@@ -121,6 +123,8 @@ static grep grepstack[maxgrep];
 static int ngrep;
 
 static grep *triglist[maxtrig];
+static int sncount;
+static int tfreq;
 
 static char **cmdhist;
 static int ncmdhist;
@@ -548,7 +552,9 @@ static void redraw(pwstate *pw)
     updln = 1;
     for (int i = 0; i < pw->nlines; i++)
       snapshot[0][i] = dsref(pw->circbuf[i]);
-    pw->stat &= ~(stat_dirty | stat_trgrd);
+    if ((pw->stat & stat_oneshot))
+      pw->stat |= stat_susp;
+    pw->stat &= ~(stat_dirty | stat_trgrd | stat_oneshot);
     updln = 1;
   } else if ((pw->stat & stat_force)) {
     pw->stat &= ~stat_force;
@@ -760,6 +766,34 @@ static void execute(pwstate *pw, char *cmd)
   case 'B':
     regex_flags = 0;
     break;
+  case 'c':
+    if (arg[0] == 0) {
+      pw->sncount = 0;
+    } else  {
+      char *err = 0;
+      int val = getznn(arg, &err);
+      if (val < 0)
+        snprintf(cmd, cmdsize, "bad trigger count: %s", err);
+      else
+        sncount = val;
+      dsdrop(err);
+    }
+    break;
+  case 'f':
+    if (arg[0] == 0) {
+      sprintf(cmd, "frequency argument required!");
+      break;
+    }
+    {
+      char *err = 0;
+      int val = getznn(arg, &err);
+      if (val < 0)
+        snprintf(cmd, cmdsize, "bad trigger freq: %s", err);
+      else
+        tfreq = val;
+      dsdrop(err);
+    }
+    break;
   default:
     sprintf(cmd, "bad command");
     break;
@@ -1079,36 +1113,47 @@ int main(int argc, char **argv)
         }
         if (line) {
           if (pw.nlines == maxlines) {
+            int trig = 0;
             dsdrop(pw.circbuf[0]);
             memmove(pw.circbuf, pw.circbuf + 1,
                     (pw.nlines - 1) * sizeof *pw.circbuf);
             pw.circbuf[pw.nlines - 1] = line;
             pw.stat |= stat_dirty;
-            if ((pw.stat & stat_ttmode)) {
+            if ((pw.stat & (stat_ttmode | stat_susp)) == stat_ttmode) {
               int lim = min(maxtrig, pw.nlines);
-              int trig = 1;
+              int match = 1;
               for (int i = 0; i < lim; i++) {
                 grep *gr = triglist[i];
                 if (gr && !grexec(gr, pw.circbuf[i])) {
-                  trig = 0;
+                  match = 0;
                   break;
                 }
               }
-              if (trig)
-                pw.stat |= stat_trgrd;
-            } else if ((pw.stat & stat_htmode)) {
-              int trig = 1;
+              trig = match;
+            } else if ((pw.stat & (stat_htmode | stat_susp)) == stat_htmode) {
+              int match = 1;
               for (int j = pw.nlines- 1, i = 0; j >= 0 && i < maxtrig;
                    j--, i++)
               {
                 grep *gr = triglist[i];
                 if (gr && !grexec(gr, pw.circbuf[j])) {
-                  trig = 0;
+                  match = 0;
                   break;
                 }
               }
-              if (trig)
-                pw.stat |= stat_trgrd;
+              trig = match;
+            }
+
+            if (trig) {
+              if (tfreq == 0 || ++pw.tcount >= tfreq) {
+                pw.tcount = 0;
+                if (sncount == 0 || ++pw.sncount < sncount) {
+                  pw.stat |= stat_trgrd;
+                } else if (sncount) {
+                  pw.stat |= stat_trgrd | stat_oneshot;
+                  pw.sncount = 0;
+                }
+              }
             }
           } else {
             pw.circbuf[pw.nlines++] = line;