// Copyright (C) 2020 Free Software Foundation, Inc.

// This file is part of GCC.

// GCC is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 3, or (at your option) any later
// version.

// GCC is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
// for more details.

// You should have received a copy of the GNU General Public License
// along with GCC; see the file COPYING3.  If not see
// <http://www.gnu.org/licenses/>.

#ifndef RUST_COMPILE_FNPARAM
#define RUST_COMPILE_FNPARAM

#include "rust-compile-base.h"

namespace Rust {
namespace Compile {

class CompileFnParam : public HIRCompileBase
{
public:
  static Bvariable *compile (Context *ctx, Bfunction *fndecl,
			     HIR::FunctionParam *param, Btype *decl_type,
			     Location locus)
  {
    CompileFnParam compiler (ctx, fndecl, decl_type, locus);
    param->get_param_name ()->accept_vis (compiler);
    return compiler.translated;
  }

  void visit (HIR::IdentifierPattern &pattern)
  {
    if (!pattern.is_mut)
      decl_type = ctx->get_backend ()->immutable_type (decl_type);

    translated
      = ctx->get_backend ()->parameter_variable (fndecl, pattern.variable_ident,
						 decl_type,
						 false /* address_taken */,
						 locus);
  }

private:
  CompileFnParam (Context *ctx, ::Bfunction *fndecl, ::Btype *decl_type,
		  Location locus)
    : HIRCompileBase (ctx), fndecl (fndecl), decl_type (decl_type),
      locus (locus), translated (nullptr)
  {}

  ::Bfunction *fndecl;
  ::Btype *decl_type;
  Location locus;
  ::Bvariable *translated;
};

class CompileSelfParam : public HIRCompileBase
{
public:
  static Bvariable *compile (Context *ctx, Bfunction *fndecl,
			     HIR::SelfParam &self, Btype *decl_type,
			     Location locus)
  {
    if (!self.get_is_mut ())
      decl_type = ctx->get_backend ()->immutable_type (decl_type);

    return ctx->get_backend ()->parameter_variable (fndecl, "self", decl_type,
						    false /* address_taken */,
						    locus);
  }
};

} // namespace Compile
} // namespace Rust

#endif // RUST_COMPILE_FNPARAM