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
    // clang-format off
    SIP_PYOBJECT __repr__();
    % MethodCode
    QString str = u"<QgsPointCloudAttribute: %1 (%2)>"_s.arg( sipCpp->name() ).arg( sipCpp->displayType() );
    sipRes = PyUnicode_FromString( str.toUtf8().constData() );
    % End
// clang-format on
#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