water.tools.MojoConvertTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of h2o-algos Show documentation
H2O Algorithms
There is a newer version: 3.46.0.6
package water.tools;

import hex.generic.Generic;
import hex.generic.GenericModel;
import water.ExtensionManager;
import water.H2O;
import water.Paxos;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

/**
 * Convenience command line tool for converting H2O MOJO to POJO
 */
public class MojoConvertTool {

    private final File _mojo_file;
    private final File _pojo_file;

    public MojoConvertTool(File mojoFile, File pojoFile) {
        _mojo_file = mojoFile;
        _pojo_file = pojoFile;
    }

    void convert() throws IOException {
        GenericModel mojo = Generic.importMojoModel(_mojo_file.getAbsolutePath(), true);
        String pojo = mojo.toJava(false, true);
        Path pojoPath = Paths.get(_pojo_file.toURI());
        Files.write(pojoPath, pojo.getBytes(StandardCharsets.UTF_8));
    }

    private static void usage() {
        System.err.println("java -cp h2o.jar " + MojoConvertTool.class.getName() + " source_mojo.zip target_pojo.java");
    }

    public static void main(String[] args) throws IOException {
        if (args.length < 2) {
            usage();
            System.exit(1);
        }

        File mojoFile = new File(args[0]);
        if (!mojoFile.isFile()) {
            System.err.println("Specified MOJO file (" + mojoFile.getAbsolutePath() + ") doesn't exist!");
            System.exit(2);
        }
        File pojoFile = new File(args[1]);
        if (pojoFile.isDirectory() || (pojoFile.getParentFile() != null && !pojoFile.getParentFile().isDirectory())) {
            System.err.println("Invalid target POJO file (" + pojoFile.getAbsolutePath() + ")! Please specify a file in an existing directory.");
            System.exit(3);
        }

        System.out.println();
        System.out.println("Starting local H2O instance to facilitate MOJO to POJO conversion.");
        System.out.println();

        H2O.main(new String[]{"-disable_web", "-ip", "localhost", "-disable_net"});
        ExtensionManager.getInstance().registerRestApiExtensions();

        H2O.waitForCloudSize(1, 60_000);
        Paxos.lockCloud("H2O is started in a single node configuration.");

        System.out.println();
        System.out.println("Converting " + mojoFile + " to " + pojoFile + "...");
        new MojoConvertTool(mojoFile, pojoFile).convert();
        System.out.println("DONE");
    }

}