001    /*
002     * Sonar, open source software quality management tool.
003     * Copyright (C) 2008-2011 SonarSource
004     * mailto:contact AT sonarsource DOT com
005     *
006     * Sonar is free software; you can redistribute it and/or
007     * modify it under the terms of the GNU Lesser General Public
008     * License as published by the Free Software Foundation; either
009     * version 3 of the License, or (at your option) any later version.
010     *
011     * Sonar is distributed in the hope that it will be useful,
012     * but WITHOUT ANY WARRANTY; without even the implied warranty of
013     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
014     * Lesser General Public License for more details.
015     *
016     * You should have received a copy of the GNU Lesser General Public
017     * License along with Sonar; if not, write to the Free Software
018     * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02
019     */
020    
021    /**
022     * BSD-style license; for more info see http://pmd.sourceforge.net/license.html
023     */
024    package net.sourceforge.pmd.cpd;
025    
026    import org.sonar.duplications.cpd.Match;
027    
028    import java.util.ArrayList;
029    import java.util.Iterator;
030    import java.util.List;
031    
032    public class Tokens {
033    
034        private List<TokenEntry> tokens = new ArrayList<TokenEntry>();
035    
036        public void add(TokenEntry tokenEntry) {
037            this.tokens.add(tokenEntry);
038        }
039    
040        public Iterator<TokenEntry> iterator() {
041            return tokens.iterator();
042        }
043    
044        private TokenEntry get(int index) {
045            return tokens.get(index);
046        }
047    
048        public int size() {
049            return tokens.size();
050        }
051    
052        public int getLineCount(TokenEntry mark, Match match) {
053            TokenEntry endTok = get(mark.getIndex() + match.getTokenCount() - 1);
054            if (endTok == TokenEntry.EOF) {
055                endTok = get(mark.getIndex() + match.getTokenCount() - 2);
056            }
057            return endTok.getBeginLine() - mark.getBeginLine() + 1;
058        }
059    
060        public List<TokenEntry> getTokens() {
061            return tokens;
062        }
063    
064    }