

#ifndef INCLUDED_PDAL_DRIVER_PGPOINTCLOUD_READER_HPP
#define INCLUDED_PDAL_DRIVER_PGPOINTCLOUD_READER_HPP

#include <pdal/Reader.hpp>
#include <pdal/ReaderIterator.hpp>
#include <pdal/PointBuffer.hpp>

#include <pdal/drivers/pgpointcloud/common.hpp>
#include <boost/scoped_ptr.hpp>
#include <boost/scoped_array.hpp>

#include <vector>


namespace pdal
{
namespace drivers
{
namespace pgpointcloud
{


class PDAL_DLL Reader : public pdal::Reader
{
public:
    SET_STAGE_NAME("drivers.pgpointcloud.reader", "PostgreSQL Pointcloud Database Reader")
    SET_STAGE_LINK("http://pdal.io/stages/drivers.pgpointcloud.reader.html")
    
    Reader(const Options&);
    ~Reader();

    virtual void initialize();
    static Options getDefaultOptions();

    bool supportsIterator(StageIteratorType t) const
    {
        if (t == StageIterator_Sequential) return true;
        return false;
    }
    
    virtual boost::uint64_t getNumPoints() const;
    boost::uint64_t getMaxPoints() const;
    std::string getDataQuery() const;
    void getSession() const;

    pdal::StageSequentialIterator* createSequentialIterator(PointBuffer& buffer) const;


private:

    Reader& operator=(const Reader&); // not implemented
    Reader(const Reader&); // not implemented

    pdal::SpatialReference fetchSpatialReference() const;
    boost::uint32_t fetchPcid() const;
    pdal::Schema fetchSchema() const;

    PGconn* m_session;
    std::string m_connection;
    std::string m_table_name;
    std::string m_schema_name;
    std::string m_column_name;
    std::string m_where;
    mutable boost::uint32_t m_pcid;
    mutable boost::uint64_t m_cached_point_count;
    mutable boost::uint64_t m_cached_max_points;

}; // pdal.drivers.pgpointcloud.Reader


namespace iterators
{
namespace sequential
{

typedef boost::shared_ptr<PointBuffer> BufferPtr;


class Iterator : public pdal::StageSequentialIterator
{
public:
    Iterator(const pdal::drivers::pgpointcloud::Reader& reader, PointBuffer& buffer);
    ~Iterator();

protected:


private:
    //
    // Methods
    //
    const pdal::drivers::pgpointcloud::Reader& getReader() const;

    // Skip count points, return number of points skipped
    boost::uint64_t skipImpl(boost::uint64_t count);

    // Fill the provided pointbuffer, return the number of points written
    boost::uint32_t readBufferImpl(PointBuffer& data);

    // True when there are no more points to read
    bool atEndImpl() const;

    // Internal functions for managing scroll cursor
    bool CursorSetup();
    bool CursorTeardown();
    bool NextBuffer();

    //
    // Members
    //
    const pdal::drivers::pgpointcloud::Reader& m_reader;
    bool m_at_end;
    pdal::PointBuffer* m_buffer;
    boost::uint64_t m_buffer_position;

    bool m_cursor;
    std::string m_patch_hex;
    boost::uint32_t m_patch_npoints;
    PGconn* m_session;
    pdal::schema::DimensionMap* m_dimension_map;

    boost::uint32_t m_cur_row;
    boost::uint32_t m_cur_nrows;
    PGresult* m_cur_result;

}; // pdal.drivers.pgpointcloud.sequential.iterators.Iterator


} // sequential
} // iterators


} // pgpointcloud
} // driver
} // pdal


#endif // INCLUDED_PDAL_DRIVER_PGPOINTCLOUD_READER_HPP
