mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
8 years ago
committed by
GitHub
GitHub
21 changed files with 504 additions and 395 deletions
@ -0,0 +1,186 @@ |
|||||
|
// Copyright (c) Six Labors and contributors.
|
||||
|
// Licensed under the Apache License, Version 2.0.
|
||||
|
|
||||
|
using System; |
||||
|
using System.Collections.Concurrent; |
||||
|
using System.Collections.Generic; |
||||
|
using System.Linq; |
||||
|
using System.Text; |
||||
|
|
||||
|
namespace SixLabors.ImageSharp.Formats |
||||
|
{ |
||||
|
/// <summary>
|
||||
|
/// Collection of Image Formats to be used in <see cref="Configuration" /> class.
|
||||
|
/// </summary>
|
||||
|
public class ImageFormatManager |
||||
|
{ |
||||
|
/// <summary>
|
||||
|
/// The list of supported <see cref="IImageEncoder"/> keyed to mime types.
|
||||
|
/// </summary>
|
||||
|
private readonly ConcurrentDictionary<IImageFormat, IImageEncoder> mimeTypeEncoders = new ConcurrentDictionary<IImageFormat, IImageEncoder>(); |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// The list of supported <see cref="IImageEncoder"/> keyed to mime types.
|
||||
|
/// </summary>
|
||||
|
private readonly ConcurrentDictionary<IImageFormat, IImageDecoder> mimeTypeDecoders = new ConcurrentDictionary<IImageFormat, IImageDecoder>(); |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// The list of supported <see cref="IImageFormat"/>s.
|
||||
|
/// </summary>
|
||||
|
private readonly ConcurrentBag<IImageFormat> imageFormats = new ConcurrentBag<IImageFormat>(); |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// The list of supported <see cref="IImageFormatDetector"/>s.
|
||||
|
/// </summary>
|
||||
|
private ConcurrentBag<IImageFormatDetector> imageFormatDetectors = new ConcurrentBag<IImageFormatDetector>(); |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Initializes a new instance of the <see cref="ImageFormatManager" /> class.
|
||||
|
/// </summary>
|
||||
|
public ImageFormatManager() |
||||
|
{ |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the maximum header size of all the formats.
|
||||
|
/// </summary>
|
||||
|
internal int MaxHeaderSize { get; private set; } |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the currently registered <see cref="IImageFormat"/>s.
|
||||
|
/// </summary>
|
||||
|
public IEnumerable<IImageFormat> ImageFormats => this.imageFormats; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the currently registered <see cref="IImageFormatDetector"/>s.
|
||||
|
/// </summary>
|
||||
|
internal IEnumerable<IImageFormatDetector> FormatDetectors => this.imageFormatDetectors; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the currently registered <see cref="IImageDecoder"/>s.
|
||||
|
/// </summary>
|
||||
|
internal IEnumerable<KeyValuePair<IImageFormat, IImageDecoder>> ImageDecoders => this.mimeTypeDecoders; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the currently registered <see cref="IImageEncoder"/>s.
|
||||
|
/// </summary>
|
||||
|
internal IEnumerable<KeyValuePair<IImageFormat, IImageEncoder>> ImageEncoders => this.mimeTypeEncoders; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Registers a new format provider.
|
||||
|
/// </summary>
|
||||
|
/// <param name="format">The format to register as a known format.</param>
|
||||
|
public void AddImageFormat(IImageFormat format) |
||||
|
{ |
||||
|
Guard.NotNull(format, nameof(format)); |
||||
|
Guard.NotNull(format.MimeTypes, nameof(format.MimeTypes)); |
||||
|
Guard.NotNull(format.FileExtensions, nameof(format.FileExtensions)); |
||||
|
this.imageFormats.Add(format); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// For the specified file extensions type find the e <see cref="IImageFormat"/>.
|
||||
|
/// </summary>
|
||||
|
/// <param name="extension">The extension to discover</param>
|
||||
|
/// <returns>The <see cref="IImageFormat"/> if found otherwise null</returns>
|
||||
|
public IImageFormat FindFormatByFileExtension(string extension) |
||||
|
{ |
||||
|
return this.imageFormats.FirstOrDefault(x => x.FileExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase)); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// For the specified mime type find the <see cref="IImageFormat"/>.
|
||||
|
/// </summary>
|
||||
|
/// <param name="mimeType">The mime-type to discover</param>
|
||||
|
/// <returns>The <see cref="IImageFormat"/> if found; otherwise null</returns>
|
||||
|
public IImageFormat FindFormatByMimeType(string mimeType) |
||||
|
{ |
||||
|
return this.imageFormats.FirstOrDefault(x => x.MimeTypes.Contains(mimeType, StringComparer.OrdinalIgnoreCase)); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Sets a specific image encoder as the encoder for a specific image format.
|
||||
|
/// </summary>
|
||||
|
/// <param name="imageFormat">The image format to register the encoder for.</param>
|
||||
|
/// <param name="encoder">The encoder to use,</param>
|
||||
|
public void SetEncoder(IImageFormat imageFormat, IImageEncoder encoder) |
||||
|
{ |
||||
|
Guard.NotNull(imageFormat, nameof(imageFormat)); |
||||
|
Guard.NotNull(encoder, nameof(encoder)); |
||||
|
this.AddImageFormat(imageFormat); |
||||
|
this.mimeTypeEncoders.AddOrUpdate(imageFormat, encoder, (s, e) => encoder); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Sets a specific image decoder as the decoder for a specific image format.
|
||||
|
/// </summary>
|
||||
|
/// <param name="imageFormat">The image format to register the encoder for.</param>
|
||||
|
/// <param name="decoder">The decoder to use,</param>
|
||||
|
public void SetDecoder(IImageFormat imageFormat, IImageDecoder decoder) |
||||
|
{ |
||||
|
Guard.NotNull(imageFormat, nameof(imageFormat)); |
||||
|
Guard.NotNull(decoder, nameof(decoder)); |
||||
|
this.AddImageFormat(imageFormat); |
||||
|
this.mimeTypeDecoders.AddOrUpdate(imageFormat, decoder, (s, e) => decoder); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Removes all the registered image format detectors.
|
||||
|
/// </summary>
|
||||
|
public void ClearImageFormatDetectors() |
||||
|
{ |
||||
|
this.imageFormatDetectors = new ConcurrentBag<IImageFormatDetector>(); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Adds a new detector for detecting mime types.
|
||||
|
/// </summary>
|
||||
|
/// <param name="detector">The detector to add</param>
|
||||
|
public void AddImageFormatDetector(IImageFormatDetector detector) |
||||
|
{ |
||||
|
Guard.NotNull(detector, nameof(detector)); |
||||
|
this.imageFormatDetectors.Add(detector); |
||||
|
this.SetMaxHeaderSize(); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// For the specified mime type find the decoder.
|
||||
|
/// </summary>
|
||||
|
/// <param name="format">The format to discover</param>
|
||||
|
/// <returns>The <see cref="IImageDecoder"/> if found otherwise null</returns>
|
||||
|
public IImageDecoder FindDecoder(IImageFormat format) |
||||
|
{ |
||||
|
Guard.NotNull(format, nameof(format)); |
||||
|
if (this.mimeTypeDecoders.TryGetValue(format, out IImageDecoder decoder)) |
||||
|
{ |
||||
|
return decoder; |
||||
|
} |
||||
|
|
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// For the specified mime type find the encoder.
|
||||
|
/// </summary>
|
||||
|
/// <param name="format">The format to discover</param>
|
||||
|
/// <returns>The <see cref="IImageEncoder"/> if found otherwise null</returns>
|
||||
|
public IImageEncoder FindEncoder(IImageFormat format) |
||||
|
{ |
||||
|
Guard.NotNull(format, nameof(format)); |
||||
|
if (this.mimeTypeEncoders.TryGetValue(format, out IImageEncoder encoder)) |
||||
|
{ |
||||
|
return encoder; |
||||
|
} |
||||
|
|
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Sets the max header size.
|
||||
|
/// </summary>
|
||||
|
private void SetMaxHeaderSize() |
||||
|
{ |
||||
|
this.MaxHeaderSize = this.imageFormatDetectors.Max(x => x.HeaderSize); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,129 @@ |
|||||
|
// Copyright (c) Six Labors and contributors.
|
||||
|
// Licensed under the Apache License, Version 2.0.
|
||||
|
|
||||
|
using System; |
||||
|
using System.Collections.Generic; |
||||
|
using System.IO; |
||||
|
using System.Linq; |
||||
|
using SixLabors.ImageSharp.Formats; |
||||
|
using SixLabors.ImageSharp.IO; |
||||
|
using SixLabors.ImageSharp.PixelFormats; |
||||
|
using SixLabors.ImageSharp.Formats.Png; |
||||
|
using SixLabors.ImageSharp.Formats.Bmp; |
||||
|
using SixLabors.ImageSharp.Formats.Jpeg; |
||||
|
using SixLabors.ImageSharp.Formats.Gif; |
||||
|
using Moq; |
||||
|
using Xunit; |
||||
|
|
||||
|
|
||||
|
namespace SixLabors.ImageSharp.Tests |
||||
|
{ |
||||
|
public class ImageFormatManagerTests |
||||
|
{ |
||||
|
public ImageFormatManager FormatsManagerEmpty { get; private set; } |
||||
|
public ImageFormatManager DefaultFormatsManager { get; private set; } |
||||
|
|
||||
|
public ImageFormatManagerTests() |
||||
|
{ |
||||
|
this.DefaultFormatsManager = Configuration.Default.ImageFormatsManager; |
||||
|
this.FormatsManagerEmpty = new ImageFormatManager(); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void IfAutoloadWellKnownFormatsIsTrueAllFormatsAreLoaded() |
||||
|
{ |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageEncoders.Select(item => item.Value).OfType<PngEncoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageEncoders.Select(item => item.Value).OfType<BmpEncoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageEncoders.Select(item => item.Value).OfType<JpegEncoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageEncoders.Select(item => item.Value).OfType<GifEncoder>().Count()); |
||||
|
|
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageDecoders.Select(item => item.Value).OfType<PngDecoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageDecoders.Select(item => item.Value).OfType<BmpDecoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageDecoders.Select(item => item.Value).OfType<JpegDecoder>().Count()); |
||||
|
Assert.Equal(1, this.DefaultFormatsManager.ImageDecoders.Select(item => item.Value).OfType<BmpDecoder>().Count()); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void AddImageFormatDetectorNullthrows() |
||||
|
{ |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.AddImageFormatDetector(null); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void RegisterNullMimeTypeEncoder() |
||||
|
{ |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetEncoder(null, new Mock<IImageEncoder>().Object); |
||||
|
}); |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetEncoder(ImageFormats.Bmp, null); |
||||
|
}); |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetEncoder(null, null); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void RegisterNullSetDecoder() |
||||
|
{ |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetDecoder(null, new Mock<IImageDecoder>().Object); |
||||
|
}); |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetDecoder(ImageFormats.Bmp, null); |
||||
|
}); |
||||
|
Assert.Throws<ArgumentNullException>(() => |
||||
|
{ |
||||
|
this.DefaultFormatsManager.SetDecoder(null, null); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void RegisterMimeTypeEncoderReplacesLast() |
||||
|
{ |
||||
|
IImageEncoder encoder1 = new Mock<IImageEncoder>().Object; |
||||
|
this.FormatsManagerEmpty.SetEncoder(TestFormat.GlobalTestFormat, encoder1); |
||||
|
IImageEncoder found = this.FormatsManagerEmpty.FindEncoder(TestFormat.GlobalTestFormat); |
||||
|
Assert.Equal(encoder1, found); |
||||
|
|
||||
|
IImageEncoder encoder2 = new Mock<IImageEncoder>().Object; |
||||
|
this.FormatsManagerEmpty.SetEncoder(TestFormat.GlobalTestFormat, encoder2); |
||||
|
IImageEncoder found2 = this.FormatsManagerEmpty.FindEncoder(TestFormat.GlobalTestFormat); |
||||
|
Assert.Equal(encoder2, found2); |
||||
|
Assert.NotEqual(found, found2); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void RegisterMimeTypeDecoderReplacesLast() |
||||
|
{ |
||||
|
IImageDecoder decoder1 = new Mock<IImageDecoder>().Object; |
||||
|
this.FormatsManagerEmpty.SetDecoder(TestFormat.GlobalTestFormat, decoder1); |
||||
|
IImageDecoder found = this.FormatsManagerEmpty.FindDecoder(TestFormat.GlobalTestFormat); |
||||
|
Assert.Equal(decoder1, found); |
||||
|
|
||||
|
IImageDecoder decoder2 = new Mock<IImageDecoder>().Object; |
||||
|
this.FormatsManagerEmpty.SetDecoder(TestFormat.GlobalTestFormat, decoder2); |
||||
|
IImageDecoder found2 = this.FormatsManagerEmpty.FindDecoder(TestFormat.GlobalTestFormat); |
||||
|
Assert.Equal(decoder2, found2); |
||||
|
Assert.NotEqual(found, found2); |
||||
|
} |
||||
|
|
||||
|
[Fact] |
||||
|
public void AddFormatCallsConfig() |
||||
|
{ |
||||
|
var provider = new Mock<IConfigurationModule>(); |
||||
|
var config = new Configuration(); |
||||
|
config.Configure(provider.Object); |
||||
|
|
||||
|
provider.Verify(x => x.Configure(config)); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue