jcuda.LibUtils Maven / Gradle / Ivy
/*
*
* * Copyright 2015 Skymind,Inc.
* *
* * Licensed under the Apache License, Version 2.0 (the "License");
* * you may not use this file except in compliance with the License.
* * You may obtain a copy of the License at
* *
* * http://www.apache.org/licenses/LICENSE-2.0
* *
* * Unless required by applicable law or agreed to in writing, software
* * distributed under the License is distributed on an "AS IS" BASIS,
* * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* * See the License for the specific language governing permissions and
* * limitations under the License.
*
*
*/
package jcuda;
import java.io.*;
import java.util.Locale;
/**
* Utility class for detecting the operating system and architecture
* types, and automatically loading the matching native library
* as a resource or from a file.
*
* The architecture and OS detection has been adapted from
* http://javablog.co.uk/2007/05/19/making-jni-cross-platform/
* and extended with http://lopica.sourceforge.net/os.html
*/
public final class LibUtils
{
/**
* Enumeration of common operating systems, independent of version
* or architecture.
*/
public static enum OSType
{
APPLE, LINUX, SUN, WINDOWS, UNKNOWN
}
/**
* Enumeration of common CPU architectures.
*/
public static enum ARCHType
{
PPC, PPC_64, SPARC, X86, X86_64, ARM, MIPS, RISC, UNKNOWN
}
/**
* Loads the specified library. The full name of the library
* is created by calling {@link LibUtils#createLibName(String)}
* with the given argument. The method will attempt to load
* the library as a as a resource (for usage within a JAR),
* and, if this fails, using the usual System.loadLibrary
* call.
*
* @param baseName The base name of the library
* @throws UnsatisfiedLinkError if the native library
* could not be loaded.
*/
public static void loadLibrary(String baseName)
{
String libName = LibUtils.createLibName(baseName);
Throwable throwable = null;
final boolean tryResource = true;
if (tryResource)
{
try
{
loadLibraryResource(libName);
return;
}
catch (Throwable t)
{
throwable = t;
}
}
try
{
System.loadLibrary(libName);
return;
}
catch (Throwable t)
{
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
pw.println("Error while loading native library \"" +
libName + "\" with base name \""+baseName+"\"");
pw.println("Operating system name: "+
System.getProperty("os.name"));
pw.println("Architecture : "+
System.getProperty("os.arch"));
pw.println("Architecture bit size: "+
System.getProperty("sun.arch.data.model"));
if (throwable != null)
{
pw.println(
"Stack trace from the attempt to " +
"load the library as a resource:");
throwable.printStackTrace(pw);
}
pw.println(
"Stack trace from the attempt to " +
"load the library as a file:");
t.printStackTrace(pw);
pw.flush();
pw.close();
throw new UnsatisfiedLinkError(
"Could not load the native library.\n"+
sw.toString());
}
}
/**
* Load the library with the given name from a resource.
* The extension for the current OS will be appended.
*
* @param libName The library name
* @throws Throwable If the library could not be loaded
*/
private static void loadLibraryResource(String libName) throws Throwable
{
String libPrefix = createLibPrefix();
String libExtension = createLibExtension();
String fullName = libPrefix + libName;
String resourceName = "/lib/" + fullName + "." + libExtension;
InputStream inputStream =
LibUtils.class.getResourceAsStream(resourceName);
if (inputStream == null)
{
throw new NullPointerException(
"No resource found with name '"+resourceName+"'");
}
File tempFile = File.createTempFile(fullName, "."+libExtension);
tempFile.deleteOnExit();
OutputStream outputStream = null;
try
{
outputStream = new FileOutputStream(tempFile);
byte[] buffer = new byte[8192];
while (true)
{
int read = inputStream.read(buffer);
if (read < 0)
{
break;
}
outputStream.write(buffer, 0, read);
}
outputStream.flush();
outputStream.close();
outputStream = null;
System.load(tempFile.toString());
}
finally
{
if (outputStream != null)
{
outputStream.close();
}
}
}
/**
* Returns the extension for dynamically linked libraries on the
* current OS. That is, returns "jnilib" on Apple, "so" on Linux
* and Sun, and "dll" on Windows.
*
* @return The library extension
*/
private static String createLibExtension()
{
OSType osType = calculateOS();
switch (osType)
{
case APPLE:
return "jnilib";
case LINUX:
return "so";
case SUN:
return "so";
case WINDOWS:
return "dll";
}
return "";
}
/**
* Returns the prefix for dynamically linked libraries on the
* current OS. That is, returns "lib" on Apple, Linux and Sun,
* and the empty String on Windows.
*
* @return The library prefix
*/
private static String createLibPrefix()
{
OSType osType = calculateOS();
switch (osType)
{
case APPLE:
case LINUX:
case SUN:
return "lib";
case WINDOWS:
return "";
}
return "";
}
/**
* Creates the name for the native library with the given base
* name for the current operating system and architecture.
* The resulting name will be of the form
* baseName-OSType-ARCHType
* where OSType and ARCHType are the lower case Strings
* of the respective enum constants. Example:
* jcuda-windows-x86
*
* @param baseName The base name of the library
* @return The library name
*/
public static String createLibName(String baseName)
{
OSType osType = calculateOS();
ARCHType archType = calculateArch();
String libName = baseName;
libName += "-" + osType.toString().toLowerCase(Locale.ENGLISH);
libName += "-" + archType.toString().toLowerCase(Locale.ENGLISH);
return libName;
}
/**
* Calculates the current OSType
*
* @return The current OSType
*/
public static OSType calculateOS()
{
String osName = System.getProperty("os.name");
osName = osName.toLowerCase(Locale.ENGLISH);
if (osName.startsWith("mac os"))
{
return OSType.APPLE;
}
if (osName.startsWith("windows"))
{
return OSType.WINDOWS;
}
if (osName.startsWith("linux"))
{
return OSType.LINUX;
}
if (osName.startsWith("sun"))
{
return OSType.SUN;
}
return OSType.UNKNOWN;
}
/**
* Calculates the current ARCHType
*
* @return The current ARCHType
*/
public static ARCHType calculateArch()
{
String osArch = System.getProperty("os.arch");
osArch = osArch.toLowerCase(Locale.ENGLISH);
if (osArch.equals("i386") ||
osArch.equals("x86") ||
osArch.equals("i686"))
{
return ARCHType.X86;
}
if (osArch.startsWith("amd64") || osArch.startsWith("x86_64"))
{
return ARCHType.X86_64;
}
if (osArch.equals("ppc") || osArch.equals("powerpc"))
{
return ARCHType.PPC;
}
if (osArch.startsWith("ppc"))
{
return ARCHType.PPC_64;
}
if (osArch.startsWith("sparc"))
{
return ARCHType.SPARC;
}
if (osArch.startsWith("arm"))
{
return ARCHType.ARM;
}
if (osArch.startsWith("mips"))
{
return ARCHType.MIPS;
}
if (osArch.contains("risc"))
{
return ARCHType.RISC;
}
return ARCHType.UNKNOWN;
}
/**
* Private constructor to prevent instantiation.
*/
private LibUtils()
{
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy