PixiEditor/PixiEditor/Models/DataHolders/BitmapPixelChanges.cs

using PixiEditor.Exceptions;
using PixiEditor.Helpers.Extensions;
using PixiEditor.Models.Position;
using SkiaSharp;
using System;
using System.Collections.Generic;
using System.Linq;

namespace PixiEditor.Models.DataHolders
{
    public struct BitmapPixelChanges
    {
        public BitmapPixelChanges(Dictionary<Coordinates, SKColor> changedPixels)
        {
            ChangedPixels = changedPixels;
            WasBuiltAsSingleColored = false;
        }

        public static BitmapPixelChanges Empty => new BitmapPixelChanges(new Dictionary<Coordinates, SKColor>());

        public bool WasBuiltAsSingleColored { get; private set; }

        public Dictionary<Coordinates, SKColor> ChangedPixels { get; set; }

        /// <summary>
        ///     Builds BitmapPixelChanges with only one color for specified coordinates.
        /// </summary>
        /// <returns>Single-colored BitmapPixelChanges.</returns>
        public static BitmapPixelChanges FromSingleColoredArray(IEnumerable<Coordinates> coordinates, SKColor color)
        {
            Dictionary<Coordinates, SKColor> dict = new Dictionary<Coordinates, SKColor>();
            foreach (Coordinates coordinate in coordinates)
            {
                if (dict.ContainsKey(coordinate))
                {
                    continue;
                }

                dict.Add(coordinate, color);
            }

            return new BitmapPixelChanges(dict) { WasBuiltAsSingleColored = true };
        }

        /// <summary>
        ///     Combines pixel changes array with overriding values.
        /// </summary>
        /// <param name="changes">BitmapPixelChanges to combine.</param>
        /// <returns>Combined BitmapPixelChanges.</returns>
        public static BitmapPixelChanges CombineOverride(BitmapPixelChanges[] changes)
        {
            if (changes == null || changes.Length == 0)
            {
                throw new ArgumentException();
            }

            BitmapPixelChanges output = Empty;

            for (int i = 0; i < changes.Length; i++)
            {
                output.ChangedPixels.AddRangeOverride(changes[i].ChangedPixels);
            }

            return output;
        }

        public static BitmapPixelChanges CombineOverride(BitmapPixelChanges changes1, BitmapPixelChanges changes2)
        {
            return CombineOverride(new[] { changes1, changes2 });
        }

        /// <summary>
        ///     Builds BitmapPixelChanges using 2 same-length enumerables of coordinates and colors.
        /// </summary>
        public static BitmapPixelChanges FromArrays(IEnumerable<Coordinates> coordinates, IEnumerable<SKColor> color)
        {
            Coordinates[] coordinateArray = coordinates.ToArray();
            SKColor[] colorArray = color.ToArray();
            if (coordinateArray.Length != colorArray.Length)
            {
                throw new ArrayLengthMismatchException();
            }

            Dictionary<Coordinates, SKColor> dict = new Dictionary<Coordinates, SKColor>();
            for (int i = 0; i < coordinateArray.Length; i++)
            {
                dict.Add(coordinateArray[i], colorArray[i]);
            }

            return new BitmapPixelChanges(dict);
        }

        public BitmapPixelChanges WithoutTransparentPixels()
        {
            return new BitmapPixelChanges(ChangedPixels.Where(x => x.Value.Alpha > 0).ToDictionary(y => y.Key, y => y.Value));
        }
    }
}