codeql/swift/codegen/templates/cpp_classes.mustache

// generated by {{generator}} from {{source}}
// clang-format off
#pragma once

#include <iostream>
#include <optional>
#include <vector>

#include "swift/extractor/trap/TrapLabel.h"
#include "swift/extractor/trap/TrapTagTraits.h"
#include "./TrapEntries.h"

namespace codeql {
{{#classes}}

struct {{name}}{{#has_bases}} : {{#bases}}{{^first}}, {{/first}}{{ref.name}}{{/bases}}{{/has_bases}} {
  {{#final}}
  explicit {{name}}(TrapLabel<{{name}}Tag> id) : id{id} {}

  TrapLabel<{{name}}Tag> id{};
  {{/final}}
  {{#fields}}
  {{type}} {{field_name}}{};
  {{/fields}}
  {{#final}}

  friend std::ostream& operator<<(std::ostream& out, const {{name}}& x) {
    x.emit(out);
    return out;
  }
  {{/final}}

 protected:
  void emit({{^final}}TrapLabel<{{name}}Tag> id, {{/final}}std::ostream& out) const {
    {{#trap_name}}
    out << {{.}}Trap{id{{#single_fields}}, {{field_name}}{{/single_fields}}} << '\n';
    {{/trap_name}}
    {{#bases}}
    {{ref.name}}::emit(id, out);
    {{/bases}}
    {{#fields}}
    {{#is_predicate}}
    if ({{field_name}}) out << {{trap_name}}Trap{id} << '\n';
    {{/is_predicate}}
    {{#is_optional}}
    {{^is_repeated}}
    if ({{field_name}}) out << {{trap_name}}Trap{id, *{{field_name}}} << '\n';
    {{/is_repeated}}
    {{/is_optional}}
    {{#is_repeated}}
    for (auto i = 0u; i < {{field_name}}.size(); ++i) {
      {{^is_optional}}
      out << {{trap_name}}Trap{id, i, {{field_name}}[i]};
      {{/is_optional}}
      {{#is_optional}}
      if ({{field_name}}[i]) out << {{trap_name}}Trap{id, i, *{{field_name}}[i]};
      {{/is_optional}}
    }
    {{/is_repeated}}
    {{/fields}}
  }
};

template <>
struct detail::ToTrapClassFunctor<{{name}}Tag> {
  using type = {{name}};
};
{{/classes}}
}