CompilationUnitTreeImpl.java revision 1739:4a6a1fd3d3dd 
1/*
2 * Copyright (c) 2015, 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;
30
31final class CompilationUnitTreeImpl extends TreeImpl
32    implements CompilationUnitTree {
33    private final FunctionNode funcNode;
34    private final List<? extends Tree> elements;
35    private final ModuleTree module;
36
37    CompilationUnitTreeImpl(final FunctionNode node,
38            final List<? extends Tree> elements,
39            final ModuleTree module) {
40        super(node);
41        this.funcNode = node;
42        assert funcNode.getKind() == FunctionNode.Kind.SCRIPT ||
43                funcNode.getKind() == FunctionNode.Kind.MODULE :
44                "script or module function expected";
45        this.elements = elements;
46        this.module = module;
47    }
48
49    @Override
50    public Tree.Kind getKind() {
51        return Tree.Kind.COMPILATION_UNIT;
52    }
53
54    @Override
55    public List<? extends Tree> getSourceElements() {
56        return elements;
57    }
58
59    @Override
60    public String getSourceName() {
61        return funcNode.getSourceName();
62    }
63
64    @Override
65    public boolean isStrict() {
66        return funcNode.isStrict();
67    }
68
69    @Override
70    public LineMap getLineMap() {
71        return new LineMapImpl(funcNode.getSource());
72    }
73
74    @Override
75    public ModuleTree getModule() {
76        return module;
77    }
78
79    @Override
80    public <R,D> R accept(final TreeVisitor<R,D> visitor, final D data) {
81        return visitor.visitCompilationUnit(this, data);
82    }
83}
84