!747

!awt {

!actionListener {

m {

  static S snippetID;

  static int maxTokens = 100;

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

  static S corpusID = "#1001006"; // Snippets DB

  static S predictorID = "#1001011";

  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(); }

  }

  static void ask() {

    corpusID = slte("Corpus ID:", corpusID);

    awt { go(); }

  }

  static void go() {

    //snippetID = formatSnippetID(snippetID);

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

    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();

    }

  }

  static void 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);

      }

    }

  }

  static void 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(); }

  }

  static void advisorShow(final S text) ctex {

    awt {

      advisorScroller.setViewportView(advisorTextPane);

      advisorTextPane.setText(text);

    }

  }

  static void 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));

      // make single prediction

      S prediction = cast call(p, "read", 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", 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")));

  }

}