hex.generic.GenericModelMojoWriter 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 hex.generic;

import hex.ModelMojoWriter;
import hex.genmodel.utils.IOUtils;
import water.api.StreamWriteOption;
import water.fvec.ByteVec;
import water.util.Log;

import java.io.*;

public class GenericModelMojoWriter extends ModelMojoWriter {

    private ByteVec _mojoBytes;

    @SuppressWarnings("unused")  // Called through reflection in ModelBuildersHandler
    public GenericModelMojoWriter() {
    }

    public GenericModelMojoWriter(ByteVec _mojoBytes) {
        this._mojoBytes = _mojoBytes;
    }


    @Override
    public String mojoVersion() {
        return "1.00";
    }

    @Override
    protected void writeModelData() throws IOException {
        // Do nothing on purpose
    }

    @Override
    public void writeTo(final OutputStream os, StreamWriteOption... options) {
        try (final InputStream inputStream = _mojoBytes.openStream(null); OutputStream outputStream = os) {
            IOUtils.copyStream(inputStream, outputStream);
        } catch (IOException e) {
            Log.throwErr(e);
        }
    }
}