OpenDiablo2/OpenDiablo2.Core/ResourceManager.cs

/*  OpenDiablo 2 - An open source re-implementation of Diablo 2 in C#
 *  
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <https://www.gnu.org/licenses/>. 
 */

using System.Collections.Generic;
using System.Linq;
using OpenDiablo2.Common;
using OpenDiablo2.Common.Enums;
using OpenDiablo2.Common.Interfaces;
using OpenDiablo2.Common.Models;

namespace OpenDiablo2.Core
{
    public sealed class ResourceManager : IResourceManager
    {
        static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);

        private readonly ICache cache;
        private readonly IMPQProvider mpqProvider;
        private readonly IEngineDataManager engineDataManager;

        public Dictionary<string, List<AnimationData>> Animations { get; private set; }

        public ResourceManager(ICache cache, IMPQProvider mpqProvider, IEngineDataManager engineDataManager)
        {
            this.cache = cache;
            this.mpqProvider = mpqProvider;
            this.engineDataManager = engineDataManager;

            Animations = AnimationData.LoadFromStream(mpqProvider.GetStream(ResourcePaths.AnimationData));
        }

        public ImageSet GetImageSet(string resourcePath)
            => cache.AddOrGetExisting($"ImageSet::{resourcePath}", () => ImageSet.LoadFromStream(mpqProvider.GetStream(resourcePath)));

        public MPQFont GetMPQFont(string resourcePath)
            => cache.AddOrGetExisting($"Font::{resourcePath}", () => MPQFont.LoadFromStream(mpqProvider.GetStream($"{resourcePath}.DC6"), mpqProvider.GetStream($"{resourcePath}.tbl")));

        public MPQDS1 GetMPQDS1(string resourcePath, LevelPreset level, LevelDetail levelDetail, LevelType levelType)
            => cache.AddOrGetExisting($"DS1::{resourcePath}::{level}::{levelDetail}::{levelType}", ()
                => new MPQDS1(mpqProvider.GetStream(resourcePath), level, levelDetail, levelType, engineDataManager, this) { MapFile = resourcePath });

        public Palette GetPalette(string paletteFile)
            => cache.AddOrGetExisting($"Palette::{paletteFile}", () =>
            {
                var paletteNameParts = paletteFile.Split('\\');
                var paletteName = paletteNameParts[paletteNameParts.Count() - 2];
                return Palette.LoadFromStream(mpqProvider.GetStream(paletteFile), paletteName);
            });

        public MPQDT1 GetMPQDT1(string resourcePath)
            => cache.AddOrGetExisting($"DT1::{resourcePath}", () => new MPQDT1(mpqProvider.GetStream(resourcePath)));

        public MPQCOF GetPlayerAnimation(eHero hero, eWeaponClass weaponClass, eMobMode mobMode)
            => cache.AddOrGetExisting($"COF::{hero}::{weaponClass}::{mobMode}", () =>
            {
                var path = $"{ResourcePaths.PlayerAnimationBase}\\{hero.ToToken()}\\COF\\{hero.ToToken()}{mobMode.ToToken()}{weaponClass.ToToken()}.cof";
                return MPQCOF.Load(mpqProvider.GetStream(path), Animations, hero, weaponClass, mobMode);
            });

        public MPQDCC GetPlayerDCC(MPQCOF.COFLayer cofLayer, eArmorType armorType, Palette palette)
        {
            // TODO: We need to cache this...
            byte[] binaryData;

            var streamPath = cofLayer.GetDCCPath(armorType);
            using (var stream = mpqProvider.GetStream(streamPath))
            {
                if (stream == null)
                {
                    log.Error($"Could not load Player DCC: {streamPath}");
                    return null;
                }

                binaryData = new byte[stream.Length];
                stream.Read(binaryData, 0, (int)stream.Length);
            }
            var result = new MPQDCC(binaryData, palette);
            return result;
        }

        /*
            => cache.AddOrGetExisting($"DCC::{cofLayer}::{armorType}::{palette.Name}", () =>
            {
                byte[] binaryData;

                using (var stream = mpqProvider.GetStream(cofLayer.GetDCCPath(armorType)))
                {
                    if (stream == null)
                        return null;

                    binaryData = new byte[stream.Length];
                    stream.Read(binaryData, 0, (int)stream.Length);
                }
                var result = new MPQDCC(binaryData, palette);
                return result;
            });
        /*
         */
    }
}