!752

static S snippetID;

static int maxTokens = 100;

static S fontID = null/*"#1000993"*/;

static S corpusID = "#1001010";

static S predictorID = "#1001028";

static boolean allTokens = true;

static JTextArea textArea;

static JLabel status;

static JButton btnSave;

static JTextArea advisorTextPane;

static JScrollPane textScroller, advisorScroller;

static volatile Object predictor;

static Font fancyFont;

static class MyLayoutManager implements LayoutManager {

  public void addLayoutComponent(String s, Component c) {}

  public void removeLayoutComponent(Component c) {}

  public Dimension preferredLayoutSize(Container c) {

    return minimumLayoutSize(c);

  }

  public Dimension minimumLayoutSize(Container c) {

    return new Dimension(500, 400);

  }

  public void layoutContainer(Container c) {

    int w = c.getWidth(), h = c.getHeight();

    int split = h*2/3;

    textScroller.setBounds(0, 0, w, split);

    advisorScroller.setBounds(0, split, w, h-split);

  }

}

p {

  if (args.length != 0) snippetID = args[0];

  thread { ask(); }

}

svoid ask() {

  corpusID = slte("Corpus ID or GitHub URL:", corpusID);

  if (corpusID != null && !isSnippetID(corpusID)) {

    Matcher m  = matcher("github.com/([^/]+/[^/]+)", corpusID);

    if (m.find())

      github(m.group(1));

    else {

      m  = matcher("([^/]+/[^/]+)", corpusID);

      if (m.matches())

        github(corpusID);

      else {

        print("Unknown corpus reference: " + corpusID);

        corpusID = null;

      }

    }

  }

  awt { go(); }

}

svoid go() {

  //snippetID = formatSnippetID(snippetID);

  JFrame frame = new JFrame("Predictive Editor v2");

  JPanel panel2 = new JPanel(new MyLayoutManager());

  advisorTextPane = new JTextArea();

  advisorTextPane.setEditable(false);

  advisorTextPane.setLineWrap(true);

  advisorTextPane.setWrapStyleWord(true);

  // Prevents scrolling down?

  DefaultCaret caret = (DefaultCaret) advisorTextPane.getCaret();

  caret.setUpdatePolicy(DefaultCaret.NEVER_UPDATE);

  if (fontID != null) try {

    fancyFont = loadFont(fontID, 24);

    advisorTextPane.setFont(fancyFont);

  } catch (Throwable e) { e.printStackTrace(); }

  //advisorScroller = new JScrollPane(advisorTextPane);

  advisorScroller = new JScrollPane(makeAnimation());

  advisorScroller.setMinimumSize(new Dimension(100, 300));

  textArea = new JTextArea();

  textScroller = new JScrollPane(textArea);

  panel2.add(textScroller);

  panel2.add(advisorScroller);

  status = new JLabel(" ");

  btnSave = new JButton("Save " + snippetID);

  btnSave.setEnabled(false);

  btnSave.addActionListener(actionListener {

    saveSnippet();

  });

  JPanel bottom = new JPanel(new BorderLayout);

  bottom.add(BorderLayout.CENTER, status);

  //bottom.add(BorderLayout.EAST, btnSave);

  JPanel panel = new JPanel(new BorderLayout);

  panel.add(BorderLayout.CENTER, panel2);

  panel.add(BorderLayout.SOUTH, bottom);

  textArea.getDocument().addDocumentListener(new DocumentListener() {

    public void insertUpdate(DocumentEvent e) {

      btnSave.setEnabled(true);

      predict();

    }

    public void removeUpdate(DocumentEvent e) {

      btnSave.setEnabled(true);

      predict();

    }

    public void changedUpdate(DocumentEvent e) {

      // apparently it doesn't occur

    }

  });

  textArea.addCaretListener(new CaretListener() {

    public void caretUpdate(CaretEvent e) {

      predict();

    }

  });

  if (snippetID != null) {

    textArea.setEditable(false); // still loading

    status.setText("Loading...");

    thread {

      final S s = loadSnippet(snippetID);

      awt {

        status.setText("Loaded " + s.length() + " chars.");

        textArea.setText(s);

        //textArea.scrollRectToVisible(new Rectangle(0, 0, 1, 1));

        textArea.setCaretPosition(0);

        btnSave.setEnabled(false);

        textArea.setEditable(true);

        textArea.requestFocus();

      }

    }

  }

  frame.addWindowListener(new WindowAdapter() {

    public void windowOpened(WindowEvent e) {

      textArea.requestFocus();

    }

  });

  frame.add(panel);

  frame.setBounds(200, 150, 500, 420);

  frame.setVisible(true);

  exitOnFrameClose(frame);

  thread {

    loadPredictor();

  }

}

svoid saveSnippet() {

  final S text = textArea.getText();

  btnSave.setEnabled(false);

  status.setText("Saving...");

  thread {

    final S url = "http://tinybrain.de:8080/tb-int/update_snippet_text.php";

    S user = loadTextFile(new File(userHome(), ".tinybrain/username").getPath(), null);

    S pass = loadTextFile(new File(userHome(), ".tinybrain/userpass").getPath(), null);

    S query = "id=" + parseSnippetID(snippetID) + "&text=" + urlencode(text) + "&_user=" + urlencode(user) + "&_pass=" + urlencode(pass);

    final S page = doPost(query, url);

    awt {

      status.setText("Saved snippet: " + page);

    }

  }

}

svoid loadPredictor() ctex {

  // We have the animation instead.

  //advisorShow("Loading predictor...");

  Class learner = hotwire(predictorID);

  set(learner, "showGUI", false);

  if (corpusID != null)

    set(learner, "corpusID", corpusID);

  callMain(learner);

  predictor = get(get(learner, "collector"), "winner");

  print("Predictor: " + predictor);

  awt { predict(); }

}

svoid advisorShow(final S text) ctex {

  awt {

    advisorScroller.setViewportView(advisorTextPane);

    advisorTextPane.setText(text);

  }

}

svoid predict() {

  try {

    long startTime = now();

    Object p = predictor;

    if (p == null) return;

    //p = clone(p);

    p = call(p, "derive"); // properly make a local copy

    S text = textArea.getText();

    int idx = textArea.getCaretPosition();

    S previous = safeSubstring(text, 0, idx);

    previous ="\n==\n" + previous; // indicate start of snippet to predictor

    L<S> tok = javaTok(previous);

    if (!allTokens) tok = codeTokensOnly(tok);

    //if (tok.size() != 0) tok.remove(tok.size()-1);

    print(structure(tok));

    S fileName = "#new#";

    // make single prediction

    S prediction = cast call(p, "read", fileName, tok);

    new StringBuilder buf;

    if (prediction != null) {

      //buf.append(prediction + "\n\n");

      new L<S> multiPrediction;

      S pred = prediction;

      int i = 0;

      while (true) {

        multiPrediction.add(pred);

        if (++i >= maxTokens) break;

        tok.add(pred);

        pred = (String) call(p, "read", fileName, tok);

        if (pred == null) break;

      }

      if (multiPrediction.size() > 1)

        buf.append(join(multiPrediction) + "\n");

    }

    advisorShow(buf.toString());

    print((now()-startTime) + " ms");

  } catch (Throwable e) {

    e.printStackTrace();

    // TODO: use "error" font :))))

    advisorShow(e.toString());

  }

}

static L<S> codeTokensOnly(L<S> tok) {

  new L<S> l;

  for (int i = 1; i < tok.size(); i += 2)

    l.add(tok.get(i));

  return l;

}

!include #1001014 // ScalablePane

static JComponent makeAnimation() ctex {

  JLabel label = new JLabel(new ImageIcon(loadLibrary("#1001015").toURI().toURL()));

  label.setText("Loading predictor...");

  label.setVerticalTextPosition(SwingConstants.BOTTOM);

  label.setHorizontalTextPosition(SwingConstants.CENTER);

  /*label.setBackground(Color.white);

  label.setOpaque(true);*/

  return label;

  //return new ScalablePane(ImageIO.read(loadLibrary("#1001013")));

}

svoid github(S githubPath) ctex {

  S url = "https://github.com/" + githubPath + "/archive/master.zip";

  S found = findSnippetNamed("Data: " + url);

  print("Found: " + found);

  if (found != null)

    corpusID = found;

  else {

    S addURL = "http://tinybrain.de:8080/tb-int/add_snippet.php";

    S query = "type=37&high=1&public=1&url=" + urlencode(url);

    corpusID = doPost(query, addURL);

    print(corpusID);

    if (!isSnippetID(corpusID)) {

      print("Github upload failed");

      corpusID = null;

    }

  }

}