151 lines
2.9 KiB
C++
151 lines
2.9 KiB
C++
/**
|
|
* Copyright 2013 (c) Anna Schumaker.
|
|
*/
|
|
#include <core/library.h>
|
|
#include <core/playlist.h>
|
|
|
|
|
|
class PlaylistQueue : public Queue {
|
|
public:
|
|
|
|
PlaylistQueue() : Queue(Q_ENABLED | Q_REPEAT)
|
|
{
|
|
sort(SORT_ARTIST, true);
|
|
sort(SORT_YEAR, false);
|
|
sort(SORT_TRACK, false);
|
|
set_flag(Q_NO_SORT);
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
while (size() > 0)
|
|
del((unsigned)0);
|
|
}
|
|
|
|
void fill(IndexEntry *ent)
|
|
{
|
|
std::set<unsigned int>::iterator it;
|
|
|
|
clear();
|
|
for (it = ent->begin(); it != ent->end(); it++)
|
|
add(tags :: get_track(*it));
|
|
}
|
|
|
|
unsigned int find_average_count()
|
|
{
|
|
Track *track;
|
|
unsigned int total = 0, count = 0;
|
|
|
|
for (unsigned int i = 0; i < tags :: track_size(); i++) {
|
|
track = tags :: get_track(i);
|
|
if (track != NULL) {
|
|
total += track->count();
|
|
count++;
|
|
}
|
|
}
|
|
return total / count;
|
|
}
|
|
|
|
void dynamic_add(const std::string &name, Track *track, unsigned int avg)
|
|
{
|
|
if ((name == "Unplayed") && (track->count() == 0))
|
|
add(track);
|
|
if ((name == "Most Played") && (track->count() > avg))
|
|
add(track);
|
|
if ((name == "Least Played") && (track->count() < avg))
|
|
add(track);
|
|
}
|
|
|
|
void dynamic_fill(const std::string &name)
|
|
{
|
|
Track *track;
|
|
unsigned int avg = 0;
|
|
|
|
if ((name == "Most Played") || (name == "Least Played"))
|
|
avg = find_average_count();
|
|
|
|
clear();
|
|
for (unsigned int i = 0; i < tags :: track_size(); i++) {
|
|
track = tags :: get_track(i);
|
|
if (track != NULL)
|
|
dynamic_add(name, track, avg);
|
|
}
|
|
}
|
|
|
|
};
|
|
|
|
|
|
static Index playlist_db("playlist.db", true);
|
|
static PlaylistQueue playlist_q;
|
|
static std::string cur_plist;
|
|
|
|
|
|
void playlist :: init()
|
|
{
|
|
std::set<unsigned int>::iterator it;
|
|
|
|
playlist_db.load();
|
|
|
|
IndexEntry *ent = get_tracks("Banned");
|
|
if (!ent)
|
|
return;
|
|
|
|
for (it = ent->begin(); it != ent->end(); it++)
|
|
library :: get_queue()->del(tags :: get_track(*it));
|
|
}
|
|
|
|
bool playlist :: has(Track *track, const std::string &name)
|
|
{
|
|
std::set<unsigned int>::iterator it;
|
|
IndexEntry *ent = playlist_db.find(name);
|
|
|
|
if (ent == NULL)
|
|
return false;
|
|
return ent->has(track->index());
|
|
}
|
|
|
|
void playlist :: add(Track *track, const std::string &name)
|
|
{
|
|
if (!( (name == "Banned") || (name == "Favorites") ))
|
|
return;
|
|
|
|
if (!has(track, name)) {
|
|
playlist_db.insert(name, track->index());
|
|
if (cur_plist == name)
|
|
playlist_q.add(track);
|
|
if (name == "Banned")
|
|
library :: get_queue()->del(track);
|
|
}
|
|
}
|
|
|
|
void playlist :: del(Track *track, const std::string &name)
|
|
{
|
|
playlist_db.remove(name, track->index());
|
|
if (cur_plist == name)
|
|
playlist_q.del(track);
|
|
if (name == "Banned")
|
|
library :: get_queue()->add(track);
|
|
}
|
|
|
|
void playlist :: select(const std::string &name)
|
|
{
|
|
IndexEntry *ent = playlist_db.find(name);
|
|
|
|
if (ent != NULL)
|
|
playlist_q.fill(ent);
|
|
else
|
|
playlist_q.dynamic_fill(name);
|
|
|
|
cur_plist = name;
|
|
}
|
|
|
|
IndexEntry *playlist :: get_tracks(const std::string &name)
|
|
{
|
|
return playlist_db.find(name);
|
|
}
|
|
|
|
Queue *playlist :: get_queue()
|
|
{
|
|
return &playlist_q;
|
|
}
|