feat: Add lengthof generated extern for lookup tables

lhearachel · Jan 15, 2025 · ec95481 · ec95481
1 parent a0b6865
commit ec95481
Show file tree

Hide file tree

Showing 14 changed files with 37 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -176,10 +176,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/src/generators/generate_c.c b/src/generators/generate_c.c
@@ -65,10 +65,12 @@ static const char *lookup_branch_fmt = ""
     "\n"
     "#ifndef %sLOOKUP_IMPL\n"
     "\n"
+    "extern const long lengthof__%s;\n"
     "extern const entry__%s lookup__%s[];\n"
     "\n"
     "#else\n"
     "\n"
+    "const long lengthof__%s = %d;\n"
     "const entry__%s lookup__%s[] = {\n"
     "";
 
@@ -133,13 +135,15 @@ bool generate_c(enumerator *input, options *opts, FILE *fout)
     }
 
     fprintf(fout, lookup_branch_fmt,
-            guardp.buf,
-            guardp.buf,
-            opts->tag.buf,
-            opts->tag.buf,
-            guardp.buf,
-            opts->tag.buf, opts->tag.buf,
-            opts->tag.buf, opts->tag.buf);
+            guardp.buf,                    // "#endif /* %sENUM */\n"
+            guardp.buf,                    // "#ifdef %sLOOKUP\n"
+            opts->tag.buf,                 // "typedef struct entry__%s {\n"
+            opts->tag.buf,                 // "} entry__%s;\n"
+            guardp.buf,                    // "#ifndef %sLOOKUP_IMPL\n"
+            opts->tag.buf,                 // "extern const long lengthof__%s;\n"
+            opts->tag.buf, opts->tag.buf,  // "extern const entry__%s lookup__%s[];\n"
+            opts->tag.buf, input->count,   // "const long lengthof__%s = %d;\n"
+            opts->tag.buf, opts->tag.buf); // "const entry__%s lookup__%s[] = {\n"
 
     for (usize i = 0; i < input->count; i++) {
         usize padding = input->max_ident_len - genned->table[i].len + 1;

diff --git a/tests/enum/append.test b/tests/enum/append.test
@@ -85,10 +85,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 17;
 const entry__stdin lookup__stdin[] = {
     { APPENDED,        "APPENDED",        },
     { APPENDED_PLUS_1, "APPENDED_PLUS_1", },

diff --git a/tests/enum/base.test b/tests/enum/base.test
@@ -75,10 +75,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/tests/enum/direct_assignment.test b/tests/enum/direct_assignment.test
@@ -75,10 +75,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/tests/enum/guard.test b/tests/enum/guard.test
@@ -77,10 +77,12 @@ typedef struct entry__stdin {
 
 #ifndef ALTERNATIVE_GUARD_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/tests/enum/leader.test b/tests/enum/leader.test
@@ -77,10 +77,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { ALTERNATIVE_LEADER_BLASTOISE,  "ALTERNATIVE_LEADER_BLASTOISE",  },
     { ALTERNATIVE_LEADER_BULBASAUR,  "ALTERNATIVE_LEADER_BULBASAUR",  },

diff --git a/tests/enum/prepend.test b/tests/enum/prepend.test
@@ -85,10 +85,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 17;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,        "BLASTOISE",        },
     { BULBASAUR,        "BULBASAUR",        },

diff --git a/tests/enum/start_from.test b/tests/enum/start_from.test
@@ -77,10 +77,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 14;
 const entry__stdin lookup__stdin[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/tests/enum/tag_name.test b/tests/enum/tag_name.test
@@ -77,10 +77,12 @@ typedef struct entry__Alternative_TagName {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__Alternative_TagName;
 extern const entry__Alternative_TagName lookup__Alternative_TagName[];
 
 #else
 
+const long lengthof__Alternative_TagName = 14;
 const entry__Alternative_TagName lookup__Alternative_TagName[] = {
     { BLASTOISE,  "BLASTOISE",  },
     { BULBASAUR,  "BULBASAUR",  },

diff --git a/tests/mask/base.test b/tests/mask/base.test
@@ -79,10 +79,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 16;
 const entry__stdin lookup__stdin[] = {
     { ANY,        "ANY",        },
     { BLASTOISE,  "BLASTOISE",  },

diff --git a/tests/mask/guard.test b/tests/mask/guard.test
@@ -81,10 +81,12 @@ typedef struct entry__stdin {
 
 #ifndef ALTERNATIVE_GUARD_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 16;
 const entry__stdin lookup__stdin[] = {
     { ANY,        "ANY",        },
     { BLASTOISE,  "BLASTOISE",  },

diff --git a/tests/mask/leader.test b/tests/mask/leader.test
@@ -81,10 +81,12 @@ typedef struct entry__stdin {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__stdin;
 extern const entry__stdin lookup__stdin[];
 
 #else
 
+const long lengthof__stdin = 16;
 const entry__stdin lookup__stdin[] = {
     { ALTERNATIVE_LEADER_ANY,        "ALTERNATIVE_LEADER_ANY",        },
     { ALTERNATIVE_LEADER_BLASTOISE,  "ALTERNATIVE_LEADER_BLASTOISE",  },

diff --git a/tests/mask/tag_name.test b/tests/mask/tag_name.test
@@ -81,10 +81,12 @@ typedef struct entry__Alternative_TagName {
 
 #ifndef METANG_LOOKUP_IMPL
 
+extern const long lengthof__Alternative_TagName;
 extern const entry__Alternative_TagName lookup__Alternative_TagName[];
 
 #else
 
+const long lengthof__Alternative_TagName = 16;
 const entry__Alternative_TagName lookup__Alternative_TagName[] = {
     { ANY,        "ANY",        },
     { BLASTOISE,  "BLASTOISE",  },