ExternalFunctionClassLoader.java

/* 
 * This file is part or JMathLib 
 * 
 * Check it out at http://www.jmathlib.de
 *
 * Author:   
 * (c) 2005-2009   
 */
package jmathlib.core.functions;

import java.io.IOException;
import java.io.File;
import java.io.FileInputStream;
import jmathlib.core.interpreter.*;

/**Class to load any External functions used*/
public class ExternalFunctionClassLoader extends ClassLoader
{
    private FunctionPathBroker pathBroker;

    /**Create the class loader
       @param _classDir = the name of the base class directory*/
    public ExternalFunctionClassLoader(FunctionPathBroker _pathBroker)
    {
        pathBroker = _pathBroker;        
    }

    /** Returns a class name that refers to the functionName.
     * The class name (rather than simply the function name) must be used when calling any
     * Classloader methods.
     */
    public String getClassnameForFunction(String functionName) {
      File classFile = pathBroker.findFunction(functionName);
      if (classFile != null) {
        String baseDir = pathBroker.getBaseDirectory().toString();            
        String className = classFile.toString();
        className = className.substring(baseDir.length() + 1, className.length());
        className = className.replace('/',  '.');
        className = className.replace('\\', '.');
        if (className.toLowerCase().endsWith(".class"))
          className = className.substring(0, className.length()-6);        
        return className;
      }
      return null;
    }
    
    /* checks the specified directory for the class file, if this fails it
    calls the system file to load the class
    @param className = the class for to look for*/  
    public Class findClass(String className) throws ClassNotFoundException
    {
        ErrorLogger.debugLine("ext func loader loadClass "+ className);
        Class newClass = null;

        // try loading the class
        try
        {
            File baseDir = pathBroker.getBaseDirectory();
            
            String classFilename = className.replace('.', '/');
            classFilename = classFilename + ".class";
            
            File classFile = new File(baseDir, classFilename);            
            
            byte[] classData = getClassData(classFile);
            
            if (classData != null) {
              newClass = this.defineClass(className, classData, 0, classData.length);
            }
        }
        catch(Exception readError)
        {
          throw new ClassNotFoundException(className);
        }
        
        return newClass;
    }

    /**read the data for the class from the specified directory and any subdirectories
    @param className = the class to load*/
    protected byte[] getClassData(File classFile) throws IOException
    {
        byte[] classData = null;
        
        if (classFile == null)
        {
            ErrorLogger.debugLine("ext func loader: NOTFOUND");
            return null;
        }
        else if (!classFile.getName().endsWith(".class")){            
            ErrorLogger.debugLine("ext func loader: Non-class file attempted load");
            return null;
        }else
        {
            classData = new byte[((int)classFile.length())];
            
            //now open the input stream
            FileInputStream inFile = new FileInputStream(classFile);
            try {            
              inFile.read(classData);
            } finally {
              inFile.close();
            }
        }
        
        return classData;
    }
}