qgspointcloudattribute.h

Go to the documentation of this file.

/***************************************************************************
                         qgspointcloudattribute.h
                         ---------------------
    begin                : October 2020
    copyright            : (C) 2020 by Peter Petrik
    email                : zilolv 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 QGSPOINTCLOUDATTRIBUTE_H
#define QGSPOINTCLOUDATTRIBUTE_H
 
#include "qgis.h"
#include "qgis_core.h"
#include "qgsfields.h"
#include "qgsvector3d.h"
 
#include <QSet>
#include <QString>
#include <QVector>
 
using namespace Qt::StringLiterals;
 
class QgsPointCloudAttributeCollection;

class CORE_EXPORT QgsPointCloudAttribute
{
  public:
    enum DataType
    {
      Char, 
      UChar, 
      Short, 
      UShort, 
      Int32, 
      UInt32, 
      Int64, 
      UInt64, 
      Float, 
      Double, 
    };
 
    QgsPointCloudAttribute();
    QgsPointCloudAttribute( const QString &name, DataType type );

    QString name() const { return mName; }

    int size() const { return mSize; }

    DataType type() const { return mType; }

    QMetaType::Type variantType() const;

    QString displayType() const;

    double convertValueToDouble( const char *ptr ) const SIP_SKIP;

    static bool isNumeric( DataType type );
 
#ifdef SIP_RUN
    SIP_PYOBJECT __repr__();
    % MethodCode
    QString str = u"<QgsPointCloudAttribute: %1 (%2)>"_s.arg( sipCpp->name() ).arg( sipCpp->displayType() );
    sipRes = PyUnicode_FromString( str.toUtf8().constData() );
    % End
#endif

    static void getPointXYZ( const char *ptr, int i, std::size_t pointRecordSize, int xOffset, QgsPointCloudAttribute::DataType xType,
                             int yOffset, QgsPointCloudAttribute::DataType yType,
                             int zOffset, QgsPointCloudAttribute::DataType zType,
                             const QgsVector3D &indexScale, const QgsVector3D &indexOffset, double &x, double &y, double &z ) SIP_SKIP;

    static QVariantMap getAttributeMap( const char *data, std::size_t recordOffset, const QgsPointCloudAttributeCollection &attributeCollection ) SIP_SKIP;
 
  private:
    void updateSize();
 
    QString mName;
    int mSize = 0;
    DataType mType = DataType::Char;
};

class CORE_EXPORT QgsPointCloudAttributeCollection
{
  public:
 
    QgsPointCloudAttributeCollection();
    QgsPointCloudAttributeCollection( const QVector<QgsPointCloudAttribute> &attributes );
    void push_back( const QgsPointCloudAttribute &attribute );

    void extend( const QgsPointCloudAttributeCollection &otherCollection, const QSet<QString> &matchingNames );

    QVector<QgsPointCloudAttribute> attributes() const;

    int count() const { return mAttributes.size(); }

    const QgsPointCloudAttribute &at( int index ) const { return mAttributes.at( index ); }

    const QgsPointCloudAttribute *find( const QString &attributeName, int &offset ) const;

    int indexOf( const QString &name ) const;

    int pointRecordSize() const { return mSize; }

    QgsFields toFields() const;
 
  private:
    int mSize = 0;
    QVector<QgsPointCloudAttribute> mAttributes;
 
    struct CachedAttributeData
    {
      int index;
      int offset;
      CachedAttributeData( int index, int offset )
        : index( index )
        , offset( offset )
      {}
    };
 
    QMap< QString, CachedAttributeData > mCachedAttributes;
};
 
#endif // QGSPOINTCLOUDATTRIBUTE_H