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

Go to download
package com.expleague.ml.io;

import com.expleague.ml.dynamicGrid.impl.BFDynamicGrid;
import com.expleague.ml.dynamicGrid.impl.StaticRow;
import com.expleague.ml.dynamicGrid.interfaces.DynamicRow;
import com.expleague.commons.func.Converter;
import com.expleague.commons.seq.CharSeqReader;
import com.expleague.ml.dynamicGrid.interfaces.DynamicGrid;
import gnu.trove.list.array.TDoubleArrayList;

import java.io.IOException;
import java.io.LineNumberReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class DynamicGridStringConverter implements Converter {
  @Override
  public DynamicGrid convertFrom(final CharSequence source) {
    final List rows = new ArrayList(1000);
    final LineNumberReader reader = new LineNumberReader(new CharSeqReader(source));
    String line;
    try {
      final TDoubleArrayList borders = new TDoubleArrayList();
//      int bfIndex = 0;
      int lineIndex = 0;
      while ((line = reader.readLine()) != null) {
        final StringTokenizer tok = new StringTokenizer(line, " \t");
        borders.clear();
        while (tok.hasMoreElements()) {
          borders.add(Double.parseDouble(tok.nextToken()));
        }
        rows.add(new StaticRow(null, lineIndex, borders.toArray()));
//        bfIndex += borders.size();
        lineIndex++;
      }
    } catch (IOException ioe) { //skip
    }
    return new BFDynamicGrid(rows.toArray(new StaticRow[rows.size()]));
  }

  @Override
  public CharSequence convertTo(final DynamicGrid grid) {
    final StringBuilder builder = new StringBuilder();
    for (int rowIndex = 0; rowIndex < grid.rows(); rowIndex++) {
      final DynamicRow row = grid.row(rowIndex);
      final int rowBfCount = row.size();
      for (int border = 0; border < rowBfCount; border++) {
        builder.append(border > 0 ? "\t" : "");
        builder.append(row.bf(border).condition());
      }
      builder.append("\n");
    }
    return builder;
  }

}