12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031 |
- 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);
- // These are just jpg files renamed as tbn
- if (string.Equals(inputFormat, "tbn", StringComparison.OrdinalIgnoreCase))
- {
- return new Tuple<string, DateTime>(originalImagePath, dateModified);
- }
- if (!_imageEncoder.SupportedInputFormats.Contains(inputFormat, StringComparer.OrdinalIgnoreCase))
- {
- try
- {
- var filename = (originalImagePath + dateModified.Ticks.ToString(UsCulture)).GetMD5().ToString("N");
- var cacheExtension = _mediaEncoder().SupportsEncoder("libwebp") ? ".webp" : ".png";
- var outputPath = Path.Combine(_appPaths.ImageCachePath, "converted-images", filename + cacheExtension);
- 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);
- }
- }
- }
- }
|