• Home
• io.github.repir
• RepIR
• 0.25
• source code
• TopicReaderSessionMapping.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.github.repir.TestSet.Topic.TopicReaderSessionMapping Maven / Gradle / Ivy

package io.github.repir.TestSet.Topic;

import io.github.repir.Repository.Repository;
import io.github.repir.tools.io.Datafile;
import io.github.repir.tools.io.struct.StructuredTextCSV;
import io.github.repir.tools.io.struct.StructuredTextXML;
import io.github.repir.tools.lib.Log;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;

/**
 * Topic reader for TREC Web Track 2010-2013 ad-hoc topics.
 *
 * @author jer
 */
public class TopicReaderSessionMapping extends StructuredTextCSV {

   public static Log log = new Log(TopicReaderSessionMapping.class);
   public FolderNode session = this.createRoot();
   public IntField sessionid = this.addInt("sessionid");
   public IntField topicid = this.addInt("topicid");
   public IntField subtopic = this.addInt("subtopic");

   private TopicReaderSessionMapping(Repository repository) {
      super(new Datafile(repository.configuredString("testset.mappings")));
      datafile.setBufferSize((int) datafile.getLength());
   }

   public static void assignTopics(Repository repository, HashMap topics) {
      TopicReaderSessionMapping m = new TopicReaderSessionMapping(repository);
      m.openRead();
      while (m.nextRecord()) {
         TestSetTopic get = topics.get(m.sessionid.get());
         get.qrelid = m.topicid.get();
      }
      m.closeRead();
   }
}