qgsmeshlayerutils.h

Go to the documentation of this file.

/***************************************************************************
                         qgsmeshlayerutils.h
                         --------------------------
    begin                : August 2018
    copyright            : (C) 2018 by Martin Dobias
    email                : wonder dot sk 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 QGSMESHLAYERUTILS_H
#define QGSMESHLAYERUTILS_H
 
#define SIP_NO_FILE
 
#include "qgis_core.h"
#include "qgsrectangle.h"
#include "qgsmaptopixel.h"
#include "qgsmeshdataprovider.h"
#include "qgsmeshrenderersettings.h"
 
#include <QVector>
#include <QSize>
 
 
class QgsMeshTimeSettings;
class QgsTriangularMesh;
class QgsMeshDataBlock;
class QgsMesh3dAveragingMethod;
class QgsMeshDatasetValue;
class QgsMeshLayer;
 
class CORE_EXPORT QgsMeshLayerUtils
{
  public:
 
    static int datasetValuesCount( const QgsMesh *mesh, QgsMeshDatasetGroupMetadata::DataType dataType );
 
    static QgsMeshDatasetGroupMetadata::DataType datasetValuesType( const QgsMeshDatasetGroupMetadata::DataType &type );
 
    static QgsMeshDataBlock datasetValues(
      const QgsMeshLayer *meshLayer,
      QgsMeshDatasetIndex index,
      int valueIndex,
      int count );
 
    static QVector<QgsVector> griddedVectorValues(
      const QgsMeshLayer *meshLayer,
      const QgsMeshDatasetIndex index,
      double xSpacing,
      double ySpacing,
      const QSize &size,
      const QgsPointXY &minCorner );
 
    static QVector<double> calculateMagnitudes( const QgsMeshDataBlock &block );
 
    static void boundingBoxToScreenRectangle(
      const QgsMapToPixel &mtp,
      const QSize &outputSize,
      const QgsRectangle &bbox,
      int &leftLim, int &rightLim, int &topLim, int &bottomLim );
 
    static double interpolateFromVerticesData(
      double fraction,
      double val1, double val2
    );
 
    static QgsMeshDatasetValue interpolateFromVerticesData( double fraction,
        const QgsMeshDatasetValue &val1, const QgsMeshDatasetValue &val2
                                                          );
 
    static double interpolateFromVerticesData(
      const QgsPointXY &p1, const QgsPointXY &p2, const QgsPointXY &p3,
      double val1, double val2, double val3, const QgsPointXY &pt
    );
 
    static QgsVector interpolateVectorFromVerticesData(
      const QgsPointXY &p1, const QgsPointXY &p2, const QgsPointXY &p3,
      QgsVector vect1, QgsVector vect2, QgsVector vect3, const QgsPointXY &pt
    );
 
    static double interpolateFromFacesData(
      const QgsPointXY &p1, const QgsPointXY &p2, const QgsPointXY &p3,
      double val, const QgsPointXY &pt );
 
    static QgsVector interpolateVectorFromFacesData(
      const QgsPointXY &p1, const QgsPointXY &p2, const QgsPointXY &p3,
      QgsVector vect, const QgsPointXY &pt );
 
    static QVector<double> interpolateFromFacesData(
      QVector<double> valuesOnFaces,
      const QgsMesh *nativeMesh,
      const QgsTriangularMesh *triangularMesh,
      QgsMeshDataBlock *active,
      QgsMeshRendererScalarSettings::DataResamplingMethod method
    );
 
    static QVector<double> resampleFromVerticesToFaces(
      const QVector<double> valuesOnVertices,
      const QgsMesh *nativeMesh,
      const QgsTriangularMesh *triangularMesh,
      const QgsMeshDataBlock *active,
      QgsMeshRendererScalarSettings::DataResamplingMethod method
    );
 
    static QVector<double> calculateMagnitudeOnVertices(
      const QgsMeshLayer *meshLayer,
      const QgsMeshDatasetIndex index,
      QgsMeshDataBlock *activeFaceFlagValues,
      const QgsMeshRendererScalarSettings::DataResamplingMethod method = QgsMeshRendererScalarSettings::NeighbourAverage );
 
    static QgsRectangle triangleBoundingBox( const QgsPointXY &p1, const QgsPointXY &p2, const QgsPointXY &p3 );
 
    static QString formatTime( double hours, const QDateTime &referenceTime, const QgsMeshTimeSettings &settings );
 
    static QVector<QVector3D> calculateNormals(
      const QgsTriangularMesh &triangularMesh,
      const QVector<double> &verticalMagnitude,
      bool isRelative );
 
    static void  createDatasetGroupTreeItems( const QList<QgsMeshDatasetGroupMetadata> &metadataList,
        QgsMeshDatasetGroupTreeItem *rootItem, int startingIndex )
    {
      QMap<QString, QgsMeshDatasetGroupTreeItem *> mNameToItem;
      for ( int i = 0; i < metadataList.count(); ++i )
      {
        int groupIndex = startingIndex + i;
        const QgsMeshDatasetGroupMetadata meta = metadataList.at( i );
        const QString name = meta.name();
        const QStringList subdatasets = name.split( '/' );
 
        QString displayName = name;
        QgsMeshDatasetGroupTreeItem *parent = rootItem;
 
        if ( subdatasets.size() == 2 )
        {
          auto it = mNameToItem.find( subdatasets[0] );
          if ( it == mNameToItem.end() )
            QgsDebugMsg( QStringLiteral( "Unable to find parent group for %1." ).arg( name ) );
          else
          {
            displayName = subdatasets[1];
            parent = it.value();
          }
        }
        else if ( subdatasets.size() != 1 )
          QgsDebugMsg( QStringLiteral( "Ignoring too deep child group name %1." ).arg( name ) );
 
        QgsMeshDatasetGroupTreeItem *item = new QgsMeshDatasetGroupTreeItem( displayName, meta.isVector(), groupIndex );
        parent->appendChild( item );
        if ( mNameToItem.contains( name ) )
          QgsDebugMsg( QStringLiteral( "Group %1 is not unique" ).arg( displayName ) );
        mNameToItem[name] = item;
      }
    }
};
 
 
#endif // QGSMESHLAYERUTILS_H