@@@ major overhaul, new primitives

[mLib] / utils / t / control-test.c

/* -*-c-*-
 *
 * Test the control-flow metaprogramming macros
 *
 * (c) 2022 Straylight/Edgeware
 */

/*----- Licensing notice --------------------------------------------------*
 *
 * This file is part of the mLib utilities library.
 *
 * mLib is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Library General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * mLib is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with mLib.  If not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
 * USA.
 */

/*----- Header files ------------------------------------------------------*/

#include <stdio.h>
#include <stdlib.h>

#include "control.h"

/*----- Main code ---------------------------------------------------------*/

static int step = 0;
static int rc = 0;

#define STEP(s) check_step(s, __FILE__ ": " STR(__LINE__))
#define MISSTEP STEP(-1)
static void check_step(int s, const char *where)
{
  if (step != s) {
    fprintf(stderr, "misstep at %s: expected %d but found %d\n",
	    where, step, s);
    rc = 2;
  }
  step++;
}

#define LASTSTEP(s) laststep(s, __FILE__ ": " STR(__LINE__))
static void laststep(int s, const char *where)
  { check_step(s, where); step = 0; }

#define FORELSE(head)							\
			MC_GOTO(top)					\
  MC_LABEL(out)		MC_ACT({ ; })					\
  MC_LABEL(top)		ALLOWELSE(els)					\
			AFTER(outer, { GOELSE(els); })			\
			for (head)					\
			  WRAP(inner, { ; },				\
				      { ; },				\
				      { MC_GOTO(out); })

#define FOR_FIZZBUZZ(var, base, limit)					\
			MC_GOTO(top)					\
  MC_LABEL(out)		MC_ACT({ ; })					\
  MC_LABEL(top)		DECL(bounds,					\
			     int _i = base COMMA _limit = limit)	\
			for (; _i < _limit; _i++)			\
			  DECL(buf, char _buf[24])			\
			  DECL(var, const char *var)			\
			  WRAP(wrap, {					\
			    switch (_i%15) {				\
			      case 0:					\
				var = "fizzbuzz";			\
				break;					\
			      case 3: case 6: case 9: case 12:		\
				var = "fizz";				\
				break;					\
			      case 5: case 10:				\
				var = "buzz";				\
				break;					\
			      default:					\
				sprintf(_buf, "%d", _i); var = _buf;	\
				break;					\
			    }						\
			  },						\
			  { ; },					\
			  { MC_GOTO(out); })

int main(void)
{
  int i;

  BEFORE(before0, { STEP(0); }) STEP(1);
  AFTER(after0, { STEP(3); }) STEP(2);
  LASTSTEP(4);

  WRAP(wrap0, { STEP(0); }, { STEP(2); }, { MISSTEP; }) STEP(1);
  WRAP(wrap1, { STEP(3); }, { MISSTEP; }, { STEP(5); }) { STEP(4); break; }
  LASTSTEP(6);

  STEP(0);
  for (;;) {
    AFTER(after1, { STEP(2); break; }) STEP(1);
    MISSTEP; break;
  }
  LASTSTEP(3);

  FORELSE (i = 0; i < 10; i++) {
    STEP(i);
    if (i == 7) break;
  } else
    MISSTEP;
  LASTSTEP(8);

  FORELSE (i = 0; i < 10; i++) {
    STEP(i);
    if (i == 12) break;
  } else
    STEP(10);
  LASTSTEP(11);

#define TEST								\
			MC_ACT({ STEP(0); MC_GOTO(in_plain); })		\
  MC_LABEL(done_plain)	MC_ACT({ STEP(5); GOELSE(elsie); })		\
  MC_LABEL(in_plain)	WRAP(outer_wrap, { STEP(1); },			\
					 { STEP(7); },			\
					 { MISSTEP; })			\
			ALLOWELSE(elsie)				\
			  WRAP(inner_wrap, { STEP(2); },		\
					   { STEP(4);			\
					     MC_GOTO(done_plain); },	\
					   { MISSTEP; })		\
			  STEP(3);					\
			else						\
			  STEP(6);					\
			LASTSTEP(8);
  TEST
#undef TEST

#if __STDC_VERSION__ >= 199901 || defined(__cplusplus)
  STEP(0);
  DECL(decl0, int j = 1) STEP(j);
  LASTSTEP(2);
#endif

  FOR_FIZZBUZZ(fb, 19, 32) printf("%s\n", fb);

  return (rc);
}

/*----- That's all, folks -------------------------------------------------*/