X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2FIOUtils.java;h=439964dfc11dee8a317bf4e63a5251deaef7f17f;hb=0ff6a0aa1ec0a37d3fa0fbc4f76b9629fb2fd7ef;hp=3d252eac126df091a7fa8abf55999d95af6f288f;hpb=7a55041dd8940f529ab47284ee722e286b2e47e6;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/IOUtils.java b/src/be/nikiroo/utils/IOUtils.java
index 3d252ea..439964d 100644
--- a/src/be/nikiroo/utils/IOUtils.java
+++ b/src/be/nikiroo/utils/IOUtils.java
@@ -403,6 +403,34 @@ public class IOUtils {
 
 		return loader.getResourceAsStream(name);
 	}
+	
+	/**
+	 * Return the running directory/file, that is, the root binary directory for
+	 * running java classes or the running JAR file for JAR files.
+	 * 
+	 * @param clazz
+	 *            a Class from the running program (will only have an impact
+	 *            when not running from a JAR file)
+	 * @param base
+	 *            return the base directory (the one where the binary root or
+	 *            the JAR file resides)
+	 * 
+	 * @return the directory or file
+	 */
+	public static File getRunningDirectory(
+			@SuppressWarnings("rawtypes") Class clazz, boolean base) {
+		String uri = clazz.getProtectionDomain().getCodeSource().getLocation()
+				.toString();
+		if (uri.startsWith("file:"))
+			uri = uri.substring("file:".length());
+		File root = new File(uri);
+
+		if (base) {
+			root = root.getParentFile();
+		}
+
+		return root;
+	}
 
 	/**
 	 * Return a resetable {@link InputStream} from this stream, and reset it.