@Override
    public @Nullable UsageType getUsageType(@NotNull PsiElement element) {
      VirtualFile file = PsiUtilCore.getVirtualFile(element);
      RootType rootType = ScratchFileService.getInstance().getRootType(file);
      return rootType == null ? null : ourUsageTypes.get(rootType);
    }
@Override
    public @Nullable UsageType getUsageType(@NotNull PsiElement element)