NativeResource.java

Go to the documentation of this file.

package com.neuronrobotics.imageprovider;
 
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
 
public class NativeResource {
    
    private boolean loaded = false;
    public synchronized void load(String libraryName) throws NativeResourceException {  
        if(loaded)
            return;
        loaded = true;
        if(System.getProperty(libraryName + ".userlib") != null) {
            try {
                if(System.getProperty(libraryName + ".userlib").equalsIgnoreCase("sys")) {
                    System.loadLibrary(libraryName);
                } else {
                    System.load(System.getProperty(libraryName + ".userlib"));
                }
                return;
            } catch (Exception e){
                e.printStackTrace();
                throw new NativeResourceException("Unable to load native resource from given path.\n" + e.getLocalizedMessage());
            }
        }
        loadLib(libraryName);   
    }
    
 
    
    private void inJarLoad(String name)throws UnsatisfiedLinkError, NativeResourceException{
        //start by assuming the library can be loaded from the jar
        InputStream resourceSource = locateResource(name);
        File resourceLocation;
        try {
            resourceLocation = inJarLoad(resourceSource,name);
            loadResource(resourceLocation);
            testNativeCode();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
    }
    @SuppressWarnings("rawtypes")
    public static File inJarLoad(Class inputClass, String name) throws IOException{
        InputStream resourceSource = inputClass.getResourceAsStream(name);
        File resourceLocation = prepResourceLocation(name);
        //System.out.println("Resource selected "+resourceSource);
        //System.out.println("Resource target "+resourceLocation);
 
        copyResource(resourceSource, resourceLocation);
        return resourceLocation;
    }
    public static File inJarLoad(InputStream inputStream, String name) throws IOException{
        InputStream resourceSource = inputStream;
        File resourceLocation = prepResourceLocation(name);
        //System.out.println("Resource selected "+resourceSource);
        //System.out.println("Resource target "+resourceLocation);
 
        copyResource(resourceSource, resourceLocation);
        return resourceLocation;
    }
    private void loadLib(String name) throws NativeResourceException {
 
 
        inJarLoad(name);
        
        return;
    
    }
    
    private void testNativeCode()throws UnsatisfiedLinkError {
        
    }
 
    private InputStream locateResource(String name) {
        name +=  getExtension();
        String file="";
        if( isOSX()) {
            file="/native/osx/" + name;
        }else if( isWindows()) {
            if( is64Bit()){
                file="/native/windows/x86_64/" + name;
            }else {
                file="/native/windows/x86_32/" + name;
            }
        }else if( isLinux()) {
            if( isARM()) {
                file = "/native/linux/ARM/" + name;
            }else {
                if( is64Bit()) {
                    file="/native/linux/x86_64/" + name;
                }else {
                    file="/native/linux/x86_32/" + name;
                }
            }
        }else{
            System.err.println("Can't load native file: "+name+" for os arch: "+ getOsArch());
            return null;
        }
        //System.out.println("Loading "+file);
        return getClass().getResourceAsStream(file);
    }
    
    private void loadResource(File resource) {
        if(!resource.canRead())
            throw new RuntimeException("Cant open JNI file: "+resource.getAbsolutePath());
        //System.out.println("Loading: "+resource.getAbsolutePath());
        System.load(resource.getAbsolutePath());
    }
 
    public static void copyResource(InputStream io, File file) throws IOException {
        FileOutputStream fos = new FileOutputStream(file);
        
        
        byte[] buf = new byte[256];
        int read = 0;
        while ((read = io.read(buf)) > 0) {
            fos.write(buf, 0, read);
        }
        fos.close();
        io.close();
    }
 
    public static File prepResourceLocation(String fileName) throws NativeResourceException {       
        String tmpDir = System.getProperty("java.io.tmpdir");
        //String tmpDir = "M:\\";
        if ((tmpDir == null) || (tmpDir.length() == 0)) {
            tmpDir = "tmp";
        }
        
        String displayName = new File(fileName).getName().split("\\.")[0];
        
        String user = System.getProperty("user.name");
        
        File fd = null;
        File dir = null;
        
        for(int i = 0; i < 10; i++) {
            dir = new File(tmpDir, displayName + "_" + user + "_" + (i));
            if (dir.exists()) {
                if (!dir.isDirectory()) {
                    continue;
                }
                
                try {
                    File[] files = dir.listFiles();
                    for (int j = 0; j < files.length; j++) {
                        if (!files[j].delete()) {
                            continue;
                        }
                    }
                } catch (Throwable e) {
                    
                }
            }
            
            if ((!dir.exists()) && (!dir.mkdirs())) {
                continue;
            }
            
            try {
                dir.deleteOnExit();
            } catch (Throwable e) {
                // Java 1.1 or J9
            }
            
            fd = new File(dir, fileName + getExtension());
            if ((fd.exists()) && (!fd.delete())) {
                continue;
            }
            
            try {
                if (!fd.createNewFile()) {
                    continue;
                }
            } catch (IOException e) {
                continue;
            } catch (Throwable e) {
                // Java 1.1 or J9
            }
            
            break;
        }
        
        if(fd == null || !fd.canRead()) {
            throw new NativeResourceException("Unable to deploy native resource");
        }
        //System.out.println("Local file: "+fd.getAbsolutePath());
        return fd;
    }
    
 
        public static boolean is64Bit() {
            return getOsArch().startsWith("x86_64") || getOsArch().startsWith("amd64");
        }
        public static boolean isARM() {
            return getOsArch().startsWith("arm");
        }
        public static boolean isCortexA8(){
            if(isARM()){
                //TODO check for cortex a8 vs arm9 generic
                return true;
            }
            return false;
        }
        public static boolean isWindows() {
            return getOsName().toLowerCase().startsWith("windows") ||getOsName().toLowerCase().startsWith("microsoft") || getOsName().toLowerCase().startsWith("ms");
        }
        
        public static boolean isLinux() {
            return getOsName().toLowerCase().startsWith("linux");
        }
        
        public static boolean isOSX() {
            return getOsName().toLowerCase().startsWith("mac");
        }
        
        public static String getExtension() {
            return "";
        }
        
        public static String getOsName() {  
            return System.getProperty("os.name");
        }
        
        public static String getOsArch() {
            return System.getProperty("os.arch");
        }
        
        @SuppressWarnings("unused")
        public static String getIdentifier() {
            return getOsName() + " : " + getOsArch();
        }
    
 
}