Daniel@0: # Part of DML (Digital Music Laboratory)
Daniel@0: # Copyright 2014-2015 Samer Abdallah, University College 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: # -*- coding: utf-8 -*-
Daniel@0: __author__="samer"
Daniel@0: 
Daniel@0: from rdfutils import *
Daniel@0: from n3Parser import *
Daniel@0: from pitchutils import *
Daniel@0: from aggregate import *
Daniel@0: from csvutils import csv_map_rows
Daniel@0: from warnings import warn
Daniel@0: 
Daniel@0: def tonic_from_n3(filename):
Daniel@0:     graph=get_rdf_graph_from_n3(filename)
Daniel@0:     max_time = 0
Daniel@0:     tonic = None
Daniel@0:     for ev, time in graph.subjects_objects(ev_ns.time):
Daniel@0:         t = parse_xsd_duration(graph.value(time,tl_ns.at))
Daniel@0:         if t>max_time: tonic = graph.value(ev,af_ns.feature)
Daniel@0:     return tonic-1
Daniel@0: 
Daniel@0: def tonic_from_csv(filename):
Daniel@0:     # format: time, pitch_class_number, pitch_class_name
Daniel@0:     return int(csv_map_rows(filename,3,lambda row:row[1])[-1])-1
Daniel@0: 
Daniel@0: 
Daniel@0: def aggregate(inputs):
Daniel@0:     parser_table = { 'n3':tonic_from_n3, 
Daniel@0:                      'csv':tonic_from_csv }
Daniel@0:     hist = 12*[0]
Daniel@0:     def accum(f): hist[decode_tagged(parser_table,f)] += 1
Daniel@0:     stats=for_each(inputs,accum)
Daniel@0:     return { 'result': discrete_hist(pitch_class_names,hist), 'stats':stats }