src/terralib/ws/ogc/wms/qtplugin/Plugin.cpp

Go to the documentation of this file.

/*  Copyright (C) 2008 National Institute For Space Research (INPE) - Brazil.

    This file is part of the TerraLib - a Framework for building GIS enabled applications.

    TerraLib is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License,
    or (at your option) any later version.

    TerraLib 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with TerraLib. See COPYING. If not, write to
    TerraLib Team at <terralib-team@terralib.org>.
 */

/*!
  \file terralib/ws/ogc/wms/qtplugin/Plugin.cpp

  \brief Plugin implementation for the OGC Web Coverage Service (WMS) data source widget.

  \author Emerson Moraes
*/

#include "Plugin.h"

#include <iostream>

#include "../../../../core/translator/Translator.h"
#include "../../../../core/logger/Logger.h"
#include "../../../../dataaccess/datasource/DataSourceInfoManager.h"
#include "../../../../qt/widgets/datasource/core/DataSourceTypeManager.h"
#include "../../../../qt/af/events/ApplicationEvents.h"
#include "../../../../qt/af/events/LayerEvents.h"
#include "../../../../qt/af/ApplicationController.h"
#include "../../../../qt/widgets/layer/explorer/LayerItem.h"
#include "../../../../qt/widgets/layer/explorer/LayerItemView.h"
#include "../qt/WMSType.h"

void GetAllWMSLayers(te::qt::widgets::LayerItemView* view, const QModelIndex& parent,
                        QModelIndexList& layersIdx)
{
  QAbstractItemModel* model = view->model();
  int cS = model->rowCount(parent);

  for(int i = 0; i < cS; i++)
  {
    QModelIndex idx = model->index(i, 0, parent);

    te::qt::widgets::TreeItem* child =
        static_cast<te::qt::widgets::TreeItem*>(idx.internalPointer());

    if(child->getType() == "FOLDER")
      GetAllWMSLayers(view, idx, layersIdx);
    else if(child->getType() == "LAYER")
    {
      if(((te::qt::widgets::LayerItem*)child)->getLayer()->getType() == "OGCWMSLAYER")
        layersIdx.push_back(idx);
    }
  }
}

std::list<te::map::AbstractLayerPtr> GetLayers(const QModelIndexList& lst)
{
  std::list<te::map::AbstractLayerPtr> res;

  for(QModelIndexList::const_iterator it = lst.begin(); it != lst.end(); ++it)
  {
    QModelIndex idx = *it;
    te::qt::widgets::TreeItem* item = static_cast<te::qt::widgets::TreeItem*>(idx.internalPointer());
    res.push_back(((te::qt::widgets::LayerItem*)item)->getLayer());
  }

  return res;
}

te::ws::ogc::wms::qtplugin::Plugin::Plugin(const te::core::PluginInfo &pluginInfo)
  : te::core::CppPlugin(pluginInfo),
    m_delegate(nullptr)
{
}

te::ws::ogc::wms::qtplugin::Plugin::~Plugin() = default;

void te::ws::ogc::wms::qtplugin::Plugin::startup()
{
  if(m_initialized)
    return;

  te::qt::widgets::DataSourceTypeManager::getInstance().add(new te::ws::ogc::wms::qt::WMSType);

  TE_LOG_TRACE(TE_TR("TerraLib Qt OGC Web Map Service (WMS) widget startup!"));

  m_initialized = true;

  te::qt::af::AppCtrlSingleton::getInstance().addListener(this, te::qt::af::SENDER);

  updateDelegate(true);
}

void te::ws::ogc::wms::qtplugin::Plugin::shutdown()
{
  if(!m_initialized)
    return;

  te::da::DataSourceInfoManager::getInstance().removeByType("WMS2");
  te::qt::widgets::DataSourceTypeManager::getInstance().remove("WMS2");

  TE_LOG_TRACE(TE_TR("TerraLib Qt OGC Web Map Service (WMS) widget shutdown!"));

  m_initialized = false;

  updateDelegate(false);

  QModelIndexList wls;

  te::qt::af::evt::GetLayerExplorer e;

  emit triggered(&e);

  if(e.m_layerExplorer == nullptr)
    return;

  GetAllWMSLayers(e.m_layerExplorer, QModelIndex(), wls);

  if(!wls.isEmpty())
  {
    std::list<te::map::AbstractLayerPtr> lst = GetLayers(wls);

    e.m_layerExplorer->removeItems(wls);

    te::qt::af::evt::LayerRemoved evt(lst, false);

    emit triggered(&evt);
  }

  te::qt::af::AppCtrlSingleton::getInstance().removeListener(this);
}

void te::ws::ogc::wms::qtplugin::Plugin::updateDelegate(const bool &add)
{
  te::qt::af::evt::GetLayerExplorer e;

  emit triggered(&e);

  te::qt::widgets::LayerItemView* view = e.m_layerExplorer;

  if(view == nullptr)
    return;

  if(add)
  {
    m_delegate = new te::ws::ogc::wms::qt::WMSItemDelegate((QStyledItemDelegate*)view->itemDelegate(), this);
    view->setItemDelegate(m_delegate);
  }
  else
  {
    view->removeDelegate(m_delegate);
    delete m_delegate;
    m_delegate = nullptr;
  }
}

TERRALIB_PLUGIN_CALL_BACK_IMPL(te::ws::ogc::wms::qtplugin::Plugin)