/**
 * Stone of Orthanc
 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
 * Department, University Hospital of Liege, Belgium
 * Copyright (C) 2017 Osimis, 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/>.
 **/


#include "BasicApplicationContext.h"

#include "../../Framework/Toolbox/OrthancSeriesLoader.h"
#include "../../Framework/Volumes/VolumeImageSimplePolicy.h"
#include "../../Framework/Volumes/VolumeImageProgressivePolicy.h"

namespace OrthancStone
{
  BasicApplicationContext::BasicApplicationContext(OrthancPlugins::IOrthancConnection& orthanc) :
    orthanc_(orthanc)
  {
  }


  BasicApplicationContext::~BasicApplicationContext()
  {
    for (Interactors::iterator it = interactors_.begin(); it != interactors_.end(); ++it)
    {
      assert(*it != NULL);
      delete *it;
    }

    for (Volumes::iterator it = volumes_.begin(); it != volumes_.end(); ++it)
    {
      assert(*it != NULL);
      delete *it;
    }

    for (StructureSets::iterator it = structureSets_.begin(); it != structureSets_.end(); ++it)
    {
      assert(*it != NULL);
      delete *it;
    }
  }


  IWidget& BasicApplicationContext::SetCentralWidget(IWidget* widget)   // Takes ownership
  {
    viewport_.SetCentralWidget(widget);
    return *widget;
  }


  VolumeImage& BasicApplicationContext::AddSeriesVolume(const std::string& series,
                                                        bool isProgressiveDownload,
                                                        size_t downloadThreadCount)
  {
    std::auto_ptr<VolumeImage> volume(new VolumeImage(new OrthancSeriesLoader(orthanc_, series)));

    if (isProgressiveDownload)
    {
      volume->SetDownloadPolicy(new VolumeImageProgressivePolicy);
    }
    else
    {
      volume->SetDownloadPolicy(new VolumeImageSimplePolicy);
    }

    volume->SetThreadCount(downloadThreadCount);

    VolumeImage& result = *volume;
    volumes_.push_back(volume.release());

    return result;
  }


  DicomStructureSet& BasicApplicationContext::AddStructureSet(const std::string& instance)
  {
    std::auto_ptr<DicomStructureSet> structureSet(new DicomStructureSet(orthanc_, instance));

    DicomStructureSet& result = *structureSet;
    structureSets_.push_back(structureSet.release());

    return result;
  }


  IWorldSceneInteractor& BasicApplicationContext::AddInteractor(IWorldSceneInteractor* interactor)
  {
    if (interactor == NULL)
    {
      throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
    }

    interactors_.push_back(interactor);

    return *interactor;
  }


  void BasicApplicationContext::Start()
  {
    for (Volumes::iterator it = volumes_.begin(); it != volumes_.end(); ++it)
    {
      assert(*it != NULL);
      (*it)->Start();
    }

    viewport_.Start();
  }


  void BasicApplicationContext::Stop()
  {
    viewport_.Stop();

    for (Volumes::iterator it = volumes_.begin(); it != volumes_.end(); ++it)
    {
      assert(*it != NULL);
      (*it)->Stop();
    }
  }
}