src/modules/previewpane/BgcodeThumbnailProvider/BgcodeThumbnailProvider.cs - main - githubmirror/PowerToys

src/modules/previewpane/BgcodeThumbnailProvider/BgcodeThumbnailProvider.cs
146 строк5 KB
Pedro Lamas
Adds BgcodeThumbnailProvider and BgcodePreviewHandler (#38667)
10 июл 2025, 12:20
10 июл 2025, 12:20071f5d7
// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System.Drawing.Drawing2D;
using System.Drawing.Imaging;

using Microsoft.PowerToys.FilePreviewCommon;

namespace Microsoft.PowerToys.ThumbnailHandler.Bgcode
{
    /// <summary>
    /// Binary G-code Thumbnail Provider.
    /// </summary>
    public class BgcodeThumbnailProvider
    {
        public BgcodeThumbnailProvider(string filePath)
        {
            FilePath = filePath;
            Stream = new FileStream(filePath, FileMode.Open, FileAccess.Read);
        }

        /// <summary>
        /// Gets the file path to the file creating thumbnail for.
        /// </summary>
        public string FilePath { get; private set; }

        /// <summary>
        /// Gets the stream object to access file.
        /// </summary>
        public Stream Stream { get; private set; }

        /// <summary>
        ///  The maximum dimension (width or height) thumbnail we will generate.
        /// </summary>
        private const uint MaxThumbnailSize = 10000;

        /// <summary>
        /// Reads the Binary G-code content searching for thumbnails and returns the largest.
        /// </summary>
        /// <param name="reader">The BinaryReader instance for the Binary G-code content.</param>
        /// <param name="cx">The maximum thumbnail size, in pixels.</param>
        /// <returns>A thumbnail extracted from the Binary G-code content.</returns>
        public static Bitmap GetThumbnail(BinaryReader reader, uint cx)
        {
            if (cx > MaxThumbnailSize || reader == null || reader.BaseStream.Length == 0)
            {
                return null;
            }

            Bitmap thumbnail = null;

            try
            {
                var bgcodeThumbnail = BgcodeHelper.GetBestThumbnail(reader);

                thumbnail = bgcodeThumbnail?.GetBitmap();
            }
            catch (Exception)
            {
                // TODO: add logger
            }

            if (thumbnail != null && (
                ((thumbnail.Width != cx || thumbnail.Height > cx) && (thumbnail.Height != cx || thumbnail.Width > cx)) ||
                thumbnail.PixelFormat != PixelFormat.Format32bppArgb))
            {
                // We are not the appropriate size for caller.  Resize now while
                // respecting the aspect ratio.
                float scale = Math.Min((float)cx / thumbnail.Width, (float)cx / thumbnail.Height);
                int scaleWidth = (int)(thumbnail.Width * scale);
                int scaleHeight = (int)(thumbnail.Height * scale);
                thumbnail = ResizeImage(thumbnail, scaleWidth, scaleHeight);
            }

            return thumbnail;
        }

        /// <summary>
        /// Resize the image with high quality to the specified width and height.
        /// </summary>
        /// <param name="image">The image to resize.</param>
        /// <param name="width">The width to resize to.</param>
        /// <param name="height">The height to resize to.</param>
        /// <returns>The resized image.</returns>
        public static Bitmap ResizeImage(Image image, int width, int height)
        {
            if (width <= 0 ||
                height <= 0 ||
                width > MaxThumbnailSize ||
                height > MaxThumbnailSize ||
                image == null)
            {
                return null;
            }

            Bitmap destImage = new Bitmap(width, height, PixelFormat.Format32bppArgb);

            destImage.SetResolution(image.HorizontalResolution, image.VerticalResolution);

            using (var graphics = Graphics.FromImage(destImage))
            {
                graphics.CompositingMode = CompositingMode.SourceCopy;
                graphics.CompositingQuality = CompositingQuality.HighQuality;
                graphics.InterpolationMode = InterpolationMode.HighQualityBicubic;
                graphics.SmoothingMode = SmoothingMode.HighQuality;
                graphics.PixelOffsetMode = PixelOffsetMode.HighQuality;

                graphics.DrawImage(image, 0, 0, width, height);
            }

            image.Dispose();

            return destImage;
        }

        /// <summary>
        /// Generate thumbnail bitmap for provided Bgcode file/stream.
        /// </summary>
        /// <param name="cx">Maximum thumbnail size, in pixels.</param>
        /// <returns>Generated bitmap</returns>
        public Bitmap GetThumbnail(uint cx)
        {
            if (cx == 0 || cx > MaxThumbnailSize)
            {
                return null;
            }

            if (global::PowerToys.GPOWrapper.GPOWrapper.GetConfiguredBgcodeThumbnailsEnabledValue() == global::PowerToys.GPOWrapper.GpoRuleConfigured.Disabled)
            {
                // GPO is disabling this utility.
                return null;
            }

            using (var reader = new BinaryReader(this.Stream))
            {
                Bitmap thumbnail = GetThumbnail(reader, cx);
                if (thumbnail != null && thumbnail.Size.Width > 0 && thumbnail.Size.Height > 0)
                {
                    return thumbnail;
                }
            }

            return null;
        }
    }
}