QGIS/tests/bench/qgsbench.cpp
2024-05-31 08:46:33 +10:00

358 lines
10 KiB
C++

/***************************************************************************
qgsbench.cpp - Benchmark
-------------------
begin : 2011-11-15
copyright : (C) 2011 Radim Blazek
email : radim dot blazek 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. *
* *
***************************************************************************/
#include <QtGlobal>
#include <cmath>
#include <ctime>
#include <iostream>
#include <cstdio>
#ifndef Q_OS_WIN
#include <sys/resource.h>
#endif
#include <ctime>
#include <cmath>
#include <QFile>
#include <QFileInfo>
#include <QPainter>
#include <QSettings>
#include <QString>
#include <QTextStream>
#include <QTime>
#ifndef QGSVERSION
#include "qgsversion.h"
#endif
#include "qgsbench.h"
#include "qgslogger.h"
#include "qgsmaprendererparalleljob.h"
#include "qgsmaprenderersequentialjob.h"
#include "qgsproject.h"
const char *pre[] = { "user", "sys", "total", "wall" };
#ifdef Q_OS_WIN
// slightly adapted from http://anoncvs.postgresql.org/cvsweb.cgi/pgsql/src/port/getrusage.c?rev=1.18;content-type=text%2Fplain
#include <winsock2.h>
#include <errno.h>
#define RUSAGE_SELF 0
struct rusage
{
struct timeval ru_utime; /* user time used */
struct timeval ru_stime; /* system time used */
};
/*-------------------------------------------------------------------------
*
* getrusage.c
* get information about resource utilisation
*
* Portions Copyright (c) 1996-2010, PostgreSQL Global Development Group
* Portions Copyright (c) 1994, Regents of the University of California
*
*
* IDENTIFICATION
* $PostgreSQL: pgsql/src/port/getrusage.c,v 1.18 2010-01-02 16:58:13 momjian Exp $
*
*-------------------------------------------------------------------------
*/
int getrusage( int who, struct rusage *rusage )
{
FILETIME starttime;
FILETIME exittime;
FILETIME kerneltime;
FILETIME usertime;
ULARGE_INTEGER li;
if ( who != RUSAGE_SELF )
{
/* Only RUSAGE_SELF is supported in this implementation for now */
errno = EINVAL;
return -1;
}
if ( !rusage )
{
errno = EFAULT;
return -1;
}
memset( rusage, 0, sizeof( struct rusage ) );
if ( GetProcessTimes( GetCurrentProcess(),
&starttime, &exittime, &kerneltime, &usertime ) == 0 )
{
// _dosmaperr(GetLastError());
return -1;
}
/* Convert FILETIMEs (0.1 us) to struct timeval */
memcpy( &li, &kerneltime, sizeof( FILETIME ) );
li.QuadPart /= 10L; /* Convert to microseconds */
rusage->ru_stime.tv_sec = li.QuadPart / 1000000L;
rusage->ru_stime.tv_usec = li.QuadPart % 1000000L;
memcpy( &li, &usertime, sizeof( FILETIME ) );
li.QuadPart /= 10L; /* Convert to microseconds */
rusage->ru_utime.tv_sec = li.QuadPart / 1000000L;
rusage->ru_utime.tv_usec = li.QuadPart % 1000000L;
return 0;
}
#endif
QgsBench::QgsBench( int width, int height, int iterations )
: mWidth( width )
, mHeight( height )
, mIterations( iterations )
, mSetExtent( false )
, mUserStart( 0.0 )
, mSysStart( 0.0 )
, mParallel( false )
{
QgsDebugMsgLevel( QStringLiteral( "mIterations = %1" ).arg( mIterations ), 1 );
connect( QgsProject::instance(), &QgsProject::readProject,
this, &QgsBench::readProject );
}
bool QgsBench::openProject( const QString &fileName )
{
if ( ! QgsProject::instance()->read( fileName ) )
{
return false;
}
mLogMap.insert( QStringLiteral( "project" ), fileName );
return true;
}
void QgsBench::readProject( const QDomDocument &doc )
{
const QDomNodeList nodes = doc.elementsByTagName( QStringLiteral( "mapcanvas" ) );
if ( nodes.count() )
{
QDomNode node = nodes.item( 0 );
mMapSettings.readXml( node );
}
else
{
fprintf( stderr, "Cannot read mapcanvas from project\n" );
}
}
void QgsBench::setExtent( const QgsRectangle &extent )
{
mExtent = extent;
mSetExtent = true;
}
void QgsBench::render()
{
QgsDebugMsgLevel( "extent: " + mMapSettings.extent().toString(), 1 );
const QMap<QString, QgsMapLayer *> layersMap = QgsProject::instance()->mapLayers();
mMapSettings.setLayers( layersMap.values() );
if ( mSetExtent )
{
mMapSettings.setExtent( mExtent );
}
// Maybe in future
//outputCRS = QgsCrsCache::instance()->crsByAuthId( crsId );
//mMapRenderer->setMapUnits( outputCRS.mapUnits() );
//mMapRenderer->setDestinationCrs( outputCRS );
// Enable labeling
mMapSettings.setFlag( Qgis::MapSettingsFlag::DrawLabeling );
mMapSettings.setOutputSize( QSize( mWidth, mHeight ) );
// TODO: do we need the other QPainter flags?
mMapSettings.setFlag( Qgis::MapSettingsFlag::Antialiasing, mRendererHints.testFlag( QPainter::Antialiasing ) );
for ( int i = 0; i < mIterations; i++ )
{
QgsMapRendererQImageJob *job = nullptr;
if ( mParallel )
job = new QgsMapRendererParallelJob( mMapSettings );
else
job = new QgsMapRendererSequentialJob( mMapSettings );
start();
job->start();
job->waitForFinished();
elapsed();
mImage = job->renderedImage();
delete job;
}
mLogMap.insert( QStringLiteral( "iterations" ), mTimes.size() );
mLogMap.insert( QStringLiteral( "revision" ), QGSVERSION );
// Calc stats: user, sys, total
double min[4] = {std::numeric_limits<double>::max()};
double max[4] = { std::numeric_limits<double>::lowest()};
double stdev[4] = {0.};
double maxdev[4] = {0.};
double avg[4] = {0.};
for ( int t = 0; t < 4; t++ )
{
for ( int i = 0; i < mTimes.size(); i++ )
{
avg[t] += mTimes.at( i )[t];
if ( i == 0 || mTimes.at( i )[t] < min[t] ) min[t] = mTimes.at( i )[t];
if ( i == 0 || mTimes.at( i )[t] > max[t] ) max[t] = mTimes.at( i )[t];
}
avg[t] /= mTimes.size();
}
QMap<QString, QVariant> timesMap;
for ( int t = 0; t < 4; t++ )
{
if ( mIterations > 1 )
{
for ( int i = 0; i < mTimes.size(); i++ )
{
const double d = std::fabs( avg[t] - mTimes.at( i )[t] );
stdev[t] += std::pow( d, 2 );
if ( i == 0 || d > maxdev[t] ) maxdev[t] = d;
}
stdev[t] = std::sqrt( stdev[t] / mTimes.size() );
}
QMap<QString, QVariant> map;
map.insert( QStringLiteral( "min" ), min[t] );
map.insert( QStringLiteral( "max" ), max[t] );
map.insert( QStringLiteral( "avg" ), avg[t] );
map.insert( QStringLiteral( "stdev" ), stdev[t] );
map.insert( QStringLiteral( "maxdev" ), maxdev[t] );
timesMap.insert( pre[t], map );
}
mLogMap.insert( QStringLiteral( "times" ), timesMap );
}
void QgsBench::saveSnapsot( const QString &fileName )
{
// If format is 0, QImage will attempt to guess the format by looking at fileName's suffix.
mImage.save( fileName );
}
void QgsBench::printLog( const QString &printTime )
{
std::cout << "iterations: " << mLogMap[QStringLiteral( "iterations" )].toString().toLatin1().constData() << std::endl;
bool validPrintTime = false;
for ( int x = 0; x < 4; ++x )
if ( printTime == pre[x] )
validPrintTime = true;
if ( !validPrintTime )
{
std::cout << "invalid --print option: " << printTime.toLatin1().data() << std::endl;
return;
}
QMap<QString, QVariant> timesMap = mLogMap[QStringLiteral( "times" )].toMap();
QMap<QString, QVariant> totalMap = timesMap[printTime].toMap();
QMap<QString, QVariant>::iterator i = totalMap.begin();
while ( i != totalMap.end() )
{
const QString s = printTime + '_' + i.key() + ": " + i.value().toString();
std::cout << s.toLatin1().constData() << std::endl;
++i;
}
}
QString QgsBench::serialize( const QMap<QString, QVariant> &map, int level )
{
QStringList list;
const QString space = QStringLiteral( " " ).repeated( level * 2 );
const QString space2 = QStringLiteral( " " ).repeated( level * 2 + 2 );
QMap<QString, QVariant>::const_iterator i = map.constBegin();
while ( i != map.constEnd() )
{
switch ( static_cast< QMetaType::Type >( i.value().userType() ) )
{
case QMetaType::Int:
list.append( space2 + '\"' + i.key() + "\": " + QString::number( i.value().toInt() ) );
break;
case QMetaType::Double:
list.append( space2 + '\"' + i.key() + "\": " + QStringLiteral( "%1" ).arg( i.value().toDouble(), 0, 'f', 3 ) );
break;
case QMetaType::QString:
list.append( space2 + '\"' + i.key() + "\": \"" + i.value().toString().replace( '\\', QLatin1String( "\\\\" ) ).replace( '\"', QLatin1String( "\\\"" ) ) + '\"' );
break;
//case QMetaType::QMap: QMap is not in QMetaType
default:
list.append( space2 + '\"' + i.key() + "\": " + serialize( i.value().toMap(), level + 1 ) );
break;
}
++i;
}
return space + "{\n" + list.join( QLatin1String( ",\n" ) ) + '\n' + space + '}';
}
void QgsBench::saveLog( const QString &fileName )
{
QFile file( fileName );
if ( !file.open( QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate ) )
return;
QTextStream out( &file );
out << serialize( mLogMap ).toLatin1().constData() << '\n';
file.close();
}
void QgsBench::start()
{
struct rusage usage;
getrusage( RUSAGE_SELF, &usage );
mUserStart = usage.ru_utime.tv_sec + usage.ru_utime.tv_usec / 1000000.;
mSysStart = usage.ru_stime.tv_sec + usage.ru_stime.tv_usec / 1000000.;
mWallTime.start();
}
void QgsBench::elapsed()
{
struct rusage usage;
getrusage( RUSAGE_SELF, &usage );
const double userEnd = usage.ru_utime.tv_sec + usage.ru_utime.tv_usec / 1000000.;
const double sysEnd = usage.ru_stime.tv_sec + usage.ru_stime.tv_usec / 1000000.;
double *t = new double[4];
t[0] = userEnd - mUserStart;
t[1] = sysEnd - mSysStart;
t[2] = t[0] + t[1];
t[3] = mWallTime.elapsed() / 1000.;
mTimes.append( t );
}