001    /*
002     * Copyright 2005-2008 Sun Microsystems, Inc.  All Rights Reserved.
003     * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
004     *
005     * This code is free software; you can redistribute it and/or modify it
006     * under the terms of the GNU General Public License version 2 only, as
007     * published by the Free Software Foundation.  Sun designates this
008     * particular file as subject to the "Classpath" exception as provided
009     * by Sun in the LICENSE file that accompanied this code.
010     *
011     * This code is distributed in the hope that it will be useful, but WITHOUT
012     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
013     * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
014     * version 2 for more details (a copy is included in the LICENSE file that
015     * accompanied this code).
016     *
017     * You should have received a copy of the GNU General Public License version
018     * 2 along with this work; if not, write to the Free Software Foundation,
019     * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
020     *
021     * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
022     * CA 95054 USA or visit www.sun.com if you need additional information or
023     * have any questions.
024     */
025    
026    package com.sun.tools.javac.parser;
027    
028    import java.util.Map;
029    import java.util.HashMap;
030    import com.sun.tools.javac.tree.JCTree;
031    import com.sun.tools.javac.tree.TreeInfo;
032    
033    import static com.sun.tools.javac.tree.JCTree.*;
034    
035    /**
036     * This class is similar to Parser except that it stores ending
037     * positions for the tree nodes.
038     *
039     * <p><b>This is NOT part of any API supported by Sun Microsystems.
040     * If you write code that depends on this, you do so at your own risk.
041     * This code and its internal interfaces are subject to change or
042     * deletion without notice.</b></p>
043     */
044    public class EndPosParser extends JavacParser {
045    
046        public EndPosParser(ParserFactory fac, Lexer S, boolean keepDocComments, boolean keepLineMap) {
047            super(fac, S, keepDocComments, keepLineMap);
048            this.S = S;
049            endPositions = new HashMap<JCTree,Integer>();
050        }
051    
052        private Lexer S;
053    
054        /** A hashtable to store ending positions
055         *  of source ranges indexed by the tree nodes.
056         *  Defined only if option flag genEndPos is set.
057         */
058        Map<JCTree, Integer> endPositions;
059    
060        /** {@inheritDoc} */
061        @Override
062        protected void storeEnd(JCTree tree, int endpos) {
063            int errorEndPos = getErrorEndPos();
064            endPositions.put(tree, errorEndPos > endpos ? errorEndPos : endpos);
065        }
066    
067        /** {@inheritDoc} */
068        @Override
069        protected <T extends JCTree> T to(T t) {
070            storeEnd(t, S.endPos());
071            return t;
072        }
073    
074        /** {@inheritDoc} */
075        @Override
076        protected <T extends JCTree> T toP(T t) {
077            storeEnd(t, S.prevEndPos());
078            return t;
079        }
080    
081        @Override
082        public JCCompilationUnit parseCompilationUnit() {
083            JCCompilationUnit t = super.parseCompilationUnit();
084            t.endPositions = endPositions;
085            return t;
086        }
087    
088        /** {@inheritDoc} */
089        @Override
090        JCExpression parExpression() {
091            int pos = S.pos();
092            JCExpression t = super.parExpression();
093            return toP(F.at(pos).Parens(t));
094        }
095    
096        /** {@inheritDoc} */
097        @Override
098        public int getEndPos(JCTree tree) {
099            return TreeInfo.getEndPos(tree, endPositions);
100        }
101    
102    }