io.anserini.doc.GenerateRegressionDocsTest.java Source code

Introduction

Here is the source code for io.anserini.doc.GenerateRegressionDocsTest.java

Source

/**
 * Anserini: An information retrieval toolkit built on Lucene
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.anserini.doc;

import java.io.File;
import java.net.URL;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.text.StrSubstitutor;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import org.junit.Test;

public class GenerateRegressionDocsTest {
    @Test
    public void main() throws Exception {
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        URL resource = GenerateRegressionDocsTest.class.getResource("/regression/all.yaml");
        DataModel data = mapper.readValue(Paths.get(resource.toURI()).toFile(), DataModel.class);
        //System.out.println(ReflectionToStringBuilder.toString(data, ToStringStyle.MULTI_LINE_STYLE));

        for (String collection : data.getCollections().keySet()) {
            Map<String, String> valuesMap = new HashMap<>();
            valuesMap.put("index_cmds", data.generateIndexingCommand(collection));
            valuesMap.put("ranking_cmds", data.generateRankingCommand(collection));
            valuesMap.put("eval_cmds", data.generateEvalCommand(collection));
            valuesMap.put("effectiveness", data.generateEffectiveness(collection));
            StrSubstitutor sub = new StrSubstitutor(valuesMap);
            URL template = GenerateRegressionDocsTest.class
                    .getResource(String.format("/docgen/templates/%s.template", collection));
            Scanner scanner = new Scanner(Paths.get(template.toURI()).toFile(), "UTF-8");
            String text = scanner.useDelimiter("\\A").next();
            scanner.close();
            String resolvedString = sub.replace(text);

            FileUtils.writeStringToFile(new File(String.format("docs/experiments-%s.md", collection)),
                    resolvedString, "UTF-8");
        }
    }
}