@Nullable
    @Override
    ElementContext<T> getContext(@NotNull PsiElement origin) {
      T elementToSort = getElementToSort(origin);
      if (elementToSort == null) return null;
      return new ElementContext<>(elementToSort);
    }
@Nullable
    @Override
    ElementContext<T> getContext(@NotNull PsiElement origin)