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. import java.util.stream.Stream;

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

  33.     /**
  34.      * Counts how many times each element provided occurred in an array and
  35.      * returns a map with the element as key and the count as value.
  36.      *
  37.      * @param tokens array of tokens.
  38.      * @return a map, where the elements are key, and the count the value.
  39.      */
  40.     public static Map<CharSequence, Integer> of(final CharSequence[] tokens) {
  41.         final Map<CharSequence, Integer> map = new HashMap<>();
  42.         Stream.of(tokens).forEach(token -> map.compute(token, (k, v) -> v != null ? v + 1 : 1));
  43.         return map;
  44.     }

  45.     /**
  46.      * Hidden constructor.
  47.      */
  48.     private Counter() {
  49.     }

  50. }