SyntheticDataWriter.java

package com.datascience.galc.dataGenerator;

import com.datascience.core.base.AssignedLabel;
import com.datascience.core.base.ContValue;
import com.datascience.core.base.LObject;
import com.datascience.core.base.Worker;
import com.datascience.core.results.WorkerContResults;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Collection;
import java.util.Map;
import java.util.Set;

public class SyntheticDataWriter {
	
	private final SyntheticData data;
	
	public SyntheticDataWriter(SyntheticData data) {
		this.data = data;
	}
	
	private BufferedWriter openFile(String filename) throws IOException {
		File outfile = new File(filename);
		if (outfile.getParent() != null) {
			File parentDir = new File(outfile.getParent());
			if (!parentDir.exists()) {
				parentDir.mkdirs();
			}
		}
		return new BufferedWriter(new FileWriter(outfile));
	}

	public void writeAssignedLabelsToFile(Collection<AssignedLabel<ContValue>> assigns,
			String filename) throws IOException {

		BufferedWriter bw = openFile(filename);
		try {
			for (AssignedLabel<ContValue> al : assigns) {
				String line = al.getWorker().getName() + "\t" + 
						al.getLobject().getName() + "\t" + 
						// TODO: is it correct?
						al.getLabel().getValue() + "\n";
				bw.write(line);
			}
		} finally {
			bw.close();
		}
	}

	public void writeTrueWorkerDataToFile(
			Map<Worker, WorkerContResults> workerContResults,
			String filename) throws IOException {

		BufferedWriter bw = openFile(filename);
		try {
			for (Map.Entry<Worker, WorkerContResults> e : workerContResults.entrySet()) {
				Worker w = e.getKey();
				WorkerContResults wcr = e.getValue();
				String line = w.getName() + "\t" +
						wcr.getTrueRho() + "\t" +
						wcr.getTrueMu() + "\t" +
						wcr.getTrueSigma() + "\t"
						+ "\n";
				bw.write(line);
			}
		} finally {
			bw.close();
		}
	}
	
	public void writeTrueObjectDataToFile(Collection<LObject<ContValue>> lObjects,
			String filename) throws IOException {

		BufferedWriter bw = openFile(filename);
		try {
			for (LObject<ContValue> lo : lObjects) {
				if (lo.isEvaluation()) {
					ContValue contValue = lo.getEvaluationLabel();
					String line = lo.getName() + "\t" +
							contValue.getValue() + "\t" +
							contValue.getZeta() + "\n";
					bw.write(line);
				}
			}
		} finally {
			bw.close();
		}
	}

	public void writeGoldObjectDataToFile(Collection<LObject<ContValue>> lObjects,
			String filename) throws IOException {

		BufferedWriter bw = openFile(filename);
		try {
			for (LObject<ContValue> lo : lObjects) {
				if (lo.isGold()) {
					ContValue contValue = lo.getGoldLabel();
					String line = lo.getName() + "\t" +
							contValue.getValue() + "\t" +
							contValue.getZeta() + "\n";
					bw.write(line);
				}
			}
		} finally {
			bw.close();
		}
	}
	
}