pcc/runtime/purescript.cc · rowtype-yoga/purescript-native

git.iko.soy/rowtype-yoga/purescript-native

on 4e7cfee help sign in

rowtype-yoga/purescript-native/pcc/runtime/purescript.cc

239 lines 8.0 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 (*static_cast<const Closure*>(POINTER_FROM_MEMBER(variant.p)))(arg);

32 }

33 assert(variant.tag == Tag::Function);

34 return (*variant.f)(arg);

35}

37auto any::operator()(const as_thunk) const -> const any& {

38 assert(tag == Tag::Thunk);

39 return (*t)(unthunk);

40}

42auto any::operator()() const -> any {

43 const any& variant = unthunkVariant(*this);

44 if (variant.tag == Tag::EffClosure) {

45 return (*static_cast<const EffClosure*>(POINTER_FROM_MEMBER(variant.p)))();

46 }

47 assert(variant.tag == Tag::EffFunction);

48 return (*variant.e)();

49}

51any::operator int() const {

52 RETURN_VALUE(Tag::Integer, i,)

53}

55any::operator double() const {

56 RETURN_VALUE(Tag::Double, d,)

57}

59any::operator bool() const {

60 RETURN_VALUE(Tag::Boolean, b,)

61}

63any::operator char32_t() const {

64 RETURN_VALUE(Tag::Character, c,)

65}

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);

71}

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);

77}

79any::operator const string&() const {

80 return *static_cast<const string*>(extractPointer(IF_DEBUG(Tag::String)));

81}

83any::operator const char *() const {

84 return static_cast<const string*>(extractPointer(IF_DEBUG(Tag::String)))->c_str();

85}

87any::operator const array&() const {

88 return *static_cast<const array*>(extractPointer(IF_DEBUG(Tag::Array)));

89}

91any::operator const record&() const {

92 return *static_cast<const record*>(extractPointer(IF_DEBUG(Tag::Record)));

93}

95auto any::extractPointer(IF_DEBUG(const any::Tag t)) const -> void* {

96 RETURN_VALUE(t, POINTER_FROM_MEMBER(p),)

97}

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;

105}

106

107auto any::operator[](const size_t rhs) const -> const any& {

108 const auto& xs = static_cast<const array&>(*this);

109 return xs[rhs];

110}

111

112auto any::size() const -> size_t {

113 return static_cast<const array&>(*this).size();

114}

115

116auto any::empty() const -> bool {

117 return static_cast<const array&>(*this).empty();

118}

119

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;

130}

131

132auto any::cstr_cmp::operator ()(const char * a, const char * b) const -> bool {

133 return a == b ? false : std::strcmp(a, b) < 0;

134}

135

136auto any::at(const char * key) const -> const any& {

137 return static_cast<const record&>(*this).at(key);

138}

139

140//-----------------------------------------------------------------------------

141// Operator helper macros

142//-----------------------------------------------------------------------------

143

144#define DEFINE_COMPARISON_OPERATOR(op) \

145 auto operator op (const any& lhs_, const any& rhs) -> bool { \

146 const any& lhs = any::unthunkVariant(lhs_); \

147 switch (lhs.tag) { \

148 case any::Tag::Integer: return lhs.i op static_cast<decltype(any::i)>(rhs); \

149 case any::Tag::Double: return lhs.d op static_cast<decltype(any::d)>(rhs); \

150 case any::Tag::Character: return lhs.c op static_cast<decltype(any::c)>(rhs); \

151 case any::Tag::Boolean: return lhs.b op static_cast<decltype(any::b)>(rhs); \

152 case any::Tag::String: return *static_cast<const string*>(POINTER_FROM_MEMBER(lhs.p)) \

153 op static_cast<const string&>(rhs); \

154 case any::Tag::RawPointer: return lhs.v op cast<decltype(any::v)>(rhs); \

155 case any::Tag::Pointer: return lhs.p op any::unthunkVariant(rhs).p; \

156 default: assert(false && "Unsupported tag for operator " #op); \

157 } \

158 return false; \

159 }

160

161//-----------------------------------------------------------------------------

162// Operator definitions

163//-----------------------------------------------------------------------------

164

165DEFINE_COMPARISON_OPERATOR(==)

166DEFINE_COMPARISON_OPERATOR(!=)

167DEFINE_COMPARISON_OPERATOR(<)

168DEFINE_COMPARISON_OPERATOR(<=)

169DEFINE_COMPARISON_OPERATOR(>)

170DEFINE_COMPARISON_OPERATOR(>=)

171

172auto operator+(const any& lhs_, const any& rhs) -> any {

173 const any& lhs = any::unthunkVariant(lhs_);

174 switch (lhs.tag) {

175 case any::Tag::Integer: return lhs.i + static_cast<decltype(any::i)>(rhs);

176 case any::Tag::Double: return lhs.d + static_cast<decltype(any::d)>(rhs);

177 case any::Tag::Character: return decltype(any::c){lhs.c + static_cast<decltype(any::c)>(rhs)};

178 case any::Tag::String: return *static_cast<const string*>(POINTER_FROM_MEMBER(lhs.p))

179 + static_cast<const string&>(rhs);

180 default: assert(false && "Unsupported tag for '+' operator");

181 }

182 return nullptr;

183}

184

185auto operator-(const any& lhs_, const any& rhs) -> any {

186 const any& lhs = any::unthunkVariant(lhs_);

187 switch (lhs.tag) {

188 case any::Tag::Integer: return lhs.i - static_cast<decltype(any::i)>(rhs);

189 case any::Tag::Double: return lhs.d - static_cast<decltype(any::d)>(rhs);

190 case any::Tag::Character: return decltype(any::c){lhs.c - static_cast<decltype(any::c)>(rhs)};

191 default: assert(false && "Unsupported tag for '-' operator");

192 }

193 return nullptr;

194}

195

196auto operator*(const any& lhs_, const any& rhs) -> any {

197 const any& lhs = any::unthunkVariant(lhs_);

198 switch (lhs.tag) {

199 case any::Tag::Integer: return lhs.i * static_cast<decltype(any::i)>(rhs);

200 case any::Tag::Double: return lhs.d * static_cast<decltype(any::d)>(rhs);

201 case any::Tag::Character: return decltype(lhs.c){lhs.c - static_cast<decltype(any::c)>(rhs)};

202 default: assert(false && "Unsupported tag for '*' operator");

203 }

204 return nullptr;

205}

206

207auto operator/(const any& lhs_, const any& rhs) -> any {

208 const any& lhs = any::unthunkVariant(lhs_);

209 switch (lhs.tag) {

210 case any::Tag::Integer: return lhs.i / static_cast<decltype(any::i)>(rhs);

211 case any::Tag::Double: return lhs.d / static_cast<decltype(any::d)>(rhs);

212 case any::Tag::Character: return decltype(any::c){lhs.c / static_cast<decltype(any::c)>(rhs)};

213 default: assert(false && "Unsupported tag for '/' operator");

214 }

215 return nullptr;

216}

217

218auto operator%(const any& lhs_, const any& rhs) -> any {

219 const any& lhs = any::unthunkVariant(lhs_);

220 switch (lhs.tag) {

221 case any::Tag::Integer: return lhs.i % static_cast<decltype(any::i)>(rhs);

222 case any::Tag::Character: return decltype(any::c){lhs.c % static_cast<decltype(any::c)>(rhs)};

223 default: assert(false && "Unsupported tag for '%' operator");

224 }

225 return nullptr;

226}

227

228// unary negate

229auto operator-(const any& rhs_) -> any {

230 const any& rhs = any::unthunkVariant(rhs_);

231 switch (rhs.tag) {

232 case any::Tag::Integer: return (- rhs.i);

233 case any::Tag::Double: return (- rhs.d);

234 default: assert(false && "Unsupported tag for unary '-' operator");

235 }

236 return nullptr;

237}

238

239} // namespace PureScript