comparison src/Silvet.cpp @ 294:19fd6cb033c7

Add pitch activation matrix output

     d.unit = "";
     d.hasFixedBinCount = true;
     d.binCount = m_instruments[0].templateHeight;
     d.binNames.clear();
     if (m_cq) {
-        char name[20];
+        char name[50];
         for (int i = 0; i < m_instruments[0].templateHeight; ++i) {
             // We have a 600-bin (10 oct 60-bin CQ) of which the
             // lowest-frequency 55 bins have been dropped, for a
             // 545-bin template. The native CQ bins go high->low
             // frequency though, so these are still the first 545 bins

     d.isQuantized = false;
     d.sampleType = OutputDescriptor::FixedSampleRate;
     d.sampleRate = m_colsPerSec;
     d.hasDuration = false;
     m_fcqOutputNo = list.size();
+    list.push_back(d);
+
+    d.identifier = "pitchactivation";
+    d.name = "Pitch activation distribution";
+    d.description = "Pitch activation distribution resulting from expectation-maximisation algorithm, prior to note extraction.";
+    d.unit = "";
+    d.hasFixedBinCount = true;
+    d.binCount = m_instruments[0].templateNoteCount;
+    d.binNames.clear();
+    if (m_cq) {
+        for (int i = 0; i < m_instruments[0].templateNoteCount; ++i) {
+            d.binNames.push_back(noteName(i, 0, 1));
+        }
+    }
+    d.hasKnownExtents = false;
+    d.isQuantized = false;
+    d.sampleType = OutputDescriptor::FixedSampleRate;
+    d.sampleRate = m_colsPerSec;
+    d.hasDuration = false;
+    m_pitchOutputNo = list.size();
     list.push_back(d);
 
     return list;
 }
 

                 m_pianoRollShifts.push_back(map<int, int>());
             }
             continue;
         }
 
-        postProcess(localPitches[i], localBestShifts[i], wantShifts);
+        vector<double> filtered = postProcess
+            (localPitches[i], localBestShifts[i], wantShifts);
+
+        Feature f;
+        for (int j = 0; j < (int)filtered.size(); ++j) {
+            float v(filtered[j]);
+            if (v < pack.levelThreshold) v = 0.f;
+            f.values.push_back(v);
+        }
+        fs[m_pitchOutputNo].push_back(f);
         
         FeatureList noteFeatures = noteTrack(shiftCount);
 
         for (FeatureList::const_iterator fi = noteFeatures.begin();
              fi != noteFeatures.end(); ++fi) {

     }
 
     return out;
 }
     
-void
+vector<double>
 Silvet::postProcess(const vector<double> &pitches,
                     const vector<int> &bestShifts,
                     bool wantShifts)
 {
     const InstrumentPack &pack = m_instruments[m_instrument];

     m_pianoRoll.push_back(active);
 
     if (wantShifts) {
         m_pianoRollShifts.push_back(activeShifts);
     }
+
+    return filtered;
 }
 
 Vamp::Plugin::FeatureList
 Silvet::noteTrack(int shiftCount)
 {