jellyfin/Jellyfin.Drawing.Skia/StripCollageBuilder.cs

207 lines
7.8 KiB
C#
Raw Normal View History

2017-05-09 19:53:46 +00:00
using System;
using System.Collections.Generic;
2017-05-09 19:53:46 +00:00
using System.IO;
using SkiaSharp;
2017-05-09 19:53:46 +00:00
2019-01-26 19:43:13 +00:00
namespace Jellyfin.Drawing.Skia
2017-05-09 19:53:46 +00:00
{
/// <summary>
/// Used to build collages of multiple images arranged in vertical strips.
/// </summary>
2017-05-09 19:53:46 +00:00
public class StripCollageBuilder
{
private readonly SkiaEncoder _skiaEncoder;
2017-05-09 19:53:46 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="StripCollageBuilder"/> class.
/// </summary>
/// <param name="skiaEncoder">The encoder to use for building collages.</param>
public StripCollageBuilder(SkiaEncoder skiaEncoder)
2017-05-09 19:53:46 +00:00
{
_skiaEncoder = skiaEncoder;
2017-05-09 19:53:46 +00:00
}
/// <summary>
/// Check which format an image has been encoded with using its filename extension.
/// </summary>
/// <param name="outputPath">The path to the image to get the format for.</param>
/// <returns>The image format.</returns>
public static SKEncodedImageFormat GetEncodedFormat(string outputPath)
2017-05-09 19:53:46 +00:00
{
if (outputPath == null)
{
throw new ArgumentNullException(nameof(outputPath));
}
var ext = Path.GetExtension(outputPath);
2017-05-09 19:53:46 +00:00
if (string.Equals(ext, ".jpg", StringComparison.OrdinalIgnoreCase)
|| string.Equals(ext, ".jpeg", StringComparison.OrdinalIgnoreCase))
{
2017-05-09 19:53:46 +00:00
return SKEncodedImageFormat.Jpeg;
}
2017-05-09 19:53:46 +00:00
if (string.Equals(ext, ".webp", StringComparison.OrdinalIgnoreCase))
{
2017-05-09 19:53:46 +00:00
return SKEncodedImageFormat.Webp;
}
2017-05-09 19:53:46 +00:00
if (string.Equals(ext, ".gif", StringComparison.OrdinalIgnoreCase))
{
2017-05-09 19:53:46 +00:00
return SKEncodedImageFormat.Gif;
}
2017-05-09 19:53:46 +00:00
if (string.Equals(ext, ".bmp", StringComparison.OrdinalIgnoreCase))
{
2017-05-09 19:53:46 +00:00
return SKEncodedImageFormat.Bmp;
}
2017-05-09 19:53:46 +00:00
// default to png
return SKEncodedImageFormat.Png;
}
/// <summary>
/// Create a square collage.
/// </summary>
/// <param name="paths">The paths of the images to use in the collage.</param>
/// <param name="outputPath">The path at which to place the resulting collage image.</param>
/// <param name="width">The desired width of the collage.</param>
/// <param name="height">The desired height of the collage.</param>
2017-05-09 19:53:46 +00:00
public void BuildSquareCollage(string[] paths, string outputPath, int width, int height)
{
2020-07-19 18:12:53 +00:00
using var bitmap = BuildSquareCollageBitmap(paths, width, height);
using var outputStream = new SKFileWStream(outputPath);
using var pixmap = new SKPixmap(new SKImageInfo(width, height), bitmap.GetPixels());
pixmap.Encode(outputStream, GetEncodedFormat(outputPath), 90);
2017-05-09 19:53:46 +00:00
}
/// <summary>
/// Create a thumb collage.
/// </summary>
/// <param name="paths">The paths of the images to use in the collage.</param>
/// <param name="outputPath">The path at which to place the resulting image.</param>
/// <param name="width">The desired width of the collage.</param>
/// <param name="height">The desired height of the collage.</param>
/// <param name="libraryName">The name of the library to draw on the collage.</param>
public void BuildThumbCollage(string[] paths, string outputPath, int width, int height, string? libraryName)
2017-05-09 19:53:46 +00:00
{
using var bitmap = BuildThumbCollageBitmap(paths, width, height, libraryName);
2020-07-19 18:12:53 +00:00
using var outputStream = new SKFileWStream(outputPath);
using var pixmap = new SKPixmap(new SKImageInfo(width, height), bitmap.GetPixels());
pixmap.Encode(outputStream, GetEncodedFormat(outputPath), 90);
2017-05-09 19:53:46 +00:00
}
private SKBitmap BuildThumbCollageBitmap(string[] paths, int width, int height, string? libraryName)
2017-05-09 19:53:46 +00:00
{
var bitmap = new SKBitmap(width, height);
2017-05-09 19:53:46 +00:00
2020-07-19 18:12:53 +00:00
using var canvas = new SKCanvas(bitmap);
canvas.Clear(SKColors.Black);
2017-05-09 19:53:46 +00:00
using var backdrop = GetNextValidImage(paths, 0, out _);
if (backdrop == null)
2020-07-19 18:12:53 +00:00
{
return bitmap;
2018-09-12 17:26:21 +00:00
}
// resize to the same aspect as the original
var backdropHeight = Math.Abs(width * backdrop.Height / backdrop.Width);
using var residedBackdrop = SkiaEncoder.ResizeImage(backdrop, new SKImageInfo(width, backdropHeight, backdrop.ColorType, backdrop.AlphaType, backdrop.ColorSpace));
// draw the backdrop
canvas.DrawImage(residedBackdrop, 0, 0);
// draw shadow rectangle
var paintColor = new SKPaint
{
Color = SKColors.Black.WithAlpha(0x78),
Style = SKPaintStyle.Fill
};
canvas.DrawRect(0, 0, width, height, paintColor);
// draw library name
var textPaint = new SKPaint
{
Color = SKColors.White,
Style = SKPaintStyle.Fill,
2020-08-21 18:36:56 +00:00
TextSize = 112,
TextAlign = SKTextAlign.Center,
Typeface = SKTypeface.FromFamilyName("sans-serif", SKFontStyleWeight.Bold, SKFontStyleWidth.Normal, SKFontStyleSlant.Upright),
IsAntialias = true
};
2020-08-22 12:31:28 +00:00
// scale down text to 90% of the width if text is larger than 95% of the width
var textWidth = textPaint.MeasureText(libraryName);
if (textWidth > width * 0.95)
{
textPaint.TextSize = 0.9f * width * textPaint.TextSize / textWidth;
}
2020-08-21 18:36:56 +00:00
canvas.DrawText(libraryName, width / 2f, (height / 2f) + (textPaint.FontMetrics.XHeight / 2), textPaint);
2018-09-12 17:26:21 +00:00
return bitmap;
}
private SKBitmap? GetNextValidImage(string[] paths, int currentIndex, out int newIndex)
2018-09-12 17:26:21 +00:00
{
2019-01-13 20:37:13 +00:00
var imagesTested = new Dictionary<int, int>();
SKBitmap? bitmap = null;
2018-09-12 17:26:21 +00:00
while (imagesTested.Count < paths.Length)
{
if (currentIndex >= paths.Length)
{
currentIndex = 0;
}
bitmap = _skiaEncoder.Decode(paths[currentIndex], false, null, out _);
2018-09-12 17:26:21 +00:00
imagesTested[currentIndex] = 0;
2018-09-12 17:26:21 +00:00
currentIndex++;
2018-09-12 17:26:21 +00:00
if (bitmap != null)
{
break;
2017-05-09 19:53:46 +00:00
}
}
2017-05-09 19:53:46 +00:00
2018-09-12 17:26:21 +00:00
newIndex = currentIndex;
return bitmap;
2017-05-09 19:53:46 +00:00
}
private SKBitmap BuildSquareCollageBitmap(string[] paths, int width, int height)
{
var bitmap = new SKBitmap(width, height);
var imageIndex = 0;
var cellWidth = width / 2;
var cellHeight = height / 2;
2020-07-19 18:12:53 +00:00
using var canvas = new SKCanvas(bitmap);
for (var x = 0; x < 2; x++)
2017-05-09 19:53:46 +00:00
{
2020-07-19 18:12:53 +00:00
for (var y = 0; y < 2; y++)
2017-05-09 19:53:46 +00:00
{
2020-07-19 18:12:53 +00:00
using var currentBitmap = GetNextValidImage(paths, imageIndex, out int newIndex);
imageIndex = newIndex;
if (currentBitmap == null)
2017-05-09 19:53:46 +00:00
{
2020-07-19 18:12:53 +00:00
continue;
2017-05-09 19:53:46 +00:00
}
2020-07-19 18:12:53 +00:00
// Scale image. The FromBitmap creates a copy
var imageInfo = new SKImageInfo(cellWidth, cellHeight, currentBitmap.ColorType, currentBitmap.AlphaType, currentBitmap.ColorSpace);
2020-08-09 20:59:31 +00:00
using var resizedBitmap = SKBitmap.FromImage(SkiaEncoder.ResizeImage(currentBitmap, imageInfo));
2020-07-19 18:12:53 +00:00
// draw this image into the strip at the next position
var xPos = x * cellWidth;
var yPos = y * cellHeight;
canvas.DrawBitmap(resizedBitmap, xPos, yPos);
2017-05-09 19:53:46 +00:00
}
}
return bitmap;
}
}
2018-12-13 21:34:28 +00:00
}