comparison transform/FeatureExtractionPluginTransform.cpp @ 134:7a5ba9dadbf7

* Fix #1706927 NaNs from plugin outputs should not be used

 //    std::cerr << "FeatureExtractionPluginTransform: output sample type "
 //	      << m_descriptor->sampleType << std::endl;
 
     int binCount = 1;
     float minValue = 0.0, maxValue = 0.0;
+    bool haveExtents = false;
     
     if (m_descriptor->hasFixedBinCount) {
 	binCount = m_descriptor->binCount;
     }
 

 //	      << binCount << std::endl;
 
     if (binCount > 0 && m_descriptor->hasKnownExtents) {
 	minValue = m_descriptor->minValue;
 	maxValue = m_descriptor->maxValue;
+        haveExtents = true;
     }
 
     size_t modelRate = m_input->getSampleRate();
     size_t modelResolution = 1;
     

 	m_output = new SparseOneDimensionalModel(modelRate, modelResolution,
 						 false);
 
     } else if (binCount == 1) {
 
-        SparseTimeValueModel *model = new SparseTimeValueModel
-            (modelRate, modelResolution, minValue, maxValue, false);
+        SparseTimeValueModel *model;
+        if (haveExtents) {
+            model = new SparseTimeValueModel
+                (modelRate, modelResolution, minValue, maxValue, false);
+        } else {
+            model = new SparseTimeValueModel
+                (modelRate, modelResolution, false);
+        }
         model->setScaleUnits(outputs[m_outputFeatureNo].unit.c_str());
 
         m_output = model;
 
     } else if (m_descriptor->sampleType ==

         // treat the first as pitch, second as duration in frames,
         // third (if present) as velocity. (Our note model doesn't
         // yet store velocity.)
         //!!! todo: ask the user!
 	
-        NoteModel *model = new NoteModel
-            (modelRate, modelResolution, minValue, maxValue, false);
+        NoteModel *model;
+        if (haveExtents) {
+            model = new NoteModel
+                (modelRate, modelResolution, minValue, maxValue, false);
+        } else {
+            model = new NoteModel
+                (modelRate, modelResolution, false);
+        }            
         model->setScaleUnits(outputs[m_outputFeatureNo].unit.c_str());
 
         m_output = model;
 
     } else {
-	
-	m_output = new EditableDenseThreeDimensionalModel
+
+        EditableDenseThreeDimensionalModel *model =
+            new EditableDenseThreeDimensionalModel
             (modelRate, modelResolution, binCount, false);
 
 	if (!m_descriptor->binNames.empty()) {
 	    std::vector<QString> names;
 	    for (size_t i = 0; i < m_descriptor->binNames.size(); ++i) {
 		names.push_back(m_descriptor->binNames[i].c_str());
 	    }
-	    (dynamic_cast<EditableDenseThreeDimensionalModel *>(m_output))
-		->setBinNames(names);
-	}
+	    model->setBinNames(names);
+	}
+        
+        m_output = model;
     }
 }
 
 FeatureExtractionPluginTransform::~FeatureExtractionPluginTransform()
 {