osu-lazer/osu.Game/Screens/Edit/Verify/IssueList.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System.Collections.Generic;
using System.Linq;
using osu.Framework.Allocation;
using osu.Framework.Bindables;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Graphics.Shapes;
using osu.Game.Beatmaps;
using osu.Game.Graphics.Containers;
using osu.Game.Graphics.UserInterface;
using osu.Game.Overlays;
using osu.Game.Rulesets.Edit;
using osu.Game.Rulesets.Edit.Checks.Components;
using osuTK;

namespace osu.Game.Screens.Edit.Verify
{
    public class IssueList : CompositeDrawable
    {
        private IssueTable table;

        [Resolved]
        private EditorClock clock { get; set; }

        [Resolved]
        private IBindable<WorkingBeatmap> workingBeatmap { get; set; }

        [Resolved]
        private EditorBeatmap beatmap { get; set; }

        [Resolved]
        private Bindable<Issue> selectedIssue { get; set; }

        public Dictionary<IssueType, Bindable<bool>> ShowType { get; set; }

        public Bindable<DifficultyRating> InterpretedDifficulty { get; set; }

        private IBeatmapVerifier rulesetVerifier;
        private BeatmapVerifier generalVerifier;
        private BeatmapVerifierContext context;

        [BackgroundDependencyLoader]
        private void load(OverlayColourProvider colours)
        {
            // Reflects the user interface. Only types in this dictionary have configurable visibility.
            ShowType = new Dictionary<IssueType, Bindable<bool>>
            {
                { IssueType.Warning, new Bindable<bool>(true) },
                { IssueType.Error, new Bindable<bool>(true) },
                { IssueType.Negligible, new Bindable<bool>(false) }
            };

            generalVerifier = new BeatmapVerifier();
            rulesetVerifier = beatmap.BeatmapInfo.Ruleset?.CreateInstance()?.CreateBeatmapVerifier();

            InterpretedDifficulty = new Bindable<DifficultyRating>(beatmap.BeatmapInfo.DifficultyRating);

            context = new BeatmapVerifierContext(workingBeatmap.Value);

            RelativeSizeAxes = Axes.Both;

            InternalChildren = new Drawable[]
            {
                new Box
                {
                    Colour = colours.Background2,
                    RelativeSizeAxes = Axes.Both,
                },
                new OsuScrollContainer
                {
                    RelativeSizeAxes = Axes.Both,
                    Child = table = new IssueTable(),
                },
                new FillFlowContainer
                {
                    AutoSizeAxes = Axes.Both,
                    Anchor = Anchor.BottomRight,
                    Origin = Anchor.BottomRight,
                    Margin = new MarginPadding(20),
                    Children = new Drawable[]
                    {
                        new TriangleButton
                        {
                            Text = "Refresh",
                            Action = Refresh,
                            Size = new Vector2(120, 40),
                            Anchor = Anchor.BottomRight,
                            Origin = Anchor.BottomRight,
                        },
                    }
                },
            };
        }

        protected override void LoadComplete()
        {
            base.LoadComplete();

            Refresh();
        }

        public void Refresh()
        {
            var issues = generalVerifier.Run(beatmap, context);

            if (rulesetVerifier != null)
                issues = issues.Concat(rulesetVerifier.Run(beatmap, context));

            issues = filter(issues);

            table.Issues = issues
                           .OrderBy(issue => issue.Template.Type)
                           .ThenBy(issue => issue.Check.Metadata.Category);
        }

        private IEnumerable<Issue> filter(IEnumerable<Issue> issues)
        {
            foreach (IssueType issueType in ShowType.Keys)
            {
                if (!ShowType[issueType].Value)
                    issues = issues.Where(issue => issue.Template.Type != issueType);
            }

            return issues;
        }
    }
}