git.iko.soy/rowtype-yoga/purescript-native
on 5bc7fab help sign in
233 lines 7.4 kB history raw
1///////////////////////////////////////////////////////////////////////////////
2//
3// Module : purescript.cc
4// Copyright : (c) Andy Arvanitis 2015
5// License : MIT
6//
7// Maintainer : Andy Arvanitis <andy.arvanitis@gmail.com>
8// Stability : experimental
9// Portability :
10//
11// Basic types and functions to support purescript-to-C++1x rendering
12//
13///////////////////////////////////////////////////////////////////////////////
14//
15#include <cstring>
16#include "PureScript.hh"
18namespace PureScript {
20any::Closure::~Closure() {}
21any::EffClosure::~EffClosure() {}
23#define RETURN_VALUE(TAG, ACCESSOR, INDIRECTION) \
24 const any& variant = unthunkVariant(*this); \
25 assert(variant.tag == TAG); \
26 return INDIRECTION(variant.ACCESSOR); \
28auto any::operator()(const any& arg) const -> any {
29 const any& variant = unthunkVariant(*this);
30 if (variant.tag == Tag::Closure) {
31 return (*variant.l)(arg);
32 }
33 assert(variant.tag == Tag::Function);
34 return (*variant.f)(arg);
37auto any::operator()(const as_thunk) const -> const any& {
38 assert(tag == Tag::Thunk);
39 return (*t)(unthunk);
42auto any::operator()() const -> any {
43 const any& variant = unthunkVariant(*this);
44 if (variant.tag == Tag::EffClosure) {
45 return (*variant.k)();
46 }
47 assert(variant.tag == Tag::EffFunction);
48 return (*variant.e)();
51any::operator int() const {
52 RETURN_VALUE(Tag::Integer, i,)
55any::operator double() const {
56 RETURN_VALUE(Tag::Double, d,)
59any::operator bool() const {
60 RETURN_VALUE(Tag::Boolean, b,)
63any::operator char32_t() const {
64 RETURN_VALUE(Tag::Character, c,)
67any::operator char() const {
68 const auto val = static_cast<decltype(c)>(*this);
69 assert(val <= std::numeric_limits<char>::max());
70 return static_cast<char>(val);
73any::operator size_t() const {
74 const auto val = static_cast<decltype(i)>(*this);
75 assert(val >= 0);
76 return static_cast<size_t>(val);
79any::operator const string&() const {
80 RETURN_VALUE(Tag::String, s, *)
83any::operator const char *() const {
84 RETURN_VALUE(Tag::String, s->c_str(),)
87any::operator const array&() const {
88 RETURN_VALUE(Tag::Array, a, *)
91any::operator const record&() const {
92 RETURN_VALUE(Tag::Record, r, *)
95auto any::extractPointer(IF_DEBUG(const any::Tag t)) const -> void* {
96 RETURN_VALUE(t, POINTER_FROM_MEMBER(p),)
99auto any::unthunkVariant(const any& a) -> const any& {
100 const any * variant = &a;
101 while (variant->tag == Tag::Thunk) {
102 variant = &((*variant->t)(unthunk));
103 }
104 return *variant;
107auto any::operator[](const size_t rhs) const -> const any& {
108 const auto& xs = static_cast<const array&>(*this);
109 return xs[rhs];
112auto any::size() const -> size_t {
113 return static_cast<const array&>(*this).size();
116auto any::empty() const -> bool {
117 return static_cast<const array&>(*this).empty();
120auto any::contains(const Private::Symbol key) const -> bool {
121 // TODO: assumes at least one element -- safe assumption?
122 const auto& m = cast<dict<unknown_size>>(*this);
123 std::remove_reference<decltype(m)>::type::size_type i = 0;
124 do {
125 if (m[i].first == key) {
126 return true;
127 }
128 } while (m[++i].first != nullptr);
129 return false;
132auto any::at(const std::string& key) const -> const any& {
133 return static_cast<const record&>(*this).at(key);
136//-----------------------------------------------------------------------------
137// Operator helper macros
138//-----------------------------------------------------------------------------
140#define DEFINE_COMPARISON_OPERATOR(op) \
141 auto operator op (const any& lhs_, const any& rhs) -> bool { \
142 const any& lhs = any::unthunkVariant(lhs_); \
143 switch (lhs.tag) { \
144 case any::Tag::Integer: return lhs.i op static_cast<decltype(any::i)>(rhs); \
145 case any::Tag::Double: return lhs.d op static_cast<decltype(any::d)>(rhs); \
146 case any::Tag::Character: return lhs.c op static_cast<decltype(any::c)>(rhs); \
147 case any::Tag::Boolean: return lhs.b op static_cast<decltype(any::b)>(rhs); \
148 case any::Tag::String: return *lhs.s op static_cast<const string&>(rhs); \
149 case any::Tag::RawPointer: return lhs.v op cast<decltype(any::v)>(rhs); \
150 case any::Tag::Pointer: return lhs.p op any::unthunkVariant(rhs).p; \
151 default: assert(false && "Unsupported tag for operator " #op); \
152 } \
153 return false; \
154 }
156//-----------------------------------------------------------------------------
157// Operator definitions
158//-----------------------------------------------------------------------------
160DEFINE_COMPARISON_OPERATOR(==)
161DEFINE_COMPARISON_OPERATOR(!=)
162DEFINE_COMPARISON_OPERATOR(<)
163DEFINE_COMPARISON_OPERATOR(<=)
164DEFINE_COMPARISON_OPERATOR(>)
165DEFINE_COMPARISON_OPERATOR(>=)
167auto operator+(const any& lhs_, const any& rhs) -> any {
168 const any& lhs = any::unthunkVariant(lhs_);
169 switch (lhs.tag) {
170 case any::Tag::Integer: return lhs.i + static_cast<decltype(any::i)>(rhs);
171 case any::Tag::Double: return lhs.d + static_cast<decltype(any::d)>(rhs);
172 case any::Tag::Character: return decltype(any::c){lhs.c + static_cast<decltype(any::c)>(rhs)};
173 case any::Tag::String: return *lhs.s + static_cast<const string&>(rhs);
174 default: assert(false && "Unsupported tag for '+' operator");
175 }
176 return nullptr;
179auto operator-(const any& lhs_, const any& rhs) -> any {
180 const any& lhs = any::unthunkVariant(lhs_);
181 switch (lhs.tag) {
182 case any::Tag::Integer: return lhs.i - static_cast<decltype(any::i)>(rhs);
183 case any::Tag::Double: return lhs.d - static_cast<decltype(any::d)>(rhs);
184 case any::Tag::Character: return decltype(any::c){lhs.c - static_cast<decltype(any::c)>(rhs)};
185 default: assert(false && "Unsupported tag for '-' operator");
186 }
187 return nullptr;
190auto operator*(const any& lhs_, const any& rhs) -> any {
191 const any& lhs = any::unthunkVariant(lhs_);
192 switch (lhs.tag) {
193 case any::Tag::Integer: return lhs.i * static_cast<decltype(any::i)>(rhs);
194 case any::Tag::Double: return lhs.d * static_cast<decltype(any::d)>(rhs);
195 case any::Tag::Character: return decltype(lhs.c){lhs.c - static_cast<decltype(any::c)>(rhs)};
196 default: assert(false && "Unsupported tag for '*' operator");
197 }
198 return nullptr;
201auto operator/(const any& lhs_, const any& rhs) -> any {
202 const any& lhs = any::unthunkVariant(lhs_);
203 switch (lhs.tag) {
204 case any::Tag::Integer: return lhs.i / static_cast<decltype(any::i)>(rhs);
205 case any::Tag::Double: return lhs.d / static_cast<decltype(any::d)>(rhs);
206 case any::Tag::Character: return decltype(any::c){lhs.c / static_cast<decltype(any::c)>(rhs)};
207 default: assert(false && "Unsupported tag for '/' operator");
208 }
209 return nullptr;
212auto operator%(const any& lhs_, const any& rhs) -> any {
213 const any& lhs = any::unthunkVariant(lhs_);
214 switch (lhs.tag) {
215 case any::Tag::Integer: return lhs.i % static_cast<decltype(any::i)>(rhs);
216 case any::Tag::Character: return decltype(any::c){lhs.c % static_cast<decltype(any::c)>(rhs)};
217 default: assert(false && "Unsupported tag for '%' operator");
218 }
219 return nullptr;
222// unary negate
223auto operator-(const any& rhs_) -> any {
224 const any& rhs = any::unthunkVariant(rhs_);
225 switch (rhs.tag) {
226 case any::Tag::Integer: return (- rhs.i);
227 case any::Tag::Double: return (- rhs.d);
228 default: assert(false && "Unsupported tag for unary '-' operator");
229 }
230 return nullptr;
233} // namespace PureScript