//   Copyright Naoki Shibata and contributors 2010 - 2025.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE.txt or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

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

#ifndef ENABLE_STREAM
#error ENABLE_STREAM not defined
#endif

int main(int argc, char **argv) {
  if (argc < 3) {
    fprintf(stderr, "Usage : %s <base type> <base type ID> <maxbutwidth> <minshift> <maxshift> <isa> ...\n", argv[0]);
    exit(-1);
  }

  const char *baseType = argv[1];
  const char *baseTypeID = argv[2];
  const int maxbutwidth = atoi(argv[3]);
  const int minshift = atoi(argv[4]);
  const int maxshift = atoi(argv[5]);
  const int isastart = 6;
  const int isamax = argc - isastart;

#if ENABLE_STREAM == 1
  const int enable_stream = 1;
#else
  const int enable_stream = 0;
#endif

  printf("#define MAXBUTWIDTH%s %d\n", baseTypeID, maxbutwidth);
  printf("#define MINSHIFT%s %d\n", baseTypeID, minshift);
  printf("#define MAXSHIFT%s %d\n", baseTypeID, maxshift);
  printf("#define CONFIGMAX 4\n");
  printf("#define ISAMAX %d\n", isamax);
  printf("\n");

  if (strcmp(baseType, "paramonly") == 0) exit(0);

  for(int k=isastart;k<argc;k++) {
    for(int config=0;config<4;config++) {
#if ENABLE_STREAM == 0
      if ((config & 1) != 0) continue;
#endif
      for(int j=1;j<=maxbutwidth;j++) {
        printf("void dft%df_%d_%s(%s *, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType);
        printf("void dft%db_%d_%s(%s *, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType);
        printf("void tbut%df_%d_%s(%s *, uint32_t *, const %s *, const int, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType, baseType);
        printf("void tbut%db_%d_%s(%s *, uint32_t *, const %s *, const int, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType, baseType);
        printf("void but%df_%d_%s(%s *, uint32_t *, const int, const %s *, const int, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType, baseType);
        printf("void but%db_%d_%s(%s *, uint32_t *, const int, const %s *, const int, const %s *, const int);\n", 1 << j, config, argv[k], baseType, baseType, baseType);

        for(int s=minshift;s<maxshift;s++) {
          printf("void dft%df_%d_%d_%s(%s *, const %s *);\n", 1 << j, s, config, argv[k], baseType, baseType);
          printf("void dft%db_%d_%d_%s(%s *, const %s *);\n", 1 << j, s, config, argv[k], baseType, baseType);
          printf("void tbut%df_%d_%d_%s(%s *, uint32_t *, const %s *, const %s *, const int);\n", 1 << j, s, config, argv[k], baseType, baseType, baseType);
          printf("void tbut%db_%d_%d_%s(%s *, uint32_t *, const %s *, const %s *, const int);\n", 1 << j, s, config, argv[k], baseType, baseType, baseType);
        }

        for(int s=0;s<maxshift;s++) {
          printf("void but%df_%d_%d_%s(%s *, uint32_t *, const %s *, const int, const %s *, const int);\n", 1 << j, s, config, argv[k], baseType, baseType, baseType);
          printf("void but%db_%d_%d_%s(%s *, uint32_t *, const %s *, const int, const %s *, const int);\n", 1 << j, s, config, argv[k], baseType, baseType, baseType);
        }
      }
    }
    printf("void realSub0_%s(%s *, const %s *, const int, const %s *, const %s *);\n", argv[k], baseType, baseType, baseType, baseType);
    printf("void realSub1_%s(%s *, const %s *, const int, const %s *, const %s *, const int);\n", argv[k], baseType, baseType, baseType, baseType);
    printf("int getInt_%s(int);\n", argv[k]);
    printf("const void *getPtr_%s(int);\n", argv[k]);
  }

  printf("\n");

  printf("void (*dftf_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          printf("dft%df_%d_%s, ", 1 << i, config, argv[k]);
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*dftb_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          if (i == 1) {
            printf("dft%df_%d_%s, ", 1 << i, config, argv[k]);
          } else {
            printf("dft%db_%d_%s, ", 1 << i, config, argv[k]);
          }
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*tbutf_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const %s *, const int, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          printf("tbut%df_%d_%s, ", 1 << i, config, argv[k]);
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*tbutb_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const %s *, const int, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          printf("tbut%db_%d_%s, ", 1 << i, config, argv[k]);
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*butf_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const int, const %s *, const int, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          printf("but%df_%d_%s, ", 1 << i, config, argv[k]);
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*butb_%s[CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const int, const %s *, const int, const %s *, const int) = {\n", baseType, baseTypeID, baseType, baseType, baseType);
  for(int config=0;config<4;config++) {
    printf("  {\n");
    for(int k=isastart;k<argc;k++) {
      printf("    {NULL, ");
      for(int i=1;i<=maxbutwidth;i++) {
        if (enable_stream || (config & 1) == 0) {
          printf("but%db_%d_%s, ", 1 << i, config, argv[k]);
        } else {
          printf("NULL, ");
        }
      }
      printf("},\n");
    }
    printf("},\n");
  }
  printf("};\n\n");

  printf("void (*tbutfs_%s[MAXSHIFT%s][CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const %s *, const %s *, const int) = {\n", baseType, baseTypeID, baseTypeID, baseType, baseType, baseType);
  for(int s=0;s<maxshift;s++) {
    printf("  {\n");
    for(int config=0;config<4;config++) {
      printf("    {\n");
      for(int k=isastart;k<argc;k++) {
        printf("      {NULL, ");
        for(int i=1;i<=maxbutwidth;i++) {
          if ((enable_stream || (config & 1) == 0) && s >= minshift) {
            printf("tbut%df_%d_%d_%s, ", 1 << i, s, config, argv[k]);
          } else {
            printf("NULL, ");
          }
        }
        printf("},\n");
      }
      printf("    },\n");
    }
    printf("  },\n");
  }
  printf("};\n\n");

  printf("void (*tbutbs_%s[MAXSHIFT%s][CONFIGMAX][ISAMAX][MAXBUTWIDTH%s+1])(%s *, uint32_t *, const %s *, const %s *, const int) = {\n", baseType, baseTypeID, baseTypeID, baseType, baseType, baseType);
  for(int s=0;s<maxshift;s++) {
    printf("  {\n");
    for(int config=0;config<4;config++) {
      printf("    {\n");
      for(int k=isastart;k<argc;k++) {
        printf("      {NULL, ");
        for(int i=1;i<=maxbutwidth;i++) {
          if ((enable_stream || (config & 1) == 0) && s >= minshift) {
            printf("tbut%db_%d_%d_%s, ", 1 << i, s, config, argv[k]);
          } else {
            printf("NULL, ");
          }
        }
        printf("},\n");
      }
      printf("    },\n");
    }
    printf("  },\n");
  }
  printf("};\n\n");

  //

  printf("void (*realSub0_%s[ISAMAX])(%s *, const %s *, const int, const %s *, const %s *) = {\n  ", baseType, baseType, baseType, baseType, baseType);
  for(int k=isastart;k<argc;k++) printf("realSub0_%s, ", argv[k]);
  printf("\n};\n\n");

  printf("void (*realSub1_%s[ISAMAX])(%s *, const %s *, const int, const %s *, const %s *, const int) = {\n  ", baseType, baseType, baseType, baseType, baseType);
  for(int k=isastart;k<argc;k++) printf("realSub1_%s, ", argv[k]);
  printf("\n};\n\n");

  printf("int (*getInt_%s[16])(int) = {\n  ", baseType);
  for(int k=isastart;k<argc;k++) printf("getInt_%s, ", argv[k]);
  for(int k=0;k<16-(argc-isastart);k++) printf("NULL, ");
  printf("\n};\n\n");

  printf("const void *(*getPtr_%s[16])(int) = {\n  ", baseType);
  for(int k=isastart;k<argc;k++) printf("getPtr_%s, ", argv[k]);
  for(int k=0;k<16-(argc-isastart);k++) printf("NULL, ");
  printf("\n};\n\n");
}
