pl.poznan.put.structure.formats.AbstractDotBracket Maven / Gradle / Ivy

Go to download
package pl.poznan.put.structure.formats;

import pl.poznan.put.structure.DotBracketSymbol;
import pl.poznan.put.structure.ImmutableDotBracketSymbol;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

abstract class AbstractDotBracket implements DotBracket {
  @Override
  public List symbols() {
    return IntStream.range(0, sequence().length())
        .mapToObj(i -> ImmutableDotBracketSymbol.of(sequence().charAt(i), structure().charAt(i), i))
        .collect(Collectors.toList());
  }

  protected final List> candidatesToCombine() {
    final List> result = new ArrayList<>();
    final List toCombine = new ArrayList<>();
    int level = 0;

    for (final Strand strand : strands()) {
      toCombine.add(strand);

      for (final DotBracketSymbol symbol : strand.symbols()) {
        level += symbol.isOpening() ? 1 : 0;
        level -= symbol.isClosing() ? 1 : 0;
      }

      if (level == 0) {
        result.add(new ArrayList<>(toCombine));
        toCombine.clear();
      }
    }

    return result;
  }
}