Chris@34: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@34: 
Chris@34: /*
Chris@34:   Silvet
Chris@34: 
Chris@34:   A Vamp plugin for note transcription.
Chris@34:   Centre for Digital Music, Queen Mary University of London.
Chris@34:     
Chris@34:   This program is free software; you can redistribute it and/or
Chris@34:   modify it under the terms of the GNU General Public License as
Chris@34:   published by the Free Software Foundation; either version 2 of the
Chris@34:   License, or (at your option) any later version.  See the file
Chris@34:   COPYING included with this distribution for more information.
Chris@34: */
Chris@34: 
Chris@34: #include "EM.h"
Chris@34: 
Chris@34: #include "data/include/templates.h"
Chris@34: 
Chris@35: static double epsilon = 1e-16;
Chris@35: 
Chris@35: EM::EM() :
Chris@35:     m_notes(SILVET_TEMPLATE_NOTE_COUNT),
Chris@35:     m_bins(SILVET_TEMPLATE_HEIGHT),
Chris@35:     m_instruments(SILVET_TEMPLATE_COUNT)
Chris@35: {
Chris@35:     m_lowest = 0;
Chris@35:     m_highest = m_notes - 1;
Chris@35: 
Chris@35:     for (int i = 0; i < m_instruments; ++i) {
Chris@35:         if (i == 0 || silvet_templates[i].lowest < m_lowest) {
Chris@35:             m_lowest = silvet_templates[i].lowest;
Chris@35:         }
Chris@35:         if (i == 0 || silvet_templates[i].highest > m_highest) {
Chris@35:             m_highest = silvet_templates[i].highest;
Chris@35:         }
Chris@35:     }
Chris@35: 
Chris@35:     m_pitches = V(m_notes);
Chris@35: 
Chris@35:     for (int n = 0; n < m_notes; ++i) {
Chris@35:         m_pitches[n] = drand48();
Chris@35:     }
Chris@35:     
Chris@35:     m_sources = Grid(m_instruments);
Chris@35:     
Chris@35:     for (int i = 0; i < m_instruments; ++i) {
Chris@35:         m_sources[i] = V(m_notes);
Chris@35:         for (int n = 0; n < m_notes; ++n) {
Chris@35:             m_sources[i][n] = (inRange(i, n) ? 1.0 : 0.0);
Chris@35:         }
Chris@35:     }
Chris@35: 
Chris@35:     m_q = V(m_bins);
Chris@35:     
Chris@35:     for (int w = 0; w < m_bins; ++w) {
Chris@35:         m_q[w] = epsilon;
Chris@35:     }
Chris@35: }
Chris@35: 
Chris@35: EM::~EM()
Chris@35: {
Chris@35: }
Chris@35: 
Chris@35: bool
Chris@35: EM::inRange(int instrument, int note)
Chris@35: {
Chris@35:     return (note >= silvet_templates[instrument].lowest &&
Chris@35:             note <= silvet_templates[instrument].highest);
Chris@35: }
Chris@35: