Daniel@0: /* Part of DML (Digital Music Laboratory)
Daniel@0: 	Copyright 2014-2015 Samer Abdallah, University of London
Daniel@0: 	 
Daniel@0: 	This program is free software; you can redistribute it and/or
Daniel@0: 	modify it under the terms of the GNU General Public License
Daniel@0: 	as published by the Free Software Foundation; either version 2
Daniel@0: 	of the License, or (at your option) any later version.
Daniel@0: 
Daniel@0: 	This program is distributed in the hope that it will be useful,
Daniel@0: 	but WITHOUT ANY WARRANTY; without even the implied warranty of
Daniel@0: 	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Daniel@0: 	GNU General Public License for more details.
Daniel@0: 
Daniel@0: 	You should have received a copy of the GNU General Public
Daniel@0: 	License along with this library; if not, write to the Free Software
Daniel@0: 	Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
Daniel@0: */
Daniel@0: 
Daniel@0: :- module(beets_p2r, []).
Daniel@0: 
Daniel@0: /** <module> Access to beets database
Daniel@0:  */
Daniel@0: 
Daniel@0: 
Daniel@0: :- use_module(library(prosqlite)).
Daniel@0: :- use_module(library(musicbrainz)).
Daniel@0: :- use_module(library(semweb/rdf_db)).
Daniel@0: :- use_module(library(rdfutils)).
Daniel@0: :- use_module(entailment(p2r)).
Daniel@0: 
Daniel@0: :- set_prolog_flag(double_quotes,string).
Daniel@0: 
Daniel@0: :- rdf_register_prefix(beets,'http://dml.org/beets/props/item/').
Daniel@0: :- rdf_register_prefix(beets_album,'http://dml.org/beets/props/album/').
Daniel@0: :- rdf_register_prefix(audio,'audio:').
Daniel@0: :- rdf_register_prefix(dml,'http://dml.org/dml/').
Daniel@0: :- rdf_register_prefix(mb,'http://musicbrainz.org/').
Daniel@0: 
Daniel@0: :- setting(db_file,string,"~/lib/beets/music-ro.db","Location of beets database").
Daniel@0: :- setting(audio_root,string,"/usr/local/share/Music/","Root of audio file directory tree").
Daniel@0: 
Daniel@0: 
Daniel@0: rdf(audio:tail(Rel), rdf:type, mo:'AudioFile'),
Daniel@0: rdf(audio:tail(Rel), mo:encoding, literal(Enc)) <==
Daniel@0:    setting(audio_root,Root),
Daniel@0:    expand_file_name(Root,[Root1]),
Daniel@0:    item(Id,path,Path),
Daniel@0:    item(Id,format,Enc),
Daniel@0:    atom_concat(Root1,Rel,Path).
Daniel@0: 
Daniel@0: rdf(audio:tail(Rel), beets:enc(Prop), literal(Val)) <==
Daniel@0:    audio_item_path(Id,Rel),
Daniel@0:    item(Id,Prop,Val1), 
Daniel@0:    (  Prop=path -> exists_file(Val1); true),
Daniel@0:    as_typed_literal(Val1,Val).
Daniel@0: 
Daniel@0: rdf(Signal,rdf:type,mo:'Signal') <==
Daniel@0:    item(_,mb_trackid,ID), % track id is actually a Recording.
Daniel@0:    mb_id_uri(recording,ID,Signal).
Daniel@0: 
Daniel@0: rdf(Signal,mo:available_as,audio:tail(Rel)) <==
Daniel@0:    setting(audio_root,Root),
Daniel@0:    expand_file_name(Root,[Root1]),
Daniel@0:    item(Id,path,Path),
Daniel@0:    exists_file(Path),
Daniel@0:    atom_concat(Root1,Rel,Path),
Daniel@0:    item(Id,mb_trackid,ID), % track id is actually a Recording.
Daniel@0:   % need to ask Linkedbrainz to get Signal URI
Daniel@0:    mb_id_uri(recording,ID,Signal).
Daniel@0:    
Daniel@0: rdf(audio:tail(Rel), dml:file_artist, URI) <==
Daniel@0:    audio_item_path(Id,Rel),
Daniel@0:    item(Id,mb_artistid,ID),
Daniel@0:    mb_id_uri(artist,ID,URI).
Daniel@0: 
Daniel@0: rdf(audio:tail(Rel), dml:file_release, URI) <==
Daniel@0:    audio_item_path(Id,Rel),
Daniel@0:    item(Id,mb_albumid,ID),
Daniel@0:    mb_id_uri(release,ID,URI).
Daniel@0: 
Daniel@0: rdf(audio:tail(Rel), dml:file_recording, URI) <==
Daniel@0:    audio_item_path(Id,Rel),
Daniel@0:    item(Id,mb_trackid,ID),
Daniel@0:    mb_id_uri(recording,ID,URI).
Daniel@0: 
Daniel@0: % rdf(URI,mo:release_type,mo:
Daniel@0: rdf(URI,rdf:type,mo:'Release') <==
Daniel@0:    item(_,mb_albumid,ID),
Daniel@0:    mb_id_uri(release,ID,URI).
Daniel@0: 
Daniel@0: audio_item_path(Id,Rel) :-
Daniel@0:    setting(audio_root,Root),
Daniel@0:    expand_file_name(Root,[Root1]),
Daniel@0:    item(Id,path,Path),
Daniel@0:    atom_concat(Root1,Rel,Path).
Daniel@0: 
Daniel@0: :- public import/0.
Daniel@0: import :- with_beets_db(assert_all(beets_p2r)).
Daniel@0: 
Daniel@0: with_beets_db(Goal) :-
Daniel@0:    setting(db_file,DBFile),
Daniel@0:    expand_file_name(DBFile,[Path|_]),
Daniel@0:    setup_call_cleanup(
Daniel@0:       sqlite_connect(Path,Con,[alias(beets),ext(''),as_predicates(true),arity(unary),at_module(beets)]),
Daniel@0:       Goal,
Daniel@0:       sqlite_disconnect(Con)).
Daniel@0: 
Daniel@0: :- dynamic beets:items/1.
Daniel@0: item(Id) :- beets:items([id=Id]).
Daniel@0: 
Daniel@0: item(Id,path,Path) :-
Daniel@0:    (  var(Path) -> beets:items([id=Id,path=Path])
Daniel@0:    ;  var(Id)   -> sqlite_format_query(beets,"select id from items where path like '~s'"-[Path],row(Id))
Daniel@0:    ;  sqlite_format_query(beets,"select null from items where id=~w and path like '~s'"-[Id,Path],row(_))
Daniel@0:    ).
Daniel@0: 
Daniel@0: item(Id,Prop,Val) :-
Daniel@0:    table_column(items,Prop), Prop\=path, Prop\=comp,
Daniel@0:    beets:items([id=Id,Prop=Val]), 
Daniel@0:    \+invalid(Prop,Val).
Daniel@0:    
Daniel@0: 
Daniel@0: % album(Id) :- beets:albums([id=Id]).
Daniel@0: % album(Id,Prop,Val) :-
Daniel@0: %    table_column(albums,Prop),
Daniel@0: %    beets:albums([id=Id,Prop=Val]), Val\=''.
Daniel@0: 
Daniel@0: table_column(T,C) :-
Daniel@0:    sqlite_table_column(beets,T,C1), C=C1.
Daniel@0: 
Daniel@0: invalid(_,'').
Daniel@0: invalid(Prop,0) :- nonzero(Prop).
Daniel@0: invalid(mtime,0.0).
Daniel@0: 
Daniel@0: nonzero(bitdepth).
Daniel@0: nonzero(bitrate).
Daniel@0: nonzero(samplerate).
Daniel@0: nonzero(track).
Daniel@0: nonzero(tracktotal).
Daniel@0: nonzero(disc).
Daniel@0: nonzero(disctotal).
Daniel@0: nonzero(bpm).
Daniel@0: nonzero(day).
Daniel@0: nonzero(month).
Daniel@0: nonzero(year).
Daniel@0: nonzero(original_day).
Daniel@0: nonzero(original_month).
Daniel@0: nonzero(original_year).
Daniel@0: 
Daniel@0: :- public audio_file/3.
Daniel@0: 
Daniel@0: audio_file(URI,File,just(T0)) :-
Daniel@0:    rdf(URI,beets:path,literal(File)), !,
Daniel@0:    rdf(URI,beets:format,literal(F0)),
Daniel@0:    format_type(F0,T0).
Daniel@0: 
Daniel@0: format_type('MP3',mp3).
Daniel@0: format_type('OGG',ogg).
Daniel@0: format_type('AAC',aac).
Daniel@0: format_type('ALAC',mp4). % !!! ??