1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#ifndef AFTL_MTP_METADATA_LIBRARY_H
#define AFTL_MTP_METADATA_LIBRARY_H
#include <mtp/ptp/ObjectId.h>
#include <mtp/ptp/ObjectFormat.h>
#include <mtp/ByteArray.h>
#include <mtp/types.h>
#include <functional>
#include <memory>
#include <string>
#include <unordered_map>
#include <unordered_set>
namespace mtp
{
class Session;
DECLARE_PTR(Session);
class Library
{
SessionPtr _session;
StorageId _storage;
public:
enum struct State {
Initialising,
QueryingArtists,
LoadingArtists,
QueryingAlbums,
LoadingAlbums,
Loaded
};
using ProgressReporter = std::function<void (State, u64, u64)>;
struct Artist
{
ObjectId Id;
ObjectId MusicFolderId;
std::string Name;
};
DECLARE_PTR(Artist);
struct Album
{
ObjectId Id;
ObjectId MusicFolderId;
ArtistPtr Artist;
std::string Name;
time_t Year = 0;
bool RefsLoaded = false;
void LoadRefs();
std::unordered_set<ObjectId> Refs;
std::unordered_multimap<std::string, int> Tracks;
};
DECLARE_PTR(Album);
struct NewTrackInfo
{
ObjectId Id;
std::string Name;
int Index;
};
private:
ObjectId _artistsFolder;
ObjectId _albumsFolder;
ObjectId _musicFolder;
bool _artistSupported;
bool _albumDateAuthoredSupported;
bool _albumCoverSupported;
using ArtistMap = std::unordered_map<std::string, ArtistPtr>;
ArtistMap _artists;
using AlbumKey = std::pair<ArtistPtr, std::string>;
struct AlbumKeyHash
{ size_t operator() (const AlbumKey & key) const {
return std::hash<ArtistPtr>()(key.first) + std::hash<std::string>()(key.second);
}};
using AlbumMap = std::unordered_map<AlbumKey, AlbumPtr, AlbumKeyHash>;
AlbumMap _albums;
using NameToObjectIdMap = std::unordered_map<std::string, ObjectId>;
NameToObjectIdMap ListAssociations(ObjectId parentId);
ObjectId GetOrCreate(ObjectId parentId, const std::string &name);
void LoadRefs(AlbumPtr album);
public:
Library(const mtp::SessionPtr & session, ProgressReporter && reporter = ProgressReporter());
~Library();
static bool Supported(const mtp::SessionPtr & session);
//search by Metadata?
ArtistPtr GetArtist(std::string name);
ArtistPtr CreateArtist(std::string name);
AlbumPtr GetAlbum(const ArtistPtr & artist, std::string name);
AlbumPtr CreateAlbum(const ArtistPtr & artist, std::string name, int year);
bool HasTrack(const AlbumPtr & album, const std::string &name, int trackIndex);
NewTrackInfo CreateTrack(const ArtistPtr & artist, const AlbumPtr & album, ObjectFormat type, std::string name, const std::string & genre, int trackIndex, const std::string &filename, size_t size);
void AddTrack(AlbumPtr album, const NewTrackInfo &ti);
void AddCover(AlbumPtr album, const mtp::ByteArray &data);
};
DECLARE_PTR(Library);
}
#endif
|
