Add conversion functions to ClassSymbol

Signed-off-by: Roberto Raggi <[email protected]>

Author: robertoraggi

src/parser/cxx/scope.cc

@@ -27,6 +27,7 @@
 #include <algorithm>
 #include <cassert>
 #include <functional>
+#include <print>
 
 namespace cxx {
 
@@ -111,6 +112,14 @@ void Scope::addSymbol(Symbol* symbol) {
   symbol->setEnclosingScope(this);
   symbols_.push_back(symbol);
 
+  if (name_cast<ConversionFunctionId>(symbol->name())) {
+    if (auto functionSymbol = symbol_cast<FunctionSymbol>(symbol)) {
+      if (auto classSymbol = symbol_cast<ClassSymbol>(owner_)) {
+        classSymbol->addConversionFunction(functionSymbol);
+      }
+    }
+  }
+
   if (3 * symbols_.size() >= 2 * buckets_.size()) {
     rehash();
   } else {

src/parser/cxx/symbols.cc

@@ -215,6 +215,15 @@ auto ClassSymbol::hasBaseClass(
   return false;
 }
 
+auto ClassSymbol::conversionFunctions() const
+    -> const std::vector<FunctionSymbol*>& {
+  return conversionFunctions_;
+}
+
+void ClassSymbol::addConversionFunction(FunctionSymbol* conversionFunction) {
+  conversionFunctions_.push_back(conversionFunction);
+}
+
 auto ClassSymbol::declaration() const -> SpecifierAST* { return specifier_; }
 
 void ClassSymbol::setDeclaration(SpecifierAST* specifier) {

src/parser/cxx/symbols.h

@@ -246,6 +246,11 @@ class ClassSymbol final : public ScopedSymbol {
 
   void addConstructor(FunctionSymbol* constructor);
 
+  [[nodiscard]] auto conversionFunctions() const
+      -> const std::vector<FunctionSymbol*>&;
+
+  void addConversionFunction(FunctionSymbol* conversionFunction);
+
   [[nodiscard]] auto isFinal() const -> bool;
   void setFinal(bool isFinal);
 
@@ -316,6 +321,7 @@ class ClassSymbol final : public ScopedSymbol {
  private:
   std::vector<BaseClassSymbol*> baseClasses_;
   std::vector<FunctionSymbol*> constructors_;
+  std::vector<FunctionSymbol*> conversionFunctions_;
   std::unique_ptr<TemplateInfo<ClassSymbol>> templateInfo_;
   SpecifierAST* specifier_ = nullptr;
   TemplateDeclarationAST* templateDeclaration_ = nullptr;