public @NotNull LeafState getLeafState(@NotNull Object element) {
    return isAlwaysLeaf(element) ? LeafState.ALWAYS : LeafState.get(element);
  }
public @NotNull LeafState getLeafState(@NotNull Object element)