org.mozilla.javascript.ScriptOrFnNode Maven / Gradle / Ivy
Go to download
Rhino is an open-source implementation of JavaScript written entirely in Java. It is typically embedded into Java applications to provide scripting to end users.
/* -*- Mode: java; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
*
* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is Rhino code, released
* May 6, 1999.
*
* The Initial Developer of the Original Code is
* Netscape Communications Corporation.
* Portions created by the Initial Developer are Copyright (C) 1997-1999
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* Igor Bukanov
* Bob Jervis
* Norris Boyd
*
* Alternatively, the contents of this file may be used under the terms of
* the GNU General Public License Version 2 or later (the "GPL"), in which
* case the provisions of the GPL are applicable instead of those above. If
* you wish to allow use of your version of this file only under the terms of
* the GPL and not to allow others to use your version of this file under the
* MPL, indicate your decision by deleting the provisions above and replacing
* them with the notice and other provisions required by the GPL. If you do
* not delete the provisions above, a recipient may use your version of this
* file under either the MPL or the GPL.
*
* ***** END LICENSE BLOCK ***** */
package org.mozilla.javascript;
import java.util.ArrayList;
public class ScriptOrFnNode extends Node.Scope {
public ScriptOrFnNode(int nodeType) {
super(nodeType);
symbols = new ArrayList(4);
setParent(null);
}
public final String getSourceName() { return sourceName; }
public final void setSourceName(String sourceName) {
this.sourceName = sourceName;
}
public final int getEncodedSourceStart() { return encodedSourceStart; }
public final int getEncodedSourceEnd() { return encodedSourceEnd; }
public final void setEncodedSourceBounds(int start, int end) {
this.encodedSourceStart = start;
this.encodedSourceEnd = end;
}
public final int getBaseLineno() { return this.lineno; }
public final void setBaseLineno(int lineno) {
// One time action
if (lineno < 0 || this.lineno >= 0) Kit.codeBug();
this.lineno = lineno;
}
public final int getEndLineno() { return endLineno; }
public final void setEndLineno(int lineno) {
// One time action
if (lineno < 0 || endLineno >= 0) Kit.codeBug();
endLineno = lineno;
}
public final int getFunctionCount() {
if (functions == null) { return 0; }
return functions.size();
}
public final FunctionNode getFunctionNode(int i) {
return (FunctionNode)functions.get(i);
}
public final int addFunction(FunctionNode fnNode) {
if (fnNode == null) Kit.codeBug();
if (functions == null) { functions = new ObjArray(); }
functions.add(fnNode);
return functions.size() - 1;
}
public final int getRegexpCount() {
if (regexps == null) { return 0; }
return regexps.size() / 2;
}
public final String getRegexpString(int index) {
return (String)regexps.get(index * 2);
}
public final String getRegexpFlags(int index) {
return (String)regexps.get(index * 2 + 1);
}
public final int addRegexp(String string, String flags) {
if (string == null) Kit.codeBug();
if (regexps == null) { regexps = new ObjArray(); }
regexps.add(string);
regexps.add(flags);
return regexps.size() / 2 - 1;
}
public int getIndexForNameNode(Node nameNode) {
if (variableNames == null) throw Kit.codeBug();
Node.Scope node = nameNode.getScope();
Symbol symbol = node == null ? null
: node.getSymbol(nameNode.getString());
if (symbol == null)
return -1;
return symbol.index;
}
public final String getParamOrVarName(int index) {
if (variableNames == null) throw Kit.codeBug();
return variableNames[index];
}
public final int getParamCount() {
return paramCount;
}
public final int getParamAndVarCount() {
if (variableNames == null) throw Kit.codeBug();
return symbols.size();
}
public final String[] getParamAndVarNames() {
if (variableNames == null) throw Kit.codeBug();
return variableNames;
}
public final boolean[] getParamAndVarConst() {
if (variableNames == null) throw Kit.codeBug();
return isConsts;
}
void addSymbol(Symbol symbol) {
if (variableNames != null) throw Kit.codeBug();
if (symbol.declType == Token.LP) {
paramCount++;
}
symbols.add(symbol);
}
/**
* Assign every symbol a unique integer index. Generate arrays of variable
* names and constness that can be indexed by those indices.
*
* @param flattenAllTables if true, flatten all symbol tables, included
* nested block scope symbol tables. If false, just flatten the script's
* or function's symbol table.
*/
void flattenSymbolTable(boolean flattenAllTables) {
if (!flattenAllTables) {
ArrayList newSymbols = new ArrayList();
if (this.symbolTable != null) {
// Just replace "symbols" with the symbols in this object's
// symbol table. Can't just work from symbolTable map since
// we need to retain duplicate parameters.
for (int i=0; i < symbols.size(); i++) {
Symbol symbol = (Symbol) symbols.get(i);
if (symbol.containingTable == this) {
newSymbols.add(symbol);
}
}
}
symbols = newSymbols;
}
variableNames = new String[symbols.size()];
isConsts = new boolean[symbols.size()];
for (int i=0; i < symbols.size(); i++) {
Symbol symbol = (Symbol) symbols.get(i);
variableNames[i] = symbol.name;
isConsts[i] = symbol.declType == Token.CONST;
symbol.index = i;
}
}
public final Object getCompilerData()
{
return compilerData;
}
public final void setCompilerData(Object data)
{
if (data == null) throw new IllegalArgumentException();
// Can only call once
if (compilerData != null) throw new IllegalStateException();
compilerData = data;
}
public String getNextTempName()
{
return "$" + tempNumber++;
}
private int encodedSourceStart;
private int encodedSourceEnd;
private String sourceName;
private int endLineno = -1;
private ObjArray functions;
private ObjArray regexps;
private ArrayList symbols;
private int paramCount = 0;
private String[] variableNames;
private boolean[] isConsts;
private Object compilerData;
private int tempNumber = 0;
}