Replace.cpp

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