!752

sS text = [[
Yo Spain! I said:

"Let's do that."

And he said

"Yeah sure"

And they said

"Yeah nice"

And we said

"That's cool!"
]];

static Color markedColor = Color.green;
static Color unmarkedColor = Color.white;

concepts.

concept Marking {
  S text;
  new BitSet marking;
}

static JTextPane pane, guessPane;
static Marking marking;
static Guesser guesser = new MarkLineByFirstChar;

sinterface Guesser {
  BitSet guess(S text, BitSet initial);
}

p-substance {
  loadAndAutoSaveConcepts();
  text = text.trim();
  marking = uniqueConcept(Marking.class, "text", text);
  pane = new JTextPane;
  guessPane = new JTextPane;
  showFrame(hgrid(
    withTitle("Your marking", pane),
    withTitle("My guess", guessPane)));
  update();
  addButtonsToWindow(pane,
    textComponentDependButton(pane, jbutton("Mark quote", "mark")),
    textComponentDependButton(pane, jbutton("Unmark quote", "unmark")));
}

svoid mark { mark(true); }
svoid unmark { mark(false); }

svoid mark(bool b) {
  int from = pane.getSelectionStart(), to = pane.getSelectionEnd();
  if (from < 0) ret;
  for (int i = from; i < to; i++)
    marking.marking.set(i, b);
  change();
  update();
}

svoid update {
  showMarkedText(pane, marking.marking);
  BitSet guess = guess();
  showMarkedText(guessPane, guess);
}

static BitSet guess() {
  try {
    if (guesser != null)
      ret unnull(guesser.guess(text, marking.marking));
  } catch e {
    print(e);
  }
  ret new BitSet;
}

svoid showMarkedText(JTextPane pane, BitSet marking) {
  new L mtext;
  Color c = unmarkedColor;
  new StringBuilder buf;
  for i over text: {
    Color cc = marking.get(i) ? markedColor : unmarkedColor;
    if (neq(cc, c)) {
      if (nempty(buf))
        mtext.addAll(ll(c, str(buf)));
      buf.setLength(0);
      c = cc;
    }
    buf.append(text.charAt(i));
  }
  if (nempty(buf))
    mtext.addAll(ll(c, str(buf)));
    
  // Update JTextPane with new colors, preserve selection
  int from = pane.getSelectionStart(), to = pane.getSelectionEnd();
  showColoredText(pane, mtext);
  pane.setSelectionStart(from);
  pane.setSelectionEnd(to);
}

sclass DummyGuesser implements Guesser {
  public BitSet guess(S text, BitSet initial) {
    BitSet bs = cloneBitSet(initial);
    for (int i = 0; i < l(text); i += 2)
      bs.set(i);
    ret bs;
  }
}

sclass LineStartsTest implements Guesser {
  public BitSet guess(S text, BitSet initial) {
    L<Int> l = lineStartIndexes(text);
    ret bitSet(l);
  }
}

sclass MarkLineByFirstChar implements Guesser {
  public BitSet guess(S text, BitSet initial) {
    L<S> lines = filterNempty(extractMarkedLines(text, initial));
    S pref = commonPrefixMulti(lines);
    if (empty(pref)) ret null;
    pref = substring(pref, 0, 1);
    print("Prefix: " + pref);
    new BitSet bs;
    for (int i : lineStartIndexes(text)) {
      S line = extractLine(text, i);
      if (startsWith(line, pref))
        setBitRange(bs, i, min(l(text), smartIndexOf(text, "\n", i)+1));
    }
    ret bs;
  }
}