diff --git a/misc/codegen/templates/ql_db.mustache b/misc/codegen/templates/ql_db.mustache index e63e0aae903..a82ff78ec51 100644 --- a/misc/codegen/templates/ql_db.mustache +++ b/misc/codegen/templates/ql_db.mustache @@ -29,9 +29,79 @@ module Raw { {{type}} {{getter}}({{#is_indexed}}int index{{/is_indexed}}) { {{tablename}}({{#tableparams}}{{^first}}, {{/first}}{{param}}{{/tableparams}}) } + {{#is_indexed}} + /** + * Gets the number of {{doc_plural}}. + {{#internal}} + * INTERNAL: Do not use. + {{/internal}} + */ + int getNumberOf{{plural}}() { + result = count(int i | {{tablename}}(this, i, _)) + } + {{/is_indexed}} {{/synth}} {{/properties}} } + + {{#final}} + private Element getImmediateChildOf{{name}}({{name}} e, int index) { + {{^has_children}}none(){{/has_children}} + {{#has_children}} + {{! n is the base offset 0, for ease of generation }} + {{! n is constructed to be strictly greater than the indexes for children }} + exists(int n{{#all_children}}, int n{{property.singular}}{{/all_children}} | + n = 0 + {{#all_children}} + {{#property}} + {{#synth}} + and + n{{singular}} = n{{prev}} + {{/synth}} + {{^synth}} + {{! n is defined on top of the previous definition }} + {{! for single and optional properties it adds 1 (regardless of whether the optional property exists) }} + {{! for repeated it adds 1 + the maximum index (which works for repeated optional as well) }} + and + n{{singular}} = n{{prev}} {{^is_repeated}}+ 1{{/is_repeated}}{{#is_repeated}}+ e.getNumberOf{{plural}}(){{/is_repeated}} + {{/synth}} + {{/property}} + {{/all_children}} and ( + none() + {{#all_children}} + {{#property}} + {{^synth}} + or + {{#is_repeated}} + result = e.get{{singular}}(index - n{{prev}}) + {{/is_repeated}} + {{^is_repeated}} + index = n{{prev}} and result = e.get{{singular}}() + {{/is_repeated}} + {{/synth}} + {{/property}} + {{/all_children}} + )) + {{/has_children}} + } + {{/final}} + {{/classes}} + + /** + * Gets the immediate child indexed at `index`. Indexes are not guaranteed to be contiguous, but are guaranteed to be distinct. + */ + pragma[nomagic] + Element getImmediateChild(Element e, int index) { + // why does this look more complicated than it should? + // * none() simplifies generation, as we can append `or ...` without a special case for the first item + none() + {{#classes}} + {{#final}} + or + result = getImmediateChildOf{{name}}(e, index) + {{/final}} + {{/classes}} + } } diff --git a/misc/codegen/templates/ql_parent.mustache b/misc/codegen/templates/ql_parent.mustache index 84bc0d79a0b..91c5de98c47 100644 --- a/misc/codegen/templates/ql_parent.mustache +++ b/misc/codegen/templates/ql_parent.mustache @@ -23,7 +23,12 @@ private module Impl { {{! for single and optional properties it adds 1 (regardless of whether the optional property exists) }} {{! for repeated it adds 1 + the maximum index (which works for repeated optional as well) }} and - n{{singular}} = n{{prev}} + 1{{#is_repeated}}+ max(int i | i = -1 or exists(e.get{{#type_is_hideable}}Immediate{{/type_is_hideable}}{{singular}}(i)) | i){{/is_repeated}} + {{#type_is_hideable}} + n{{singular}} = n{{prev}} + 1{{#is_repeated}}+ max(int i | i = -1 or exists(e.getImmediate{{singular}}(i)) | i){{/is_repeated}} + {{/type_is_hideable}} + {{^type_is_hideable}} + n{{singular}} = n{{prev}} {{^is_repeated}}+ 1{{/is_repeated}}{{#is_repeated}}+ e.getNumberOf{{plural}}(){{/is_repeated}} + {{/type_is_hideable}} {{/property}} {{/all_children}} and ( none() diff --git a/misc/codegen/templates/ql_synth_types.mustache b/misc/codegen/templates/ql_synth_types.mustache index c76654b8ea9..14a48f635e5 100644 --- a/misc/codegen/templates/ql_synth_types.mustache +++ b/misc/codegen/templates/ql_synth_types.mustache @@ -32,9 +32,28 @@ module Synth { {{/root}} {{/non_final_classes}} + /** + * INTERNAL: Do not use. + * + * Gets the parent of synthetic element `e`. + */ + Raw::{{root}} getSynthParent(T{{root}} e) { + none() + {{#final_classes}} + {{#is_fresh_synth}} + {{#has_params}} + or + e = T{{name}}({{#params}}{{#first}}result{{/first}}{{^first}}, _{{/first}}{{/params}}) + {{/has_params}} + {{/is_fresh_synth}} + {{/final_classes}} + } + + {{#final_classes}} /** * INTERNAL: Do not use. + * * Converts a raw element to a synthesized `T{{name}}`, if possible. */ T{{name}} convert{{name}}FromRaw(Raw::Element e) {