X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/d09e527c381abbc21b5ad603c3b55fe07c35d505..1504c9b837a7561d8ae2c81f1c4d57fe0f138781:/Output.cpp?ds=sidebyside diff --git a/Output.cpp b/Output.cpp index 1e4d437..d6c3a03 100644 --- a/Output.cpp +++ b/Output.cpp @@ -1,9 +1,45 @@ +/* Cycript - Inlining/Optimizing JavaScript Compiler + * Copyright (C) 2009 Jay Freeman (saurik) +*/ + +/* Modified BSD License {{{ */ +/* + * Redistribution and use in source and binary + * forms, with or without modification, are permitted + * provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the + * above copyright notice, this list of conditions + * and the following disclaimer. + * 2. Redistributions in binary form must reproduce the + * above copyright notice, this list of conditions + * and the following disclaimer in the documentation + * and/or other materials provided with the + * distribution. + * 3. The name of the author may not be used to endorse + * or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, + * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF + * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR + * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN + * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF + * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ +/* }}} */ + +#include "cycript.hpp" #include "Parser.hpp" -#include -#include - -#include #include _finline CYFlags operator ~(CYFlags rhs) { @@ -23,7 +59,7 @@ _finline CYFlags &operator |=(CYFlags &lhs, CYFlags rhs) { } _finline CYFlags CYLeft(CYFlags flags) { - return flags & ~CYNoDangle; + return flags & ~(CYNoDangle | CYNoInteger); } _finline CYFlags CYRight(CYFlags flags) { @@ -34,8 +70,6 @@ _finline CYFlags CYCenter(CYFlags flags) { return CYLeft(CYRight(flags)); } -#define CYPA 16 - void CYOutput::Terminate() { out_ << ';'; mode_ = NoMode; @@ -51,14 +85,22 @@ CYOutput &CYOutput::operator <<(char rhs) { for (unsigned i(0); i != indent_; ++i) out_ << " "; else goto done; - else goto work; - + else if (rhs == '\r') { + if (right_) { + out_ << '\n'; + right_ = false; + } goto done; + } else goto work; + + right_ = true; mode_ = NoMode; goto done; work: - if (mode_ == Terminated && rhs != '}') + if (mode_ == Terminated && rhs != '}') { + right_ = true; out_ << ';'; + } if (rhs == ';') { if (pretty_) @@ -82,6 +124,7 @@ CYOutput &CYOutput::operator <<(char rhs) { } else none: mode_ = NoMode; + right_ = true; out_ << rhs; done: return *this; @@ -107,24 +150,11 @@ CYOutput &CYOutput::operator <<(const char *rhs) { else mode_ = NoMode; + right_ = true; out_ << rhs; return *this; } -void OutputBody(CYOutput &out, CYStatement *body) { - out << ' ' << '{' << '\n'; - ++out.indent_; - if (body != NULL) - body->Multiple(out); - --out.indent_; - out << '\t' << '}'; -} - -void CYAddressOf::Output(CYOutput &out, CYFlags flags) const { - rhs_->Output(out, 1, CYLeft(flags)); - out << ".$cya()"; -} - void CYArgument::Output(CYOutput &out) const { if (name_ != NULL) { out << *name_; @@ -145,21 +175,7 @@ void CYArray::Output(CYOutput &out, CYFlags flags) const { } void CYArrayComprehension::Output(CYOutput &out, CYFlags flags) const { - // XXX: I don't necc. need the ()s - out << "(function($cyv"; - for (CYComprehension *comprehension(comprehensions_); comprehension != NULL; comprehension = comprehension->next_) - if (const char *name = comprehension->Name()) - out << ',' << name; - out << "){"; - out << "$cyv=[];"; - comprehensions_->Output(out); - out << "$cyv.push("; - expression_->Output(out, CYPA, CYNoFlags); - out << ");"; - for (CYComprehension *comprehension(comprehensions_); comprehension != NULL; comprehension = comprehension->next_) - comprehension->End_(out); - out << "return $cyv;"; - out << "}())"; + out << '[' << *expression_ << ' ' << *comprehensions_ << ']'; } void CYAssignment::Output(CYOutput &out, CYFlags flags) const { @@ -168,8 +184,22 @@ void CYAssignment::Output(CYOutput &out, CYFlags flags) const { rhs_->Output(out, Precedence(), CYRight(flags)); } +void CYBlock::Output(CYOutput &out) const { + out << '{' << '\n'; + ++out.indent_; + if (statements_ != NULL) + statements_->Multiple(out); + --out.indent_; + out << '\t' << '}'; +} + void CYBlock::Output(CYOutput &out, CYFlags flags) const { - statements_->Single(out, flags); + if (statements_ == NULL) + out.Terminate(); + else if (statements_->next_ == NULL) + statements_->Single(out, flags); + else + Output(out); } void CYBoolean::Output(CYOutput &out, CYFlags flags) const { @@ -193,56 +223,20 @@ void CYCall::Output(CYOutput &out, CYFlags flags) const { out << ')'; } -void CYCatch::Output(CYOutput &out) const { - out << "catch" << ' ' << '(' << *name_ << ')' << ' ' << '{'; - if (code_ != NULL) - code_->Multiple(out); - out << '}'; -} +namespace cy { +namespace Syntax { -void CYCategory::Output(CYOutput &out, CYFlags flags) const { - out << "(function($cys,$cyp,$cyc,$cyn,$cyt){"; - out << "$cyp=object_getClass($cys);"; - out << "$cyc=$cys;"; - if (messages_ != NULL) - messages_->Output(out, true); - out << "})("; - name_->ClassName(out, true); - out << ')'; - out << ';'; +void Catch::Output(CYOutput &out) const { + out << ' ' << "catch" << ' ' << '(' << *name_ << ')' << ' ' << code_; } -void CYClass::Output(CYOutput &out, CYFlags flags) const { - // XXX: I don't necc. need the ()s - out << "(function($cys,$cyp,$cyc,$cyn,$cyt,$cym){"; - out << "$cyp=object_getClass($cys);"; - out << "$cyc=objc_allocateClassPair($cys,"; - if (name_ != NULL) - name_->ClassName(out, false); - else - out << "$cyq(\"CY$\")"; - out << ",0);"; - out << "$cym=object_getClass($cyc);"; - if (fields_ != NULL) - fields_->Output(out); - if (messages_ != NULL) - messages_->Output(out, false); - out << "objc_registerClassPair($cyc);"; - out << "return $cyc;"; - out << "}("; - if (super_ != NULL) - super_->Output(out, CYPA, CYNoFlags); - else - out << "null"; - out << "))"; -} +} } -void CYClassExpression::Output(CYOutput &out, CYFlags flags) const { - CYClass::Output(out, flags); -} - -void CYClassStatement::Output(CYOutput &out, CYFlags flags) const { - CYClass::Output(out, flags); +void CYComment::Output(CYOutput &out, CYFlags flags) const { + out << '\r'; + out.out_ << value_; + out.right_ = true; + out << '\r'; } void CYCompound::Output(CYOutput &out, CYFlags flags) const { @@ -261,11 +255,6 @@ void CYCompound::Output(CYOutput &out, CYFlags flags) const { expression->Output(out, flags); } -void CYComprehension::Output(CYOutput &out) const { - Begin_(out); - out << next_; -} - void CYCondition::Output(CYOutput &out, CYFlags flags) const { test_->Output(out, Precedence() - 1, CYLeft(flags)); out << ' ' << '?' << ' '; @@ -288,13 +277,13 @@ void CYClause::Output(CYOutput &out) const { else out << "default"; out << ':' << '\n'; - if (code_ != NULL) - code_->Multiple(out, CYNoFlags); + if (statements_ != NULL) + statements_->Multiple(out); out << next_; } const char *CYDeclaration::ForEachIn() const { - return identifier_->Value(); + return identifier_->Word(); } void CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const { @@ -302,12 +291,9 @@ void CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const { Output(out, CYRight(flags)); } -void CYDeclaration::ForEachIn(CYOutput &out) const { - out << *identifier_; -} - void CYDeclaration::Output(CYOutput &out, CYFlags flags) const { out << *identifier_; + //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_; if (initialiser_ != NULL) { out << ' ' << '=' << ' '; initialiser_->Output(out, CYPA, CYRight(flags)); @@ -340,7 +326,7 @@ void CYDeclarations::Output(CYOutput &out, CYFlags flags) const { } void CYDirectMember::Output(CYOutput &out, CYFlags flags) const { - object_->Output(out, Precedence(), CYLeft(flags)); + object_->Output(out, Precedence(), CYLeft(flags) | CYNoInteger); if (const char *word = property_->Word()) out << '.' << word; else @@ -386,10 +372,6 @@ void CYExpression::For(CYOutput &out) const { Output(out, CYNoIn); } -void CYExpression::ForEachIn(CYOutput &out) const { - Output(out, CYPA, CYNoRightHand); -} - void CYExpression::ForIn(CYOutput &out, CYFlags flags) const { Output(out, flags | CYNoRightHand); } @@ -405,15 +387,8 @@ void CYExpression::Output(CYOutput &out, unsigned precedence, CYFlags flags) con Output(out, flags); } -void CYField::Output(CYOutput &out) const { - // XXX: implement! -} - void CYFinally::Output(CYOutput &out) const { - out << "finally" << ' ' << '{'; - if (code_ != NULL) - code_->Multiple(out); - out << '}'; + out << ' ' << "finally" << ' ' << code_; } void CYFor::Output(CYOutput &out, CYFlags flags) const { @@ -429,47 +404,26 @@ void CYFor::Output(CYOutput &out, CYFlags flags) const { } void CYForEachIn::Output(CYOutput &out, CYFlags flags) const { - out << "with({$cys:0,$cyt:0}){"; - - out << "$cys="; - set_->Output(out, CYPA, CYNoFlags); - out << ';'; - - out << "for($cyt in $cys){"; - - initialiser_->ForEachIn(out); - out << "=$cys[$cyt];"; - - code_->Multiple(out); - - out << '}'; - - out << '}'; -} - -void CYForEachInComprehension::Begin_(CYOutput &out) const { - out << "(function($cys){"; - out << "$cys="; - set_->Output(out, CYPA, CYNoFlags); - out << ';'; - - out << "for(" << *name_ << " in $cys){"; - out << *name_ << "=$cys[" << *name_ << "];"; + out << "for" << ' ' << "each" << ' ' << '('; + initialiser_->ForIn(out, CYNoIn); + out << "in" << *set_ << ')'; + code_->Single(out, CYRight(flags)); } -void CYForEachInComprehension::End_(CYOutput &out) const { - out << "}}());"; +void CYForEachInComprehension::Output(CYOutput &out) const { + out << "for" << ' ' << "each" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')' << next_; } void CYForIn::Output(CYOutput &out, CYFlags flags) const { out << "for" << ' ' << '('; - initialiser_->ForIn(out, CYNoIn); + if (initialiser_ != NULL) + initialiser_->ForIn(out, CYNoIn); out << "in" << *set_ << ')'; code_->Single(out, CYRight(flags)); } -void CYForInComprehension::Begin_(CYOutput &out) const { - out << "for" << ' ' << '(' << *name_ << "in" << *set_ << ')'; +void CYForInComprehension::Output(CYOutput &out) const { + out << "for" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')'; } void CYFunction::Output(CYOutput &out, CYFlags flags) const { @@ -481,7 +435,7 @@ void CYFunction::Output(CYOutput &out, CYFlags flags) const { if (name_ != NULL) out << ' ' << *name_; out << '(' << parameters_ << ')'; - OutputBody(out, body_); + out << ' ' << code_; if (protect) out << ')'; } @@ -500,6 +454,10 @@ void CYFunctionParameter::Output(CYOutput &out) const { out << ',' << ' ' << *next_; } +const char *CYIdentifier::Word() const { + return replace_ == NULL || replace_ == this ? CYWord::Word() : replace_->Word(); +} + void CYIf::Output(CYOutput &out, CYFlags flags) const { bool protect(false); if (false_ == NULL && (flags & CYNoDangle) != 0) { @@ -517,10 +475,9 @@ void CYIf::Output(CYOutput &out, CYFlags flags) const { else jacks |= protect ? CYNoFlags : CYCenter(flags); - bool single(true_->Single(out, jacks)); + true_->Single(out, jacks); if (false_ != NULL) { - out << (single ? '\t' : ' '); out << "else"; false_->Single(out, right); } @@ -529,22 +486,16 @@ void CYIf::Output(CYOutput &out, CYFlags flags) const { out << '}'; } -void CYIfComprehension::Begin_(CYOutput &out) const { - out << "if" << '(' << *test_ << ')'; -} - -void CYIndirect::Output(CYOutput &out, CYFlags flags) const { - rhs_->Output(out, 1, CYLeft(flags)); - out << ".$cyi"; +void CYIfComprehension::Output(CYOutput &out) const { + out << "if" << ' ' << '(' << *test_ << ')' << next_; } void CYIndirectMember::Output(CYOutput &out, CYFlags flags) const { object_->Output(out, Precedence(), CYLeft(flags)); - out << ".$cyi"; if (const char *word = property_->Word()) - out << '.' << word; + out << "->" << word; else - out << '[' << *property_ << ']'; + out << "->" << '[' << *property_ << ']'; } void CYInfix::Output(CYOutput &out, CYFlags flags) const { @@ -561,34 +512,13 @@ void CYInfix::Output(CYOutput &out, CYFlags flags) const { out << ')'; } -void CYLet::Output(CYOutput &out, CYFlags flags) const { - out << "let" << ' ' << '(' << *declarations_ << ')' << ' ' << '{'; - if (statements_ != NULL) - statements_->Multiple(out); - out << '}'; +void CYLabel::Output(CYOutput &out, CYFlags flags) const { + out << *name_ << ':' << ' '; + statement_->Single(out, CYRight(flags)); } -void CYMessage::Output(CYOutput &out, bool replace) const { - if (next_ != NULL) - next_->Output(out, replace); - out << "$cyn=new Selector(\""; - for (CYMessageParameter *parameter(parameter_); parameter != NULL; parameter = parameter->next_) - if (parameter->tag_ != NULL) { - out << *parameter->tag_; - if (parameter->name_ != NULL) - out << ':'; - } - out << "\");"; - out << "$cyt=$cyn.type($cy" << (instance_ ? 's' : 'p') << ");"; - out << "class_" << (replace ? "replace" : "add") << "Method($cy" << (instance_ ? 'c' : 'm') << ",$cyn,"; - out << "new Functor(function(self,_cmd"; - for (CYMessageParameter *parameter(parameter_); parameter != NULL; parameter = parameter->next_) - if (parameter->name_ != NULL) - out << ',' << *parameter->name_; - out << "){return function(){"; - if (body_ != NULL) - body_->Multiple(out); - out << "}.call(self);},$cyt),$cyt);"; +void CYLet::Output(CYOutput &out, CYFlags flags) const { + out << "let" << ' ' << '(' << *declarations_ << ')' << ' ' << code_; } void CYNew::Output(CYOutput &out, CYFlags flags) const { @@ -604,9 +534,13 @@ void CYNull::Output(CYOutput &out, CYFlags flags) const { } void CYNumber::Output(CYOutput &out, CYFlags flags) const { - char value[32]; - sprintf(value, "%.17g", Value()); - out << value; + std::ostringstream str; + CYNumerify(str, Value()); + std::string value(str.str()); + out << value.c_str(); + // XXX: this should probably also handle hex conversions and exponents + if ((flags & CYNoInteger) != 0 && value.find('.') == std::string::npos) + out << '.'; } void CYNumber::PropertyName(CYOutput &out) const { @@ -619,7 +553,7 @@ void CYObject::Output(CYOutput &out, CYFlags flags) const { out << '('; out << '{' << '\n'; ++out.indent_; - out << property_; + out << properties_; --out.indent_; out << '\t' << '}'; if (protect) @@ -639,6 +573,11 @@ void CYPrefix::Output(CYOutput &out, CYFlags flags) const { rhs_->Output(out, Precedence(), CYRight(flags)); } +void CYProgram::Output(CYOutput &out) const { + if (statements_ != NULL) + statements_->Multiple(out); +} + void CYProperty::Output(CYOutput &out) const { out << '\t'; name_->PropertyName(out); @@ -661,40 +600,6 @@ void CYReturn::Output(CYOutput &out, CYFlags flags) const { out << ';'; } -void CYSelector::Output(CYOutput &out, CYFlags flags) const { - out << "new Selector(\""; - if (name_ != NULL) - name_->Output(out); - out << "\")"; -} - -void CYSelectorPart::Output(CYOutput &out) const { - out << name_; - if (value_) - out << ':'; - out << next_; -} - -void CYSend::Output(CYOutput &out, CYFlags flags) const { - out << "objc_msgSend("; - self_->Output(out, CYPA, CYNoFlags); - out << ','; - std::ostringstream name; - for (CYArgument *argument(arguments_); argument != NULL; argument = argument->next_) - if (argument->name_ != NULL) { - name << *argument->name_; - if (argument->value_ != NULL) - name << ':'; - } - out.out_ << reinterpret_cast(sel_registerName(name.str().c_str())); - for (CYArgument *argument(arguments_); argument != NULL; argument = argument->next_) - if (argument->value_ != NULL) { - out << ','; - argument->value_->Output(out, CYPA, CYNoFlags); - } - out << ')'; -} - void CYStatement::Multiple(CYOutput &out, CYFlags flags) const { bool first(true); for (const CYStatement *next(this); next != NULL; next = next->next_) { @@ -707,70 +612,19 @@ void CYStatement::Multiple(CYOutput &out, CYFlags flags) const { } } -bool CYStatement::Single(CYOutput &out, CYFlags flags) const { - if (next_ != NULL) { - out << ' ' << '{' << '\n'; - ++out.indent_; - Multiple(out); - --out.indent_; - out << '\t' << '}'; - return false; - } else { - for (CYLabel *label(labels_); label != NULL; label = label->next_) - out << ' ' << *label->name_ << ':'; - out << '\n'; - ++out.indent_; - out << '\t'; - Output(out, flags); - out << '\n'; - --out.indent_; - return true; - } +void CYStatement::Single(CYOutput &out, CYFlags flags) const { + _assert(next_ == NULL); + out << '\n'; + ++out.indent_; + out << '\t'; + Output(out, flags); + out << '\n'; + --out.indent_; } void CYString::Output(CYOutput &out, CYFlags flags) const { - unsigned quot(0), apos(0); - for (const char *value(value_), *end(value_ + size_); value != end; ++value) - if (*value == '"') - ++quot; - else if (*value == '\'') - ++apos; - - bool single(quot > apos); - std::ostringstream str; - - str << (single ? '\'' : '"'); - for (const char *value(value_), *end(value_ + size_); value != end; ++value) - switch (*value) { - case '\\': str << "\\\\"; break; - case '\b': str << "\\b"; break; - case '\f': str << "\\f"; break; - case '\n': str << "\\n"; break; - case '\r': str << "\\r"; break; - case '\t': str << "\\t"; break; - case '\v': str << "\\v"; break; - - case '"': - if (!single) - str << "\\\""; - else goto simple; - break; - - case '\'': - if (single) - str << "\\'"; - else goto simple; - break; - - default: - if (*value < 0x20 || *value >= 0x7f) - str << "\\x" << std::setbase(16) << std::setw(2) << std::setfill('0') << unsigned(*value); - else simple: - str << *value; - } - str << (single ? '\'' : '"'); - + CYStringify(str, value_, size_); out << str.str().c_str(); } @@ -800,8 +654,6 @@ static const char *Reserved_[] = { "let", "yield", - "each", - NULL }; @@ -828,22 +680,22 @@ void CYThis::Output(CYOutput &out, CYFlags flags) const { CYWord::Output(out); } -void CYThrow::Output(CYOutput &out, CYFlags flags) const { +namespace cy { +namespace Syntax { + +void Throw::Output(CYOutput &out, CYFlags flags) const { out << "throw"; if (value_ != NULL) out << ' ' << *value_; out << ';'; } -void CYTry::Output(CYOutput &out, CYFlags flags) const { - out << "try" << ' ' << '{'; - if (code_ != NULL) - code_->Multiple(out); - out << '}'; - out << catch_; - out << finally_; +void Try::Output(CYOutput &out, CYFlags flags) const { + out << "try" << ' ' << code_ << catch_ << finally_; } +} } + void CYVar::Output(CYOutput &out, CYFlags flags) const { out << "var"; declarations_->Output(out, flags); @@ -867,15 +719,21 @@ void CYWith::Output(CYOutput &out, CYFlags flags) const { void CYWord::ClassName(CYOutput &out, bool object) const { if (object) out << "objc_getClass("; - out << '"' << Value() << '"'; + out << '"' << Word() << '"'; if (object) out << ')'; } void CYWord::Output(CYOutput &out) const { - out << Value(); + out << Word(); + if (out.options_.verbose_) + out.out_ << '@' << this; } void CYWord::PropertyName(CYOutput &out) const { Output(out); } + +const char *CYWord::Word() const { + return word_; +}