annotate rdf/RDFFeatureWriter.cpp @ 586:4ab8e8f7d7ed

* Replace not very useful signal-uri option with more useful audiofile-uri
author Chris Cannam
date Wed, 13 May 2009 12:49:22 +0000
parents cbc884787a08
children 94054c97c76a
rev   line source
Chris@498 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@498 2
Chris@498 3 /*
Chris@498 4 Sonic Annotator
Chris@498 5 A utility for batch feature extraction from audio files.
Chris@498 6 Mark Levy, Chris Sutton and Chris Cannam, Queen Mary, University of London.
Chris@498 7 Copyright 2007-2008 QMUL.
Chris@498 8
Chris@498 9 This program is free software; you can redistribute it and/or
Chris@498 10 modify it under the terms of the GNU General Public License as
Chris@498 11 published by the Free Software Foundation; either version 2 of the
Chris@498 12 License, or (at your option) any later version. See the file
Chris@498 13 COPYING included with this distribution for more information.
Chris@498 14 */
Chris@498 15
Chris@498 16 #include <fstream>
Chris@498 17
Chris@498 18 #include "vamp-hostsdk/PluginHostAdapter.h"
Chris@498 19 #include "vamp-hostsdk/PluginLoader.h"
Chris@498 20
Chris@498 21 #include "RDFFeatureWriter.h"
Chris@498 22 #include "RDFTransformFactory.h"
Chris@498 23
Chris@498 24 #include <QTextStream>
Chris@498 25 #include <QUrl>
Chris@508 26 #include <QFileInfo>
Chris@498 27 #include <QRegExp>
Chris@498 28
Chris@498 29 using namespace std;
Chris@498 30 using Vamp::Plugin;
Chris@498 31 using Vamp::PluginBase;
Chris@498 32
Chris@498 33 RDFFeatureWriter::RDFFeatureWriter() :
Chris@498 34 FileFeatureWriter(SupportOneFilePerTrackTransform |
Chris@498 35 SupportOneFilePerTrack |
Chris@498 36 SupportOneFileTotal,
Chris@498 37 "n3"),
Chris@498 38 m_plain(false),
Chris@498 39 m_count(0)
Chris@498 40 {
Chris@498 41 }
Chris@498 42
Chris@498 43 RDFFeatureWriter::~RDFFeatureWriter()
Chris@498 44 {
Chris@498 45 }
Chris@498 46
Chris@498 47 RDFFeatureWriter::ParameterList
Chris@498 48 RDFFeatureWriter::getSupportedParameters() const
Chris@498 49 {
Chris@498 50 ParameterList pl = FileFeatureWriter::getSupportedParameters();
Chris@498 51 Parameter p;
Chris@498 52
Chris@498 53 p.name = "plain";
Chris@498 54 p.description = "Use \"plain\" RDF even if transform metadata is available.";
Chris@498 55 p.hasArg = false;
Chris@498 56 pl.push_back(p);
Chris@498 57
Chris@586 58 p.name = "audiofile-uri";
Chris@586 59 p.description = "Link the output RDF to the given audio file URI instead of its actual location.";
Chris@498 60 p.hasArg = true;
Chris@498 61 pl.push_back(p);
Chris@498 62
Chris@498 63 return pl;
Chris@498 64 }
Chris@498 65
Chris@498 66 void
Chris@498 67 RDFFeatureWriter::setParameters(map<string, string> &params)
Chris@498 68 {
Chris@498 69 FileFeatureWriter::setParameters(params);
Chris@498 70
Chris@498 71 for (map<string, string>::iterator i = params.begin();
Chris@498 72 i != params.end(); ++i) {
Chris@498 73 if (i->first == "plain") {
Chris@498 74 m_plain = true;
Chris@498 75 }
Chris@586 76 if (i->first == "audiofile-uri") {
Chris@498 77 m_suri = i->second.c_str();
Chris@498 78 }
Chris@498 79 }
Chris@498 80 }
Chris@498 81
Chris@504 82 void
Chris@504 83 RDFFeatureWriter::setTrackMetadata(QString trackId,
Chris@504 84 TrackMetadata metadata)
Chris@504 85 {
Chris@504 86 m_metadata[trackId] = metadata;
Chris@504 87 }
Chris@504 88
Chris@504 89 void
Chris@510 90 RDFFeatureWriter::setFixedEventTypeURI(QString uri)
Chris@510 91 {
Chris@510 92 m_fixedEventTypeURI = uri;
Chris@510 93 }
Chris@510 94
Chris@510 95 void
Chris@504 96 RDFFeatureWriter::write(QString trackId,
Chris@504 97 const Transform &transform,
Chris@504 98 const Plugin::OutputDescriptor& output,
Chris@504 99 const Plugin::FeatureList& features,
Chris@504 100 std::string summaryType)
Chris@498 101 {
Chris@498 102 QString pluginId = transform.getPluginIdentifier();
Chris@498 103
Chris@498 104 if (m_rdfDescriptions.find(pluginId) == m_rdfDescriptions.end()) {
Chris@498 105
Chris@498 106 m_rdfDescriptions[pluginId] = PluginRDFDescription(pluginId);
Chris@498 107
Chris@498 108 if (m_rdfDescriptions[pluginId].haveDescription()) {
Chris@498 109 cerr << "NOTE: Have RDF description for plugin ID \""
Chris@498 110 << pluginId.toStdString() << "\"" << endl;
Chris@498 111 } else {
Chris@498 112 cerr << "NOTE: Do not have RDF description for plugin ID \""
Chris@498 113 << pluginId.toStdString() << "\"" << endl;
Chris@498 114 }
Chris@498 115 }
Chris@498 116
Chris@498 117 // Need to select appropriate output file for our track/transform
Chris@498 118 // combination
Chris@498 119
Chris@498 120 QTextStream *stream = getOutputStream(trackId, transform.getIdentifier());
Chris@512 121 if (!stream) {
Chris@512 122 std::cerr << "RDFFeatureWriter::write: ERROR: No output stream for track id \""
Chris@512 123 << trackId.toStdString() << "\" and transform \""
Chris@512 124 << transform.getIdentifier().toStdString() << "\"" << std::endl;
Chris@513 125 exit(1);
Chris@512 126 }
Chris@498 127
Chris@498 128 if (m_startedStreamTransforms.find(stream) ==
Chris@498 129 m_startedStreamTransforms.end()) {
Chris@530 130 cerr << "This stream is new, writing prefixes" << endl;
Chris@498 131 writePrefixes(stream);
Chris@498 132 if (m_singleFileName == "" && !m_stdout) {
Chris@498 133 writeSignalDescription(stream, trackId);
Chris@498 134 }
Chris@498 135 }
Chris@498 136
Chris@498 137 if (m_startedStreamTransforms[stream].find(transform) ==
Chris@498 138 m_startedStreamTransforms[stream].end()) {
Chris@498 139 m_startedStreamTransforms[stream].insert(transform);
Chris@498 140 writeLocalFeatureTypes
Chris@498 141 (stream, transform, output, m_rdfDescriptions[pluginId]);
Chris@498 142 }
Chris@498 143
Chris@498 144 if (m_singleFileName != "" || m_stdout) {
Chris@498 145 if (m_startedTrackIds.find(trackId) == m_startedTrackIds.end()) {
Chris@498 146 writeSignalDescription(stream, trackId);
Chris@498 147 m_startedTrackIds.insert(trackId);
Chris@498 148 }
Chris@498 149 }
Chris@498 150
Chris@498 151 QString timelineURI = m_trackTimelineURIs[trackId];
Chris@498 152
Chris@498 153 if (timelineURI == "") {
Chris@498 154 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing features without having established a timeline URI!" << endl;
Chris@498 155 exit(1);
Chris@498 156 }
Chris@498 157
Chris@498 158 if (summaryType != "") {
Chris@498 159
Chris@498 160 writeSparseRDF(stream, transform, output, features,
Chris@498 161 m_rdfDescriptions[pluginId], timelineURI);
Chris@498 162
Chris@498 163 } else if (m_rdfDescriptions[pluginId].haveDescription() &&
Chris@498 164 m_rdfDescriptions[pluginId].getOutputDisposition
Chris@498 165 (output.identifier.c_str()) ==
Chris@498 166 PluginRDFDescription::OutputDense) {
Chris@498 167
Chris@498 168 QString signalURI = m_trackSignalURIs[trackId];
Chris@498 169
Chris@498 170 if (signalURI == "") {
Chris@498 171 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing dense features without having established a signal URI!" << endl;
Chris@498 172 exit(1);
Chris@498 173 }
Chris@498 174
Chris@498 175 writeDenseRDF(stream, transform, output, features,
Chris@498 176 m_rdfDescriptions[pluginId], signalURI, timelineURI);
Chris@498 177
Chris@507 178 } else if (!m_plain &&
Chris@507 179 m_rdfDescriptions[pluginId].haveDescription() &&
Chris@507 180 m_rdfDescriptions[pluginId].getOutputDisposition
Chris@507 181 (output.identifier.c_str()) ==
Chris@507 182 PluginRDFDescription::OutputTrackLevel &&
Chris@507 183 m_rdfDescriptions[pluginId].getOutputFeatureAttributeURI
Chris@507 184 (output.identifier.c_str()) != "") {
Chris@507 185
Chris@507 186 QString signalURI = m_trackSignalURIs[trackId];
Chris@507 187
Chris@507 188 if (signalURI == "") {
Chris@507 189 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing track-level features without having established a signal URI!" << endl;
Chris@507 190 exit(1);
Chris@507 191 }
Chris@507 192
Chris@507 193 writeTrackLevelRDF(stream, transform, output, features,
Chris@507 194 m_rdfDescriptions[pluginId], signalURI);
Chris@507 195
Chris@498 196 } else {
Chris@498 197
Chris@498 198 writeSparseRDF(stream, transform, output, features,
Chris@498 199 m_rdfDescriptions[pluginId], timelineURI);
Chris@498 200 }
Chris@498 201 }
Chris@498 202
Chris@498 203 void
Chris@498 204 RDFFeatureWriter::writePrefixes(QTextStream *sptr)
Chris@498 205 {
Chris@498 206 QTextStream &stream = *sptr;
Chris@498 207
Chris@498 208 stream << "@prefix dc: <http://purl.org/dc/elements/1.1/> .\n"
Chris@498 209 << "@prefix mo: <http://purl.org/ontology/mo/> .\n"
Chris@498 210 << "@prefix af: <http://purl.org/ontology/af/> .\n"
Chris@504 211 << "@prefix foaf: <http://xmlns.com/foaf/0.1/> . \n"
Chris@498 212 << "@prefix event: <http://purl.org/NET/c4dm/event.owl#> .\n"
Chris@498 213 << "@prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#> .\n"
Chris@498 214 << "@prefix rdfs: <http://www.w3.org/2000/01/rdf-schema#> .\n"
Chris@498 215 << "@prefix xsd: <http://www.w3.org/2001/XMLSchema#> .\n"
Chris@498 216 << "@prefix tl: <http://purl.org/NET/c4dm/timeline.owl#> .\n"
Chris@498 217 << "@prefix vamp: <http://purl.org/ontology/vamp/> .\n"
Chris@498 218 << "@prefix : <#> .\n\n";
Chris@498 219 }
Chris@498 220
Chris@498 221 void
Chris@498 222 RDFFeatureWriter::writeSignalDescription(QTextStream *sptr,
Chris@498 223 QString trackId)
Chris@498 224 {
Chris@530 225 std::cerr << "RDFFeatureWriter::writeSignalDescription" << std::endl;
Chris@530 226
Chris@498 227 QTextStream &stream = *sptr;
Chris@498 228
Chris@498 229 /*
Chris@498 230 * Describe signal we're analysing (AudioFile, Signal, TimeLine, etc.)
Chris@498 231 */
Chris@498 232
Chris@498 233 QUrl url(trackId);
Chris@498 234 QString scheme = url.scheme().toLower();
Chris@498 235 bool local = (scheme == "" || scheme == "file" || scheme.length() == 1);
Chris@498 236
Chris@498 237 if (local) {
Chris@498 238 if (scheme == "") {
Chris@498 239 url.setScheme("file");
Chris@508 240 url.setPath(QFileInfo(url.path()).absoluteFilePath());
Chris@498 241 } else if (scheme.length() == 1) { // DOS drive letter!
Chris@498 242 url.setScheme("file");
Chris@498 243 url.setPath(scheme + ":" + url.path());
Chris@498 244 }
Chris@498 245 }
Chris@498 246
Chris@498 247 //!!! FIX: If we are appending, we need to start counting after
Chris@498 248 //all of the existing counts that are already in the file!
Chris@498 249
Chris@498 250 uint64_t signalCount = m_count++;
Chris@498 251
Chris@498 252 if (m_trackSignalURIs.find(trackId) == m_trackSignalURIs.end()) {
Chris@498 253 m_trackSignalURIs[trackId] = QString(":signal_%1").arg(signalCount);
Chris@498 254 }
Chris@498 255 QString signalURI = m_trackSignalURIs[trackId];
Chris@498 256
Chris@585 257 if (m_trackTrackURIs.find(trackId) == m_trackTrackURIs.end()) {
Chris@585 258 m_trackTrackURIs[trackId] = QString(":track_%1").arg(signalCount);
Chris@585 259 }
Chris@585 260 QString trackURI = m_trackTrackURIs[trackId];
Chris@585 261
Chris@498 262 if (m_trackTimelineURIs.find(trackId) == m_trackTimelineURIs.end()) {
Chris@498 263 m_trackTimelineURIs[trackId] = QString(":signal_timeline_%1").arg(signalCount);
Chris@498 264 }
Chris@498 265 QString timelineURI = m_trackTimelineURIs[trackId];
Chris@498 266
Chris@586 267 QString afURI = url.toEncoded().data();
Chris@586 268 if (m_suri != NULL) afURI = m_suri;
Chris@586 269
Chris@585 270 if (m_metadata.find(trackId) != m_metadata.end()) {
Chris@585 271 TrackMetadata tm = m_metadata[trackId];
Chris@585 272 if (tm.title != "" || tm.maker != "") {
Chris@585 273 // We only write a Track at all if we have some
Chris@585 274 // title/artist metadata to put in it. Otherwise we can't
Chris@585 275 // be sure that what we have is a Track, in the
Chris@585 276 // publication sense -- it may just be a fragment, a test
Chris@585 277 // file, whatever. Since we'd have no metadata to
Chris@585 278 // associate with our Track, the only effect of including
Chris@585 279 // a Track would be to assert that this was one, which is
Chris@585 280 // the one thing we wouldn't know...
Chris@585 281 stream << trackURI << " a mo:Track ";
Chris@585 282 if (tm.title != "") {
Chris@585 283 stream << ";\n dc:title \"\"\"" << tm.title << "\"\"\" ";
Chris@585 284 }
Chris@585 285 if (tm.maker != "") {
Chris@585 286 stream << ";\n foaf:maker [ a mo:MusicArtist; foaf:name \"\"\"" << tm.maker << "\"\"\" ] ";
Chris@585 287 }
Chris@585 288 if (trackId != "") {
Chris@586 289 stream << ";\n mo:available_as <" << afURI << "> ";
Chris@585 290 }
Chris@585 291 stream << ".\n\n";
Chris@585 292 }
Chris@585 293 }
Chris@585 294
Chris@500 295 if (trackId != "") {
Chris@586 296 stream << "<" << afURI << "> a mo:AudioFile ;\n";
Chris@585 297 stream << " mo:encodes " << signalURI << ".\n\n";
Chris@500 298 }
Chris@500 299
Chris@500 300 stream << signalURI << " a mo:Signal ;\n";
Chris@500 301
Chris@500 302 stream << " mo:time [\n"
Chris@498 303 << " a tl:Interval ;\n"
Chris@498 304 << " tl:onTimeLine "
Chris@498 305 << timelineURI << "\n ] .\n\n";
Chris@585 306
Chris@585 307 stream << timelineURI << " a tl:Timeline .\n";
Chris@498 308 }
Chris@498 309
Chris@498 310 void
Chris@498 311 RDFFeatureWriter::writeLocalFeatureTypes(QTextStream *sptr,
Chris@498 312 const Transform &transform,
Chris@498 313 const Plugin::OutputDescriptor &od,
Chris@498 314 PluginRDFDescription &desc)
Chris@498 315 {
Chris@498 316 QString outputId = od.identifier.c_str();
Chris@498 317 QTextStream &stream = *sptr;
Chris@498 318
Chris@507 319 // There is no "needFeatureType" for track-level outputs, because
Chris@507 320 // we can't meaningfully write a feature at all if we don't know
Chris@507 321 // what property to use for it. If the output is track level but
Chris@507 322 // there is no feature type given, we have to revert to events.
Chris@507 323
Chris@498 324 bool needEventType = false;
Chris@498 325 bool needSignalType = false;
Chris@498 326
Chris@498 327 //!!! bin names, extents and so on can be written out using e.g. vamp:bin_names ( "a" "b" "c" )
Chris@498 328
Chris@498 329 if (desc.getOutputDisposition(outputId) ==
Chris@498 330 PluginRDFDescription::OutputDense) {
Chris@498 331
Chris@498 332 // no feature events, so may need signal type but won't need
Chris@498 333 // event type
Chris@498 334
Chris@498 335 if (m_plain) {
Chris@498 336
Chris@498 337 needSignalType = true;
Chris@498 338
Chris@498 339 } else if (desc.getOutputSignalTypeURI(outputId) == "") {
Chris@498 340
Chris@498 341 needSignalType = true;
Chris@498 342 }
Chris@498 343
Chris@507 344 } else if (desc.getOutputDisposition(outputId) ==
Chris@507 345 PluginRDFDescription::OutputTrackLevel) {
Chris@507 346
Chris@507 347 // see note above -- need to generate an event type if no
Chris@507 348 // feature type given, or if in plain mode
Chris@507 349
Chris@507 350 if (m_plain) {
Chris@507 351
Chris@507 352 needEventType = true;
Chris@507 353
Chris@507 354 } else if (desc.getOutputFeatureAttributeURI(outputId) == "") {
Chris@507 355
Chris@507 356 if (desc.getOutputEventTypeURI(outputId) == "") {
Chris@507 357
Chris@507 358 needEventType = true;
Chris@507 359 }
Chris@507 360 }
Chris@507 361
Chris@498 362 } else {
Chris@498 363
Chris@498 364 // may need event type but won't need signal type
Chris@498 365
Chris@498 366 if (m_plain) {
Chris@498 367
Chris@498 368 needEventType = true;
Chris@498 369
Chris@498 370 } else if (desc.getOutputEventTypeURI(outputId) == "") {
Chris@498 371
Chris@498 372 needEventType = true;
Chris@498 373 }
Chris@498 374 }
Chris@498 375
Chris@498 376 QString transformUri;
Chris@498 377 if (m_transformURIs.find(transform) != m_transformURIs.end()) {
Chris@498 378 transformUri = m_transformURIs[transform];
Chris@498 379 } else {
Chris@498 380 transformUri = QString(":transform_%1_%2").arg(m_count++).arg(outputId);
Chris@498 381 m_transformURIs[transform] = transformUri;
Chris@498 382 }
Chris@498 383
Chris@500 384 if (transform.getIdentifier() != "") {
Chris@508 385 stream << endl
Chris@508 386 << RDFTransformFactory::writeTransformToRDF(transform, transformUri)
Chris@500 387 << endl;
Chris@500 388 }
Chris@498 389
Chris@510 390 if (needEventType && m_fixedEventTypeURI == "") {
Chris@498 391
Chris@498 392 QString uri;
Chris@498 393 if (m_syntheticEventTypeURIs.find(transform) !=
Chris@498 394 m_syntheticEventTypeURIs.end()) {
Chris@498 395 uri = m_syntheticEventTypeURIs[transform];
Chris@498 396 } else {
Chris@498 397 uri = QString(":event_type_%1").arg(m_count++);
Chris@498 398 m_syntheticEventTypeURIs[transform] = uri;
Chris@498 399 }
Chris@498 400
Chris@498 401 stream << uri
Chris@498 402 << " rdfs:subClassOf event:Event ;" << endl
Chris@498 403 << " dc:title \"" << od.name.c_str() << "\" ;" << endl
Chris@498 404 << " dc:format \"" << od.unit.c_str() << "\" ;" << endl
Chris@498 405 << " dc:description \"" << od.description.c_str() << "\" ."
Chris@498 406 << endl << endl;
Chris@498 407 }
Chris@498 408
Chris@498 409 if (needSignalType) {
Chris@498 410
Chris@498 411 QString uri;
Chris@498 412 if (m_syntheticSignalTypeURIs.find(transform) !=
Chris@498 413 m_syntheticSignalTypeURIs.end()) {
Chris@498 414 uri = m_syntheticSignalTypeURIs[transform];
Chris@498 415 } else {
Chris@498 416 uri = QString(":signal_type_%1").arg(m_count++);
Chris@498 417 m_syntheticSignalTypeURIs[transform] = uri;
Chris@498 418 }
Chris@498 419
Chris@498 420 stream << uri
Chris@498 421 << " rdfs:subClassOf af:Signal ;" << endl
Chris@498 422 << " dc:title \"" << od.name.c_str() << "\" ;" << endl
Chris@498 423 << " dc:format \"" << od.unit.c_str() << "\" ;" << endl
Chris@498 424 << " dc:description \"" << od.description.c_str() << "\" ."
Chris@498 425 << endl << endl;
Chris@498 426 }
Chris@498 427 }
Chris@498 428
Chris@498 429 void
Chris@498 430 RDFFeatureWriter::writeSparseRDF(QTextStream *sptr,
Chris@498 431 const Transform &transform,
Chris@498 432 const Plugin::OutputDescriptor& od,
Chris@498 433 const Plugin::FeatureList& featureList,
Chris@498 434 PluginRDFDescription &desc,
Chris@498 435 QString timelineURI)
Chris@498 436 {
Chris@512 437 // std::cerr << "RDFFeatureWriter::writeSparseRDF: have " << featureList.size() << " features" << std::endl;
Chris@512 438
Chris@498 439 if (featureList.empty()) return;
Chris@498 440 QTextStream &stream = *sptr;
Chris@498 441
Chris@498 442 bool plain = (m_plain || !desc.haveDescription());
Chris@498 443
Chris@498 444 QString outputId = od.identifier.c_str();
Chris@498 445
Chris@498 446 // iterate through FeatureLists
Chris@498 447
Chris@498 448 for (int i = 0; i < featureList.size(); ++i) {
Chris@498 449
Chris@498 450 const Plugin::Feature &feature = featureList[i];
Chris@498 451 uint64_t featureNumber = m_count++;
Chris@498 452
Chris@498 453 stream << ":event_" << featureNumber << " a ";
Chris@498 454
Chris@510 455 if (m_fixedEventTypeURI != "") {
Chris@510 456 stream << m_fixedEventTypeURI << " ;\n";
Chris@510 457 } else {
Chris@510 458 QString eventTypeURI = desc.getOutputEventTypeURI(outputId);
Chris@510 459 if (plain || eventTypeURI == "") {
Chris@510 460 if (m_syntheticEventTypeURIs.find(transform) !=
Chris@510 461 m_syntheticEventTypeURIs.end()) {
Chris@510 462 stream << m_syntheticEventTypeURIs[transform] << " ;\n";
Chris@510 463 } else {
Chris@510 464 stream << ":event_type_" << outputId << " ;\n";
Chris@510 465 }
Chris@498 466 } else {
Chris@510 467 stream << "<" << eventTypeURI << "> ;\n";
Chris@498 468 }
Chris@498 469 }
Chris@498 470
Chris@498 471 QString timestamp = feature.timestamp.toString().c_str();
Chris@498 472 timestamp.replace(QRegExp("^ +"), "");
Chris@498 473
Chris@498 474 if (feature.hasDuration && feature.duration > Vamp::RealTime::zeroTime) {
Chris@498 475
Chris@498 476 QString duration = feature.duration.toString().c_str();
Chris@498 477 duration.replace(QRegExp("^ +"), "");
Chris@498 478
Chris@498 479 stream << " event:time [ \n"
Chris@498 480 << " a tl:Interval ;\n"
Chris@498 481 << " tl:onTimeLine " << timelineURI << " ;\n"
Chris@498 482 << " tl:beginsAt \"PT" << timestamp
Chris@498 483 << "S\"^^xsd:duration ;\n"
Chris@498 484 << " tl:duration \"PT" << duration
Chris@498 485 << "S\"^^xsd:duration ;\n"
Chris@498 486 << " ] ";
Chris@498 487
Chris@498 488 } else {
Chris@498 489
Chris@498 490 stream << " event:time [ \n"
Chris@498 491 << " a tl:Instant ;\n" //location of the event in time
Chris@498 492 << " tl:onTimeLine " << timelineURI << " ;\n"
Chris@498 493 << " tl:at \"PT" << timestamp
Chris@498 494 << "S\"^^xsd:duration ;\n ] ";
Chris@498 495 }
Chris@498 496
Chris@500 497 if (transform.getIdentifier() != "") {
Chris@500 498 stream << ";\n";
Chris@500 499 stream << " vamp:computed_by " << m_transformURIs[transform] << " ";
Chris@500 500 }
Chris@498 501
Chris@498 502 if (feature.label.length() > 0) {
Chris@498 503 stream << ";\n";
Chris@510 504 stream << " rdfs:label \"\"\"" << feature.label.c_str() << "\"\"\" ";
Chris@498 505 }
Chris@498 506
Chris@498 507 if (!feature.values.empty()) {
Chris@498 508 stream << ";\n";
Chris@498 509 //!!! named bins?
Chris@498 510 stream << " af:feature \"" << feature.values[0];
Chris@498 511 for (int j = 1; j < feature.values.size(); ++j) {
Chris@498 512 stream << " " << feature.values[j];
Chris@498 513 }
Chris@498 514 stream << "\" ";
Chris@498 515 }
Chris@498 516
Chris@498 517 stream << ".\n";
Chris@498 518 }
Chris@498 519 }
Chris@498 520
Chris@498 521 void
Chris@507 522 RDFFeatureWriter::writeTrackLevelRDF(QTextStream *sptr,
Chris@507 523 const Transform &transform,
Chris@507 524 const Plugin::OutputDescriptor& od,
Chris@507 525 const Plugin::FeatureList& featureList,
Chris@507 526 PluginRDFDescription &desc,
Chris@507 527 QString signalURI)
Chris@507 528 {
Chris@507 529 if (featureList.empty()) return;
Chris@507 530 QTextStream &stream = *sptr;
Chris@507 531
Chris@507 532 bool plain = (m_plain || !desc.haveDescription());
Chris@507 533
Chris@507 534 QString outputId = od.identifier.c_str();
Chris@507 535 QString featureUri = desc.getOutputFeatureAttributeURI(outputId);
Chris@507 536
Chris@507 537 if (featureUri == "") {
Chris@507 538 cerr << "RDFFeatureWriter::writeTrackLevelRDF: ERROR: No feature URI available -- this function should not have been called!" << endl;
Chris@507 539 return;
Chris@507 540 }
Chris@507 541
Chris@507 542 for (int i = 0; i < featureList.size(); ++i) {
Chris@507 543
Chris@507 544 const Plugin::Feature &feature = featureList[i];
Chris@507 545
Chris@507 546 if (feature.values.empty()) {
Chris@507 547
Chris@507 548 if (feature.label == "") continue;
Chris@507 549
Chris@508 550 stream << signalURI << " " << featureUri << " \"\"\""
Chris@508 551 << feature.label.c_str() << "\"\"\" .\n";
Chris@507 552
Chris@507 553 } else {
Chris@507 554
Chris@507 555 stream << signalURI << " " << featureUri << " \""
Chris@507 556 << feature.values[0] << "\"^^xsd:float .\n";
Chris@507 557 }
Chris@507 558 }
Chris@507 559 }
Chris@507 560
Chris@507 561 void
Chris@498 562 RDFFeatureWriter::writeDenseRDF(QTextStream *sptr,
Chris@498 563 const Transform &transform,
Chris@498 564 const Plugin::OutputDescriptor& od,
Chris@498 565 const Plugin::FeatureList& featureList,
Chris@498 566 PluginRDFDescription &desc,
Chris@498 567 QString signalURI,
Chris@498 568 QString timelineURI)
Chris@498 569 {
Chris@498 570 if (featureList.empty()) return;
Chris@498 571
Chris@498 572 StringTransformPair sp(signalURI, transform);
Chris@498 573
Chris@498 574 if (m_openDenseFeatures.find(sp) == m_openDenseFeatures.end()) {
Chris@498 575
Chris@498 576 StreamBuffer b(sptr, "");
Chris@498 577 m_openDenseFeatures[sp] = b;
Chris@498 578
Chris@498 579 QString &str(m_openDenseFeatures[sp].second);
Chris@498 580 QTextStream stream(&str);
Chris@498 581
Chris@498 582 bool plain = (m_plain || !desc.haveDescription());
Chris@498 583 QString outputId = od.identifier.c_str();
Chris@498 584
Chris@498 585 uint64_t featureNumber = m_count++;
Chris@498 586
Chris@498 587 // need to write out feature timeline map -- for this we need
Chris@498 588 // the sample rate, window length and hop size from the
Chris@498 589 // transform
Chris@498 590
Chris@498 591 stream << "\n:feature_timeline_" << featureNumber << " a tl:DiscreteTimeLine .\n\n";
Chris@498 592
Chris@498 593 size_t stepSize = transform.getStepSize();
Chris@498 594 if (stepSize == 0) {
Chris@498 595 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing dense features without having set the step size properly!" << endl;
Chris@498 596 return;
Chris@498 597 }
Chris@498 598
Chris@498 599 size_t blockSize = transform.getBlockSize();
Chris@498 600 if (blockSize == 0) {
Chris@498 601 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing dense features without having set the block size properly!" << endl;
Chris@498 602 return;
Chris@498 603 }
Chris@498 604
Chris@498 605 float sampleRate = transform.getSampleRate();
Chris@498 606 if (sampleRate == 0.f) {
Chris@498 607 cerr << "RDFFeatureWriter: INTERNAL ERROR: writing dense features without having set the sample rate properly!" << endl;
Chris@498 608 return;
Chris@498 609 }
Chris@498 610
Chris@498 611 stream << ":feature_timeline_map_" << featureNumber
Chris@498 612 << " a tl:UniformSamplingWindowingMap ;\n"
Chris@498 613 << " tl:rangeTimeLine :feature_timeline_" << featureNumber << " ;\n"
Chris@498 614 << " tl:domainTimeLine " << timelineURI << " ;\n"
Chris@498 615 << " tl:sampleRate \"" << int(sampleRate) << "\"^^xsd:int ;\n"
Chris@498 616 << " tl:windowLength \"" << blockSize << "\"^^xsd:int ;\n"
Chris@498 617 << " tl:hopSize \"" << stepSize << "\"^^xsd:int .\n\n";
Chris@498 618
Chris@498 619 stream << signalURI << " af:signal_feature :feature_"
Chris@498 620 << featureNumber << " ." << endl << endl;
Chris@498 621
Chris@498 622 stream << ":feature_" << featureNumber << " a ";
Chris@498 623
Chris@498 624 QString signalTypeURI = desc.getOutputSignalTypeURI(outputId);
Chris@498 625 if (plain || signalTypeURI == "") {
Chris@498 626 if (m_syntheticSignalTypeURIs.find(transform) !=
Chris@498 627 m_syntheticSignalTypeURIs.end()) {
Chris@498 628 stream << m_syntheticSignalTypeURIs[transform] << " ;\n";
Chris@498 629 } else {
Chris@498 630 stream << ":signal_type_" << outputId << " ;\n";
Chris@498 631 }
Chris@498 632 } else {
Chris@498 633 stream << signalTypeURI << " ;\n";
Chris@498 634 }
Chris@498 635
Chris@498 636 stream << " mo:time ["
Chris@498 637 << "\n a tl:Interval ;"
Chris@498 638 << "\n tl:onTimeLine :feature_timeline_" << featureNumber << " ;";
Chris@498 639
Chris@498 640 RealTime startrt = transform.getStartTime();
Chris@498 641 RealTime durationrt = transform.getDuration();
Chris@498 642
Chris@498 643 int start = RealTime::realTime2Frame(startrt, sampleRate) / stepSize;
Chris@498 644 int duration = RealTime::realTime2Frame(durationrt, sampleRate) / stepSize;
Chris@498 645
Chris@498 646 if (start != 0) {
Chris@498 647 stream << "\n tl:start \"" << start << "\"^^xsd:int ;";
Chris@498 648 }
Chris@498 649 if (duration != 0) {
Chris@498 650 stream << "\n tl:duration \"" << duration << "\"^^xsd:int ;";
Chris@498 651 }
Chris@498 652
Chris@498 653 stream << "\n ] ;\n";
Chris@498 654
Chris@584 655 if (transform.getIdentifier() != "") {
Chris@584 656 stream << " vamp:computed_by " << m_transformURIs[transform] << " ;\n";
Chris@584 657 }
Chris@584 658
Chris@498 659 if (od.hasFixedBinCount) {
Chris@498 660 // We only know the height, so write the width as zero
Chris@498 661 stream << " af:dimensions \"" << od.binCount << " 0\" ;\n";
Chris@498 662 }
Chris@498 663
Chris@498 664 stream << " af:value \"";
Chris@498 665 }
Chris@498 666
Chris@498 667 QString &str = m_openDenseFeatures[sp].second;
Chris@498 668 QTextStream stream(&str);
Chris@498 669
Chris@498 670 for (int i = 0; i < featureList.size(); ++i) {
Chris@498 671
Chris@498 672 const Plugin::Feature &feature = featureList[i];
Chris@498 673
Chris@498 674 for (int j = 0; j < feature.values.size(); ++j) {
Chris@498 675 stream << feature.values[j] << " ";
Chris@498 676 }
Chris@498 677 }
Chris@498 678 }
Chris@498 679
Chris@498 680 void RDFFeatureWriter::finish()
Chris@498 681 {
Chris@578 682 // cerr << "RDFFeatureWriter::finish()" << endl;
Chris@498 683
Chris@498 684 // close any open dense feature literals
Chris@498 685
Chris@498 686 for (map<StringTransformPair, StreamBuffer>::iterator i =
Chris@498 687 m_openDenseFeatures.begin();
Chris@498 688 i != m_openDenseFeatures.end(); ++i) {
Chris@512 689 // cerr << "closing a stream" << endl;
Chris@498 690 StreamBuffer &b = i->second;
Chris@498 691 *(b.first) << b.second << "\" ." << endl;
Chris@498 692 }
Chris@498 693
Chris@498 694 m_openDenseFeatures.clear();
Chris@530 695 m_startedStreamTransforms.clear();
Chris@530 696
Chris@530 697 FileFeatureWriter::finish();
Chris@498 698 }
Chris@498 699
Chris@498 700