Counter.java

  1. /*
  2.  * Licensed to the Apache Software Foundation (ASF) under one or more
  3.  * contributor license agreements.  See the NOTICE file distributed with
  4.  * this work for additional information regarding copyright ownership.
  5.  * The ASF licenses this file to You under the Apache License, Version 2.0
  6.  * (the "License"); you may not use this file except in compliance with
  7.  * the License.  You may obtain a copy of the License at
  8.  *
  9.  *      http://www.apache.org/licenses/LICENSE-2.0
  10.  *
  11.  * Unless required by applicable law or agreed to in writing, software
  12.  * distributed under the License is distributed on an "AS IS" BASIS,
  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14.  * See the License for the specific language governing permissions and
  15.  * limitations under the License.
  16.  */
  17. package org.apache.commons.text.similarity;

  18. import java.util.HashMap;
  19. import java.util.Map;

  20. /**
  21.  * Java implementation of Python's collections Counter module.
  22.  *
  23.  * <p>It counts how many times each element provided occurred in an array and
  24.  * returns a dict with the element as key and the count as value.</p>
  25.  *
  26.  * @see <a href="https://docs.python.org/dev/library/collections.html#collections.Counter">
  27.  * https://docs.python.org/dev/library/collections.html#collections.Counter</a>
  28.  *
  29.  * @since 1.0
  30.  */
  31. final class Counter {

  32.     /**
  33.      * Hidden constructor.
  34.      */
  35.     private Counter() {
  36.         super();
  37.     }

  38.     /**
  39.      * It counts how many times each element provided occurred in an array and
  40.      * returns a dict with the element as key and the count as value.
  41.      *
  42.      * @param tokens array of tokens
  43.      * @return dict, where the elements are key, and the count the value
  44.      */
  45.     public static Map<CharSequence, Integer> of(final CharSequence[] tokens) {
  46.         final Map<CharSequence, Integer> innerCounter = new HashMap<>();
  47.         for (final CharSequence token : tokens) {
  48.             if (innerCounter.containsKey(token)) {
  49.                 int value = innerCounter.get(token);
  50.                 innerCounter.put(token, ++value);
  51.             } else {
  52.                 innerCounter.put(token, 1);
  53.             }
  54.         }
  55.         return innerCounter;
  56.     }

  57. }