source: src/Parser/ExpressionNode.cc @ 35f9114

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// ExpressionNode.cc --
//
// Author           : Rodolfo G. Esteves
// Created On       : Sat May 16 13:17:07 2015
// Last Modified By : Peter A. Buhr
// Last Modified On : Fri Aug  5 15:07:19 2016
// Update Count     : 409
//

#include <cassert>
#include <cctype>
#include <algorithm>
#include <sstream>
#include <cstdio>

#include "ParseNode.h"
#include "TypeData.h"
#include "SynTree/Constant.h"
#include "SynTree/Expression.h"
#include "SynTree/Declaration.h"
#include "Common/UnimplementedError.h"
#include "parseutility.h"
#include "Common/utility.h"

using namespace std;

ExpressionNode::ExpressionNode() : ParseNode() {}

ExpressionNode::ExpressionNode( const string *name ) : ParseNode( name ) {}

ExpressionNode::ExpressionNode( const ExpressionNode &other ) : ParseNode( other.name ), extension( other.extension ) {
        if ( other.argName ) {
                argName = other.argName->clone();
        } else {
                argName = 0;
        } // if
}

ExpressionNode * ExpressionNode::set_argName( const std::string *aName ) {
        argName = new VarRefNode( aName );
        return this;
}

ExpressionNode * ExpressionNode::set_argName( ExpressionNode *aDesignator ) {
        argName = aDesignator;
        return this;
}

void ExpressionNode::printDesignation( std::ostream &os, int indent ) const {
        if ( argName ) {
                os << string( indent, ' ' ) << "(designated by:  ";
                argName->printOneLine( os, indent );
                os << ")" << std::endl;
        } // if
}

//##############################################################################

NullExprNode::NullExprNode() {}

NullExprNode *NullExprNode::clone() const {
        return new NullExprNode();
}

void NullExprNode::print( std::ostream & os, int indent ) const {
        printDesignation( os );
        os << "null expression";
}

void NullExprNode::printOneLine( std::ostream & os, int indent ) const {
        printDesignation( os );
        os << "null";
}

Expression *NullExprNode::build() const {
        return 0;
}

// CommaExprNode *ExpressionNode::add_to_list( ExpressionNode *exp ) {
//      return new CommaExprNode( this, exp );
// }

//##############################################################################

ConstantNode::ConstantNode( ConstantExpr *expr ) : expr( expr ) {
} // ConstantNode::ConstantNode

ConstantNode *ConstantNode::appendstr( const std::string *newValue ) {
        assert( newValue != 0 );

        string value = expr->get_constant()->get_value();

        // "abc" "def" "ghi" => "abcdefghi", remove new text from quotes and insert before last quote in old string.
        value.insert( value.length() - 1, newValue->substr( 1, newValue->length() - 2 ) );
        expr->get_constant()->set_value( value );

        delete newValue;                                                                        // allocated by lexer
        return this;
}

void ConstantNode::printOneLine( std::ostream &os, int indent ) const {
        // os << string( indent, ' ' );
        // printDesignation( os );

        // switch ( type ) {
        //   case Integer:
        //   case Float:
        //      os << value ;
        //      break;
        //   case Character:
        //      os << "'" << value << "'";
        //      break;
        //   case String:
        //      os << '"' << value << '"';
        //      break;
        // } // switch

        // os << ' ';
}

void ConstantNode::print( std::ostream &os, int indent ) const {
        printOneLine( os, indent );
        os << endl;
}

Expression *ConstantNode::build() const {
        return expr->clone();
}

//##############################################################################

VarRefNode::VarRefNode() : isLabel( false ) {}

VarRefNode::VarRefNode( const string *name_, bool labelp ) : ExpressionNode( name_ ), isLabel( labelp ) {}

VarRefNode::VarRefNode( const VarRefNode &other ) : ExpressionNode( other ), isLabel( other.isLabel ) {
}

Expression *VarRefNode::build() const {
        return new NameExpr( get_name(), maybeBuild< Expression >( get_argName() ) );
}

void VarRefNode::printOneLine( std::ostream &os, int indent ) const {
        printDesignation( os );
        os << get_name() << ' ';
}

void VarRefNode::print( std::ostream &os, int indent ) const {
        printDesignation( os );
        os << string( indent, ' ' ) << "Referencing: ";
        os << "Variable: " << get_name();
        os << endl;
}

//##############################################################################

DesignatorNode::DesignatorNode( ExpressionNode *expr, bool isArrayIndex ) : isArrayIndex( isArrayIndex ) {
        set_argName( expr );
        assert( get_argName() );

        if ( ! isArrayIndex ) {
                if ( VarRefNode * var = dynamic_cast< VarRefNode * >( expr ) ) {

                        stringstream ss( var->get_name() );
                        double value;
                        if ( ss >> value ) {
                                // this is a floating point constant. It MUST be
                                // ".0" or ".1", otherwise the program is invalid
                                if ( ! (var->get_name() == ".0" || var->get_name() == ".1") ) {
                                        throw SemanticError( "invalid designator name: " + var->get_name() );
                                } // if
                                var->set_name( var->get_name().substr(1) );
                        } // if
                } // if
        } // if
}

DesignatorNode::DesignatorNode( const DesignatorNode &other ) : ExpressionNode( other ), isArrayIndex( other.isArrayIndex ) {
}

class DesignatorFixer : public Mutator {
public:
        virtual Expression* mutate( NameExpr *nameExpr ) {
                if ( nameExpr->get_name() == "0" || nameExpr->get_name() == "1" ) {
                        Constant val( new BasicType( Type::Qualifiers(), BasicType::SignedInt ), nameExpr->get_name() );
                        delete nameExpr;
                        return new ConstantExpr( val );
                }
                return nameExpr;
        }
};

Expression *DesignatorNode::build() const {
        Expression * ret = maybeBuild<Expression>(get_argName());

        if ( isArrayIndex ) {
                // need to traverse entire structure and change any instances of 0 or 1 to
                // ConstantExpr
                DesignatorFixer fixer;
                ret = ret->acceptMutator( fixer );
        } // if

        return ret;
}

void DesignatorNode::printOneLine( std::ostream &os, int indent ) const {
        if ( get_argName() ) {
                if ( isArrayIndex ) {
                        os << "[";
                        get_argName()->printOneLine( os, indent );
                        os << "]";
                } else {
                        os << ".";
                        get_argName()->printOneLine( os, indent );
                }
        } // if
}

void DesignatorNode::print( std::ostream &os, int indent ) const {
        if ( get_argName() ) {
                if ( isArrayIndex ) {
                        os << "[";
                        get_argName()->print( os, indent );
                        os << "]";
                } else {
                        os << ".";
                        get_argName()->print( os, indent );
                }
        } // if
}

//##############################################################################

static const char *OperName[] = {
        // diadic
        "SizeOf", "AlignOf", "OffsetOf", "Attr", "?+?", "?-?", "?*?", "?/?", "?%?", "||", "&&",
        "?|?", "?&?", "?^?", "Cast", "?<<?", "?>>?", "?<?", "?>?", "?<=?", "?>=?", "?==?", "?!=?",
        "?=?", "?*=?", "?/=?", "?%=?", "?+=?", "?-=?", "?<<=?", "?>>=?", "?&=?", "?^=?", "?|=?",
        "?[?]", "...",
        // monadic
        "+?", "-?", "AddressOf", "*?", "!?", "~?", "++?", "?++", "--?", "?--", "&&"
};

//##############################################################################

CompositeExprNode::CompositeExprNode( Expression *expr ) : expr( expr ) {}
CompositeExprNode::CompositeExprNode( const CompositeExprNode &other ) : expr( other.expr->clone() ) {}
CompositeExprNode::~CompositeExprNode() { delete expr; }
void CompositeExprNode::print( std::ostream &, int indent ) const { assert( false ); }
void CompositeExprNode::printOneLine( std::ostream &, int indent ) const { assert( false ); }

Expression *build_cast( TypeValueNode * arg, ExpressionNode *expr_node ) {
        DeclarationNode *decl_node = arg->get_decl();

        Type *targetType = decl_node->buildType();
        if ( dynamic_cast< VoidType* >( targetType ) ) {
                delete targetType;
                return new CastExpr( maybeBuild<Expression>(expr_node) );
        } else {
                return new CastExpr( maybeBuild<Expression>(expr_node), targetType );
        } // if
}

Expression *build_fieldSel( ExpressionNode *expr_node, VarRefNode *member ) {
        NameExpr* memberExpr = dynamic_cast<NameExpr*> ( maybeBuild<Expression>( member) );
        assert( memberExpr );
        UntypedMemberExpr *ret = new UntypedMemberExpr( memberExpr->get_name(), maybeBuild<Expression>(expr_node) );
        delete member;
        return ret;
}

Expression *build_pfieldSel( ExpressionNode *expr_node, VarRefNode *member ) {
        NameExpr* memberExpr = dynamic_cast<NameExpr*> ( maybeBuild<Expression>( member) );
        assert( memberExpr );
        UntypedExpr *deref = new UntypedExpr( new NameExpr( "*?" ) );
        deref->get_args().push_back( maybeBuild<Expression>(expr_node) );
        UntypedMemberExpr *ret = new UntypedMemberExpr( memberExpr->get_name(), deref );
        delete member;
        return ret;
}

Expression *build_addressOf( ExpressionNode *expr_node ) {
                return new AddressExpr( maybeBuild<Expression>(expr_node) );
}
Expression *build_sizeOf( ExpressionNode *expr_node ) {
        if ( TypeValueNode * arg = dynamic_cast<TypeValueNode *>( expr_node ) ) {
                return new SizeofExpr( arg->get_decl()->buildType() );
        } else {
                return new SizeofExpr( maybeBuild<Expression>(expr_node) );
        } // if
}
Expression *build_alignOf( ExpressionNode *expr_node ) {
        if ( TypeValueNode * arg = dynamic_cast<TypeValueNode *>( expr_node ) ) {
                return new AlignofExpr( arg->get_decl()->buildType() );
        } else {
                return new AlignofExpr( maybeBuild<Expression>(expr_node) );
        } // if
}
Expression *build_offsetOf( TypeValueNode * arg, VarRefNode *member ) {
        NameExpr *memberExpr = dynamic_cast<NameExpr *>( maybeBuild<Expression>( member ) );
        assert( memberExpr );
        return new UntypedOffsetofExpr( arg->get_decl()->buildType(), memberExpr->get_name() );
}

Expression *build_and_or( ExpressionNode *expr_node1, ExpressionNode *expr_node2, bool kind ) {
        return new LogicalExpr( notZeroExpr( maybeBuild<Expression>(expr_node1) ), notZeroExpr( maybeBuild<Expression>(expr_node2) ), kind );
}

Expression *build_unary_val( OperKinds op, ExpressionNode *expr_node ) {
        std::list<Expression *> args;
        args.push_back( maybeBuild<Expression>(expr_node) );
        return new UntypedExpr( new NameExpr( OperName[ (int)op ] ), args );
}
Expression *build_unary_ptr( OperKinds op, ExpressionNode *expr_node ) {
        std::list<Expression *> args;
        args.push_back( new AddressExpr( maybeBuild<Expression>(expr_node) ) );
        return new UntypedExpr( new NameExpr( OperName[ (int)op ] ), args );
}
Expression *build_binary_val( OperKinds op, ExpressionNode *expr_node1, ExpressionNode *expr_node2 ) {
        std::list<Expression *> args;
        args.push_back( maybeBuild<Expression>(expr_node1) );
        args.push_back( maybeBuild<Expression>(expr_node2) );
        return new UntypedExpr( new NameExpr( OperName[ (int)op ] ), args );
}
Expression *build_binary_ptr( OperKinds op, ExpressionNode *expr_node1, ExpressionNode *expr_node2 ) {
        std::list<Expression *> args;
        args.push_back( new AddressExpr( maybeBuild<Expression>(expr_node1) ) );
        args.push_back( maybeBuild<Expression>(expr_node2) );
        return new UntypedExpr( new NameExpr( OperName[ (int)op ] ), args );
}

Expression *build_cond( ExpressionNode *expr_node1, ExpressionNode *expr_node2, ExpressionNode *expr_node3 ) {
        return new ConditionalExpr( notZeroExpr( maybeBuild<Expression>(expr_node1) ), maybeBuild<Expression>(expr_node2), maybeBuild<Expression>(expr_node3) );
}

Expression *build_comma( ExpressionNode *expr_node1, ExpressionNode *expr_node2 ) {
        return new CommaExpr( maybeBuild<Expression>(expr_node1), maybeBuild<Expression>(expr_node2) );
}

Expression *build_attr( VarRefNode *var, ExpressionNode * expr ) {
        if ( TypeValueNode * arg = dynamic_cast<TypeValueNode *>( expr ) ) {
                return new AttrExpr( maybeBuild<Expression>(var), arg->get_decl()->buildType() );
        } else {
                return new AttrExpr( maybeBuild<Expression>(var), maybeBuild<Expression>(expr) );
        } // if
}

Expression *build_tuple( ExpressionNode * expr ) {
        TupleExpr *ret = new TupleExpr();
        buildList( expr, ret->get_exprs() );
        return ret;
}

Expression *build_func( ExpressionNode * function, ExpressionNode * expr ) {
        std::list<Expression *> args;

        buildList( expr, args );
        return new UntypedExpr( maybeBuild<Expression>(function), args, nullptr );
}

Expression *build_range( ExpressionNode * low, ExpressionNode *high ) {
        Expression *low_cexpr = maybeBuild<Expression>( low );
        Expression *high_cexpr = maybeBuild<Expression>( high );
        return new RangeExpr( low_cexpr, high_cexpr );
}

//##############################################################################

Expression *AsmExprNode::build() const {
        return new AsmExpr( maybeBuild< Expression >( inout ), (ConstantExpr *)maybeBuild<Expression>(constraint), maybeBuild<Expression>(operand) );
}

void AsmExprNode::print( std::ostream &os, int indent ) const {
        os << string( indent, ' ' ) << "Assembler Expression:" << endl;
        if ( inout ) {
                os << string( indent, ' ' ) << "inout: " << std::endl;
                inout->print( os, indent + 2 );
        } // if
        if ( constraint ) {
                os << string( indent, ' ' ) << "constraint: " << std::endl;
                constraint->print( os, indent + 2 );
        } // if
        if ( operand ) {
                os << string( indent, ' ' ) << "operand: " << std::endl;
                operand->print( os, indent + 2 );
        } // if
}

void AsmExprNode::printOneLine( std::ostream &os, int indent ) const {
        printDesignation( os );
        os << "( ";
        if ( inout ) inout->printOneLine( os, indent + 2 );
        os << ", ";
        if ( constraint ) constraint->printOneLine( os, indent + 2 );
        os << ", ";
        if ( operand ) operand->printOneLine( os, indent + 2 );
        os << ") ";
}

//##############################################################################

void LabelNode::print( std::ostream &os, int indent ) const {}

void LabelNode::printOneLine( std::ostream &os, int indent ) const {}

//##############################################################################

ValofExprNode::ValofExprNode( StatementNode *s ): body( s ) {}

ValofExprNode::ValofExprNode( const ValofExprNode &other ) : ExpressionNode( other ), body( maybeClone( body ) ) {
}

ValofExprNode::~ValofExprNode() {
        delete body;
}

void ValofExprNode::print( std::ostream &os, int indent ) const {
        printDesignation( os );
        os << string( indent, ' ' ) << "Valof Expression:" << std::endl;
        get_body()->print( os, indent + 4);
}

void ValofExprNode::printOneLine( std::ostream &, int indent ) const {
        assert( false );
}

Expression *ValofExprNode::build() const {
        return new UntypedValofExpr ( maybeBuild<Statement>(get_body()), maybeBuild< Expression >( get_argName() ) );
}

//##############################################################################

ForCtlExprNode::ForCtlExprNode( ParseNode *init_, ExpressionNode *cond, ExpressionNode *incr ) throw ( SemanticError ) : condition( cond ), change( incr ) {
        if ( init_ == 0 )
                init = 0;
        else {
                DeclarationNode *decl;
                ExpressionNode *exp;

                if (( decl = dynamic_cast<DeclarationNode *>(init_) ) != 0)
                        init = new StatementNode( decl );
                else if (( exp = dynamic_cast<ExpressionNode *>( init_)) != 0)
                        init = new StatementNode( StatementNode::Exp, exp );
                else
                        throw SemanticError("Error in for control expression");
        }
}

ForCtlExprNode::ForCtlExprNode( const ForCtlExprNode &other )
        : ExpressionNode( other ), init( maybeClone( other.init ) ), condition( maybeClone( other.condition ) ), change( maybeClone( other.change ) ) {
}

ForCtlExprNode::~ForCtlExprNode() {
        delete init;
        delete condition;
        delete change;
}

Expression *ForCtlExprNode::build() const {
        // this shouldn't be used!
        assert( false );
        return 0;
}

void ForCtlExprNode::print( std::ostream &os, int indent ) const{
        os << string( indent,' ' ) << "For Control Expression -- :" << endl;

        os << string( indent + 2, ' ' ) << "initialization:" << endl;
        if ( init != 0 )
                init->printList( os, indent + 4 );

        os << string( indent + 2, ' ' ) << "condition: " << endl;
        if ( condition != 0 )
                condition->print( os, indent + 4 );
        os << string( indent + 2, ' ' ) << "increment: " << endl;
        if ( change != 0 )
                change->print( os, indent + 4 );
}

void ForCtlExprNode::printOneLine( std::ostream &, int indent ) const {
        assert( false );
}

//##############################################################################

TypeValueNode::TypeValueNode( DeclarationNode *decl ) : decl( decl ) {
}

TypeValueNode::TypeValueNode( const TypeValueNode &other ) : ExpressionNode( other ), decl( maybeClone( other.decl ) ) {
}

Expression *TypeValueNode::build() const {
        return new TypeExpr( decl->buildType() );
}

void TypeValueNode::print( std::ostream &os, int indent ) const {
        os << std::string( indent, ' ' ) << "Type:";
        get_decl()->print( os, indent + 2);
}

void TypeValueNode::printOneLine( std::ostream &os, int indent ) const {
        os << "Type:";
        get_decl()->print( os, indent + 2);
}


CompoundLiteralNode::CompoundLiteralNode( DeclarationNode *type, InitializerNode *kids ) : type( type ), kids( kids ) {}
CompoundLiteralNode::CompoundLiteralNode( const CompoundLiteralNode &other ) : ExpressionNode( other ), type( other.type ), kids( other.kids ) {}

CompoundLiteralNode::~CompoundLiteralNode() {
        delete kids;
        delete type;
}

CompoundLiteralNode *CompoundLiteralNode::clone() const {
        return new CompoundLiteralNode( *this );
}

void CompoundLiteralNode::print( std::ostream &os, int indent ) const {
        os << string( indent,' ' ) << "CompoundLiteralNode:" << endl;

        os << string( indent + 2, ' ' ) << "type:" << endl;
        if ( type != 0 )
                type->print( os, indent + 4 );

        os << string( indent + 2, ' ' ) << "initialization:" << endl;
        if ( kids != 0 )
                kids->printList( os, indent + 4 );
}

void CompoundLiteralNode::printOneLine( std::ostream &os, int indent ) const {
        os << "( ";
        if ( type ) type->print( os );
        os << ", ";
        if ( kids ) kids->printOneLine( os );
        os << ") ";
}

Expression *CompoundLiteralNode::build() const {
        Declaration * newDecl = maybeBuild<Declaration>(type); // compound literal type
        if ( DeclarationWithType * newDeclWithType = dynamic_cast< DeclarationWithType * >( newDecl ) ) { // non-sue compound-literal type
                return new CompoundLiteralExpr( newDeclWithType->get_type(), maybeBuild<Initializer>(kids) );
        // these types do not have associated type information
        } else if ( StructDecl * newDeclStructDecl = dynamic_cast< StructDecl * >( newDecl )  ) {
                return new CompoundLiteralExpr( new StructInstType( Type::Qualifiers(), newDeclStructDecl->get_name() ), maybeBuild<Initializer>(kids) );
        } else if ( UnionDecl * newDeclUnionDecl = dynamic_cast< UnionDecl * >( newDecl )  ) {
                return new CompoundLiteralExpr( new UnionInstType( Type::Qualifiers(), newDeclUnionDecl->get_name() ), maybeBuild<Initializer>(kids) );
        } else if ( EnumDecl * newDeclEnumDecl = dynamic_cast< EnumDecl * >( newDecl )  ) {
                return new CompoundLiteralExpr( new EnumInstType( Type::Qualifiers(), newDeclEnumDecl->get_name() ), maybeBuild<Initializer>(kids) );
        } else {
                assert( false );
        } // if
}

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //