1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024 |
- using MediaBrowser.Common.Extensions;
- using MediaBrowser.Controller;
- using MediaBrowser.Controller.Drawing;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Providers;
- using MediaBrowser.Model.Drawing;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.Logging;
- using MediaBrowser.Model.Serialization;
- using System;
- using System.Collections.Concurrent;
- using System.Collections.Generic;
- using System.Globalization;
- using System.IO;
- using System.Linq;
- using System.Threading;
- using System.Threading.Tasks;
- using MediaBrowser.Model.IO;
- using Emby.Drawing.Common;
- using MediaBrowser.Controller.Library;
- using MediaBrowser.Controller.MediaEncoding;
- using MediaBrowser.Model.Net;
- using MediaBrowser.Model.Threading;
- using MediaBrowser.Model.Extensions;
- namespace Emby.Drawing
- {
- /// <summary>
- /// Class ImageProcessor
- /// </summary>
- public class ImageProcessor : IImageProcessor, IDisposable
- {
- /// <summary>
- /// The us culture
- /// </summary>
- protected readonly CultureInfo UsCulture = new CultureInfo("en-US");
- /// <summary>
- /// The _cached imaged sizes
- /// </summary>
- private readonly ConcurrentDictionary<Guid, ImageSize> _cachedImagedSizes;
- /// <summary>
- /// Gets the list of currently registered image processors
- /// Image processors are specialized metadata providers that run after the normal ones
- /// </summary>
- /// <value>The image enhancers.</value>
- public IImageEnhancer[] ImageEnhancers { get; private set; }
- /// <summary>
- /// The _logger
- /// </summary>
- private readonly ILogger _logger;
- private readonly IFileSystem _fileSystem;
- private readonly IJsonSerializer _jsonSerializer;
- private readonly IServerApplicationPaths _appPaths;
- private IImageEncoder _imageEncoder;
- private readonly Func<ILibraryManager> _libraryManager;
- private readonly Func<IMediaEncoder> _mediaEncoder;
- public ImageProcessor(ILogger logger,
- IServerApplicationPaths appPaths,
- IFileSystem fileSystem,
- IJsonSerializer jsonSerializer,
- IImageEncoder imageEncoder,
- Func<ILibraryManager> libraryManager, ITimerFactory timerFactory, Func<IMediaEncoder> mediaEncoder)
- {
- _logger = logger;
- _fileSystem = fileSystem;
- _jsonSerializer = jsonSerializer;
- _imageEncoder = imageEncoder;
- _libraryManager = libraryManager;
- _mediaEncoder = mediaEncoder;
- _appPaths = appPaths;
- ImageEnhancers = new IImageEnhancer[] { };
- _saveImageSizeTimer = timerFactory.Create(SaveImageSizeCallback, null, Timeout.Infinite, Timeout.Infinite);
- ImageHelper.ImageProcessor = this;
- Dictionary<Guid, ImageSize> sizeDictionary;
- try
- {
- sizeDictionary = jsonSerializer.DeserializeFromFile<Dictionary<Guid, ImageSize>>(ImageSizeFile) ??
- new Dictionary<Guid, ImageSize>();
- }
- catch (FileNotFoundException)
- {
- // No biggie
- sizeDictionary = new Dictionary<Guid, ImageSize>();
- }
- catch (IOException)
- {
- // No biggie
- sizeDictionary = new Dictionary<Guid, ImageSize>();
- }
- catch (Exception ex)
- {
- logger.ErrorException("Error parsing image size cache file", ex);
- sizeDictionary = new Dictionary<Guid, ImageSize>();
- }
- _cachedImagedSizes = new ConcurrentDictionary<Guid, ImageSize>(sizeDictionary);
- }
- public IImageEncoder ImageEncoder
- {
- get { return _imageEncoder; }
- set
- {
- if (value == null)
- {
- throw new ArgumentNullException("value");
- }
- _imageEncoder = value;
- }
- }
- public string[] SupportedInputFormats
- {
- get
- {
- return new string[]
- {
- "tiff",
- "tif",
- "jpeg",
- "jpg",
- "png",
- "aiff",
- "cr2",
- "crw",
- "dng",
- // Remove until supported
- //"nef",
- "orf",
- "pef",
- "arw",
- "webp",
- "gif",
- "bmp",
- "erf",
- "raf",
- "rw2",
- "nrw",
- "dng",
- "ico",
- "astc",
- "ktx",
- "pkm",
- "wbmp"
- };
- }
- }
- public bool SupportsImageCollageCreation
- {
- get
- {
- return _imageEncoder.SupportsImageCollageCreation;
- }
- }
- private string ResizedImageCachePath
- {
- get
- {
- return Path.Combine(_appPaths.ImageCachePath, "resized-images");
- }
- }
- private string EnhancedImageCachePath
- {
- get
- {
- return Path.Combine(_appPaths.ImageCachePath, "enhanced-images");
- }
- }
- public void AddParts(IEnumerable<IImageEnhancer> enhancers)
- {
- ImageEnhancers = enhancers.ToArray();
- }
- public async Task ProcessImage(ImageProcessingOptions options, Stream toStream)
- {
- var file = await ProcessImage(options).ConfigureAwait(false);
- using (var fileStream = _fileSystem.GetFileStream(file.Item1, FileOpenMode.Open, FileAccessMode.Read, FileShareMode.Read, true))
- {
- await fileStream.CopyToAsync(toStream).ConfigureAwait(false);
- }
- }
- public ImageFormat[] GetSupportedImageOutputFormats()
- {
- return _imageEncoder.SupportedOutputFormats;
- }
- private static readonly string[] TransparentImageTypes = new string[] { ".png", ".webp" };
- public bool SupportsTransparency(string path)
- {
- return TransparentImageTypes.Contains(Path.GetExtension(path) ?? string.Empty);
- }
- public async Task<Tuple<string, string, DateTime>> ProcessImage(ImageProcessingOptions options)
- {
- if (options == null)
- {
- throw new ArgumentNullException("options");
- }
- var originalImage = options.Image;
- IHasMetadata item = options.Item;
- if (!originalImage.IsLocalFile)
- {
- if (item == null)
- {
- item = _libraryManager().GetItemById(options.ItemId);
- }
- originalImage = await _libraryManager().ConvertImageToLocal(item, originalImage, options.ImageIndex).ConfigureAwait(false);
- }
- var originalImagePath = originalImage.Path;
- var dateModified = originalImage.DateModified;
- if (!_imageEncoder.SupportsImageEncoding)
- {
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- var supportedImageInfo = await GetSupportedImage(originalImagePath, dateModified).ConfigureAwait(false);
- originalImagePath = supportedImageInfo.Item1;
- dateModified = supportedImageInfo.Item2;
- var requiresTransparency = TransparentImageTypes.Contains(Path.GetExtension(originalImagePath) ?? string.Empty);
- if (options.Enhancers.Count > 0)
- {
- if (item == null)
- {
- item = _libraryManager().GetItemById(options.ItemId);
- }
- var tuple = await GetEnhancedImage(new ItemImageInfo
- {
- DateModified = dateModified,
- Type = originalImage.Type,
- Path = originalImagePath
- }, requiresTransparency, item, options.ImageIndex, options.Enhancers).ConfigureAwait(false);
- originalImagePath = tuple.Item1;
- dateModified = tuple.Item2;
- requiresTransparency = tuple.Item3;
- }
- var photo = item as Photo;
- var autoOrient = false;
- ImageOrientation? orientation = null;
- if (photo != null && photo.Orientation.HasValue && photo.Orientation.Value != ImageOrientation.TopLeft)
- {
- autoOrient = true;
- orientation = photo.Orientation;
- }
- if (options.HasDefaultOptions(originalImagePath) && (!autoOrient || !options.RequiresAutoOrientation))
- {
- // Just spit out the original file if all the options are default
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- ImageSize? originalImageSize = GetSavedImageSize(originalImagePath, dateModified);
- if (originalImageSize.HasValue && options.HasDefaultOptions(originalImagePath, originalImageSize.Value) && !autoOrient)
- {
- // Just spit out the original file if all the options are default
- _logger.Info("Returning original image {0}", originalImagePath);
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- var newSize = ImageHelper.GetNewImageSize(options, originalImageSize);
- var quality = options.Quality;
- var outputFormat = GetOutputFormat(options.SupportedOutputFormats, requiresTransparency);
- var cacheFilePath = GetCacheFilePath(originalImagePath, newSize, quality, dateModified, outputFormat, options.AddPlayedIndicator, options.PercentPlayed, options.UnplayedCount, options.Blur, options.BackgroundColor, options.ForegroundLayer);
- try
- {
- CheckDisposed();
- if (!_fileSystem.FileExists(cacheFilePath))
- {
- var tmpPath = Path.ChangeExtension(Path.Combine(_appPaths.TempDirectory, Guid.NewGuid().ToString("N")), Path.GetExtension(cacheFilePath));
- _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(tmpPath));
- if (options.CropWhiteSpace && !SupportsTransparency(originalImagePath))
- {
- options.CropWhiteSpace = false;
- }
- var resultPath = _imageEncoder.EncodeImage(originalImagePath, dateModified, tmpPath, autoOrient, orientation, quality, options, outputFormat);
- if (string.Equals(resultPath, originalImagePath, StringComparison.OrdinalIgnoreCase))
- {
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(cacheFilePath));
- CopyFile(tmpPath, cacheFilePath);
- return new Tuple<string, string, DateTime>(tmpPath, GetMimeType(outputFormat, cacheFilePath), _fileSystem.GetLastWriteTimeUtc(tmpPath));
- }
- return new Tuple<string, string, DateTime>(cacheFilePath, GetMimeType(outputFormat, cacheFilePath), _fileSystem.GetLastWriteTimeUtc(cacheFilePath));
- }
- catch (ArgumentOutOfRangeException ex)
- {
- // Decoder failed to decode it
- #if DEBUG
- _logger.ErrorException("Error encoding image", ex);
- #endif
- // Just spit out the original file if all the options are default
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- catch (Exception ex)
- {
- // If it fails for whatever reason, return the original image
- _logger.ErrorException("Error encoding image", ex);
- // Just spit out the original file if all the options are default
- return new Tuple<string, string, DateTime>(originalImagePath, MimeTypes.GetMimeType(originalImagePath), dateModified);
- }
- }
- private ImageFormat GetOutputFormat(ImageFormat[] clientSupportedFormats, bool requiresTransparency)
- {
- var serverFormats = GetSupportedImageOutputFormats();
- // Client doesn't care about format, so start with webp if supported
- if (serverFormats.Contains(ImageFormat.Webp) && clientSupportedFormats.Contains(ImageFormat.Webp))
- {
- return ImageFormat.Webp;
- }
- // If transparency is needed and webp isn't supported, than png is the only option
- if (requiresTransparency)
- {
- return ImageFormat.Png;
- }
- foreach (var format in clientSupportedFormats)
- {
- if (serverFormats.Contains(format))
- {
- return format;
- }
- }
- // We should never actually get here
- return ImageFormat.Jpg;
- }
- private void CopyFile(string src, string destination)
- {
- try
- {
- _fileSystem.CopyFile(src, destination, true);
- }
- catch
- {
- }
- }
- //private static int[][] OPERATIONS = new int[][] {
- // TopLeft
- //new int[] { 0, NONE},
- // TopRight
- //new int[] { 0, HORIZONTAL},
- //new int[] {180, NONE},
- // LeftTop
- //new int[] { 0, VERTICAL},
- //new int[] { 90, HORIZONTAL},
- // RightTop
- //new int[] { 90, NONE},
- //new int[] {-90, HORIZONTAL},
- //new int[] {-90, NONE},
- //};
- private string GetMimeType(ImageFormat format, string path)
- {
- if (format == ImageFormat.Bmp)
- {
- return MimeTypes.GetMimeType("i.bmp");
- }
- if (format == ImageFormat.Gif)
- {
- return MimeTypes.GetMimeType("i.gif");
- }
- if (format == ImageFormat.Jpg)
- {
- return MimeTypes.GetMimeType("i.jpg");
- }
- if (format == ImageFormat.Png)
- {
- return MimeTypes.GetMimeType("i.png");
- }
- if (format == ImageFormat.Webp)
- {
- return MimeTypes.GetMimeType("i.webp");
- }
- return MimeTypes.GetMimeType(path);
- }
- /// <summary>
- /// Increment this when there's a change requiring caches to be invalidated
- /// </summary>
- private const string Version = "3";
- /// <summary>
- /// Gets the cache file path based on a set of parameters
- /// </summary>
- private string GetCacheFilePath(string originalPath, ImageSize outputSize, int quality, DateTime dateModified, ImageFormat format, bool addPlayedIndicator, double percentPlayed, int? unwatchedCount, int? blur, string backgroundColor, string foregroundLayer)
- {
- var filename = originalPath;
- filename += "width=" + outputSize.Width;
- filename += "height=" + outputSize.Height;
- filename += "quality=" + quality;
- filename += "datemodified=" + dateModified.Ticks;
- filename += "f=" + format;
- if (addPlayedIndicator)
- {
- filename += "pl=true";
- }
- if (percentPlayed > 0)
- {
- filename += "p=" + percentPlayed;
- }
- if (unwatchedCount.HasValue)
- {
- filename += "p=" + unwatchedCount.Value;
- }
- if (blur.HasValue)
- {
- filename += "blur=" + blur.Value;
- }
- if (!string.IsNullOrEmpty(backgroundColor))
- {
- filename += "b=" + backgroundColor;
- }
- if (!string.IsNullOrEmpty(foregroundLayer))
- {
- filename += "fl=" + foregroundLayer;
- }
- filename += "v=" + Version;
- return GetCachePath(ResizedImageCachePath, filename, "." + format.ToString().ToLower());
- }
- public ImageSize GetImageSize(ItemImageInfo info, bool allowSlowMethods)
- {
- return GetImageSize(info.Path, info.DateModified, allowSlowMethods);
- }
- public ImageSize GetImageSize(ItemImageInfo info)
- {
- return GetImageSize(info.Path, info.DateModified, false);
- }
- public ImageSize GetImageSize(string path)
- {
- return GetImageSize(path, _fileSystem.GetLastWriteTimeUtc(path), false);
- }
- /// <summary>
- /// Gets the size of the image.
- /// </summary>
- /// <param name="path">The path.</param>
- /// <param name="imageDateModified">The image date modified.</param>
- /// <param name="allowSlowMethod">if set to <c>true</c> [allow slow method].</param>
- /// <returns>ImageSize.</returns>
- /// <exception cref="System.ArgumentNullException">path</exception>
- private ImageSize GetImageSize(string path, DateTime imageDateModified, bool allowSlowMethod)
- {
- if (string.IsNullOrEmpty(path))
- {
- throw new ArgumentNullException("path");
- }
- ImageSize size;
- var cacheHash = GetImageSizeKey(path, imageDateModified);
- if (!_cachedImagedSizes.TryGetValue(cacheHash, out size))
- {
- size = GetImageSizeInternal(path, allowSlowMethod);
- SaveImageSize(size, cacheHash, false);
- }
- return size;
- }
- public void SaveImageSize(string path, DateTime imageDateModified, ImageSize size)
- {
- var cacheHash = GetImageSizeKey(path, imageDateModified);
- SaveImageSize(size, cacheHash, true);
- }
- private void SaveImageSize(ImageSize size, Guid cacheHash, bool checkExists)
- {
- if (size.Width <= 0 || size.Height <= 0)
- {
- return;
- }
- if (checkExists && _cachedImagedSizes.ContainsKey(cacheHash))
- {
- return;
- }
- if (checkExists)
- {
- if (_cachedImagedSizes.TryAdd(cacheHash, size))
- {
- StartSaveImageSizeTimer();
- }
- }
- else
- {
- StartSaveImageSizeTimer();
- _cachedImagedSizes.AddOrUpdate(cacheHash, size, (keyName, oldValue) => size);
- }
- }
- private Guid GetImageSizeKey(string path, DateTime imageDateModified)
- {
- var name = path + "datemodified=" + imageDateModified.Ticks;
- return name.GetMD5();
- }
- public ImageSize? GetSavedImageSize(string path, DateTime imageDateModified)
- {
- ImageSize size;
- var cacheHash = GetImageSizeKey(path, imageDateModified);
- if (_cachedImagedSizes.TryGetValue(cacheHash, out size))
- {
- return size;
- }
- return null;
- }
- /// <summary>
- /// Gets the image size internal.
- /// </summary>
- /// <param name="path">The path.</param>
- /// <param name="allowSlowMethod">if set to <c>true</c> [allow slow method].</param>
- /// <returns>ImageSize.</returns>
- private ImageSize GetImageSizeInternal(string path, bool allowSlowMethod)
- {
- //try
- //{
- // using (var fileStream = _fileSystem.OpenRead(path))
- // {
- // using (var file = TagLib.File.Create(new StreamFileAbstraction(Path.GetFileName(path), fileStream, null)))
- // {
- // var image = file as TagLib.Image.File;
- // if (image != null)
- // {
- // var properties = image.Properties;
- // return new ImageSize
- // {
- // Height = properties.PhotoHeight,
- // Width = properties.PhotoWidth
- // };
- // }
- // }
- // }
- //}
- //catch
- //{
- //}
- try
- {
- return ImageHeader.GetDimensions(path, _logger, _fileSystem);
- }
- catch
- {
- if (allowSlowMethod)
- {
- return _imageEncoder.GetImageSize(path);
- }
- throw;
- }
- }
- private readonly ITimer _saveImageSizeTimer;
- private const int SaveImageSizeTimeout = 5000;
- private readonly object _saveImageSizeLock = new object();
- private void StartSaveImageSizeTimer()
- {
- _saveImageSizeTimer.Change(SaveImageSizeTimeout, Timeout.Infinite);
- }
- private void SaveImageSizeCallback(object state)
- {
- lock (_saveImageSizeLock)
- {
- try
- {
- var path = ImageSizeFile;
- _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(path));
- _jsonSerializer.SerializeToFile(_cachedImagedSizes, path);
- }
- catch (Exception ex)
- {
- _logger.ErrorException("Error saving image size file", ex);
- }
- }
- }
- private string ImageSizeFile
- {
- get
- {
- return Path.Combine(_appPaths.DataPath, "imagesizes.json");
- }
- }
- /// <summary>
- /// Gets the image cache tag.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="image">The image.</param>
- /// <returns>Guid.</returns>
- /// <exception cref="System.ArgumentNullException">item</exception>
- public string GetImageCacheTag(IHasMetadata item, ItemImageInfo image)
- {
- if (item == null)
- {
- throw new ArgumentNullException("item");
- }
- if (image == null)
- {
- throw new ArgumentNullException("image");
- }
- var supportedEnhancers = GetSupportedEnhancers(item, image.Type);
- return GetImageCacheTag(item, image, supportedEnhancers);
- }
- /// <summary>
- /// Gets the image cache tag.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="image">The image.</param>
- /// <param name="imageEnhancers">The image enhancers.</param>
- /// <returns>Guid.</returns>
- /// <exception cref="System.ArgumentNullException">item</exception>
- public string GetImageCacheTag(IHasMetadata item, ItemImageInfo image, List<IImageEnhancer> imageEnhancers)
- {
- if (item == null)
- {
- throw new ArgumentNullException("item");
- }
- if (imageEnhancers == null)
- {
- throw new ArgumentNullException("imageEnhancers");
- }
- if (image == null)
- {
- throw new ArgumentNullException("image");
- }
- var originalImagePath = image.Path;
- var dateModified = image.DateModified;
- var imageType = image.Type;
- // Optimization
- if (imageEnhancers.Count == 0)
- {
- return (originalImagePath + dateModified.Ticks).GetMD5().ToString("N");
- }
- // Cache name is created with supported enhancers combined with the last config change so we pick up new config changes
- var cacheKeys = imageEnhancers.Select(i => i.GetConfigurationCacheKey(item, imageType)).ToList();
- cacheKeys.Add(originalImagePath + dateModified.Ticks);
- return string.Join("|", cacheKeys.ToArray(cacheKeys.Count)).GetMD5().ToString("N");
- }
- private async Task<Tuple<string, DateTime>> GetSupportedImage(string originalImagePath, DateTime dateModified)
- {
- var inputFormat = (Path.GetExtension(originalImagePath) ?? string.Empty)
- .TrimStart('.')
- .Replace("jpeg", "jpg", StringComparison.OrdinalIgnoreCase);
- if (!_imageEncoder.SupportedInputFormats.Contains(inputFormat, StringComparer.OrdinalIgnoreCase))
- {
- try
- {
- var filename = (originalImagePath + dateModified.Ticks.ToString(UsCulture)).GetMD5().ToString("N");
- var outputPath = Path.Combine(_appPaths.ImageCachePath, "converted-images", filename + ".webp");
- var file = _fileSystem.GetFileInfo(outputPath);
- if (!file.Exists)
- {
- await _mediaEncoder().ConvertImage(originalImagePath, outputPath).ConfigureAwait(false);
- dateModified = _fileSystem.GetLastWriteTimeUtc(outputPath);
- }
- else
- {
- dateModified = file.LastWriteTimeUtc;
- }
- originalImagePath = outputPath;
- }
- catch (Exception ex)
- {
- _logger.ErrorException("Image conversion failed for {0}", ex, originalImagePath);
- }
- }
- return new Tuple<string, DateTime>(originalImagePath, dateModified);
- }
- /// <summary>
- /// Gets the enhanced image.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="imageType">Type of the image.</param>
- /// <param name="imageIndex">Index of the image.</param>
- /// <returns>Task{System.String}.</returns>
- public async Task<string> GetEnhancedImage(IHasMetadata item, ImageType imageType, int imageIndex)
- {
- var enhancers = GetSupportedEnhancers(item, imageType);
- var imageInfo = item.GetImageInfo(imageType, imageIndex);
- var inputImageSupportsTransparency = SupportsTransparency(imageInfo.Path);
- var result = await GetEnhancedImage(imageInfo, inputImageSupportsTransparency, item, imageIndex, enhancers);
- return result.Item1;
- }
- private async Task<Tuple<string, DateTime, bool>> GetEnhancedImage(ItemImageInfo image,
- bool inputImageSupportsTransparency,
- IHasMetadata item,
- int imageIndex,
- List<IImageEnhancer> enhancers)
- {
- var originalImagePath = image.Path;
- var dateModified = image.DateModified;
- var imageType = image.Type;
- try
- {
- var cacheGuid = GetImageCacheTag(item, image, enhancers);
- // Enhance if we have enhancers
- var ehnancedImageInfo = await GetEnhancedImageInternal(originalImagePath, item, imageType, imageIndex, enhancers, cacheGuid).ConfigureAwait(false);
- var ehnancedImagePath = ehnancedImageInfo.Item1;
- // If the path changed update dateModified
- if (!string.Equals(ehnancedImagePath, originalImagePath, StringComparison.OrdinalIgnoreCase))
- {
- var treatmentRequiresTransparency = ehnancedImageInfo.Item2;
- return new Tuple<string, DateTime, bool>(ehnancedImagePath, _fileSystem.GetLastWriteTimeUtc(ehnancedImagePath), treatmentRequiresTransparency);
- }
- }
- catch (Exception ex)
- {
- _logger.Error("Error enhancing image", ex);
- }
- return new Tuple<string, DateTime, bool>(originalImagePath, dateModified, inputImageSupportsTransparency);
- }
- /// <summary>
- /// Gets the enhanced image internal.
- /// </summary>
- /// <param name="originalImagePath">The original image path.</param>
- /// <param name="item">The item.</param>
- /// <param name="imageType">Type of the image.</param>
- /// <param name="imageIndex">Index of the image.</param>
- /// <param name="supportedEnhancers">The supported enhancers.</param>
- /// <param name="cacheGuid">The cache unique identifier.</param>
- /// <returns>Task<System.String>.</returns>
- /// <exception cref="ArgumentNullException">
- /// originalImagePath
- /// or
- /// item
- /// </exception>
- private async Task<Tuple<string, bool>> GetEnhancedImageInternal(string originalImagePath,
- IHasMetadata item,
- ImageType imageType,
- int imageIndex,
- List<IImageEnhancer> supportedEnhancers,
- string cacheGuid)
- {
- if (string.IsNullOrEmpty(originalImagePath))
- {
- throw new ArgumentNullException("originalImagePath");
- }
- if (item == null)
- {
- throw new ArgumentNullException("item");
- }
- var treatmentRequiresTransparency = false;
- foreach (var enhancer in supportedEnhancers)
- {
- if (!treatmentRequiresTransparency)
- {
- treatmentRequiresTransparency = enhancer.GetEnhancedImageInfo(item, originalImagePath, imageType, imageIndex).RequiresTransparency;
- }
- }
- // All enhanced images are saved as png to allow transparency
- var cacheExtension = _imageEncoder.SupportedOutputFormats.Contains(ImageFormat.Webp) ?
- ".webp" :
- (treatmentRequiresTransparency ? ".png" : ".jpg");
- var enhancedImagePath = GetCachePath(EnhancedImageCachePath, cacheGuid + cacheExtension);
- // Check again in case of contention
- if (_fileSystem.FileExists(enhancedImagePath))
- {
- return new Tuple<string, bool>(enhancedImagePath, treatmentRequiresTransparency);
- }
- _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(enhancedImagePath));
- var tmpPath = Path.Combine(_appPaths.TempDirectory, Path.ChangeExtension(Guid.NewGuid().ToString(), Path.GetExtension(enhancedImagePath)));
- _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(tmpPath));
- await ExecuteImageEnhancers(supportedEnhancers, originalImagePath, tmpPath, item, imageType, imageIndex).ConfigureAwait(false);
- try
- {
- _fileSystem.CopyFile(tmpPath, enhancedImagePath, true);
- }
- catch
- {
- }
- return new Tuple<string, bool>(tmpPath, treatmentRequiresTransparency);
- }
- /// <summary>
- /// Executes the image enhancers.
- /// </summary>
- /// <param name="imageEnhancers">The image enhancers.</param>
- /// <param name="inputPath">The input path.</param>
- /// <param name="outputPath">The output path.</param>
- /// <param name="item">The item.</param>
- /// <param name="imageType">Type of the image.</param>
- /// <param name="imageIndex">Index of the image.</param>
- /// <returns>Task{EnhancedImage}.</returns>
- private async Task ExecuteImageEnhancers(IEnumerable<IImageEnhancer> imageEnhancers, string inputPath, string outputPath, IHasMetadata item, ImageType imageType, int imageIndex)
- {
- // Run the enhancers sequentially in order of priority
- foreach (var enhancer in imageEnhancers)
- {
- await enhancer.EnhanceImageAsync(item, inputPath, outputPath, imageType, imageIndex).ConfigureAwait(false);
- // Feed the output into the next enhancer as input
- inputPath = outputPath;
- }
- }
- /// <summary>
- /// Gets the cache path.
- /// </summary>
- /// <param name="path">The path.</param>
- /// <param name="uniqueName">Name of the unique.</param>
- /// <param name="fileExtension">The file extension.</param>
- /// <returns>System.String.</returns>
- /// <exception cref="System.ArgumentNullException">
- /// path
- /// or
- /// uniqueName
- /// or
- /// fileExtension
- /// </exception>
- public string GetCachePath(string path, string uniqueName, string fileExtension)
- {
- if (string.IsNullOrEmpty(path))
- {
- throw new ArgumentNullException("path");
- }
- if (string.IsNullOrEmpty(uniqueName))
- {
- throw new ArgumentNullException("uniqueName");
- }
- if (string.IsNullOrEmpty(fileExtension))
- {
- throw new ArgumentNullException("fileExtension");
- }
- var filename = uniqueName.GetMD5() + fileExtension;
- return GetCachePath(path, filename);
- }
- /// <summary>
- /// Gets the cache path.
- /// </summary>
- /// <param name="path">The path.</param>
- /// <param name="filename">The filename.</param>
- /// <returns>System.String.</returns>
- /// <exception cref="System.ArgumentNullException">
- /// path
- /// or
- /// filename
- /// </exception>
- public string GetCachePath(string path, string filename)
- {
- if (string.IsNullOrEmpty(path))
- {
- throw new ArgumentNullException("path");
- }
- if (string.IsNullOrEmpty(filename))
- {
- throw new ArgumentNullException("filename");
- }
- var prefix = filename.Substring(0, 1);
- path = Path.Combine(path, prefix);
- return Path.Combine(path, filename);
- }
- public void CreateImageCollage(ImageCollageOptions options)
- {
- _logger.Info("Creating image collage and saving to {0}", options.OutputPath);
- _imageEncoder.CreateImageCollage(options);
- _logger.Info("Completed creation of image collage and saved to {0}", options.OutputPath);
- }
- public List<IImageEnhancer> GetSupportedEnhancers(IHasMetadata item, ImageType imageType)
- {
- var list = new List<IImageEnhancer>();
- foreach (var i in ImageEnhancers)
- {
- try
- {
- if (i.Supports(item, imageType))
- {
- list.Add(i);
- }
- }
- catch (Exception ex)
- {
- _logger.ErrorException("Error in image enhancer: {0}", ex, i.GetType().Name);
- }
- }
- return list;
- }
- private bool _disposed;
- public void Dispose()
- {
- _disposed = true;
- var disposable = _imageEncoder as IDisposable;
- if (disposable != null)
- {
- disposable.Dispose();
- }
- _saveImageSizeTimer.Dispose();
- GC.SuppressFinalize(this);
- }
- private void CheckDisposed()
- {
- if (_disposed)
- {
- throw new ObjectDisposedException(GetType().Name);
- }
- }
- }
- }
|