MultiModelMojoReader.java

package hex.genmodel;

import java.io.BufferedReader;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

public abstract class MultiModelMojoReader<M extends MojoModel> extends ModelMojoReader<M> {

  private Map<String, MojoModel> _subModels;

  @Override
  protected final void readModelData() throws IOException {
    int subModelCount = readkv("submodel_count", 0);
    HashMap<String, MojoModel> models = new HashMap<>(subModelCount);
    for (int i = 0; i < subModelCount; i++) {
      String key = readkv("submodel_key_" + i);
      String zipDirectory = readkv("submodel_dir_" + i);
      MojoModel model = ModelMojoReader.readFrom(new NestedMojoReaderBackend(zipDirectory));
      models.put(key, model);
    }
    _subModels = Collections.unmodifiableMap(models);
    readParentModelData();
  }

  protected MojoModel getModel(String key) {
    return _subModels.get(key);
  }

  protected abstract void readParentModelData() throws IOException;

  private class NestedMojoReaderBackend implements MojoReaderBackend {

    private String _zipDirectory;

    private NestedMojoReaderBackend(String zipDirectory) {
      _zipDirectory = zipDirectory;
    }

    @Override
    public BufferedReader getTextFile(String filename) throws IOException {
      return _reader.getTextFile(_zipDirectory + filename);
    }

    @Override
    public byte[] getBinaryFile(String filename) throws IOException {
      return _reader.getBinaryFile(_zipDirectory + filename);
    }

    @Override
    public boolean exists(String filename) {
      return _reader.exists(_zipDirectory + filename);
    }
  }

}