DataSetDisplay.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/myqt/widgets/mapdisplay/DataSetDisplay.cpp

  \brief A map display for a dataset.
*/

// TerraLib
#include "../../../dataaccess/dataset/DataSet.h"
#include "../../../dataaccess/dataset/DataSetType.h"
#include "../../../dataaccess/datasource/DataSource.h"
#include "../../../dataaccess/datasource/DataSourceManager.h"
#include "../../../dataaccess/datasource/DataSourceTransactor.h"
#include "../../../dataaccess/utils/Utils.h"
#include "../../../geometry/Envelope.h"
#include "../../../geometry/Geometry.h"
#include "../../../geometry/GeometryProperty.h"
#include "../../../maptools/Utils.h"
#include "../../../se/CoverageStyle.h"
#include "../../../se/Utils.h"
#include "../canvas/Canvas.h"
#include "../Utils.h"
#include "DataSetDisplay.h"

// STL
#include <cassert>

// Qt
#include <QMessageBox>
#include <QResizeEvent>

te::qt::widgets::DataSetDisplay::DataSetDisplay(QWidget * parent, Qt::WindowFlags f)
  : QFrame(parent, f)
{
}

te::qt::widgets::DataSetDisplay::~DataSetDisplay() = default;

void te::qt::widgets::DataSetDisplay::draw(const te::da::DataSetTypePtr& dataset, const te::da::DataSourceInfoPtr& ds)
{
  if(ds.get() == nullptr)
    return;

  try
  {
    te::da::DataSourcePtr auxDs = te::da::DataSourceManager::getInstance().get(ds->getId(), ds->getType(), ds->getConnInfo());

    if(dataset->hasRaster())
    {
      std::unique_ptr<te::da::DataSourceTransactor> t = auxDs->getTransactor();

      draw(dataset, auxDs, t->getDataSet(dataset->getName()).get());
    }
    else
      draw(dataset, auxDs);
    
  }
  catch(const std::exception& e)
  {
    QMessageBox::warning(this,
                         tr("TerraLib Qt Components"),
                         tr(e.what()));
  }
  catch(...)
  {
    QMessageBox::warning(this,
                         tr("TerraLib Qt Components"),
                         tr("Unknown error when displaying dataset!"));
  }
}

void te::qt::widgets::DataSetDisplay::draw(const te::da::DataSetTypePtr& dataset, const te::da::DataSourcePtr& ds, te::da::DataSet* datasetData)
{
  if((dataset.get() == nullptr) || (ds.get() == nullptr))
    return;

  if(m_canvas.get() == nullptr)
    m_canvas.reset(new Canvas(width(), height()));

  if(dataset->size() == 0)
    te::da::LoadProperties(dataset.get(), ds->getId());

  if(dataset->hasRaster())
  {
    std::size_t rpos = te::da::GetFirstPropertyPos(datasetData, te::dt::RASTER_TYPE);

    std::unique_ptr<te::rst::Raster> rst = datasetData->getRaster(rpos);

    te::gm::Envelope* mbr = rst->getExtent();

    if (mbr == nullptr)
      return;

    m_canvas->calcAspectRatio(mbr);

    m_canvas->setWindow(mbr->getLowerLeftX(), mbr->getLowerLeftY(), mbr->getUpperRightX(), mbr->getUpperRightY());

    te::se::Style* style = te::se::CreateCoverageStyle(rst->getNumberOfBands());
    te::se::CoverageStyle* cs = dynamic_cast<te::se::CoverageStyle*>(style);

    te::gm::Envelope* rstEnv = rst->getExtent();
    int rstSrid = rst->getSRID();

    bool cancel = false;

    te::map::DrawRaster(rst.get(), m_canvas.get(), *rstEnv, rstSrid, *rstEnv, rstSrid, cs, nullptr, 0, &cancel);
  }
  else if(dataset->hasGeom())
  {

    te::gm::GeometryProperty* gp = te::da::GetFirstGeomProperty(dataset.get());

    std::unique_ptr<te::gm::Envelope> mbr(te::da::GetExtent(dataset->getName(), gp->getName(), ds->getId()));

    if(mbr.get() == nullptr)
      return;

    m_canvas->calcAspectRatio(mbr.get());

    m_canvas->setWindow(mbr->getLowerLeftX(), mbr->getLowerLeftY(), mbr->getUpperRightX(), mbr->getUpperRightY());

    switch(gp->getGeometryType())
    {
      case te::gm::PolygonType:
      case te::gm::PolygonZType:
      case te::gm::PolygonMType:
      case te::gm::PolygonZMType:
      case te::gm::MultiPolygonType:
      case te::gm::MultiPolygonZType:
      case te::gm::MultiPolygonMType:
      case te::gm::MultiPolygonZMType:
      {
        Config2DrawPolygons(m_canvas.get(), Qt::red, Qt::black);
      }
      break;

      case te::gm::LineStringType:
      case te::gm::LineStringZType:
      case te::gm::LineStringMType:
      case te::gm::LineStringZMType:
      case te::gm::MultiLineStringType:
      case te::gm::MultiLineStringZType:
      case te::gm::MultiLineStringMType:
      case te::gm::MultiLineStringZMType:
      {
        Config2DrawLines(m_canvas.get(), Qt::black);
      }
      break;

      case te::gm::PointType:
      case te::gm::PointZType:
      case te::gm::PointMType:
      case te::gm::PointZMType:
      case te::gm::MultiPointType:
      case te::gm::MultiPointZType:
      case te::gm::MultiPointMType:
      case te::gm::MultiPointZMType:
      {
        Config2DrawPoints(m_canvas.get(), "circle", 1, Qt::black, Qt::transparent, 1);
      }
      break;

      default:
        break;
    }

    if(datasetData)
    {
      std::size_t gpos = te::da::GetFirstPropertyPos(datasetData, te::dt::GEOMETRY_TYPE);

      while(datasetData->moveNext())
      {
        if (datasetData->isNull(gpos))
          continue;

        std::unique_ptr<te::gm::Geometry> g(datasetData->getGeometry(gpos));

        if (g.get())
          m_canvas->draw(g.get());
      }
    }
    else
    {
      std::unique_ptr<te::da::DataSet> feature(ds->getDataSet(dataset->getName()));

      std::size_t gpos = te::da::GetFirstPropertyPos(feature.get(), te::dt::GEOMETRY_TYPE);

      while(feature->moveNext())
      {
        if (feature->isNull(gpos))
          continue;

        std::unique_ptr<te::gm::Geometry> g(feature->getGeometry(gpos));

        if (g.get())
          m_canvas->draw(g.get());
      }
    }
  }
  else
  {
    m_canvas->setBackgroundColor(te::color::RGBAColor(0, 0, 0, 255)); //  fill black
  }

  repaint();
}

void te::qt::widgets::DataSetDisplay::clear()
{
  if(m_canvas.get() == nullptr)
    m_canvas.reset(new Canvas(width(), height()));

  m_canvas->setBackgroundColor(te::color::RGBAColor(255, 255, 255, 0));
  m_canvas->clear();

  repaint();
}

void te::qt::widgets::DataSetDisplay::paintEvent(QPaintEvent* /*e*/)
{
  if(m_canvas.get() == nullptr)
    return;

  QPainter painter(this);
  
  painter.drawPixmap(0, 0, *(m_canvas->getPixmap()));

  painter.end();
 
}

void te::qt::widgets::DataSetDisplay::resizeEvent(QResizeEvent* e)
{
  assert(e);

  if(m_canvas.get() == nullptr)
    return;

  m_canvas->resize(e->size().width(), e->size().height());

  QFrame::resizeEvent(e);
}