Spicy
local-variable.h
1 // Copyright (c) 2020-2021 by the Zeek Project. See LICENSE for details.
2 
3 #pragma once
4 
5 #include <string>
6 #include <utility>
7 #include <vector>
8 
9 #include <hilti/ast/declaration.h>
10 #include <hilti/ast/expression.h>
11 #include <hilti/ast/id.h>
12 #include <hilti/ast/type.h>
13 #include <hilti/ast/types/auto.h>
14 
15 namespace hilti {
16 namespace declaration {
17 
20 public:
21  LocalVariable(ID id, ::hilti::Type type, std::optional<hilti::Expression> init, bool const_ = false,
22  Meta m = Meta())
23  : DeclarationBase(nodes(std::move(id), std::move(type), std::move(init)), std::move(m)), _const(const_) {}
24 
25  LocalVariable(ID id, ::hilti::Type type, bool const_ = false, Meta m = Meta())
26  : DeclarationBase(nodes(std::move(id), std::move(type), node::none), std::move(m)), _const(const_) {}
27 
28  LocalVariable(ID id, hilti::Expression init, bool const_ = false, Meta m = Meta())
29  : DeclarationBase(nodes(std::move(id), node::none, std::move(init)), std::move(m)), _const(const_) {}
30 
31  LocalVariable(ID id, ::hilti::Type type, std::vector<hilti::Expression> args,
32  std::optional<hilti::Expression> init = {}, bool const_ = false, Meta m = Meta())
33  : DeclarationBase(nodes(std::move(id), std::move(type), std::move(init), std::move(args)), std::move(m)),
34  _const(const_) {}
35 
36  LocalVariable(ID id, bool const_ = false, Meta m = Meta())
37  : DeclarationBase(nodes(std::move(id), type::auto_, node::none), std::move(m)), _const(const_) {}
38 
39  auto init() const { return childs()[2].tryAs<hilti::Expression>(); }
40 
41  const auto& type() const {
42  if ( auto t = childs()[1].tryAs<hilti::Type>() )
43  return *t;
44  else {
45  assert(init());
46  return init()->type();
47  }
48  }
49 
50  auto typeArguments() const { return childs<hilti::Expression>(3, -1); }
51 
52  void setInit(hilti::Expression i) { childs()[2] = std::move(i); }
53  void setType(hilti::Type t) { childs()[1] = std::move(t); }
54  void setTypeArguments(std::vector<hilti::Expression> args) {
55  auto& c = childs();
56  c.erase(c.begin() + 3, c.end());
57  for ( auto&& a : args )
58  c.emplace_back(std::move(a));
59  }
60 
61  bool operator==(const LocalVariable& other) const {
62  return id() == other.id() && type() == other.type() && init() == other.init();
63  }
64 
66  bool isConstant() const { return _const; }
68  const ID& id() const { return child<ID>(0); }
70  Linkage linkage() const { return Linkage::Private; }
72  std::string displayName() const { return "local variable"; };
74  auto isEqual(const Declaration& other) const { return node::isEqual(this, other); }
75 
77  auto properties() const { return node::Properties{{"const", _const}}; }
78 
79 private:
80  bool _const;
81 };
82 
83 } // namespace declaration
84 } // namespace hilti
Definition: local-variable.h:19
Definition: declaration.h:53
const ID & id() const
Definition: local-variable.h:68
const Node none
Definition: node.cc:14
auto isEqual(const Declaration &other) const
Definition: local-variable.h:74
Definition: meta.h:18
Definition: type.h:159
Definition: declaration.h:87
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:99
Linkage linkage() const
Definition: local-variable.h:70
bool isConstant() const
Definition: local-variable.h:66
const auto & childs() const
Definition: node.h:470
auto properties() const
Definition: local-variable.h:77
std::string displayName() const
Definition: local-variable.h:72
Definition: id.h:18