com.deque.axe.android.wrappers.CountMap Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of axe-devtools-android-core Show documentation

The Axe Devtools Android Core Library

There is a newer version: 3.2.0

Show newest version

package com.deque.axe.android.wrappers;

import androidx.annotation.NonNull;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class CountMap extends HashMap {

  public void increment(@NonNull final T key) {
    increment(key, 1);
  }

  /**
   * Increment key by more than 1 at a time.
   * @param key The key to increment.
   * @param byValue By how much.
   */
  public void increment(@NonNull final T key, int byValue) {

    if (!containsKey(key)) {
      put(key, byValue);
    } else {
      put(key, get(key) + byValue);
    }
  }

  /**
   * A list of entries, sorted by the value.
   * @return Don't make me say it again.
   */
  public List> entriesSortedByValue() {

    List> colorEntries = new ArrayList<>(entrySet());

    Collections.sort(colorEntries, (o1, o2) -> o2.getValue().compareTo(o1.getValue()));

    return colorEntries;
  }

  /**
   * Yeah, you heard me, they're sorted by value. Get it?
   * @param ifEqualByThis Oh, yeah, if they're equal, sort by this.
   * @return More sorted entries.
   */
  public List> entriesSortedByValue(final Comparator ifEqualByThis) {

    List> colorEntries = new ArrayList<>(entrySet());

    Collections.sort(colorEntries, (o1, o2) -> {
      int result = o2.getValue().compareTo(o1.getValue());

      if (result == 0) {
        return ifEqualByThis.compare(o1.getKey(), o2.getKey());
      } else {
        return result;
      }
    });

    return colorEntries;
  }
}