carbon-lang/executable_semantics/ast/function_definition.cpp

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "executable_semantics/ast/function_definition.h"

#include <iostream>

namespace Carbon {

auto MakeFunDef(int line_num, std::string name, const Expression* ret_type,
                const Expression* param_pattern, const Statement* body)
    -> FunctionDefinition {
  FunctionDefinition f;
  f.line_num = line_num;
  f.name = std::move(name);
  f.return_type = ret_type;
  f.param_pattern = param_pattern;
  f.body = body;
  return f;
}

void PrintFunDefDepth(const FunctionDefinition& f, int depth) {
  std::cout << "fn " << f.name << " ";
  PrintExp(f.param_pattern);
  std::cout << " -> ";
  PrintExp(f.return_type);
  if (f.body) {
    std::cout << " {" << std::endl;
    PrintStatement(f.body, depth);
    std::cout << std::endl << "}" << std::endl;
  } else {
    std::cout << ";" << std::endl;
  }
}

void PrintFunDef(const FunctionDefinition& f) { PrintFunDefDepth(f, -1); }

}  // namespace Carbon