package de.mpiwg.anteater.results.impl;

import java.util.List;

import de.mpiwg.anteater.AnteaterConfiguration;
import de.mpiwg.anteater.ml.MLController;
import de.mpiwg.anteater.results.ApplicantResult;
import de.mpiwg.anteater.results.IResultFinder;
import de.mpiwg.anteater.results.LocationResult;
import de.mpiwg.anteater.results.SpeciesScientificResult;
import de.mpiwg.anteater.text.TextInformation;

public class LocationResultFinder implements IResultFinder<LocationResult> {

	private AnteaterConfiguration configuration;
	private List<SpeciesScientificResult> predictedSpecies;
	private List<ApplicantResult> predictedApplicants;
	
	public LocationResultFinder(AnteaterConfiguration configuration, List<SpeciesScientificResult> predictedSpecies, List<ApplicantResult> predictedApplicants) {
		this.configuration = configuration;
		this.predictedApplicants = predictedApplicants;
		this.predictedSpecies = predictedSpecies;
	}

	@Override
	public List<LocationResult> getResults(List<TextInformation> infos) {
		MLController mlController = new MLController(configuration);
		return mlController.runLocationMLComponent(infos, predictedSpecies, predictedApplicants);
	}

}