src/jdk/nashorn/internal/ir/BlockLexicalContext.java

Thu, 24 May 2018 16:39:31 +0800

author
aoqi
date
Thu, 24 May 2018 16:39:31 +0800
changeset 1959
61ffdd1b89f2
parent 1530
fd307cc5f58c
parent 1490
d85f981c8cf8
permissions
-rw-r--r--

Merge

     1 /*
     2  * Copyright (c) 2010, 2013, Oracle and/or its affiliates. All rights reserved.
     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
     4  *
     5  * This code is free software; you can redistribute it and/or modify it
     6  * under the terms of the GNU General Public License version 2 only, as
     7  * published by the Free Software Foundation.  Oracle designates this
     8  * particular file as subject to the "Classpath" exception as provided
     9  * by Oracle in the LICENSE file that accompanied this code.
    10  *
    11  * This code is distributed in the hope that it will be useful, but WITHOUT
    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    14  * version 2 for more details (a copy is included in the LICENSE file that
    15  * accompanied this code).
    16  *
    17  * You should have received a copy of the GNU General Public License version
    18  * 2 along with this work; if not, write to the Free Software Foundation,
    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
    20  *
    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
    22  * or visit www.oracle.com if you need additional information or have any
    23  * questions.
    24  */
    26 package jdk.nashorn.internal.ir;
    28 import java.util.ArrayDeque;
    29 import java.util.ArrayList;
    30 import java.util.Deque;
    31 import java.util.List;
    33 /**
    34  * This is a subclass of lexical context used for filling
    35  * blocks (and function nodes) with statements. When popping
    36  * a block from the lexical context, any statements that have
    37  * been generated in it are committed to the block. This saves
    38  * unnecessary object mutations and lexical context replacement
    39  */
    40 public class BlockLexicalContext extends LexicalContext {
    41     /** statement stack, each block on the lexical context maintains one of these, which is
    42      *  committed to the block on pop */
    43     private final Deque<List<Statement>> sstack = new ArrayDeque<>();
    45     /** Last non debug statement emitted in this context */
    46     protected Statement lastStatement;
    48     @Override
    49     public <T extends LexicalContextNode> T push(final T node) {
    50         final T pushed = super.push(node);
    51         if (node instanceof Block) {
    52             sstack.push(new ArrayList<Statement>());
    53         }
    54         return pushed;
    55     }
    57     /**
    58      * Get the statement list from the stack, possibly filtered
    59      * @return statement list
    60      */
    61     protected List<Statement> popStatements() {
    62         return sstack.pop();
    63     }
    65     /**
    66      * Override this method to perform some additional processing on the block after its statements have been set. By
    67      * default does nothing and returns the original block.
    68      * @param block the block to operate on
    69      * @return a modified block.
    70      */
    71     protected Block afterSetStatements(final Block block) {
    72         return block;
    73     }
    75     @SuppressWarnings("unchecked")
    76     @Override
    77     public <T extends Node> T pop(final T node) {
    78         T expected = node;
    79         if (node instanceof Block) {
    80             final List<Statement> newStatements = popStatements();
    81             expected = (T)((Block)node).setStatements(this, newStatements);
    82             expected = (T)afterSetStatements((Block)expected);
    83             if (!sstack.isEmpty()) {
    84                 lastStatement = lastStatement(sstack.peek());
    85             }
    86         }
    87         return super.pop(expected);
    88     }
    90     /**
    91      * Append a statement to the block being generated
    92      * @param statement statement to add
    93      */
    94     public void appendStatement(final Statement statement) {
    95         assert statement != null;
    96         sstack.peek().add(statement);
    97         lastStatement = statement;
    98     }
   100     /**
   101      * Prepend a statement to the block being generated
   102      * @param statement statement to prepend
   103      * @return the prepended statement
   104      */
   105     public Node prependStatement(final Statement statement) {
   106         assert statement != null;
   107         sstack.peek().add(0, statement);
   108         return statement;
   109     }
   111     /**
   112      * Prepend a list of statement to the block being generated
   113      * @param statements a list of statements to prepend
   114      */
   115     public void prependStatements(final List<Statement> statements) {
   116         assert statements != null;
   117         sstack.peek().addAll(0, statements);
   118     }
   121     /**
   122      * Get the last statement that was emitted into a block
   123      * @return the last statement emitted
   124      */
   125     public Statement getLastStatement() {
   126         return lastStatement;
   127     }
   129     private static Statement lastStatement(final List<Statement> statements) {
   130         final int s = statements.size();
   131         return s == 0 ? null : statements.get(s - 1);
   132     }
   133 }

mercurial