DisplayMagician/HeliosDisplayManagement/DisplayRepresentation.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using WindowsDisplayAPI;
using WindowsDisplayAPI.DisplayConfig;
using HeliosPlus.Shared;
using HeliosPlus.Shared.Topology;

namespace HeliosPlus
{
    internal class DisplayRepresentation
    {
        public DisplayRepresentation(Display display)
        {
            Name = display.DeviceName;
            Path = display.DevicePath;
            var index = Path.IndexOf("{", StringComparison.InvariantCultureIgnoreCase);

            if (index > 0)
            {
                Path = Path.Substring(0, index).TrimEnd('#');
            }

            IsAvailable = display.IsAvailable;

            if (IsAvailable)
            {
                PossibleSettings = GetDisplay()?.GetPossibleSettings()?.ToArray() ?? new DisplayPossibleSetting[0];
            }
        }

        public DisplayRepresentation(ProfileViewportTargetDisplay display)
        {
            Name = display.DisplayName;
            Path = display.DevicePath;
            IsAvailable = GetDisplay()?.IsAvailable ?? false;

            if (IsAvailable)
            {
                PossibleSettings = GetDisplay()?.GetPossibleSettings()?.ToArray() ?? new DisplayPossibleSetting[0];
            }
        }

        public bool IsAvailable { get; }
        public string Name { get; }
        public string Path { get; }

        public DisplayPossibleSetting[] PossibleSettings { get; }

        public static IEnumerable<DisplayRepresentation> GetDisplays(Profile profile = null)
        {
            //var displays =
            //    Display.GetDisplays()
            //        .Select(display => new DisplayRepresentation(display))
            //        .OrderByDescending(representation => representation.IsAvailable)
            //        .GroupBy(representation => representation.Path)
            //        .Select(grouping => grouping.First()).ToList();
            var displays = new List<DisplayRepresentation>();

            if (profile != null)
            {
                foreach (var target in profile.Viewports.SelectMany(path => path.TargetDisplays))
                {
                    if (displays.All(display => display.Path != target.DevicePath))
                    {
                        displays.Add(new DisplayRepresentation(target));
                    }
                }
            }

            return displays;
        }

        public Display GetDisplay()
        {
            return Display.GetDisplays().FirstOrDefault(display => display.DevicePath.StartsWith(Path));
        }

        public ProfileViewport GetPathSource(Profile profile)
        {
            return profile.Viewports.FirstOrDefault(path => path.TargetDisplays.Any(target => target.DevicePath == Path));
        }

        public ProfileViewportTargetDisplay GetPathTarget(Profile profile)
        {
            return profile.Viewports.SelectMany(path => path.TargetDisplays).FirstOrDefault(target => target.DevicePath == Path);
        }

        public PathDisplayTarget GetTargetInfo()
        {
            return
                PathDisplayTarget.GetDisplayTargets()
                    .Where(target => target.DevicePath.StartsWith(Path))
                    .OrderByDescending(target => target.IsAvailable)
                    .FirstOrDefault();
        }

        public Bitmap ToBitmap(Size size, Profile profile = null)
        {
            var targetInfo = GetTargetInfo();
            var resolution = Size.Empty;

            if (targetInfo != null && targetInfo.IsAvailable)
            {
                resolution = targetInfo.PreferredResolution;
            }
            else if (profile != null)
            {
                var targetPath = GetPathSource(profile);

                if (targetPath != null)
                {
                    resolution = targetPath.Resolution;
                }
            }

            var p = new Profile {Viewports = new ProfileViewport[1]};
            p.Viewports[0] = new ProfileViewport
            {
                Resolution = resolution,
                Position = new Point(),
                TargetDisplays = new ProfileViewportTargetDisplay[1]
            };
            p.Viewports[0].TargetDisplays[0] = new ProfileViewportTargetDisplay {DevicePath = Path};

            if (profile != null)
            {
                var targetPath = GetPathTarget(profile);

                if (targetPath != null)
                {
                    p.Viewports[0].TargetDisplays[0].SurroundTopology = targetPath.SurroundTopology;
                }
            }

            return new ProfileIcon(p).ToBitmap(size.Width, size.Height);
        }
    }
}