-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathprint.hpp
More file actions
54 lines (41 loc) · 1.41 KB
/
print.hpp
File metadata and controls
54 lines (41 loc) · 1.41 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#pragma once
#include <iostream>
#include "nodes.hpp"
namespace lambda {
namespace _print {
inline void _print(std::unique_ptr<LambdaBase>& node) noexcept;
inline void printVariable(std::unique_ptr<LambdaBase>& node) noexcept {
std::cout << ((std::unique_ptr<Variable>&)node)->name;
if (((std::unique_ptr<Variable>&)node)->count)
std::cout << ((std::unique_ptr<Variable>&)node)->count;
}
inline void printFunction(std::unique_ptr<LambdaBase>& node) noexcept {
std::cout << '&';
std::cout << ((std::unique_ptr<Function>&)node)->bound.name;
if (((std::unique_ptr<Function>&)node)->bound.count)
std::cout << ((std::unique_ptr<Function>&)node)->bound.count;
std::cout << '.';
_print(((std::unique_ptr<Function>&)node)->body);
}
inline void printApplication(std::unique_ptr<LambdaBase>& node) noexcept {
std::cout << '(';
_print(((std::unique_ptr<Application>&)node)->left);
_print(((std::unique_ptr<Application>&)node)->right);
std::cout << ')';
}
inline void _print(std::unique_ptr<LambdaBase>& node) noexcept {
switch (node->getType()) {
case LambdaType::VARIABLE:
return printVariable(node);
case LambdaType::FUNCTION:
return printFunction(node);
case LambdaType::APPLICATION:
return printApplication(node);
}
}
}; // namespace _print
inline void print(std::unique_ptr<LambdaBase>& node) noexcept {
if (!node) return;
_print::_print(node);
}
} // namespace lambda