87 lines
2.1 KiB
C++
87 lines
2.1 KiB
C++
/*
|
|
* Copyright 2003-2021 The Music Player Daemon Project
|
|
* http://www.musicpd.org
|
|
*
|
|
* 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.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "Helpers.hxx"
|
|
#include "Stats.hxx"
|
|
#include "Interface.hxx"
|
|
#include "song/LightSong.hxx"
|
|
#include "tag/Tag.hxx"
|
|
|
|
#include <set>
|
|
|
|
#include <string.h>
|
|
|
|
struct StringLess {
|
|
gcc_pure
|
|
bool operator()(const char *a, const char *b) const noexcept {
|
|
return strcmp(a, b) < 0;
|
|
}
|
|
};
|
|
|
|
using StringSet = std::set<const char *, StringLess>;
|
|
|
|
static void
|
|
StatsVisitTag(DatabaseStats &stats, StringSet &artists, StringSet &albums,
|
|
const Tag &tag) noexcept
|
|
{
|
|
if (!tag.duration.IsNegative())
|
|
stats.total_duration += tag.duration;
|
|
|
|
for (const auto &item : tag) {
|
|
switch (item.type) {
|
|
case TAG_ARTIST:
|
|
artists.emplace(item.value);
|
|
break;
|
|
|
|
case TAG_ALBUM:
|
|
albums.emplace(item.value);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void
|
|
StatsVisitSong(DatabaseStats &stats, StringSet &artists, StringSet &albums,
|
|
const LightSong &song) noexcept
|
|
{
|
|
++stats.song_count;
|
|
|
|
StatsVisitTag(stats, artists, albums, song.tag);
|
|
}
|
|
|
|
DatabaseStats
|
|
GetStats(const Database &db, const DatabaseSelection &selection)
|
|
{
|
|
DatabaseStats stats;
|
|
stats.Clear();
|
|
|
|
StringSet artists, albums;
|
|
const auto f = [&](const auto &song)
|
|
{ return StatsVisitSong(stats, artists, albums, song); };
|
|
|
|
db.Visit(selection, f);
|
|
|
|
stats.artist_count = artists.size();
|
|
stats.album_count = albums.size();
|
|
return stats;
|
|
}
|