]> git.saurik.com Git - cycript.git/blame - Replace.cpp
Add macosx.sh and readline.sh for Mac OS X.
[cycript.git] / Replace.cpp
CommitLineData
b3378a02 1/* Cycript - Optimizing JavaScript Compiler/Runtime
c15969fd 2 * Copyright (C) 2009-2013 Jay Freeman (saurik)
4644480a
JF
3*/
4
c15969fd 5/* GNU General Public License, Version 3 {{{ */
4644480a 6/*
c15969fd
JF
7 * Cycript is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published
9 * by the Free Software Foundation, either version 3 of the License,
10 * or (at your option) any later version.
4644480a 11 *
c15969fd
JF
12 * Cycript is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
4644480a 16 *
c15969fd 17 * You should have received a copy of the GNU General Public License
b3378a02
JF
18 * along with Cycript. If not, see <http://www.gnu.org/licenses/>.
19**/
4644480a
JF
20/* }}} */
21
3b52fd1a 22#include "Parser.hpp"
6a981250 23#include "Replace.hpp"
3b52fd1a 24
3b52fd1a
JF
25#include <iomanip>
26
61fe0c53
JF
27CYFunctionExpression *CYNonLocalize(CYContext &context, CYFunctionExpression *function) {
28 function->nonlocal_ = context.nextlocal_;
29 return function;
30}
31
4644480a
JF
32CYExpression *CYAdd::Replace(CYContext &context) {
33 CYInfix::Replace(context);
34
35 CYExpression *lhp(lhs_->Primitive(context));
36 CYExpression *rhp(rhs_->Primitive(context));
37
38 CYString *lhs(dynamic_cast<CYString *>(lhp));
39 CYString *rhs(dynamic_cast<CYString *>(rhp));
40
41 if (lhs != NULL || rhs != NULL) {
42 if (lhs == NULL) {
43 lhs = lhp->String(context);
44 if (lhs == NULL)
029bc65b 45 return this;
4644480a
JF
46 } else if (rhs == NULL) {
47 rhs = rhp->String(context);
48 if (rhs == NULL)
029bc65b 49 return this;
4644480a
JF
50 }
51
52 return lhs->Concat(context, rhs);
53 }
54
55 if (CYNumber *lhn = lhp->Number(context))
56 if (CYNumber *rhn = rhp->Number(context))
57 return $D(lhn->Value() + rhn->Value());
58
029bc65b 59 return this;
4644480a
JF
60}
61
3b52fd1a 62CYExpression *CYAddressOf::Replace(CYContext &context) {
3b52fd1a
JF
63 return $C0($M(rhs_, $S("$cya")));
64}
65
5192746c 66CYArgument *CYArgument::Replace(CYContext &context) { $T(NULL)
3b52fd1a 67 context.Replace(value_);
5192746c
JF
68 next_ = next_->Replace(context);
69
70 if (value_ == NULL) {
71 if (next_ == NULL)
72 return NULL;
73 else
74 value_ = $U;
75 }
76
77 return this;
3b52fd1a
JF
78}
79
80CYExpression *CYArray::Replace(CYContext &context) {
81 elements_->Replace(context);
029bc65b 82 return this;
3b52fd1a
JF
83}
84
85CYExpression *CYArrayComprehension::Replace(CYContext &context) {
86 CYVariable *cyv($V("$cyv"));
87
c8a0500b 88 return $C0($F(NULL, $P1($L("$cyv"), comprehensions_->Parameters(context)), $$->*
3b52fd1a
JF
89 $E($ CYAssign(cyv, $ CYArray()))->*
90 comprehensions_->Replace(context, $E($C1($M(cyv, $S("push")), expression_)))->*
91 $ CYReturn(cyv)
92 ));
93}
94
95CYExpression *CYAssignment::Replace(CYContext &context) {
96 context.Replace(lhs_);
97 context.Replace(rhs_);
029bc65b 98 return this;
3b52fd1a
JF
99}
100
101CYStatement *CYBlock::Replace(CYContext &context) {
cde20a5a 102 context.ReplaceAll(statements_);
029bc65b
JF
103 if (statements_ == NULL)
104 return $ CYEmpty();
105 return this;
3b52fd1a
JF
106}
107
108CYStatement *CYBreak::Replace(CYContext &context) {
029bc65b 109 return this;
3b52fd1a
JF
110}
111
6c093cce
JF
112CYExpression *CYCall::AddArgument(CYContext &context, CYExpression *value) {
113 CYArgument **argument(&arguments_);
114 while (*argument != NULL)
115 argument = &(*argument)->next_;
116 *argument = $ CYArgument(value);
117 return this;
118}
119
3b52fd1a
JF
120CYExpression *CYCall::Replace(CYContext &context) {
121 context.Replace(function_);
122 arguments_->Replace(context);
029bc65b 123 return this;
3b52fd1a
JF
124}
125
37954781
JF
126namespace cy {
127namespace Syntax {
128
129void Catch::Replace(CYContext &context) { $T()
61fe0c53 130 CYScope scope(true, context, code_.statements_);
daf22a65
JF
131
132 context.Replace(name_);
133 context.scope_->Declare(context, name_, CYIdentifierCatch);
134
3b52fd1a 135 code_.Replace(context);
daf22a65 136 scope.Close();
3b52fd1a
JF
137}
138
37954781
JF
139} }
140
3b52fd1a
JF
141void CYClause::Replace(CYContext &context) { $T()
142 context.Replace(case_);
cde20a5a 143 context.ReplaceAll(statements_);
3b52fd1a
JF
144 next_->Replace(context);
145}
146
320ce753 147CYStatement *CYComment::Replace(CYContext &context) {
029bc65b 148 return this;
320ce753
JF
149}
150
3b52fd1a 151CYExpression *CYCompound::Replace(CYContext &context) {
cde20a5a 152 context.ReplaceAll(expressions_);
541654b4
JF
153 if (expressions_ == NULL)
154 return NULL;
9efd8b03 155 return this;
3b52fd1a
JF
156}
157
e06e5ee1
JF
158CYExpression *CYCompound::Primitive(CYContext &context) {
159 CYExpression *expression(expressions_);
160 if (expression == NULL)
161 return NULL;
162 while (expression->next_ != NULL)
163 expression = expression->next_;
164 return expression->Primitive(context);
165}
166
3b52fd1a
JF
167CYFunctionParameter *CYComprehension::Parameters(CYContext &context) const { $T(NULL)
168 CYFunctionParameter *next(next_->Parameters(context));
169 if (CYFunctionParameter *parameter = Parameter(context)) {
170 parameter->SetNext(next);
171 return parameter;
172 } else
173 return next;
174}
175
176CYStatement *CYComprehension::Replace(CYContext &context, CYStatement *statement) const {
177 return next_ == NULL ? statement : next_->Replace(context, statement);
178}
179
180CYExpression *CYCondition::Replace(CYContext &context) {
181 context.Replace(test_);
182 context.Replace(true_);
183 context.Replace(false_);
029bc65b 184 return this;
3b52fd1a
JF
185}
186
ab2aa221
JF
187void CYContext::NonLocal(CYStatement *&statements) {
188 CYContext &context(*this);
189
06293152 190 if (nextlocal_ != NULL && nextlocal_->identifier_ != NULL) {
daf22a65
JF
191 CYIdentifier *cye($I("$cye")->Replace(context));
192 CYIdentifier *unique(nextlocal_->identifier_->Replace(context));
193
194 CYStatement *declare(
c8a0500b 195 $ CYVar($L1($ CYDeclaration(unique, $ CYObject()))));
daf22a65
JF
196
197 cy::Syntax::Catch *rescue(
198 $ cy::Syntax::Catch(cye, $$->*
4e58e244
JF
199 $ CYIf($ CYIdentical($M($V(cye), $S("$cyk")), $V(unique)), $$->*
200 $ CYReturn($M($V(cye), $S("$cyv"))))->*
201 $ cy::Syntax::Throw($V(cye))));
daf22a65 202
577bfbfa 203 context.Replace(declare);
daf22a65 204 rescue->Replace(context);
ab2aa221
JF
205
206 statements = $$->*
daf22a65
JF
207 declare->*
208 $ cy::Syntax::Try(statements, rescue, NULL);
ab2aa221
JF
209 }
210}
211
212CYIdentifier *CYContext::Unique() {
0cbeddf8 213 return $ CYIdentifier($pool.strcat("$cy", $pool.itoa(unique_++), NULL));
ab2aa221
JF
214}
215
3b52fd1a 216CYStatement *CYContinue::Replace(CYContext &context) {
029bc65b
JF
217 return this;
218}
219
c8a0500b
JF
220CYStatement *CYDebugger::Replace(CYContext &context) {
221 return this;
222}
223
029bc65b 224CYAssignment *CYDeclaration::Assignment(CYContext &context) {
15b88a33
JF
225 if (initialiser_ == NULL)
226 return NULL;
127c1a9c
JF
227
228 CYAssignment *value($ CYAssign(Variable(context), initialiser_));
229 initialiser_ = NULL;
230 return value;
15b88a33
JF
231}
232
233CYVariable *CYDeclaration::Variable(CYContext &context) {
234 return $V(identifier_);
3b52fd1a
JF
235}
236
ad3b38bb 237CYStatement *CYDeclaration::ForEachIn(CYContext &context, CYExpression *value) {
c8a0500b 238 return $ CYVar($L1($ CYDeclaration(identifier_, value)));
3b52fd1a
JF
239}
240
029bc65b 241CYExpression *CYDeclaration::Replace(CYContext &context) {
15b88a33
JF
242 context.Replace(identifier_);
243 context.scope_->Declare(context, identifier_, CYIdentifierVariable);
244 return Variable(context);
245}
246
247void CYDeclarations::Replace(CYContext &context) { $T()
248 declaration_->Replace(context);
249 next_->Replace(context);
3b52fd1a
JF
250}
251
550ee46a 252CYProperty *CYDeclarations::Property(CYContext &context) { $T(NULL)
5192746c 253 return $ CYProperty(declaration_->identifier_, declaration_->initialiser_, next_->Property(context));
550ee46a
JF
254}
255
15b88a33 256CYFunctionParameter *CYDeclarations::Parameter(CYContext &context) { $T(NULL)
c8a0500b 257 return $ CYFunctionParameter($ CYDeclaration(declaration_->identifier_), next_->Parameter(context));
15b88a33
JF
258}
259
260CYArgument *CYDeclarations::Argument(CYContext &context) { $T(NULL)
5192746c 261 return $ CYArgument(declaration_->initialiser_, next_->Argument(context));
15b88a33
JF
262}
263
264CYCompound *CYDeclarations::Compound(CYContext &context) { $T(NULL)
265 CYCompound *compound(next_->Compound(context) ?: $ CYCompound());
029bc65b
JF
266 if (CYAssignment *assignment = declaration_->Assignment(context))
267 compound->AddPrev(assignment);
268 return compound;
3b52fd1a
JF
269}
270
271CYExpression *CYDirectMember::Replace(CYContext &context) {
ca0f097f
JF
272 context.Replace(object_);
273 context.Replace(property_);
029bc65b 274 return this;
3b52fd1a
JF
275}
276
277CYStatement *CYDoWhile::Replace(CYContext &context) {
278 context.Replace(test_);
279 context.Replace(code_);
029bc65b 280 return this;
3b52fd1a
JF
281}
282
283void CYElement::Replace(CYContext &context) { $T()
284 context.Replace(value_);
285 next_->Replace(context);
286}
287
288CYStatement *CYEmpty::Replace(CYContext &context) {
9cd63688 289 return NULL;
029bc65b
JF
290}
291
720a57a8
JF
292CYStatement *CYExpress::Replace(CYContext &context) {
293 while (CYExpress *express = dynamic_cast<CYExpress *>(next_)) {
294 CYCompound *compound(dynamic_cast<CYCompound *>(express->expression_));
295 if (compound == NULL)
296 compound = $ CYCompound(express->expression_);
297 compound->AddPrev(expression_);
298 expression_ = compound;
029bc65b
JF
299 SetNext(express->next_);
300 }
301
3b52fd1a 302 context.Replace(expression_);
029bc65b
JF
303 if (expression_ == NULL)
304 return $ CYEmpty();
720a57a8 305
029bc65b 306 return this;
3b52fd1a
JF
307}
308
6c093cce
JF
309CYExpression *CYExpression::AddArgument(CYContext &context, CYExpression *value) {
310 return $C1(this, value);
311}
312
3b52fd1a
JF
313CYExpression *CYExpression::ClassName(CYContext &context, bool object) {
314 return this;
315}
316
ad3b38bb
JF
317CYStatement *CYExpression::ForEachIn(CYContext &context, CYExpression *value) {
318 return $E($ CYAssign(this, value));
3b52fd1a
JF
319}
320
ae65d594
JF
321CYAssignment *CYExpression::Assignment(CYContext &context) {
322 return NULL;
323}
324
4644480a
JF
325CYNumber *CYFalse::Number(CYContext &context) {
326 return $D(0);
327}
328
329CYString *CYFalse::String(CYContext &context) {
330 return $S("false");
331}
332
a0be43fc
JF
333CYExpression *CYFatArrow::Replace(CYContext &context) {
334 CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_));
335 function->this_.SetNext(context.this_);
336 return function;
337}
338
3b52fd1a
JF
339void CYFinally::Replace(CYContext &context) { $T()
340 code_.Replace(context);
341}
342
343CYStatement *CYFor::Replace(CYContext &context) {
029bc65b 344 context.Replace(initialiser_);
3b52fd1a
JF
345 context.Replace(test_);
346 context.Replace(increment_);
347 context.Replace(code_);
029bc65b 348 return this;
3b52fd1a
JF
349}
350
15b88a33
JF
351CYCompound *CYForDeclarations::Replace(CYContext &context) {
352 declarations_->Replace(context);
353 return declarations_->Compound(context);
354}
355
356// XXX: this still feels highly suboptimal
3b52fd1a 357CYStatement *CYForIn::Replace(CYContext &context) {
127c1a9c 358 if (CYAssignment *assignment = initialiser_->Assignment(context))
15b88a33
JF
359 return $ CYBlock($$->*
360 $E(assignment)->*
361 this
362 );
ae65d594 363
127c1a9c
JF
364 context.Replace(initialiser_);
365 context.Replace(set_);
366 context.Replace(code_);
15b88a33 367 return this;
3b52fd1a
JF
368}
369
370CYFunctionParameter *CYForInComprehension::Parameter(CYContext &context) const {
c8a0500b 371 return $ CYFunctionParameter($ CYDeclaration(name_));
3b52fd1a
JF
372}
373
374CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *statement) const {
4e58e244 375 return $ CYForIn($V(name_), set_, CYComprehension::Replace(context, statement));
3b52fd1a
JF
376}
377
d5618df7 378CYStatement *CYForOf::Replace(CYContext &context) {
04ea132d
JF
379 if (CYAssignment *assignment = initialiser_->Assignment(context))
380 return $ CYBlock($$->*
381 $E(assignment)->*
382 this
383 );
384
2eb8215d 385 CYIdentifier *cys($I("$cys")), *cyt($I("$cyt"));
3b52fd1a 386
c8a0500b 387 return $ CYLetStatement($L2($ CYDeclaration(cys, set_), $ CYDeclaration(cyt)), $$->*
2eb8215d 388 $ CYForIn($V(cyt), $V(cys), $ CYBlock($$->*
ad3b38bb 389 initialiser_->ForEachIn(context, $M($V(cys), $V(cyt)))->*
3b52fd1a
JF
390 code_
391 ))
550ee46a 392 );
3b52fd1a
JF
393}
394
d5618df7 395CYFunctionParameter *CYForOfComprehension::Parameter(CYContext &context) const {
c8a0500b 396 return $ CYFunctionParameter($ CYDeclaration(name_));
3b52fd1a
JF
397}
398
d5618df7 399CYStatement *CYForOfComprehension::Replace(CYContext &context, CYStatement *statement) const {
2eb8215d 400 CYIdentifier *cys($I("cys"));
3b52fd1a 401
c8a0500b 402 return $E($C0($F(NULL, $P1($L("$cys")), $$->*
2eb8215d
JF
403 $E($ CYAssign($V(cys), set_))->*
404 $ CYForIn($V(name_), $V(cys), $ CYBlock($$->*
405 $E($ CYAssign($V(name_), $M($V(cys), $V(name_))))->*
3b52fd1a
JF
406 CYComprehension::Replace(context, statement)
407 ))
408 )));
409}
410
14ec9e00 411void CYFunction::Inject(CYContext &context) {
6a981250 412 context.Replace(name_);
a86e34d0 413 context.scope_->Declare(context, name_, CYIdentifierOther);
14ec9e00
JF
414}
415
416void CYFunction::Replace_(CYContext &context, bool outer) {
417 if (outer)
418 Inject(context);
419
a0be43fc
JF
420 CYThisScope *_this(context.this_);
421 context.this_ = CYGetLast(&this_);
422
06293152
JF
423 CYNonLocal *nonlocal(context.nonlocal_);
424 CYNonLocal *nextlocal(context.nextlocal_);
425
ab2aa221 426 bool localize;
06293152 427 if (nonlocal_ != NULL) {
ab2aa221 428 localize = false;
06293152
JF
429 context.nonlocal_ = nonlocal_;
430 } else {
ab2aa221
JF
431 localize = true;
432 nonlocal_ = $ CYNonLocal();
06293152 433 context.nextlocal_ = nonlocal_;
ab2aa221
JF
434 }
435
61fe0c53
JF
436 CYScope scope(!localize, context, code_.statements_);
437
14ec9e00
JF
438 if (!outer && name_ != NULL)
439 Inject(context);
440
c8a0500b 441 parameters_->Replace(context, code_);
3b52fd1a 442 code_.Replace(context);
029bc65b 443
a0be43fc
JF
444 if (CYIdentifier *identifier = this_.identifier_)
445 code_.statements_ = $$->*
446 $ CYVar($L1($ CYDeclaration(identifier, $ CYThis())))->*
447 code_.statements_;
448
ab2aa221
JF
449 if (localize)
450 context.NonLocal(code_.statements_);
06293152
JF
451
452 context.nextlocal_ = nextlocal;
ab2aa221
JF
453 context.nonlocal_ = nonlocal;
454
a0be43fc
JF
455 context.this_ = _this;
456
daf22a65 457 scope.Close();
3b52fd1a
JF
458}
459
460CYExpression *CYFunctionExpression::Replace(CYContext &context) {
14ec9e00 461 Replace_(context, false);
029bc65b
JF
462 return this;
463}
464
c8a0500b
JF
465void CYFunctionParameter::Replace(CYContext &context, CYBlock &code) { $T()
466 CYAssignment *assignment(initialiser_->Assignment(context));
467 context.Replace(initialiser_);
468
469 next_->Replace(context, code);
470
471 if (assignment != NULL)
472 // XXX: this cast is quite incorrect
473 code.AddPrev($ CYIf($ CYIdentical($ CYTypeOf(dynamic_cast<CYExpression *>(initialiser_)), $S("undefined")), $$->*
474 $E(assignment)
475 ));
3b52fd1a
JF
476}
477
478CYStatement *CYFunctionStatement::Replace(CYContext &context) {
14ec9e00 479 Replace_(context, true);
029bc65b
JF
480 return this;
481}
482
483CYIdentifier *CYIdentifier::Replace(CYContext &context) {
a846a8cd 484 if (replace_ != NULL && replace_ != this)
de9fc71b 485 return replace_->Replace(context);
a846a8cd 486 replace_ = context.scope_->Lookup(context, this);
a86e34d0 487 return replace_;
3b52fd1a
JF
488}
489
490CYStatement *CYIf::Replace(CYContext &context) {
491 context.Replace(test_);
492 context.Replace(true_);
493 context.Replace(false_);
029bc65b 494 return this;
3b52fd1a
JF
495}
496
497CYFunctionParameter *CYIfComprehension::Parameter(CYContext &context) const {
498 return NULL;
499}
500
501CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *statement) const {
502 return $ CYIf(test_, CYComprehension::Replace(context, statement));
503}
504
505CYExpression *CYIndirect::Replace(CYContext &context) {
3b52fd1a
JF
506 return $M(rhs_, $S("$cyi"));
507}
508
509CYExpression *CYIndirectMember::Replace(CYContext &context) {
3b52fd1a
JF
510 return $M($ CYIndirect(object_), property_);
511}
512
513CYExpression *CYInfix::Replace(CYContext &context) {
514 context.Replace(lhs_);
515 context.Replace(rhs_);
029bc65b 516 return this;
3b52fd1a
JF
517}
518
519CYStatement *CYLabel::Replace(CYContext &context) {
520 context.Replace(statement_);
029bc65b 521 return this;
3b52fd1a
JF
522}
523
c8a0500b 524CYStatement *CYLetStatement::Replace(CYContext &context) {
61fe0c53 525 return $E($ CYCall(CYNonLocalize(context, $ CYFunctionExpression(NULL, declarations_->Parameter(context), code_)), declarations_->Argument(context)));
550ee46a
JF
526}
527
2eb8215d
JF
528namespace cy {
529namespace Syntax {
530
531CYExpression *New::AddArgument(CYContext &context, CYExpression *value) {
bf45251b 532 CYSetLast(arguments_) = $ CYArgument(value);
6c093cce
JF
533 return this;
534}
535
2eb8215d 536CYExpression *New::Replace(CYContext &context) {
3b52fd1a
JF
537 context.Replace(constructor_);
538 arguments_->Replace(context);
029bc65b 539 return this;
3b52fd1a
JF
540}
541
2eb8215d
JF
542} }
543
4644480a
JF
544CYNumber *CYNull::Number(CYContext &context) {
545 return $D(0);
546}
547
548CYString *CYNull::String(CYContext &context) {
549 return $S("null");
550}
551
552CYNumber *CYNumber::Number(CYContext &context) {
553 return this;
554}
555
556CYString *CYNumber::String(CYContext &context) {
557 // XXX: there is a precise algorithm for this
0cbeddf8 558 return $S($pool.sprintf(24, "%.17g", Value()));
4644480a
JF
559}
560
3b52fd1a
JF
561CYExpression *CYObject::Replace(CYContext &context) {
562 properties_->Replace(context);
029bc65b 563 return this;
3b52fd1a
JF
564}
565
566CYExpression *CYPostfix::Replace(CYContext &context) {
567 context.Replace(lhs_);
029bc65b 568 return this;
3b52fd1a
JF
569}
570
571CYExpression *CYPrefix::Replace(CYContext &context) {
572 context.Replace(rhs_);
029bc65b 573 return this;
3b52fd1a
JF
574}
575
10711823 576// XXX: this is evil evil black magic. don't ask, don't tell... don't believe!
45d0c928 577#define MappingSet "0etnirsoalfucdphmgyvbxTwSNECAFjDLkMOIBPqzRH$_WXUVGYKQJZ"
10711823 578//#define MappingSet "0abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ$_"
45d0c928 579
e013809d
JF
580namespace {
581 struct IdentifierUsageLess :
582 std::binary_function<CYIdentifier *, CYIdentifier *, bool>
583 {
584 _finline bool operator ()(CYIdentifier *lhs, CYIdentifier *rhs) const {
585 if (lhs->usage_ != rhs->usage_)
586 return lhs->usage_ > rhs->usage_;
587 return lhs < rhs;
588 }
589 };
590
591 typedef std::set<CYIdentifier *, IdentifierUsageLess> IdentifierUsages;
592}
593
3b52fd1a 594void CYProgram::Replace(CYContext &context) {
61fe0c53 595 CYScope scope(true, context, statements_);
ab2aa221 596
06293152 597 context.nextlocal_ = $ CYNonLocal();
cde20a5a 598 context.ReplaceAll(statements_);
ab2aa221
JF
599 context.NonLocal(statements_);
600
daf22a65 601 scope.Close();
14ec9e00
JF
602
603 size_t offset(0);
604
6a981250 605 CYCStringSet external;
a846a8cd 606 for (CYIdentifierValueSet::const_iterator i(scope.identifiers_.begin()); i != scope.identifiers_.end(); ++i)
6a981250
JF
607 external.insert((*i)->Word());
608
e013809d
JF
609 IdentifierUsages usages;
610
a846a8cd 611 if (offset < context.rename_.size())
c2c9f509
JF
612 CYForEach (i, context.rename_[offset].identifier_)
613 usages.insert(i);
e013809d 614
14ec9e00 615 // XXX: totalling the probable occurrences and sorting by them would improve the result
a846a8cd 616 for (CYIdentifierUsageVector::const_iterator i(context.rename_.begin()); i != context.rename_.end(); ++i, ++offset) {
de9fc71b
JF
617 //std::cout << *i << ":" << (*i)->offset_ << std::endl;
618
14ec9e00
JF
619 const char *name;
620
621 if (context.options_.verbose_)
0cbeddf8 622 name = $pool.strcat("$", $pool.itoa(offset), NULL);
14ec9e00
JF
623 else {
624 char id[8];
625 id[7] = '\0';
626
627 id:
628 unsigned position(7), local(offset + 1);
629
630 do {
45d0c928
JF
631 unsigned index(local % (sizeof(MappingSet) - 1));
632 local /= sizeof(MappingSet) - 1;
633 id[--position] = MappingSet[index];
14ec9e00
JF
634 } while (local != 0);
635
6a981250 636 if (external.find(id + position) != external.end()) {
14ec9e00
JF
637 ++offset;
638 goto id;
639 }
640
b799113b 641 name = $pool.strmemdup(id + position, 7 - position);
14ec9e00
JF
642 // XXX: at some point, this could become a keyword
643 }
644
c2c9f509 645 CYForEach (identifier, i->identifier_)
14ec9e00
JF
646 identifier->Set(name);
647 }
3b52fd1a
JF
648}
649
650void CYProperty::Replace(CYContext &context) { $T()
651 context.Replace(value_);
652 next_->Replace(context);
5192746c
JF
653 if (value_ == NULL)
654 value_ = $U;
3b52fd1a
JF
655}
656
657CYStatement *CYReturn::Replace(CYContext &context) {
ab2aa221
JF
658 if (context.nonlocal_ != NULL) {
659 CYProperty *value(value_ == NULL ? NULL : $ CYProperty($S("$cyv"), value_));
660 return $ cy::Syntax::Throw($ CYObject(
4e58e244 661 $ CYProperty($S("$cyk"), $V(context.nonlocal_->Target(context)), value)
ab2aa221
JF
662 ));
663 }
664
3b52fd1a 665 context.Replace(value_);
029bc65b
JF
666 return this;
667}
668
6c093cce
JF
669CYExpression *CYRubyBlock::Replace(CYContext &context) {
670 // XXX: this needs to do something much more epic to handle return
671 return call_->AddArgument(context, proc_->Replace(context));
672}
673
674CYExpression *CYRubyProc::Replace(CYContext &context) {
61fe0c53 675 return CYNonLocalize(context, $ CYFunctionExpression(NULL, parameters_, code_));
6c093cce
JF
676}
677
61fe0c53
JF
678CYScope::CYScope(bool transparent, CYContext &context, CYStatement *&statements) :
679 transparent_(transparent),
daf22a65
JF
680 context_(context),
681 statements_(statements),
682 parent_(context.scope_)
683{
684 context_.scope_ = this;
685}
686
e4676127
JF
687CYScope::~CYScope() {
688}
689
daf22a65
JF
690void CYScope::Close() {
691 context_.scope_ = parent_;
692 Scope(context_, statements_);
693}
694
a86e34d0 695void CYScope::Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierFlags flags) {
61fe0c53 696 if (!transparent_ || flags == CYIdentifierArgument || flags == CYIdentifierCatch)
daf22a65 697 internal_.insert(CYIdentifierAddressFlagsMap::value_type(identifier, flags));
61fe0c53
JF
698 else if (parent_ != NULL)
699 parent_->Declare(context, identifier, flags);
a86e34d0
JF
700}
701
702CYIdentifier *CYScope::Lookup(CYContext &context, CYIdentifier *identifier) {
703 std::pair<CYIdentifierValueSet::iterator, bool> insert(identifiers_.insert(identifier));
704 return *insert.first;
705}
706
6a981250 707void CYScope::Merge(CYContext &context, CYIdentifier *identifier) {
10711823 708 std::pair<CYIdentifierValueSet::iterator, bool> insert(identifiers_.insert(identifier));
6a981250
JF
709 if (!insert.second) {
710 if ((*insert.first)->offset_ < identifier->offset_)
711 (*insert.first)->offset_ = identifier->offset_;
712 identifier->replace_ = *insert.first;
e013809d 713 (*insert.first)->usage_ += identifier->usage_ + 1;
029bc65b
JF
714 }
715}
716
10711823
JF
717namespace {
718 struct IdentifierOffset {
719 size_t offset_;
720 CYIdentifierFlags flags_;
e013809d 721 size_t usage_;
10711823
JF
722 CYIdentifier *identifier_;
723
e013809d
JF
724 IdentifierOffset(CYIdentifier *identifier, CYIdentifierFlags flags) :
725 offset_(identifier->offset_),
10711823 726 flags_(flags),
e013809d 727 usage_(identifier->usage_),
10711823
JF
728 identifier_(identifier)
729 {
730 }
731 };
732
733 struct IdentifierOffsetLess :
734 std::binary_function<const IdentifierOffset &, const IdentifierOffset &, bool>
735 {
736 _finline bool operator ()(const IdentifierOffset &lhs, const IdentifierOffset &rhs) const {
737 if (lhs.offset_ != rhs.offset_)
738 return lhs.offset_ < rhs.offset_;
739 if (lhs.flags_ != rhs.flags_)
740 return lhs.flags_ < rhs.flags_;
e013809d 741 /*if (lhs.usage_ != rhs.usage_)
a846a8cd 742 return lhs.usage_ < rhs.usage_;*/
10711823
JF
743 return lhs.identifier_ < rhs.identifier_;
744 }
745 };
746
747 typedef std::set<IdentifierOffset, IdentifierOffsetLess> IdentifierOffsets;
748}
749
029bc65b 750void CYScope::Scope(CYContext &context, CYStatement *&statements) {
2c81c6df
JF
751 if (parent_ == NULL)
752 return;
753
029bc65b 754 CYDeclarations *last(NULL), *curr(NULL);
14ec9e00 755
10711823 756 IdentifierOffsets offsets;
6a981250 757
10711823 758 for (CYIdentifierAddressFlagsMap::const_iterator i(internal_.begin()); i != internal_.end(); ++i)
a846a8cd 759 if (i->second != CYIdentifierMagic)
e013809d 760 offsets.insert(IdentifierOffset(i->first, i->second));
10711823
JF
761
762 size_t offset(0);
763
764 for (IdentifierOffsets::const_iterator i(offsets.begin()); i != offsets.end(); ++i) {
765 if (i->flags_ == CYIdentifierVariable) {
766 CYDeclarations *next($ CYDeclarations($ CYDeclaration(i->identifier_)));
029bc65b
JF
767 if (last == NULL)
768 last = next;
769 if (curr != NULL)
770 curr->SetNext(next);
771 curr = next;
772 }
029bc65b 773
10711823
JF
774 if (offset < i->offset_)
775 offset = i->offset_;
a846a8cd
JF
776 if (context.rename_.size() <= offset)
777 context.rename_.resize(offset + 1);
10711823 778
a846a8cd 779 CYIdentifierUsage &rename(context.rename_[offset++]);
e013809d
JF
780 i->identifier_->SetNext(rename.identifier_);
781 rename.identifier_ = i->identifier_;
782 rename.usage_ += i->identifier_->usage_ + 1;
6a981250
JF
783 }
784
029bc65b
JF
785 if (last != NULL) {
786 CYVar *var($ CYVar(last));
787 var->SetNext(statements);
788 statements = var;
789 }
790
6a981250
JF
791 for (CYIdentifierValueSet::const_iterator i(identifiers_.begin()); i != identifiers_.end(); ++i)
792 if (internal_.find(*i) == internal_.end()) {
de9fc71b 793 //std::cout << *i << '=' << offset << std::endl;
6a981250
JF
794 if ((*i)->offset_ < offset)
795 (*i)->offset_ = offset;
796 parent_->Merge(context, *i);
797 }
029bc65b
JF
798}
799
4644480a
JF
800CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
801 size_t size(size_ + rhs->size_);
2eb8215d 802 char *value($ char[size + 1]);
4644480a
JF
803 memcpy(value, value_, size_);
804 memcpy(value + size_, rhs->value_, rhs->size_);
805 value[size] = '\0';
a14eb702 806 return $S(value, size);
4644480a
JF
807}
808
809CYNumber *CYString::Number(CYContext &context) {
810 // XXX: there is a precise algorithm for this
811 return NULL;
812}
813
814CYString *CYString::String(CYContext &context) {
815 return this;
816}
817
3b52fd1a
JF
818CYStatement *CYSwitch::Replace(CYContext &context) {
819 context.Replace(value_);
820 clauses_->Replace(context);
029bc65b 821 return this;
3b52fd1a
JF
822}
823
824CYExpression *CYThis::Replace(CYContext &context) {
a0be43fc
JF
825 if (context.this_ != NULL)
826 return $V(context.this_->Identifier(context));
029bc65b 827 return this;
3b52fd1a
JF
828}
829
37954781
JF
830namespace cy {
831namespace Syntax {
832
833CYStatement *Throw::Replace(CYContext &context) {
3b52fd1a 834 context.Replace(value_);
029bc65b 835 return this;
3b52fd1a
JF
836}
837
37954781
JF
838} }
839
3b52fd1a 840CYExpression *CYTrivial::Replace(CYContext &context) {
029bc65b 841 return this;
3b52fd1a
JF
842}
843
4644480a
JF
844CYNumber *CYTrue::Number(CYContext &context) {
845 return $D(1);
846}
847
848CYString *CYTrue::String(CYContext &context) {
849 return $S("true");
850}
851
37954781
JF
852namespace cy {
853namespace Syntax {
854
855CYStatement *Try::Replace(CYContext &context) {
3b52fd1a
JF
856 code_.Replace(context);
857 catch_->Replace(context);
858 finally_->Replace(context);
029bc65b 859 return this;
3b52fd1a
JF
860}
861
37954781
JF
862} }
863
3b52fd1a 864CYStatement *CYVar::Replace(CYContext &context) {
15b88a33
JF
865 declarations_->Replace(context);
866 return $E(declarations_->Compound(context));
3b52fd1a
JF
867}
868
869CYExpression *CYVariable::Replace(CYContext &context) {
577bfbfa 870 context.Replace(name_);
029bc65b 871 return this;
3b52fd1a
JF
872}
873
874CYStatement *CYWhile::Replace(CYContext &context) {
875 context.Replace(test_);
876 context.Replace(code_);
029bc65b 877 return this;
3b52fd1a
JF
878}
879
880CYStatement *CYWith::Replace(CYContext &context) {
881 context.Replace(scope_);
882 context.Replace(code_);
029bc65b 883 return this;
3b52fd1a
JF
884}
885
886CYExpression *CYWord::ClassName(CYContext &context, bool object) {
887 CYString *name($S(this));
888 if (object)
889 return $C1($V("objc_getClass"), name);
890 else
891 return name;
892}