io.deephaven.engine.util.jpy.JpyInit Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation

Engine Table: Implementation and closely-coupled utilities

There is a newer version: 0.37.1

Show newest version

/**
 * Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
 */
package io.deephaven.engine.util.jpy;

import io.deephaven.configuration.Configuration;
import io.deephaven.internal.log.LoggerFactory;
import io.deephaven.io.logger.Logger;
import io.deephaven.jpy.JpyConfig;
import io.deephaven.jpy.JpyConfigExt;
import io.deephaven.jpy.JpyConfigFromSubprocess;
import io.deephaven.jpy.JpyConfigSource;
import org.jpy.PyLibInitializer;
import org.jpy.PyModule;
import org.jpy.PyObject;

import java.io.IOException;
import java.time.Duration;
import java.util.concurrent.TimeoutException;

/**
 * Initializes Jpy.
 *
 * A guarded initialization via {@link JpyConfigExt#startPython()}
 */
public class JpyInit {
    private static final Logger log = LoggerFactory.getLogger(JpyInit.class);

    /**
     * First, checks {@link Configuration#getInstance()} for any explicit jpy properties. If none are set, it will
     * source the configuration from {@link JpyConfigFromSubprocess#fromSubprocess(Duration)}.
     *
     * @throws IOException if an IO exception occurs
     * @throws InterruptedException if the current thread is interrupted
     * @throws TimeoutException if the command times out
     */
    public static synchronized void init() throws IOException, InterruptedException, TimeoutException {
        if (PyLibInitializer.isPyLibInitialized()) {
            return;
        }
        final JpyConfig explicitConfig = new JpyConfigLoader(Configuration.getInstance()).asJpyConfig();
        if (!explicitConfig.isEmpty()) {
            init(new JpyConfigExt(explicitConfig));
        } else {
            final JpyConfigSource fromSubprocess = JpyConfigFromSubprocess.fromSubprocess(Duration.ofSeconds(10));
            init(new JpyConfigExt(fromSubprocess.asJpyConfig()));
        }
    }

    private static void init(JpyConfigExt jpyConfig) {
        log.info().append("Loaded jpy config ").append(jpyConfig.toString()).endl();
        log.info().append("Starting Python interpreter").endl();
        jpyConfig.initPython();
        jpyConfig.startPython();
        log.info().append("Started Python interpreter").endl();
        markReadyAndCheckEnv();
    }

    private static void markReadyAndCheckEnv() {
        // noinspection EmptyTryBlock,unused
        try (
                final PyModule deephavenJpyModule = PyModule.importModule("deephaven_internal.jvm");
                final PyObject readyObj = deephavenJpyModule.callMethod("ready");
                final PyObject checkObj = deephavenJpyModule.callMethod("check_py_env");
                final PyObject initObj = deephavenJpyModule.callMethod("init_py")) {
            // empty
        }
    }

}