#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab 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.
 *
 * HeuristicLab 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 HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using HeuristicLab.PluginInfrastructure;
using Google.ProtocolBuffers;
using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;

namespace HeuristicLab.Persistence {
  public sealed class StaticCache {
    private static object locker = new object();

    private Dictionary<Guid, ITransformer> guid2Transformer;
    private Dictionary<ITransformer, Guid> transformer2Guid;
    private Dictionary<Guid, Type> guid2Type;
    private Dictionary<Type, Guid> type2Guid;
    private Dictionary<Type, TypeInfo> typeInfos;
    private ExtensionRegistry extensionRegistry;

    internal StaticCache() {
      Initialize();
    }
    public void Initialize() {
      lock (locker) {
        guid2Transformer = new Dictionary<Guid, ITransformer>();
        transformer2Guid = new Dictionary<ITransformer, Guid>();
        guid2Type = new Dictionary<Guid, Type>();
        type2Guid = new Dictionary<Type, Guid>();
        typeInfos = new Dictionary<Type, TypeInfo>();
        extensionRegistry = ExtensionRegistry.CreateInstance();

        foreach (var transformer in ApplicationManager.Manager.GetInstances<ITransformer>().OrderBy(x => x.Priority)) {
          guid2Transformer.Add(transformer.Guid, transformer);
          transformer2Guid.Add(transformer, transformer.Guid);
        }

        RegisterType(new Guid("94AD8522-3F55-4580-A6F8-2D2AAEDD4B8C"), typeof(bool));
        RegisterType(new Guid("4A1C0FD5-423D-4F96-AB22-A496578C25AC"), typeof(byte));
        RegisterType(new Guid("C4B00F0B-FED7-439F-B1B2-8A0048B64882"), typeof(sbyte));
        RegisterType(new Guid("9F451811-3DE1-43AB-8B74-D7E03851857B"), typeof(short));
        RegisterType(new Guid("46244D54-0145-49F7-9CF3-9CDB7FB5F240"), typeof(ushort));
        RegisterType(new Guid("1FDDE40C-09E3-491F-8FBB-32BB3C885E9E"), typeof(char));
        RegisterType(new Guid("EE3E0F9C-A5C2-4461-AF36-28BD8F26E6FB"), typeof(int));
        RegisterType(new Guid("69476D18-D285-43E9-BC7C-6CC9E9F2321E"), typeof(uint));
        RegisterType(new Guid("7C7BC5EC-F001-4BA0-9F85-50DCBAA9AE81"), typeof(long));
        RegisterType(new Guid("AC808D5A-63BB-457C-9B92-C9B83DA2B139"), typeof(ulong));
        RegisterType(new Guid("BF22653A-026C-4367-BBBA-2125AECF6C08"), typeof(float));
        RegisterType(new Guid("8B49821A-3ADC-4715-9DB1-08E2F3CFDF15"), typeof(double));
        RegisterType(new Guid("7BB386BF-6FD4-443D-A6C1-387096798C67"), typeof(DateTime));
        RegisterType(new Guid("724A2D49-7E7B-455B-BBA9-4214C64E8A21"), typeof(TimeSpan));
        foreach (var asm in AppDomain.CurrentDomain.GetAssemblies()) {
          foreach (var t in asm.GetTypes().Where(x => StorableClassAttribute.IsStorableClass(x)))
            RegisterType(StorableClassAttribute.GetStorableClassAttribute(t).Guid, t);
        }

        RegisterExtension(BoolBox.Bool);
        RegisterExtension(IntBox.Int);
        RegisterExtension(LongBox.Long);
        RegisterExtension(UnsignedIntBox.UnsignedInt);
        RegisterExtension(UnsignedLongBox.UnsignedLong);
        RegisterExtension(FloatBox.Float);
        RegisterExtension(DoubleBox.Double);
        RegisterExtension(StringBox.String);
        RegisterExtension(BytesBox.Bytes);
        RegisterExtension(BoolArrayBox.BoolArray);
        RegisterExtension(IntArrayBox.IntArray);
        RegisterExtension(LongArrayBox.LongArray);
        RegisterExtension(UnsignedIntArrayBox.UnsignedIntArray);
        RegisterExtension(UnsignedLongArrayBox.UnsignedLongArray);
        RegisterExtension(FloatArrayBox.FloatArray);
        RegisterExtension(DoubleArrayBox.DoubleArray);
        RegisterExtension(StringArrayBox.StringArray);
        RegisterExtension(MatrixBox.Matrix);
        RegisterExtension(BoolMatrixBox.BoolMatrix);
        RegisterExtension(IntMatrixBox.IntMatrix);
        RegisterExtension(DictionaryBox.Dictionary);
      }
    }

    public void RegisterType(Guid guid, Type type) {
      lock (locker) {
        guid2Type.Add(guid, type);
        type2Guid.Add(type, guid);
      }
    }
    public void RegisterExtension<TExtension>(GeneratedExtensionBase<TExtension> extension) {
      extensionRegistry.Add(extension);
    }

    public ITransformer GetTransformer(Guid guid) {
      return guid2Transformer[guid];
    }
    public Guid GetGuid(ITransformer transformer) {
      return transformer2Guid[transformer];
    }
    public Type GetType(Guid guid) {
      return guid2Type[guid];
    }
    public Guid GetGuid(Type type) {
      return type2Guid[type];
    }
    public TypeInfo GetTypeInfo(Type type) {
      lock (locker) {
        TypeInfo typeInfo;
        if (!typeInfos.TryGetValue(type, out typeInfo)) {
          var transformer = guid2Transformer.Values.First(x => x.CanTransformType(type));
          typeInfo = new TypeInfo(type, transformer);
          typeInfos.Add(type, typeInfo);
        }
        return typeInfo;
      }
    }
    public ExtensionRegistry GetExtensionRegistry() {
      return extensionRegistry;
    }
  }
}