/*
 * This file is part of ADDIS (Aggregate Data Drug Information System).
 * ADDIS is distributed from http://drugis.org/.
 * Copyright © 2009 Gert van Valkenhoef, Tommi Tervonen.
 * Copyright © 2010 Gert van Valkenhoef, Tommi Tervonen, Tijs Zwinkels,
 * Maarten Jacobs, Hanno Koeslag, Florin Schimbinschi, Ahmad Kamal, Daniel
 * Reid.
 * Copyright © 2011 Gert van Valkenhoef, Ahmad Kamal, Daniel Reid, Florin
 * Schimbinschi.
 * Copyright © 2012 Gert van Valkenhoef, Daniel Reid, Joël Kuiper, Wouter
 * Reckman.
 * Copyright © 2013 Gert van Valkenhoef, Joël Kuiper.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.drugis.addis.util.JSMAAintegration;

import org.drugis.common.threading.status.TaskProgressModel;

import fi.smaa.common.RandomUtil;
import fi.smaa.jsmaa.gui.jfreechart.CentralWeightsDataset;
import fi.smaa.jsmaa.gui.jfreechart.RankAcceptabilitiesDataset;
import fi.smaa.jsmaa.gui.presentation.CentralWeightTableModel;
import fi.smaa.jsmaa.gui.presentation.RankAcceptabilityTableModel;
import fi.smaa.jsmaa.model.SMAAModel;
import fi.smaa.jsmaa.simulator.SMAA2Results;
import fi.smaa.jsmaa.simulator.SMAA2Simulation;
import fi.smaa.jsmaa.simulator.SimulationBuilder;

public class BRSMAASimulationBuilder extends SimulationBuilder<SMAAModel, SMAA2Results, SMAA2Simulation>{
	private RankAcceptabilityTableModel d_resTableModel;
	private RankAcceptabilitiesDataset d_resDataset;
	private CentralWeightTableModel d_cwTableModel;
	private CentralWeightsDataset d_cwDataset;
	private TaskProgressModel d_progressModel;
	private SMAAModel d_mutableModel;

	public BRSMAASimulationBuilder(SMAAModel model, RankAcceptabilityTableModel resTableModel, RankAcceptabilitiesDataset dataSet,
				CentralWeightTableModel cwTableModel, CentralWeightsDataset cwDataSet, TaskProgressModel progressModel) {
		super(model);
		d_mutableModel = model;
		d_resTableModel = resTableModel;
		d_resDataset = dataSet;
		d_cwTableModel = cwTableModel;
		d_cwDataset = cwDataSet;
		d_progressModel = progressModel;
	}

	private void setResults(SMAA2Results results) {
		d_resTableModel.setResults(results);
		d_resDataset.setResults(results);
		d_cwTableModel.setResults(results);
		d_cwDataset.setResults(results);
	}

	@Override
	protected SMAA2Simulation generateSimulation() {
		SMAA2Simulation simulation = new SMAA2Simulation(model, RandomUtil.createWithRandomSeed(), 10000);
		d_progressModel.setTask(simulation.getTask());
		return simulation;
	}

	@Override
	protected void prepareSimulation(SMAA2Simulation simulation, SMAA2Results results) {
		setResults(results);
	}

	public TaskProgressModel getTaskProgressModel() {
		return d_progressModel;
	}

	public void resetModel() {
		this.model = d_mutableModel.deepCopy();
	}

}