jasima.core.util.AbstractResultSaver Maven / Gradle / Ivy

/*******************************************************************************
 * This file is part of jasima, v1.3, the Java simulator for manufacturing and 
 * logistics.
 *  
 * Copyright (c) 2015 		jasima solutions UG
 * Copyright (c) 2010-2015 Torsten Hildebrandt and jasima contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see .
 *******************************************************************************/
package jasima.core.util;

import jasima.core.experiment.ExperimentListenerBase;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * Base class for result saver.
 * 
 * @author Torsten Hildebrandt
 * @version 
 *          "$Id: AbstractResultSaver.java 753 2015-07-27 15:29:49Z [email protected] $"
 */
public abstract class AbstractResultSaver extends ExperimentListenerBase {

	private static final long serialVersionUID = -1643003626220225101L;
	private String resultFileName = null;
	private String resultFileNameHint = "runResults";
	private String actualResultFileName = null;

	public void setFileNameHint(String resultFileNameHint) {
		this.resultFileNameHint = resultFileNameHint;
	}

	public void setResultFileName(String resultFileName) {
		resultFileName = resultFileName.trim();
		this.resultFileName = resultFileName.isEmpty() ? null : resultFileName;
	}

	public static String findFreeFile(final String baseName,
			final String extension) {
		int index = 0;
		String retVal = baseName;

		while (true) {
			try {
				boolean createRes = new File(retVal + extension)
						.createNewFile();
				if (createRes)
					break; // while
			} catch (IOException e) {
			}

			retVal = baseName + "_" + ++index;

			if (index > 1000) {
				// give up
				throw new RuntimeException(String.format(Util.DEF_LOCALE,
						"Cant't create new file (baseName=%s,ext=%s).",
						baseName, extension));
			}
		}

		return retVal;
	}

	public String findFreeFile(final String baseName) {
		String retVal = baseName;
		int index = 0;
		while (!checkBaseName(retVal)) {
			retVal = baseName + "_" + ++index;
		}
		return retVal;
	}

	public abstract boolean checkBaseName(String base);

	public String getActualResultBaseName() {
		if (actualResultFileName == null) {
			if (resultFileName == null) {
				actualResultFileName = findFreeFile(resultFileNameHint
						.replaceFirst("\\..*?$", "")
						+ new SimpleDateFormat("_yyyyMMdd_HHmm")
								.format(new Date()));
			} else {
				actualResultFileName = findFreeFile(resultFileName
						.replaceFirst("\\..*?$", ""));
			}
		}
		return actualResultFileName;
	}
}