org.deeplearning4j.common.config.DL4JClassLoading Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-common Show documentation
The newest version!
/*
 *  ******************************************************************************
 *  *
 *  *
 *  * This program and the accompanying materials are made available under the
 *  * terms of the Apache License, Version 2.0 which is available at
 *  * https://www.apache.org/licenses/LICENSE-2.0.
 *  *
 *  *  See the NOTICE file distributed with this work for additional
 *  *  information regarding copyright ownership.
 *  * 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.
 *  *
 *  * SPDX-License-Identifier: Apache-2.0
 *  *****************************************************************************
 */

package org.deeplearning4j.common.config;

import lombok.extern.slf4j.Slf4j;
import org.nd4j.common.base.Preconditions;
import org.nd4j.common.config.ND4JClassLoading;

import java.lang.reflect.InvocationTargetException;
import java.util.Objects;
import java.util.ServiceLoader;

@Slf4j
public class DL4JClassLoading {
    private static ClassLoader dl4jClassloader = ND4JClassLoading.getNd4jClassloader();

    private DL4JClassLoading() {
    }

    public static ClassLoader getDl4jClassloader() {
        return DL4JClassLoading.dl4jClassloader;
    }

    public static void setDl4jClassloaderFromClass(Class clazz) {
        setDl4jClassloader(clazz.getClassLoader());
    }

    public static void setDl4jClassloader(ClassLoader dl4jClassloader) {
        DL4JClassLoading.dl4jClassloader = dl4jClassloader;
        log.debug("Global class-loader for DL4J was changed.");
    }

    public static boolean classPresentOnClasspath(String className) {
        return classPresentOnClasspath(className, dl4jClassloader);
    }

    public static boolean classPresentOnClasspath(String className, ClassLoader classLoader) {
        return loadClassByName(className, false, classLoader) != null;
    }

    public static  Class loadClassByName(String className) {
        return loadClassByName(className, true, dl4jClassloader);
    }

    @SuppressWarnings("unchecked")
    public static  Class loadClassByName(String className, boolean initialize, ClassLoader classLoader) {
        try {
            return (Class) Class.forName(className, initialize, classLoader);
        } catch (ClassNotFoundException classNotFoundException) {
            log.error(String.format("Cannot find class [%s] of provided class-loader.", className));
            return null;
        }
    }

    public static  T createNewInstance(String className) {
        return createNewInstance(className, Object.class, new Object[0]);//or null;
    }
    
    public static  T createNewInstance(String className, Object[] args) {
       
    	return createNewInstance(className, Object.class, args);
    }

    public static  T createNewInstance(String className, Class superclass) {
        return createNewInstance(className, superclass, new Class[]{}, new Object[]{});
    }

    public static  T createNewInstance(String className, Class superclass, Object[] args) {
        Class[] parameterTypes = new Class[args.length];
        for (int i = 0; i < args.length; i++) {
            Object arg = args[i];
            Objects.requireNonNull(arg);
            parameterTypes[i] = arg.getClass();
        }

        return createNewInstance(className, superclass, parameterTypes, args);
    }

    public static  T createNewInstance(
            String className,
            Class superclass,
            Class[] parameterTypes,
            Object [] args) {
        try {
            Class