qgscopcpointcloudindex.h

Go to the documentation of this file.

/***************************************************************************
                         qgscopcpointcloudindex.h
                         --------------------
    begin                : March 2022
    copyright            : (C) 2022 by Belgacem Nedjima
    email                : belgacem dot nedjima at gmail dot com
 ***************************************************************************/
 
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#ifndef QGSCOPCPOINTCLOUDINDEX_H
#define QGSCOPCPOINTCLOUDINDEX_H
 
#include <fstream>
#include <optional>
 
#include "lazperf/vlr.hpp"
#include "qgslazinfo.h"
#include "qgspointcloudindex.h"
#include "qgspointcloudstatistics.h"
 
#include <QFile>
#include <QHash>
#include <QList>
#include <QString>
#include <QStringList>
#include <QVector>
 
#define SIP_NO_FILE

 
class QgsCoordinateReferenceSystem;
 
class CORE_EXPORT QgsCopcPointCloudIndex : public QgsAbstractPointCloudIndex
{
  public:
    explicit QgsCopcPointCloudIndex();
    ~QgsCopcPointCloudIndex() override;
 
    void load( const QString &urlString, const QString &authcfg = QString() ) override;
 
    bool hasNode( QgsPointCloudNodeId n ) const override;
    QgsPointCloudNode getNode( QgsPointCloudNodeId id ) const override;
 
    std::unique_ptr< QgsPointCloudBlock> nodeData( QgsPointCloudNodeId n, const QgsPointCloudRequest &request ) override;
    QgsPointCloudBlockRequest *asyncNodeData( QgsPointCloudNodeId n, const QgsPointCloudRequest &request ) override;

    const QByteArray rawNodeData( QgsPointCloudNodeId n ) const;
 
    QgsCoordinateReferenceSystem crs() const override;
    qint64 pointCount() const override;
    QVariantMap originalMetadata() const override { return mOriginalMetadata; }
 
    bool isValid() const override;
    Qgis::PointCloudAccessType accessType() const override { return mAccessType; };

    bool writeStatistics( QgsPointCloudStatistics &stats ) override;

    QgsPointCloudStatistics metadataStatistics() const override;

    QVariantMap extraMetadata() const override;
 
    bool needsHierarchyFetching( const QgsPointCloudNodeId &n ) const override;
 
  protected:
    bool loadSchema( QgsLazInfo &lazInfo );
    bool loadHierarchy() const;

    bool fetchNodeHierarchy( QgsPointCloudNodeId n ) const;

    void fetchHierarchyPage( uint64_t offset, uint64_t byteSize ) const;
 
    void populateHierarchy( const char *hierarchyPageData, uint64_t byteSize ) const;

    QByteArray readRange( uint64_t offset, uint64_t length ) const;
 
    QByteArray fetchCopcStatisticsEvlrData() const;
 
    void reset();
 
    bool mIsValid = false;
    Qgis::PointCloudAccessType mAccessType = Qgis::PointCloudAccessType::Local;
    mutable std::ifstream mCopcFile;
    mutable lazperf::copc_info_vlr mCopcInfoVlr;
    mutable QHash<QgsPointCloudNodeId, QPair<uint64_t, int32_t>> mHierarchyNodePos; 
 
    mutable QMutex mFileMutex;
 
    QVariantMap mOriginalMetadata;
    mutable std::optional<QgsPointCloudStatistics> mStatistics;
 
    std::unique_ptr<QgsLazInfo> mLazInfo = nullptr;
 
    friend class QgsPointCloudLayerEditUtils;
    friend class QgsPointCloudEditingIndex;
};

#endif // QGSCOPCPOINTCLOUDINDEX_H