/**
 * 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
{
  void BasicApplicationContext::UpdateThread(BasicApplicationContext* that)
  {
    while (!that->stopped_)
    {
      {
        ViewportLocker locker(*that);
        locker.GetViewport().UpdateContent();
      }
      
      boost::this_thread::sleep(boost::posix_time::milliseconds(that->updateDelay_));
    }
  }
  

  BasicApplicationContext::BasicApplicationContext(OrthancSynchronousWebService& orthanc) :
    orthanc_(orthanc),
    stopped_(true),
    updateDelay_(100)   // By default, 100ms between each refresh of the content
  {
  }


  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(GetWebService().GetConnection(), 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(GetWebService().GetConnection(), 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();
    }

    if (viewport_.HasUpdateContent())
    {
      stopped_ = false;
      updateThread_ = boost::thread(UpdateThread, this);
    }

    viewport_.Start();
  }


  void BasicApplicationContext::Stop()
  {
    stopped_ = true;
    
    if (updateThread_.joinable())
    {
      updateThread_.join();
    }
    
    for (Volumes::iterator it = volumes_.begin(); it != volumes_.end(); ++it)
    {
      assert(*it != NULL);
      (*it)->Stop();
    }
  }
}