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

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

on 22aec46 help sign in

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

294 lines 10 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);

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 (*variant.k)();

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 = 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 sz = cast<int>(*this);

75 assert(sz >= 0);

76 return static_cast<size_t>(sz);

77}

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

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

81 if (variant.tag == Tag::StringLiteral) {

82 return variant.r;

83 }

84 assert(variant.tag == Tag::String);

85 return variant.s->c_str();

86}

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

89 RETURN_VALUE(Tag::Array, a, *)

90}

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

93 RETURN_VALUE(t, POINTER_FROM_MEMBER(p),)

94}

96auto any::unthunkVariant(const any& a) -> const any& {

97 const any * variant = &a;

98 while (variant->tag == Tag::Thunk) {

99 variant = &((*variant->t)(unthunk));

100 }

101 return *variant;

102}

103

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

105 const auto& xs = cast<array>(*this);

106 return xs[rhs];

107}

108

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

110 return cast<array>(*this).size();

111}

112

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

114 return cast<array>(*this).empty();

115}

116

117auto any::contains(const Private::Symbol key) const -> bool {

118 // TODO: assumes at least one element -- safe assumption?

119 const auto& m = cast<map<unknown_size>>(*this);

120 std::remove_reference<decltype(m)>::type::size_type i = 0;

121 do {

122 if (m[i].first == key) {

123 return true;

124 }

125 } while (m[++i].first != nullptr);

126 return false;

127}

128

129//-----------------------------------------------------------------------------

130// Operator helper macros

131//-----------------------------------------------------------------------------

132

133#define DEFINE_CSTR_EQUALS_OPERATOR() \

134 auto operator==(const any& lhs_, const char * rhs) -> bool { \

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

136 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String); \

137 if (lhs.tag == any::Tag::StringLiteral) { \

138 return (lhs.r == rhs) || (strcmp(lhs.r, rhs) == 0); \

139 } \

140 return strcmp(lhs.s->c_str(), rhs) == 0; \

141 } \

142 auto operator==(const char * lhs, const any& rhs_) -> bool { \

143 const any& rhs = any::unthunkVariant(rhs_); \

144 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String); \

145 if (rhs.tag == any::Tag::StringLiteral) { \

146 return (lhs == rhs.r) || (strcmp(lhs, rhs) == 0); \

147 } \

148 return strcmp(lhs, rhs.s->c_str()) == 0; \

149 }

150

151#define DEFINE_CSTR_COMPARISON_OPERATOR(op) \

152 auto operator op (const any& lhs, const char * rhs) -> bool { \

153 return strcmp(cast<const char *>(lhs), rhs) op 0; \

154 } \

155 auto operator op (const char * lhs, const any& rhs) -> bool { \

156 return strcmp(lhs, cast<const char *>(rhs)) op 0; \

157 }

158

159#define DEFINE_COMPARISON_OPERATOR(op) \

160 auto operator op (const any& lhs_, const any& rhs_) -> bool { \

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

162 const any& rhs = any::unthunkVariant(rhs_); \

163 switch (lhs.tag) { \

164 case any::Tag::Integer: assert(lhs.tag == any::Tag::Integer); return lhs.i op rhs.i; \

165 case any::Tag::Double: assert(lhs.tag == any::Tag::Double); return lhs.d op rhs.d; \

166 case any::Tag::Character: assert(lhs.tag == any::Tag::Character); return lhs.c op rhs.c; \

167 case any::Tag::Boolean: assert(lhs.tag == any::Tag::Boolean); return lhs.b op rhs.b; \

168 case any::Tag::StringLiteral: \

169 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String); \

170 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String); \

171 return strcmp(lhs.r, rhs.tag == any::Tag::StringLiteral ? rhs.r : rhs.s->c_str()) op 0; \

172 case any::Tag::String: \

173 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String); \

174 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String); \

175 return strcmp(lhs.s->c_str(), rhs.tag == any::Tag::StringLiteral ? rhs.r : rhs.s->c_str()) op 0; \

176 case any::Tag::Pointer: return lhs.p op rhs.p; \

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

178 } \

179 return false; \

180 }

181

182//-----------------------------------------------------------------------------

183// Operator definitions

184//-----------------------------------------------------------------------------

185

186DEFINE_COMPARISON_OPERATOR(==)

187DEFINE_COMPARISON_OPERATOR(!=)

188DEFINE_COMPARISON_OPERATOR(<)

189DEFINE_COMPARISON_OPERATOR(<=)

190DEFINE_COMPARISON_OPERATOR(>)

191DEFINE_COMPARISON_OPERATOR(>=)

192

193DEFINE_CSTR_EQUALS_OPERATOR()

194DEFINE_CSTR_COMPARISON_OPERATOR(!=)

195DEFINE_CSTR_COMPARISON_OPERATOR(<)

196DEFINE_CSTR_COMPARISON_OPERATOR(<=)

197DEFINE_CSTR_COMPARISON_OPERATOR(>)

198DEFINE_CSTR_COMPARISON_OPERATOR(>=)

199

200auto operator+(const any& lhs_, const any& rhs_) -> any {

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

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

203 switch (lhs.tag) {

204 case any::Tag::Integer: assert(lhs.tag == any::Tag::Integer); return lhs.i + rhs.i;

205 case any::Tag::Double: assert(lhs.tag == any::Tag::Double); return lhs.d + rhs.d;

206 case any::Tag::Character: assert(lhs.tag == any::Tag::Character); return decltype(any::c){lhs.c + rhs.c};

207 case any::Tag::StringLiteral:

208 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String);

209 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String);

210 return rhs.tag == any::Tag::StringLiteral ? string(lhs.r) + rhs.r : lhs.r + *rhs.s;

211 case any::Tag::String:

212 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String);

213 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String);

214 return rhs.tag == any::Tag::StringLiteral ? *lhs.s + rhs.r : *lhs.s + *rhs.s;

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

216 }

217 return nullptr;

218}

219

220auto operator+(const any& lhs_, const char * rhs) -> string {

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

222 assert(lhs.tag == any::Tag::StringLiteral || lhs.tag == any::Tag::String);

223 return lhs.tag == any::Tag::StringLiteral ? string(lhs.r) + rhs : *lhs.s + rhs;

224}

225

226auto operator+(const char * lhs, const any& rhs_) -> string {

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

228 assert(rhs.tag == any::Tag::StringLiteral || rhs.tag == any::Tag::String);

229 return rhs.tag == any::Tag::StringLiteral ? lhs + string(rhs.r) : lhs + *rhs.s;

230}

231

232auto operator-(const any& lhs_, const any& rhs_) -> any {

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

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

235 assert(lhs.tag == rhs.tag);

236 switch (lhs.tag) {

237 case any::Tag::Integer: return lhs.i - rhs.i;

238 case any::Tag::Double: return lhs.d - rhs.d;

239 case any::Tag::Character: return decltype(any::c){lhs.c - rhs.c};

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

241 }

242 return nullptr;

243}

244

245auto operator*(const any& lhs_, const any& rhs_) -> any {

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

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

248 assert(lhs.tag == rhs.tag);

249 switch (lhs.tag) {

250 case any::Tag::Integer: return lhs.i * rhs.i;

251 case any::Tag::Double: return lhs.d * rhs.d;

252 case any::Tag::Character: return decltype(lhs.c){lhs.c - rhs.c};

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

254 }

255 return nullptr;

256}

257

258auto operator/(const any& lhs_, const any& rhs_) -> any {

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

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

261 assert(lhs.tag == rhs.tag);

262 switch (lhs.tag) {

263 case any::Tag::Integer: return lhs.i / rhs.i;

264 case any::Tag::Double: return lhs.d / rhs.d;

265 case any::Tag::Character: return decltype(any::c){lhs.c / rhs.c};

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

267 }

268 return nullptr;

269}

270

271auto operator%(const any& lhs_, const any& rhs_) -> any {

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

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

274 assert(lhs.tag == rhs.tag);

275 switch (lhs.tag) {

276 case any::Tag::Integer: return lhs.i % rhs.i;

277 case any::Tag::Character: return decltype(any::c){lhs.c % rhs.c};

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

279 }

280 return nullptr;

281}

282

283// unary negate

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

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

286 switch (rhs.tag) {

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

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

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

290 }

291 return nullptr;

292}

293

294} // namespace PureScript