com.expleague.ml.io.TransJoinConversionPack Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmll
Various ML methods implemented by myself and my students
There is a newer version: 1.4.9
package com.expleague.ml.io;

import com.expleague.commons.func.types.ConversionPack;
import com.expleague.commons.math.Trans;
import com.expleague.commons.func.types.ConversionDependant;
import com.expleague.commons.func.types.ConversionRepository;
import com.expleague.commons.func.types.TypeConverter;
import com.expleague.commons.seq.CharSeqTools;
import com.expleague.ml.func.TransJoin;

/**
 * User: solar
 * Date: 12.08.13
 * Time: 17:16
 */
public class TransJoinConversionPack implements ConversionPack {

  public static abstract class BaseTo implements TypeConverter, ConversionDependant {
    private ConversionRepository repository;

    @Override
    public void setConversionRepository(final ConversionRepository repository) {
      this.repository = repository;
    }

    protected CharSequence convertModels(final F from) {
      final StringBuilder builder = new StringBuilder();
      builder.append("{").append(from.dirs.length).append(",\n");
      for (int i = 0; i < from.dirs.length; i++) {
        builder.append("{");
        final Trans model = from.dirs[i];
        builder.append(from.dirs[i].getClass().getCanonicalName()).append(",\n");
        builder.append("{");
        builder.append(repository.convert(model, CharSequence.class));
        builder.append("}");
        builder.append("},\n");
      }
      builder.delete(builder.length() - 1, builder.length());
      builder.append("}");
      return builder;
    }
  }

  public static class To extends BaseTo {
    @Override
    public CharSequence convert(final TransJoin from) {
      return convertModels(from);
    }
  }

  public static abstract class BaseFrom implements TypeConverter, ConversionDependant {
    private ConversionRepository repository;

    @Override
    public void setConversionRepository(final ConversionRepository repository) {
      this.repository = repository;
    }

    protected Trans[] convertModels(final CharSequence from) {
      int index = CharSeqTools.skipTo(from, 0, '{') + 1;
      final int modelsCount = Integer.parseInt(CharSeqTools.cut(from, index, ',').toString().trim());
      final Trans[] models = new Trans[modelsCount];
      index = CharSeqTools.skipTo(from, index, ',');
      try {
        for (int i = 0; i < models.length; i++) {
          final CharSequence modelCS = CharSeqTools.cutBetween(from, index, '{', '}');
          {
            final int mindex = 0;
            final String modelClassName = CharSeqTools.cut(modelCS, mindex, ',').toString().trim();
            final Class elementClass = (Class) Class.forName(modelClassName);
            models[i] = repository.convert(CharSeqTools.cutBetween(modelCS, mindex, '{', '}'), elementClass);
          }
          index += modelCS.length();
        }
      } catch (ClassNotFoundException e) {
        throw new RuntimeException("Element class not found!", e);
      }
      return models;
    }
  }

  public static class From extends BaseFrom {
    @Override
    public TransJoin convert(final CharSequence from) {
      return new TransJoin(convertModels(from));
    }
  }

  @Override
  public Class to() {
    return To.class;
  }

  @Override
  public Class from() {
    return From.class;
  }
}