mirror of
https://github.com/jellyfin/jellyfin.git
synced 2026-06-02 22:08:27 +01:00
isolated weather and moved drawing classes up to the controller project
This commit is contained in:
LukePulverenti
199
MediaBrowser.Controller/Drawing/ImageExtensions.cs
Normal file
199
MediaBrowser.Controller/Drawing/ImageExtensions.cs
Normal file
@@ -0,0 +1,199 @@
|
||||
using System;
|
||||
using System.Drawing;
|
||||
using System.Drawing.Drawing2D;
|
||||
using System.Drawing.Imaging;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
|
||||
namespace MediaBrowser.Controller.Drawing
|
||||
{
|
||||
/// <summary>
|
||||
/// Class ImageExtensions
|
||||
/// </summary>
|
||||
public static class ImageExtensions
|
||||
{
|
||||
/// <summary>
|
||||
/// Saves the image.
|
||||
/// </summary>
|
||||
/// <param name="outputFormat">The output format.</param>
|
||||
/// <param name="image">The image.</param>
|
||||
/// <param name="toStream">To stream.</param>
|
||||
/// <param name="quality">The quality.</param>
|
||||
public static void Save(this Image image, ImageFormat outputFormat, Stream toStream, int quality)
|
||||
{
|
||||
// Use special save methods for jpeg and png that will result in a much higher quality image
|
||||
// All other formats use the generic Image.Save
|
||||
if (ImageFormat.Jpeg.Equals(outputFormat))
|
||||
{
|
||||
SaveAsJpeg(image, toStream, quality);
|
||||
}
|
||||
else if (ImageFormat.Png.Equals(outputFormat))
|
||||
{
|
||||
image.Save(toStream, ImageFormat.Png);
|
||||
}
|
||||
else
|
||||
{
|
||||
image.Save(toStream, outputFormat);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Saves the JPEG.
|
||||
/// </summary>
|
||||
/// <param name="image">The image.</param>
|
||||
/// <param name="target">The target.</param>
|
||||
/// <param name="quality">The quality.</param>
|
||||
public static void SaveAsJpeg(this Image image, Stream target, int quality)
|
||||
{
|
||||
using (var encoderParameters = new EncoderParameters(1))
|
||||
{
|
||||
encoderParameters.Param[0] = new EncoderParameter(Encoder.Quality, quality);
|
||||
image.Save(target, GetImageCodecInfo("image/jpeg"), encoderParameters);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Gets the image codec info.
|
||||
/// </summary>
|
||||
/// <param name="mimeType">Type of the MIME.</param>
|
||||
/// <returns>ImageCodecInfo.</returns>
|
||||
private static ImageCodecInfo GetImageCodecInfo(string mimeType)
|
||||
{
|
||||
var encoders = ImageCodecInfo.GetImageEncoders();
|
||||
|
||||
return encoders.FirstOrDefault(i => i.MimeType.Equals(mimeType, StringComparison.OrdinalIgnoreCase)) ?? encoders.FirstOrDefault();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Crops an image by removing whitespace and transparency from the edges
|
||||
/// </summary>
|
||||
/// <param name="bmp">The BMP.</param>
|
||||
/// <returns>Bitmap.</returns>
|
||||
/// <exception cref="System.Exception"></exception>
|
||||
public static Bitmap CropWhitespace(this Bitmap bmp)
|
||||
{
|
||||
var width = bmp.Width;
|
||||
var height = bmp.Height;
|
||||
|
||||
var topmost = 0;
|
||||
for (int row = 0; row < height; ++row)
|
||||
{
|
||||
if (IsAllWhiteRow(bmp, row, width))
|
||||
topmost = row;
|
||||
else break;
|
||||
}
|
||||
|
||||
int bottommost = 0;
|
||||
for (int row = height - 1; row >= 0; --row)
|
||||
{
|
||||
if (IsAllWhiteRow(bmp, row, width))
|
||||
bottommost = row;
|
||||
else break;
|
||||
}
|
||||
|
||||
int leftmost = 0, rightmost = 0;
|
||||
for (int col = 0; col < width; ++col)
|
||||
{
|
||||
if (IsAllWhiteColumn(bmp, col, height))
|
||||
leftmost = col;
|
||||
else
|
||||
break;
|
||||
}
|
||||
|
||||
for (int col = width - 1; col >= 0; --col)
|
||||
{
|
||||
if (IsAllWhiteColumn(bmp, col, height))
|
||||
rightmost = col;
|
||||
else
|
||||
break;
|
||||
}
|
||||
|
||||
if (rightmost == 0) rightmost = width; // As reached left
|
||||
if (bottommost == 0) bottommost = height; // As reached top.
|
||||
|
||||
var croppedWidth = rightmost - leftmost;
|
||||
var croppedHeight = bottommost - topmost;
|
||||
|
||||
if (croppedWidth == 0) // No border on left or right
|
||||
{
|
||||
leftmost = 0;
|
||||
croppedWidth = width;
|
||||
}
|
||||
|
||||
if (croppedHeight == 0) // No border on top or bottom
|
||||
{
|
||||
topmost = 0;
|
||||
croppedHeight = height;
|
||||
}
|
||||
|
||||
// Graphics.FromImage will throw an exception if the PixelFormat is Indexed, so we need to handle that here
|
||||
var thumbnail = bmp.PixelFormat.HasFlag(PixelFormat.Indexed) ? new Bitmap(croppedWidth, croppedHeight) : new Bitmap(croppedWidth, croppedHeight, bmp.PixelFormat);
|
||||
|
||||
// Preserve the original resolution
|
||||
thumbnail.SetResolution(bmp.HorizontalResolution, bmp.VerticalResolution);
|
||||
|
||||
using (var thumbnailGraph = Graphics.FromImage(thumbnail))
|
||||
{
|
||||
thumbnailGraph.CompositingQuality = CompositingQuality.HighQuality;
|
||||
thumbnailGraph.SmoothingMode = SmoothingMode.HighQuality;
|
||||
thumbnailGraph.InterpolationMode = InterpolationMode.HighQualityBicubic;
|
||||
thumbnailGraph.PixelOffsetMode = PixelOffsetMode.HighQuality;
|
||||
thumbnailGraph.CompositingMode = CompositingMode.SourceOver;
|
||||
|
||||
thumbnailGraph.DrawImage(bmp,
|
||||
new RectangleF(0, 0, croppedWidth, croppedHeight),
|
||||
new RectangleF(leftmost, topmost, croppedWidth, croppedHeight),
|
||||
GraphicsUnit.Pixel);
|
||||
}
|
||||
return thumbnail;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Determines whether or not a row of pixels is all whitespace
|
||||
/// </summary>
|
||||
/// <param name="bmp">The BMP.</param>
|
||||
/// <param name="row">The row.</param>
|
||||
/// <param name="width">The width.</param>
|
||||
/// <returns><c>true</c> if [is all white row] [the specified BMP]; otherwise, <c>false</c>.</returns>
|
||||
private static bool IsAllWhiteRow(Bitmap bmp, int row, int width)
|
||||
{
|
||||
for (var i = 0; i < width; ++i)
|
||||
{
|
||||
if (!IsWhiteSpace(bmp.GetPixel(i, row)))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Determines whether or not a column of pixels is all whitespace
|
||||
/// </summary>
|
||||
/// <param name="bmp">The BMP.</param>
|
||||
/// <param name="col">The col.</param>
|
||||
/// <param name="height">The height.</param>
|
||||
/// <returns><c>true</c> if [is all white column] [the specified BMP]; otherwise, <c>false</c>.</returns>
|
||||
private static bool IsAllWhiteColumn(Bitmap bmp, int col, int height)
|
||||
{
|
||||
for (var i = 0; i < height; ++i)
|
||||
{
|
||||
if (!IsWhiteSpace(bmp.GetPixel(col, i)))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Determines if a color is whitespace
|
||||
/// </summary>
|
||||
/// <param name="color">The color.</param>
|
||||
/// <returns><c>true</c> if [is white space] [the specified color]; otherwise, <c>false</c>.</returns>
|
||||
private static bool IsWhiteSpace(Color color)
|
||||
{
|
||||
return (color.R == 255 && color.G == 255 && color.B == 255) || color.A == 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
216
MediaBrowser.Controller/Drawing/ImageHeader.cs
Normal file
216
MediaBrowser.Controller/Drawing/ImageHeader.cs
Normal file
@@ -0,0 +1,216 @@
|
||||
using MediaBrowser.Common.Logging;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Drawing;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
|
||||
namespace MediaBrowser.Controller.Drawing
|
||||
{
|
||||
/// <summary>
|
||||
/// Taken from http://stackoverflow.com/questions/111345/getting-image-dimensions-without-reading-the-entire-file/111349
|
||||
/// http://www.codeproject.com/Articles/35978/Reading-Image-Headers-to-Get-Width-and-Height
|
||||
/// Minor improvements including supporting unsigned 16-bit integers when decoding Jfif and added logic
|
||||
/// to load the image using new Bitmap if reading the headers fails
|
||||
/// </summary>
|
||||
public static class ImageHeader
|
||||
{
|
||||
/// <summary>
|
||||
/// The error message
|
||||
/// </summary>
|
||||
const string errorMessage = "Could not recognize image format.";
|
||||
|
||||
/// <summary>
|
||||
/// The image format decoders
|
||||
/// </summary>
|
||||
private static readonly Dictionary<byte[], Func<BinaryReader, Size>> imageFormatDecoders = new Dictionary<byte[], Func<BinaryReader, Size>>
|
||||
{
|
||||
{ new byte[] { 0x42, 0x4D }, DecodeBitmap },
|
||||
{ new byte[] { 0x47, 0x49, 0x46, 0x38, 0x37, 0x61 }, DecodeGif },
|
||||
{ new byte[] { 0x47, 0x49, 0x46, 0x38, 0x39, 0x61 }, DecodeGif },
|
||||
{ new byte[] { 0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A }, DecodePng },
|
||||
{ new byte[] { 0xff, 0xd8 }, DecodeJfif },
|
||||
};
|
||||
|
||||
/// <summary>
|
||||
/// Gets the dimensions of an image.
|
||||
/// </summary>
|
||||
/// <param name="path">The path of the image to get the dimensions of.</param>
|
||||
/// <returns>The dimensions of the specified image.</returns>
|
||||
/// <exception cref="ArgumentException">The image was of an unrecognised format.</exception>
|
||||
public static Size GetDimensions(string path)
|
||||
{
|
||||
try
|
||||
{
|
||||
using (var fs = File.OpenRead(path))
|
||||
{
|
||||
using (var binaryReader = new BinaryReader(fs))
|
||||
{
|
||||
return GetDimensions(binaryReader);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch
|
||||
{
|
||||
Logger.LogInfo("Failed to read image header for {0}. Doing it the slow way.", path);
|
||||
|
||||
using (var fs = File.OpenRead(path))
|
||||
{
|
||||
// Co it the old fashioned way
|
||||
using (var b = Image.FromStream(fs, true, false))
|
||||
{
|
||||
return b.Size;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Gets the dimensions of an image.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>Size.</returns>
|
||||
/// <exception cref="System.ArgumentException">binaryReader</exception>
|
||||
/// <exception cref="ArgumentException">The image was of an unrecognized format.</exception>
|
||||
private static Size GetDimensions(BinaryReader binaryReader)
|
||||
{
|
||||
int maxMagicBytesLength = imageFormatDecoders.Keys.OrderByDescending(x => x.Length).First().Length;
|
||||
var magicBytes = new byte[maxMagicBytesLength];
|
||||
for (int i = 0; i < maxMagicBytesLength; i += 1)
|
||||
{
|
||||
magicBytes[i] = binaryReader.ReadByte();
|
||||
foreach (var kvPair in imageFormatDecoders)
|
||||
{
|
||||
if (StartsWith(magicBytes, kvPair.Key))
|
||||
{
|
||||
return kvPair.Value(binaryReader);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
throw new ArgumentException(errorMessage, "binaryReader");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Startses the with.
|
||||
/// </summary>
|
||||
/// <param name="thisBytes">The this bytes.</param>
|
||||
/// <param name="thatBytes">The that bytes.</param>
|
||||
/// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
|
||||
private static bool StartsWith(byte[] thisBytes, byte[] thatBytes)
|
||||
{
|
||||
for (int i = 0; i < thatBytes.Length; i += 1)
|
||||
{
|
||||
if (thisBytes[i] != thatBytes[i])
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads the little endian int16.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>System.Int16.</returns>
|
||||
private static short ReadLittleEndianInt16(BinaryReader binaryReader)
|
||||
{
|
||||
var bytes = new byte[sizeof(short)];
|
||||
|
||||
for (int i = 0; i < sizeof(short); i += 1)
|
||||
{
|
||||
bytes[sizeof(short) - 1 - i] = binaryReader.ReadByte();
|
||||
}
|
||||
return BitConverter.ToInt16(bytes, 0);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads the little endian int32.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>System.Int32.</returns>
|
||||
private static int ReadLittleEndianInt32(BinaryReader binaryReader)
|
||||
{
|
||||
var bytes = new byte[sizeof(int)];
|
||||
for (int i = 0; i < sizeof(int); i += 1)
|
||||
{
|
||||
bytes[sizeof(int) - 1 - i] = binaryReader.ReadByte();
|
||||
}
|
||||
return BitConverter.ToInt32(bytes, 0);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Decodes the bitmap.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>Size.</returns>
|
||||
private static Size DecodeBitmap(BinaryReader binaryReader)
|
||||
{
|
||||
binaryReader.ReadBytes(16);
|
||||
int width = binaryReader.ReadInt32();
|
||||
int height = binaryReader.ReadInt32();
|
||||
return new Size(width, height);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Decodes the GIF.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>Size.</returns>
|
||||
private static Size DecodeGif(BinaryReader binaryReader)
|
||||
{
|
||||
int width = binaryReader.ReadInt16();
|
||||
int height = binaryReader.ReadInt16();
|
||||
return new Size(width, height);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Decodes the PNG.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>Size.</returns>
|
||||
private static Size DecodePng(BinaryReader binaryReader)
|
||||
{
|
||||
binaryReader.ReadBytes(8);
|
||||
int width = ReadLittleEndianInt32(binaryReader);
|
||||
int height = ReadLittleEndianInt32(binaryReader);
|
||||
return new Size(width, height);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Decodes the jfif.
|
||||
/// </summary>
|
||||
/// <param name="binaryReader">The binary reader.</param>
|
||||
/// <returns>Size.</returns>
|
||||
/// <exception cref="System.ArgumentException"></exception>
|
||||
private static Size DecodeJfif(BinaryReader binaryReader)
|
||||
{
|
||||
while (binaryReader.ReadByte() == 0xff)
|
||||
{
|
||||
byte marker = binaryReader.ReadByte();
|
||||
short chunkLength = ReadLittleEndianInt16(binaryReader);
|
||||
if (marker == 0xc0)
|
||||
{
|
||||
binaryReader.ReadByte();
|
||||
int height = ReadLittleEndianInt16(binaryReader);
|
||||
int width = ReadLittleEndianInt16(binaryReader);
|
||||
return new Size(width, height);
|
||||
}
|
||||
|
||||
if (chunkLength < 0)
|
||||
{
|
||||
var uchunkLength = (ushort)chunkLength;
|
||||
binaryReader.ReadBytes(uchunkLength - 2);
|
||||
}
|
||||
else
|
||||
{
|
||||
binaryReader.ReadBytes(chunkLength - 2);
|
||||
}
|
||||
}
|
||||
|
||||
throw new ArgumentException(errorMessage);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,10 +1,8 @@
|
||||
using MediaBrowser.Common.Drawing;
|
||||
using MediaBrowser.Common.Extensions;
|
||||
using MediaBrowser.Common.Extensions;
|
||||
using MediaBrowser.Common.IO;
|
||||
using MediaBrowser.Common.Kernel;
|
||||
using MediaBrowser.Controller.Entities;
|
||||
using MediaBrowser.Controller.Entities.TV;
|
||||
using MediaBrowser.Controller.Persistence;
|
||||
using MediaBrowser.Controller.Providers;
|
||||
using MediaBrowser.Model.Drawing;
|
||||
using MediaBrowser.Model.Entities;
|
||||
|
||||
Reference in New Issue
Block a user