First pass of sfdb tag searching. Not functional, but very very close.
[ardour.git] / libs / ardour / audio_library.cc
1 /*
2     Copyright (C) 2003-2006 Paul Davis
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include <sstream>
21
22 #include <libxml/uri.h>
23
24 #include <lrdf.h>
25
26 #include <pbd/compose.h>
27
28 #include <ardour/audio_library.h>
29 #include <ardour/utils.h>
30
31 #include "i18n.h"
32
33 using namespace std;
34 using namespace ARDOUR;
35
36 static char* TAG = "http://ardour.org/ontology/Tag";
37
38 AudioLibrary::AudioLibrary ()
39 {
40         src = "file:" + get_user_ardour_path() + "sfdb";
41
42         // workaround for possible bug in raptor that crashes when saving to a
43         // non-existant file.
44         touch_file(get_user_ardour_path() + "sfdb");
45
46         lrdf_read_file(src.c_str());
47 }
48
49 AudioLibrary::~AudioLibrary ()
50 {
51 }
52
53 void
54 AudioLibrary::save_changes ()
55 {
56         if (lrdf_export_by_source(src.c_str(), src.substr(5).c_str())) {
57                 PBD::warning << string_compose(_("Could not open %1.  Audio Library not saved"), src) << endmsg;
58         }
59 }
60
61 string
62 AudioLibrary::path2uri (string path)
63 {
64         xmlURI temp;
65         memset(&temp, 0, sizeof(temp));
66         
67         xmlChar *cal = xmlCanonicPath((xmlChar*) path.c_str());
68         temp.path = (char *) cal;
69         xmlChar *ret = xmlSaveUri(&temp);
70         xmlFree(cal);
71         
72         stringstream uri;
73         uri << "file:" << (const char*) ret;
74         
75         xmlFree (ret);
76         
77         return uri.str();
78 }
79
80 string
81 AudioLibrary::uri2path (string uri)
82 {
83         string path = xmlURIUnescapeString(uri.c_str(), 0, 0);
84         return path.substr(5);
85 }
86
87 void
88 AudioLibrary::set_tags (string member, vector<string> tags)
89 {
90         sort (tags.begin(), tags.end());
91         tags.erase (unique(tags.begin(), tags.end()), tags.end());
92         
93         string file_uri(path2uri(member));
94         
95         lrdf_remove_uri_matches (file_uri.c_str());
96         
97         for (vector<string>::iterator i = tags.begin(); i != tags.end(); ++i) {
98                 lrdf_add_triple (src.c_str(), file_uri.c_str(), TAG, (*i).c_str(), lrdf_literal);
99         }
100 }
101
102 vector<string>
103 AudioLibrary::get_tags (string member)
104 {
105         vector<string> tags;
106         
107         lrdf_statement pattern;
108         pattern.subject = strdup(path2uri(member).c_str());
109         pattern.predicate = TAG;
110         pattern.object = 0;
111         pattern.object_type = lrdf_literal;
112         
113         lrdf_statement* matches = lrdf_matches (&pattern);
114         free (pattern.subject);
115         
116         lrdf_statement* current = matches;
117         while (current != 0) {
118                 tags.push_back (current->object);
119                 
120                 current = current->next;
121         }
122         
123         lrdf_free_statements (matches);
124         
125         sort (tags.begin(), tags.end());
126         
127         return tags;
128 }
129
130 void
131 AudioLibrary::search_members_and (vector<string>& members, const vector<string> tags)
132 {
133         lrdf_statement **head;
134         lrdf_statement* pattern = 0;
135         lrdf_statement* old = 0;
136         head = &pattern;
137
138         vector<string>::const_iterator i;
139         for (i = tags.begin(); i != tags.end(); ++i){
140                 pattern = new lrdf_statement;
141                 pattern->subject = "?";
142                 pattern->predicate = TAG;
143                 pattern->object = strdup((*i).c_str());
144                 pattern->next = old;
145
146                 old = pattern;
147         }
148
149         if (*head != 0) {
150                 lrdf_uris* ulist = lrdf_match_multi(*head);
151                 for (uint32_t j = 0; ulist && j < ulist->count; ++j) {
152 //                      cerr << "AND: " << uri2path(ulist->items[j]) << endl;
153                         members.push_back(uri2path(ulist->items[j]));
154                 }
155                 lrdf_free_uris(ulist);
156
157             sort(members.begin(), members.end());
158             unique(members.begin(), members.end());
159         }
160
161         // memory clean up
162         pattern = *head;
163         while(pattern){
164                 free(pattern->object);
165                 old = pattern;
166                 pattern = pattern->next;
167                 delete old;
168         }
169 }