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