jnr.ffi.provider.jffi.NativeClosureManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jnr-ffi Show documentation
A library for invoking native functions from java
The newest version!
/*
 * Copyright (C) 2011 Wayne Meissner
 *
 * This file is part of the JNR project.
 *
 * 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 jnr.ffi.provider.jffi;

import jnr.ffi.Pointer;
import jnr.ffi.mapper.CachingTypeMapper;
import jnr.ffi.mapper.CompositeTypeMapper;
import jnr.ffi.mapper.SignatureTypeMapper;
import jnr.ffi.mapper.ToNativeContext;
import jnr.ffi.mapper.ToNativeConverter;
import jnr.ffi.provider.ClosureManager;

import java.util.IdentityHashMap;
import java.util.Map;

/**
 *
 */
final class NativeClosureManager implements ClosureManager {
    private volatile Map, NativeClosureFactory> factories = new IdentityHashMap, NativeClosureFactory>();
    private volatile Map asmClassLoaders = new IdentityHashMap();
    private final jnr.ffi.Runtime runtime;
    private final SignatureTypeMapper typeMapper;

    NativeClosureManager(jnr.ffi.Runtime runtime, SignatureTypeMapper typeMapper) {
        this.runtime = runtime;
        this.typeMapper = new CompositeTypeMapper(typeMapper, new CachingTypeMapper(new ClosureTypeMapper()));
    }

     NativeClosureFactory getClosureFactory(Class closureClass) {
        NativeClosureFactory factory = factories.get(closureClass);
        if (factory != null) {
            return factory;
        }
        AsmClassLoader asmCl = asmClassLoaders.get(closureClass.getClassLoader());
        if (asmCl==null) {
            asmCl = new AsmClassLoader( closureClass.getClassLoader());
            asmClassLoaders.put(closureClass.getClassLoader(), asmCl);
        }

        return initClosureFactory(closureClass, asmCl);
    }

    public  T newClosure(Class closureClass, T instance) {
        NativeClosureFactory factory = factories.get(closureClass);
        if (factory != null) {
            //return factory.newClosure(instance);
        }
        return null;
    }

    public final  jnr.ffi.Pointer getClosurePointer(Class closureClass, T instance) {
        return getClosureFactory(closureClass).getClosureReference(instance).getPointer();
    }

    synchronized  NativeClosureFactory initClosureFactory(Class closureClass, AsmClassLoader classLoader) {
        NativeClosureFactory factory = factories.get(closureClass);
        if (factory != null) {
            return factory;
        }


        factory = NativeClosureFactory.newClosureFactory(runtime, closureClass, typeMapper, classLoader);
        Map, NativeClosureFactory> factories = new IdentityHashMap, NativeClosureFactory>();
        factories.putAll(this.factories);
        factories.put(closureClass, factory);
        this.factories = factories;

        return factory;
    }

     ToNativeConverter newClosureSite(Class closureClass) {
        return new ClosureSite(getClosureFactory(closureClass));
    }

    @ToNativeConverter.NoContext
    public static final class ClosureSite implements ToNativeConverter {
        private final NativeClosureFactory factory;
        private NativeClosureFactory.ClosureReference closureReference = null;

        private ClosureSite(NativeClosureFactory factory) {
            this.factory = factory;
        }

        public Pointer toNative(T value, ToNativeContext context) {
            if (value == null) {
                return null;
            }
            
            // If passing down a function pointer, don't re-wrap it
            if (value instanceof ClosureFromNativeConverter.AbstractClosurePointer) {
                return (ClosureFromNativeConverter.AbstractClosurePointer) value;
            }

            NativeClosureFactory.ClosureReference ref = closureReference;
            // Fast path - same delegate as last call to this site - just re-use the native closure
            if (ref != null && ref.getCallable() == value) {
                return ref.getPointer();
            }

            ref = factory.getClosureReference(value);
            // Cache the new native closure, if this site has no valid native closure
            if (closureReference == null || closureReference.get() == null) {
                closureReference = ref;
            }

            return ref.getPointer();
        }

        public Class nativeType() {
            return Pointer.class;
        }
    }
}