summaryrefslogtreecommitdiffstats
path: root/src/de/fhswf/in/inf/java1/aufgabe9/WordCount.java
blob: fcd537371dd51ab57f17aa85d960ae86af11c206 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/**
 * 
 */
package de.fhswf.in.inf.java1.aufgabe9;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Map.Entry;
import java.util.TreeMap;

/**
 * A class, that counts words in files.
 *
 * @author $Author: $
 * @version $Revision: $, $Date: $ UTC
 */
public class WordCount
{

   private TreeMap<String, Integer> wordMap = new TreeMap<>();

   /**
    * Reads the file and counts the words.
    *
    * @param file
    *           The file to be counted.
    */
   public WordCount(File file)
   {
      try (BufferedReader f = new BufferedReader(new FileReader(file)))
      {
         String line = null;

         while ((line = f.readLine()) != null)
         {
            String[] lineArray = line
                  .split("[^\\p{IsAlphabetic}\\p{Digit}]+");
            for (String word : lineArray)
            {
               // Split creates empty String, if first char is a split char.
               if (!word.isEmpty())
               {
                  word = word.toLowerCase();

                  if (wordMap.containsKey(word))
                  {
                     wordMap.put(word, wordMap.get(word) + 1);
                  }
                  else
                  {
                     wordMap.put(word, new Integer(1));
                  }
               }
            }
         }
      }
      catch (IOException e)
      {
         e.printStackTrace(System.err);
      }
   }

   /*
    * (non-Javadoc)
    * 
    * @see java.lang.Object#toString()
    */
   @Override
   public final String toString()
   {
      StringBuilder ret = new StringBuilder();

      for (Entry<String, Integer> mapping : wordMap.entrySet())
      {
         ret.append(mapping.getKey() + " = " + mapping.getValue() + "\n");
      }

      return ret.toString();
   }
}