!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")));
  }
}