#ifndef COSMOPOLITAN_LIBC_TESTLIB_EZBENCH_H_
#define COSMOPOLITAN_LIBC_TESTLIB_EZBENCH_H_
#include "libc/macros.h"
#include "libc/math.h"
#include "libc/nexgen32e/bench.h"
#include "libc/nexgen32e/x86feature.h"
#include "libc/testlib/bench.h"
#include "libc/testlib/testlib.h"
COSMOPOLITAN_C_START_

#ifndef EZBENCH_COUNT
#define EZBENCH_COUNT 128
#endif

#define EZBENCH_TRIES 10

#define EZBENCH(INIT, EXPR) EZBENCH2(#EXPR, INIT, EXPR)

#define EZBENCH2(NAME, INIT, EXPR)                                        \
  do {                                                                    \
    int Core, Tries, Interrupts;                                          \
    double Speculative, MemoryStrict;                                     \
    Tries = 0;                                                            \
    do {                                                                  \
      __testlib_yield();                                                  \
      Core = __testlib_getcore();                                         \
      Interrupts = __testlib_getinterrupts();                             \
      INIT;                                                               \
      EXPR;                                                               \
      Speculative = BENCHLOOP(__startbench, __endbench, EZBENCH_COUNT, ({ \
                                INIT;                                     \
                                __polluteregisters();                     \
                              }),                                         \
                              (EXPR));                                    \
    } while (++Tries < EZBENCH_TRIES &&                                   \
             (__testlib_getcore() != Core &&                              \
              __testlib_getinterrupts() > Interrupts));                   \
    if (Tries == EZBENCH_TRIES)                                           \
      __testlib_ezbenchwarn(" speculative");                              \
    Tries = 0;                                                            \
    do {                                                                  \
      __testlib_yield();                                                  \
      Core = __testlib_getcore();                                         \
      Interrupts = __testlib_getinterrupts();                             \
      INIT;                                                               \
      EXPR;                                                               \
      MemoryStrict = BENCHLOOP(__startbench_m, __endbench_m, 32, ({       \
                                 INIT;                                    \
                                 __polluteregisters();                    \
                               }),                                        \
                               (EXPR));                                   \
    } while (++Tries < EZBENCH_TRIES &&                                   \
             (__testlib_getcore() != Core &&                              \
              __testlib_getinterrupts() > Interrupts));                   \
    if (Tries == EZBENCH_TRIES)                                           \
      __testlib_ezbenchwarn(" memory strict");                            \
    __testlib_ezbenchreport(                                              \
        NAME, MAX(.001, Speculative - __testlib_ezbenchcontrol()),        \
        MAX(.001, MemoryStrict - __testlib_ezbenchcontrol()));            \
  } while (0)

#define EZBENCH3(NAME, NUM, INIT, EXPR)                              \
  do {                                                               \
    int Core, Tries, Interrupts;                                     \
    double Speculative, MemoryStrict;                                \
    Tries = 0;                                                       \
    do {                                                             \
      __testlib_yield();                                             \
      Core = __testlib_getcore();                                    \
      Interrupts = __testlib_getinterrupts();                        \
      INIT;                                                          \
      EXPR;                                                          \
      Speculative = BENCHLOOP(__startbench, __endbench, NUM, ({      \
                                INIT;                                \
                                __polluteregisters();                \
                              }),                                    \
                              (EXPR));                               \
    } while (++Tries < EZBENCH_TRIES &&                              \
             (__testlib_getcore() != Core &&                         \
              __testlib_getinterrupts() > Interrupts));              \
    if (Tries == EZBENCH_TRIES)                                      \
      __testlib_ezbenchwarn(" speculative");                         \
    Tries = 0;                                                       \
    do {                                                             \
      __testlib_yield();                                             \
      Core = __testlib_getcore();                                    \
      Interrupts = __testlib_getinterrupts();                        \
      INIT;                                                          \
      EXPR;                                                          \
      MemoryStrict = BENCHLOOP(__startbench_m, __endbench_m, NUM, ({ \
                                 INIT;                               \
                                 __polluteregisters();               \
                               }),                                   \
                               (EXPR));                              \
    } while (++Tries < EZBENCH_TRIES &&                              \
             (__testlib_getcore() != Core &&                         \
              __testlib_getinterrupts() > Interrupts));              \
    if (Tries == EZBENCH_TRIES)                                      \
      __testlib_ezbenchwarn(" memory strict");                       \
    __testlib_ezbenchreport(                                         \
        NAME, MAX(.001, Speculative - __testlib_ezbenchcontrol()),   \
        MAX(.001, MemoryStrict - __testlib_ezbenchcontrol()));       \
  } while (0)

#define EZBENCH_C(NAME, CONTROL, EXPR)                                 \
  do {                                                                 \
    int Core, Tries, Interrupts;                                       \
    double Control, Speculative, MemoryStrict;                         \
    Tries = 0;                                                         \
    do {                                                               \
      __testlib_yield();                                               \
      Core = __testlib_getcore();                                      \
      Interrupts = __testlib_getinterrupts();                          \
      Control = BENCHLOOP(__startbench_m, __endbench_m, EZBENCH_COUNT, \
                          ({ __polluteregisters(); }), (CONTROL));     \
    } while (++Tries < EZBENCH_TRIES &&                                \
             (__testlib_getcore() != Core &&                           \
              __testlib_getinterrupts() > Interrupts));                \
    if (Tries == EZBENCH_TRIES)                                        \
      __testlib_ezbenchwarn(" control");                               \
    Tries = 0;                                                         \
    do {                                                               \
      __testlib_yield();                                               \
      Core = __testlib_getcore();                                      \
      Interrupts = __testlib_getinterrupts();                          \
      EXPR;                                                            \
      Speculative = BENCHLOOP(__startbench, __endbench, EZBENCH_COUNT, \
                              __polluteregisters(), (EXPR));           \
    } while (++Tries < EZBENCH_TRIES &&                                \
             (__testlib_getcore() != Core &&                           \
              __testlib_getinterrupts() > Interrupts));                \
    if (Tries == EZBENCH_TRIES)                                        \
      __testlib_ezbenchwarn(" speculative");                           \
    Tries = 0;                                                         \
    do {                                                               \
      __testlib_yield();                                               \
      Core = __testlib_getcore();                                      \
      Interrupts = __testlib_getinterrupts();                          \
      EXPR;                                                            \
      MemoryStrict = BENCHLOOP(__startbench_m, __endbench_m, 8,        \
                               ({ __polluteregisters(); }), (EXPR));   \
    } while (++Tries < EZBENCH_TRIES &&                                \
             (__testlib_getcore() != Core &&                           \
              __testlib_getinterrupts() > Interrupts));                \
    if (Tries == EZBENCH_TRIES)                                        \
      __testlib_ezbenchwarn(" memory strict");                         \
    __testlib_ezbenchreport(NAME, MAX(.001, Speculative - Control),    \
                            MAX(.001, MemoryStrict - Control));        \
  } while (0)

#define EZBENCH_N(NAME, N, EXPR)                                       \
  do {                                                                 \
    double Speculative, Toto;                                          \
    int Core, Tries, Interrupts;                                       \
    Tries = 0;                                                         \
    do {                                                               \
      __testlib_yield();                                               \
      Core = __testlib_getcore();                                      \
      Interrupts = __testlib_getinterrupts();                          \
      (void)Toto;                                                      \
      (void)Core;                                                      \
      (void)Interrupts;                                                \
      EXPR;                                                            \
      Speculative = BENCHLOOPER(__startbench, __endbench, 32, (EXPR)); \
    } while (++Tries < EZBENCH_TRIES && !Speculative);                 \
    if (Tries == EZBENCH_TRIES)                                        \
      __testlib_ezbenchwarn("");                                       \
    __testlib_ezbenchreport_n(NAME, 'n', N, Speculative);              \
  } while (0)

#define EZBENCH_K(NAME, K, EXPR)                                        \
  do {                                                                  \
    int Core;                                                           \
    double Speculative;                                                 \
    do {                                                                \
      __testlib_yield();                                                \
      Core = __testlib_getcore();                                       \
      EXPR;                                                             \
      Speculative =                                                     \
          BENCHLOOPER(__startbench, __endbench, EZBENCH_COUNT, (EXPR)); \
    } while (Core != __testlib_getcore());                              \
    __testlib_ezbenchreport_n(NAME, 'k', K, Speculative);               \
  } while (0)

void __polluteregisters(void);
void __testlib_yield(void);
int __testlib_getcore(void);
int64_t __testlib_getinterrupts(void);
double __testlib_ezbenchcontrol(void);
void __testlib_ezbenchwarn(const char *);
void __testlib_ezbenchreport(const char *, double, double);
void __testlib_ezbenchreport_n(const char *, char, size_t, double);

#ifdef __STRICT_ANSI__
#undef EZBENCH2
#undef EZBENCH3
#undef EZBENCH_N
#undef EZBENCH_K
#define EZBENCH2(NAME, INIT, EXPR)      (void)0
#define EZBENCH3(NAME, NUM, INIT, EXPR) (void)0
#define EZBENCH_N(NAME, N, EXPR)        (void)0
#define EZBENCH_K(NAME, K, EXPR)        (void)0
#endif

COSMOPOLITAN_C_END_
#endif /* COSMOPOLITAN_LIBC_TESTLIB_EZBENCH_H_ */
