import java.io.*;
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import java.util.*;
 
public class TextEditor extends JApplet {
 
 
  // Create menu items
  private JMenuItem jmiOpen = new JMenuItem("Open");
  private JMenuItem jmiSave = new JMenuItem("Save");
  private JMenuItem jmiClear = new JMenuItem("Clear");
  private JMenuItem jmiCount = new JMenuItem("Count Words");
  private JMenuItem jmiExit = new JMenuItem("Exit");
  private JMenuItem jmiForeground = new JMenuItem("Foreground");
  private JMenuItem jmiBackground = new JMenuItem("Background");
 
  // Create buttons to be placed in a tool bar
  private JButton jbtOpen = new JButton();
  private JButton jbtSave = new JButton();
  private JLabel jlblStatus = new JLabel();
 
  // Create a JFileChooser with the current directory
  private JFileChooser jFileChooser1
    = new JFileChooser(new File("."));
 
  // Create a text area
  private JTextArea jta = new JTextArea();
 
  public TextEditor() {
    // Add menu items to the menu
    JMenu jMenu1 = new JMenu("File");
    jMenu1.add(jmiOpen);
    jMenu1.add(jmiSave);
    jMenu1.add(jmiClear);
    jMenu1.add(jmiCount);
    jMenu1.addSeparator();
    jMenu1.add(jmiExit);
 
    // Add menu items to the menu
    JMenu jMenu2 = new JMenu("Edit");
    jMenu2.add(jmiForeground);
    jMenu2.add(jmiBackground);
 
    // Add menus to the menu bar
    JMenuBar jMenuBar1 = new JMenuBar();
    jMenuBar1.add(jMenu1);
    jMenuBar1.add(jMenu2);
 
    // Set the menu bar
    setJMenuBar(jMenuBar1);
 
    // Create tool bar
    JToolBar jToolBar1 = new JToolBar();
    jToolBar1.add(jbtOpen);
    jToolBar1.add(jbtSave);
 
    jmiOpen.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent e) {
        open();
      }
    });
 
    jmiSave.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        save();
      }
    });
    jmiCount.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        countWords();
      }
    });
    jmiClear.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        jta.setText(null);
      }
    });
 
    jmiExit.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        System.exit(0);
      }
    });
 
    jmiForeground.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        Color selectedColor =
          JColorChooser.showDialog(null, "Choose Foreground Color",
            jta.getForeground());
 
        if (selectedColor != null)
          jta.setForeground(selectedColor);
      }
    });
 
    jmiBackground.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        Color selectedColor =
          JColorChooser.showDialog(null, "Choose Background Color",
            jta.getForeground());
 
        if (selectedColor != null)
          jta.setBackground(selectedColor);
      }
    });
 
    jbtOpen.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        open();
      }
    });
 
    jbtSave.addActionListener(new ActionListener() {
      public void actionPerformed(ActionEvent evt) {
        save();
      }
    });
 
    getContentPane().add(jToolBar1, BorderLayout.NORTH);
    getContentPane().add(jlblStatus, BorderLayout.SOUTH);
    getContentPane().add(new JScrollPane(jta), BorderLayout.CENTER);
  }
 
  /** Open file */
  private void open() {
    if (jFileChooser1.showOpenDialog(this) ==
      JFileChooser.APPROVE_OPTION)
      open(jFileChooser1.getSelectedFile());
  }
 
  /** Open file with the specified File instance */
  private void open(File file) {
    try {
      // Read from the specified file and store it in jta
      BufferedInputStream in = new BufferedInputStream(
        new FileInputStream(file));
      byte[] b = new byte[in.available()];
      in.read(b, 0, b.length);
      jta.append(new String(b, 0, b.length));
      in.close();
 
      // Display the status of the Open file operation in jlblStatus
      jlblStatus.setText(file.getName() + " Opened");
    }
    catch (IOException ex) {
      jlblStatus.setText("Error opening " + file.getName());
    }
  }
 
  /** Save file */
  private void save() {
    if (jFileChooser1.showSaveDialog(this) ==
      JFileChooser.APPROVE_OPTION) {
        save(jFileChooser1.getSelectedFile());
    }
  }
 
    private void countWords()
    {
        int numwords;
        String text = jta.getText();
 
//        Scanner scan = new Scanner(text);
        //scan.useDelimiter("");
 
    //    while  (scan.hasNext());
    //     System.out.println(scan.next());
 
    // Create a hash map to hold words as key and count as value
    Map<String, Integer> hashMap = new HashMap<String, Integer>();
 
    String[] words = text.split("[ .!?\\n-]");
    numwords = words.length;
    System.out.println("Number of words: " + numwords);
 
    for (int i = 0; i < words.length; i++) {
      if (words[i].length() > 1) {
        if (hashMap.get(words[i]) != null) {
          int value = hashMap.get(words[i]).intValue();
          value++;
          hashMap.put(words[i], value);
        }
        else
          hashMap.put(words[i], 1);
      }
    }
 
    // Create a tree map from the hash map
    Map<String, Integer> treeMap =
      new TreeMap<String, Integer>(hashMap);
 
    // Display mappings
    System.out.println("Display words and their count in " +
      "ascending order of the words");
    System.out.print(treeMap);
 
    }
 
  /** Save file with specified File instance */
  private void save(File file) {
    try {
      // Write the text in jta to the specified file
      BufferedOutputStream out = new BufferedOutputStream(
        new FileOutputStream(file));
      byte[] b = (jta.getText()).getBytes();
      out.write(b, 0, b.length);
      out.close();
 
      // Display the status of the save file operation in jlblStatus
      jlblStatus.setText(file.getName()  + " Saved ");
    }
    catch (IOException ex) {
      jlblStatus.setText("Error saving " + file.getName());
    }
  }
 
  public static void main(String[] args) {
    TextEditor applet = new TextEditor();
    JFrame frame = new JFrame();
    //EXIT_ON_CLOSE == 3
    frame.setDefaultCloseOperation(3);
    frame.setTitle("TextEditor");
    frame.getContentPane().add(applet, BorderLayout.CENTER);
    applet.init();
    applet.start();
    frame.setSize(400,320);
    Dimension d = Toolkit.getDefaultToolkit().getScreenSize();
    frame.setLocation((d.width - frame.getSize().width) / 2,
      (d.height - frame.getSize().height) / 2);
    frame.setVisible(true);
  }
}