FunctionDeclarationTreeImpl.java revision 1739:4a6a1fd3d3dd 
1/*
2 * Copyright (c) 2016, 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 */
25
26package jdk.nashorn.api.tree;
27
28import java.util.List;
29import jdk.nashorn.internal.ir.FunctionNode;
30import jdk.nashorn.internal.ir.VarNode;
31
32final class FunctionDeclarationTreeImpl extends StatementTreeImpl
33    implements FunctionDeclarationTree {
34    private final FunctionNode funcNode;
35    private final IdentifierTree funcName;
36    private final List<? extends ExpressionTree> params;
37    private final BlockTree body;
38
39    FunctionDeclarationTreeImpl(final VarNode node,
40            final List<? extends ExpressionTree> params,
41            final BlockTree body) {
42        super(node);
43        assert node.getInit() instanceof FunctionNode : "function expected";
44        funcNode = (FunctionNode)node.getInit();
45        assert funcNode.isDeclared() : "function declaration expected";
46        funcName = funcNode.isAnonymous()? null : new IdentifierTreeImpl(node.getName());
47        this.params = params;
48        this.body = body;
49    }
50
51    @Override
52    public Kind getKind() {
53        return Kind.FUNCTION;
54    }
55
56    @Override
57    public IdentifierTree getName() {
58        return funcName;
59    }
60
61    @Override
62    public List<? extends ExpressionTree> getParameters() {
63        return params;
64    }
65
66    @Override
67    public BlockTree getBody() {
68        return body;
69    }
70
71    @Override
72    public boolean isStrict() {
73        return funcNode.isStrict();
74    }
75
76    @Override
77    public boolean isGenerator() {
78        return funcNode.getKind() == FunctionNode.Kind.GENERATOR;
79    }
80
81    @Override
82    public <R,D> R accept(final TreeVisitor<R,D> visitor, final D data) {
83        return visitor.visitFunctionDeclaration(this, data);
84    }
85}
86