abstracttopicreader.java

来自「dragontoolkit用于机器学习」· Java 代码 · 共 64 行

package dragon.onlinedb.trec;

import dragon.onlinedb.*;
import java.util.ArrayList;
/**
 * <p>Abstract class for TREC topic reading</p>
 * <p></p>
 * <p>Copyright: Copyright (c) 2005</p>
 * <p>Company: IST, Drexel University</p>
 * @author Davis Zhou
 * @version 1.0
 */

public abstract class AbstractTopicReader implements CollectionReader{
    private ArrayList topics;
    private int curTopic;

    public AbstractTopicReader(String topicFile) {
        topics=loadTopics(topicFile);
        curTopic=0;
    }

    protected abstract ArrayList loadTopics(String topicFile);

    public boolean loadCollection(String topicFile){
        topics=loadTopics(topicFile);
        curTopic=0;
        return true;
    }

    public ArticleParser getArticleParser(){
        return new BasicArticleParser();
    }

    public void setArticleParser(ArticleParser parser){
        //do nothing
    }

    public Article getNextArticle(){
        if(topics==null || curTopic>=topics.size())
            return null;
        else{
            curTopic++;
            return (Article) topics.get(curTopic-1);
        }
    }

    public Article getArticleByKey(String key){
        return null;
    }

    public void close(){
        topics=null;
    }

    public boolean supportArticleKeyRetrieval(){
        return false;
    }

    public void restart(){
        curTopic=0;
    }

}