/**
 * Stone of Orthanc
 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
 * Department, University Hospital of Liege, Belgium
 * Copyright (C) 2017-2019 Osimis S.A., Belgium
 *
 * This program is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero 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
 * Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 **/

#pragma once

#include <boost/shared_ptr.hpp>

#include <map>
#include <string>
#include <vector>

namespace OrthancStone
{
  class OrthancSeriesVolumeProgressiveLoader;
  class DicomVolumeImageMPRSlicer;
  class DicomStructureSetLoader;
#ifdef BGO_ENABLE_DICOMSTRUCTURESETLOADER2
  class DicomStructureSetLoader2;
  class DicomStructureSetSlicer2;
  class DicomStructureSet2;
#endif 
  //BGO_ENABLE_DICOMSTRUCTURESETLOADER2
  class OrthancMultiframeVolumeLoader;

#if ORTHANC_ENABLE_WASM == 1
  class WebAssemblyOracle;
#else
  class ThreadedOracle;
  namespace Deprecated
  {
    class LockingEmitter;
  }
#endif

  class LoaderCache
  {
  public:
#if ORTHANC_ENABLE_WASM == 1
    LoaderCache(WebAssemblyOracle& oracle);
#else
    LoaderCache(ThreadedOracle& oracle, Deprecated::LockingEmitter& lockingEmitter);
#endif

    boost::shared_ptr<OrthancSeriesVolumeProgressiveLoader>
      GetSeriesVolumeProgressiveLoader      (std::string seriesUuid);
    
    boost::shared_ptr<DicomVolumeImageMPRSlicer>
      GetMultiframeDicomVolumeImageMPRSlicer(std::string instanceUuid);

    boost::shared_ptr<OrthancMultiframeVolumeLoader>
      GetMultiframeVolumeLoader(std::string instanceUuid);

    boost::shared_ptr<DicomStructureSetLoader>
      GetDicomStructureSetLoader(
        std::string instanceUuid,
        const std::vector<std::string>& initiallyVisibleStructures);

#ifdef BGO_ENABLE_DICOMSTRUCTURESETLOADER2
    boost::shared_ptr<DicomStructureSetLoader2>
      GetDicomStructureSetLoader2(std::string instanceUuid);

    boost::shared_ptr<DicomStructureSetSlicer2>
      GetDicomStructureSetSlicer2(std::string instanceUuid);
#endif 
    //BGO_ENABLE_DICOMSTRUCTURESETLOADER2

    void ClearCache();

  private:
    
    void DebugDisplayObjRefCounts();
#if ORTHANC_ENABLE_WASM == 1
    WebAssemblyOracle& oracle_;
#else
    ThreadedOracle& oracle_;
    Deprecated::LockingEmitter& lockingEmitter_;
#endif

    std::map<std::string, boost::shared_ptr<OrthancSeriesVolumeProgressiveLoader> >
      seriesVolumeProgressiveLoaders_;
    std::map<std::string, boost::shared_ptr<OrthancMultiframeVolumeLoader> >
      multiframeVolumeLoaders_;
    std::map<std::string, boost::shared_ptr<DicomVolumeImageMPRSlicer> >
      dicomVolumeImageMPRSlicers_;
    std::map<std::string, boost::shared_ptr<DicomStructureSetLoader> >
      dicomStructureSetLoaders_;
#ifdef BGO_ENABLE_DICOMSTRUCTURESETLOADER2
    std::map<std::string, boost::shared_ptr<DicomStructureSetLoader2> >
      dicomStructureSetLoaders2_;
    std::map<std::string, boost::shared_ptr<DicomStructureSet2> >
      dicomStructureSets2_;
    std::map<std::string, boost::shared_ptr<DicomStructureSetSlicer2> >
      dicomStructureSetSlicers2_;
#endif 
    //BGO_ENABLE_DICOMSTRUCTURESETLOADER2
  };
}