Chris@161: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@161: 
Chris@161: /*
Chris@161:   Silvet
Chris@161: 
Chris@161:   A Vamp plugin for note transcription.
Chris@161:   Centre for Digital Music, Queen Mary University of London.
Chris@161:     
Chris@161:   This program is free software; you can redistribute it and/or
Chris@161:   modify it under the terms of the GNU General Public License as
Chris@161:   published by the Free Software Foundation; either version 2 of the
Chris@161:   License, or (at your option) any later version.  See the file
Chris@161:   COPYING included with this distribution for more information.
Chris@161: */
Chris@161: 
Chris@161: #include "Instruments.h"
Chris@161: 
Chris@161: #include "data/include/templates.h"
Chris@161: 
Chris@161: #include <iostream>
Chris@161: 
Chris@161: const int InstrumentPack::templateNoteCount = SILVET_TEMPLATE_NOTE_COUNT;
Chris@161: const int InstrumentPack::templateHeight = SILVET_TEMPLATE_HEIGHT;
Chris@161: const int InstrumentPack::templateMaxShift = SILVET_TEMPLATE_MAX_SHIFT;
Chris@161: const int InstrumentPack::templateSize = SILVET_TEMPLATE_SIZE;
Chris@161: 
Chris@161: using std::string;
Chris@161: using std::vector;
Chris@161: using std::cerr;
Chris@161: using std::endl;
Chris@161: 
Chris@161: const char *simpleInstruments[] = {
Chris@161:     "Guitar", "guitar",
Chris@161:     "Violin", "violin",
Chris@161:     "Cello", "cello",
Chris@161:     "Horn", "horn",
Chris@161:     "Flute", "flute",
Chris@161:     "Oboe", "oboe",
Chris@161:     "Clarinet", "clarinet",
Chris@161:     "Tenor Sax", "tenorsax",
Chris@161:     "Bassoon", "bassoon",
Chris@161: };
Chris@161: 
Chris@161: static InstrumentPack::Templates
Chris@161: templatesFor(string name)
Chris@161: {
Chris@161:     for (int i = 0; i < SILVET_TEMPLATE_COUNT; ++i) {
Chris@161: 
Chris@161: 	if (string(silvet_templates[i].name) == name) {
Chris@161: 
Chris@161: 	    silvet_template_t *t = &silvet_templates[i];
Chris@161: 
Chris@161: 	    InstrumentPack::Templates rt;
Chris@161: 	    rt.lowestNote = t->lowest;
Chris@161: 	    rt.highestNote = t->highest;
Chris@161: 	    rt.data = vector<vector<float> >
Chris@161: 		(SILVET_TEMPLATE_NOTE_COUNT,
Chris@161: 		 vector<float>(SILVET_TEMPLATE_SIZE, 0.f));
Chris@161: 
Chris@161: 	    for (int j = 0; j < SILVET_TEMPLATE_NOTE_COUNT; ++j) {
Chris@161: 		for (int k = 0; k < SILVET_TEMPLATE_SIZE; ++k) {
Chris@161: 		    rt.data[j][k] = t->data[j][k];
Chris@161: 		}
Chris@161: 	    }
Chris@161: 
Chris@161: 	    return rt;
Chris@161: 	}
Chris@161:     }
Chris@161: 
Chris@161:     return InstrumentPack::Templates();
Chris@161: }
Chris@161: 
Chris@161: static bool 
Chris@161: isOK(InstrumentPack &d)
Chris@161: {
Chris@161:     if (d.name == "") {
Chris@161: 	cerr << "ERROR: Silvet::InstrumentPack: Empty name in instrument definition" << endl;
Chris@161: 	return false;
Chris@161:     }
Chris@161:     if (d.templates.empty()) {
Chris@161: 	cerr << "ERROR: Silvet::InstrumentPack: Instrument definition \"" << d.name << "\" contains no templates!" << endl;
Chris@161: 	return false;
Chris@161:     }
Chris@161:     for (int i = 0; i < int(d.templates.size()); ++i) {
Chris@161: 	if (d.templates[i].data.empty()) {
Chris@161: 	    cerr << "ERROR: Silvet::InstrumentPack: Instrument definition \"" << d.name << "\" contains one or more empty templates!" << endl;
Chris@161: 	    return false;
Chris@161: 	}
Chris@161:     }
Chris@161:     return true;
Chris@161: }
Chris@161: 
Chris@161: vector<InstrumentPack> 
Chris@161: InstrumentPack::listInstrumentPacks()
Chris@161: {
Chris@161:     vector<InstrumentPack> ii;
Chris@161: 
Chris@161:     vector<Templates> allTemplates;
Chris@161:     //!!! is piano-maps-SptkBGCl the same as one of piano1, piano2, piano3?
Chris@161:     allTemplates.push_back(templatesFor("piano1"));
Chris@161: 
Chris@161:     vector<Templates> pianoTemplates;
Chris@161:     pianoTemplates.push_back(templatesFor("piano1"));
Chris@161:     pianoTemplates.push_back(templatesFor("piano2"));
Chris@161:     pianoTemplates.push_back(templatesFor("piano3"));
Chris@161:     InstrumentPack piano(silvet_templates_lowest_note,
Chris@161: 			 silvet_templates_highest_note,
Chris@161: 			 "Piano",
Chris@161: 			 pianoTemplates);
Chris@161:     if (isOK(piano)) {
Chris@161: 	ii.push_back(piano);
Chris@161:     }
Chris@161: 
Chris@161:     for (int i = 0;
Chris@161: 	 i < int(sizeof(simpleInstruments)/sizeof(simpleInstruments[0]));
Chris@161: 	 i += 2) {
Chris@161: 	vector<Templates> tt;
Chris@161: 	Templates t = templatesFor(simpleInstruments[i+1]);
Chris@161: 	tt.push_back(t);
Chris@161: 	allTemplates.push_back(t);
Chris@161: 	InstrumentPack instr(t.lowestNote,
Chris@161: 			     t.highestNote,
Chris@161: 			     simpleInstruments[i],
Chris@161: 			     tt);
Chris@161: 	if (isOK(instr)) {
Chris@161: 	    ii.push_back(instr);
Chris@161: 	}
Chris@161:     }
Chris@161: 
Chris@161:     InstrumentPack all(silvet_templates_lowest_note,
Chris@161: 		       silvet_templates_highest_note,
Chris@161: 		       "Multiple or unknown instruments",
Chris@161: 		       allTemplates);
Chris@161:     if (isOK(all)) {
Chris@161: 	ii.insert(ii.begin(), all);
Chris@161:     }
Chris@161: 
Chris@161:     return ii;
Chris@161: }
Chris@161: