ConnectedDataSet.h

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/postgis/ConnectedDataSet.h

  \brief Implementation of a connected dataset for the PostGIS driver.
*/

#ifndef __TERRALIB_POSTGIS_INTERNAL_CONNECTED_DATASET_H
#define __TERRALIB_POSTGIS_INTERNAL_CONNECTED_DATASET_H

// TerraLib
#include "../dataaccess/dataset/DataSet.h"
#include "Connection.h"
#include "Config.h"

// Forward declaration for libpq
//extern "C"
//{
//  struct pg_conn;
//  typedef struct pg_conn PGconn;
//  struct pg_result;
//  typedef struct pg_result PGresult;
//}

namespace te
{
// Forward declaration
  namespace da
  {
    class GeometryProperty;
  }

  namespace pgis
  {
    // Forward declaration
    class DataSource;

    /*!
      \class ConnectedDataSet

      \brief Implementation of a connected dataset for the PostGIS driver.

      \sa te::da::ConnectedDataSet
    */
    class TEPGISEXPORT ConnectedDataSet : public te::da::DataSet
    {
      public:
       /*!
         \brief Constructor.

         \param result        The internal PGresult.
         \param dataSource    The DataSource to get the connection.
         \param connectionId  The connection id used by this dataset.
         \param id_cursor     The id for the current cursor.
         \param size          The number of rows the layer has.
         \param ptypes        The types of each property.
         \param timeIsInteger A flag that indicates if the postgis stores,
         internally, the time and timestamp as an integer.
       */
        ConnectedDataSet(PGresult* result,
                        DataSource* dataSource,
                        const int& connectionId,
                        const std::string& id_cursor,
                        const int& size,
                        const std::vector<int>& ptypes,
                        bool timeIsInteger = true,
                        int newFetchSize = 5000);

        /*! \brief The destructor will clear the internal PGresult. */
        ~ConnectedDataSet();

        te::common::TraverseType getTraverseType() const;

        te::common::AccessPolicy getAccessPolicy() const;

        std::size_t getNumProperties() const;

        int getPropertyDataType(std::size_t i) const;

        std::string getPropertyName(std::size_t i) const;

        std::string getDatasetNameOfProperty(std::size_t i) const;

        bool isEmpty() const;

        bool isConnected() const;

        std::size_t size() const;

        std::unique_ptr<te::gm::Envelope> getExtent(std::size_t i);

        bool moveNext();

        bool movePrevious();

        bool moveBeforeFirst();

        bool moveFirst();

        bool moveLast();

        bool move(std::size_t i);

        bool isAtBegin() const;

        bool isBeforeBegin() const;

        bool isAtEnd() const;

        bool isAfterEnd() const;

        bool isPositionValid() const;

        char getChar(std::size_t i) const;

        unsigned char getUChar(std::size_t i) const;

        boost::int16_t getInt16(std::size_t i) const;

        boost::int32_t getInt32(std::size_t i) const;

        boost::int64_t getInt64(std::size_t i) const;

        bool getBool(std::size_t i) const;

        float getFloat(std::size_t i) const;

        double getDouble(std::size_t i) const;

        std::string getNumeric(std::size_t i) const;

        std::string getString(std::size_t i) const;

        std::unique_ptr<te::dt::ByteArray> getByteArray(std::size_t i) const;

        std::unique_ptr<te::gm::Geometry> getGeometry(std::size_t i) const;

        std::unique_ptr<te::rst::Raster> getRaster(std::size_t i) const;

        std::unique_ptr<te::dt::DateTime> getDateTime(std::size_t i) const; 

        std::unique_ptr<te::dt::Array> getArray(std::size_t i) const;

        bool isNull(std::size_t i) const;

        /** @name PostGIS Extended Methods
         *  Methods that exists only in the PostGIS DataSet implementation.
         */
        //@{

        /*!
          \brief It returns the internal pg result.

          \return The internal pg result.
        */
        PGresult* getPGResult() const { return m_result; }

        //Connection* getConnection() const { return m_conn; }

        //@}

      protected:

        int m_iFetch;               //!< The index of the current row of current fetch.
        int m_iLayer;               //!< The index of total layer of the current row.
        int m_size;                 //!< The number of datasets in the collection.
        PGresult* m_result;         //!< The internal buffer with the result query.
        DataSource* m_ds;           //!< The DataSource to get the connection.
        int m_connectionId;         //!< The connection id used by this dataset.
        int m_fetchSize;            //!< The size of cursor fetch.
        std::string m_idcursor;     //!< The id of current cursor.
        std::vector<int> m_ptypes;  //!< The list of property types.
        te::gm::Envelope* m_mbr;    //!< The dataset extent.
        bool m_timeIsInteger;       //!< It indicates if the postgis stores, internally, the time and timestamp as an integer.
        int m_newFetchSize;         //!< The size to be used every time we need to fetch more date
    };

  } // end namespace pgis
}   // end namespace te


#endif  // __TERRALIB_POSTGIS_INTERNAL_CONNECTED_DATASET_H