comparison host/vamp-simple-host.cpp @ 318:2ed36547e1bf

Patch from Dan Stowell to permit vamp-simple-host to read data of unknown length (for streaming input)

     SF_INFO sfinfo;
     memset(&sfinfo, 0, sizeof(SF_INFO));
 
     sndfile = sf_open(wavname.c_str(), SFM_READ, &sfinfo);
     if (!sndfile) {
-	cerr << myname << ": ERROR: Failed to open input file \""
+        cerr << myname << ": ERROR: Failed to open input file \""
              << wavname << "\": " << sf_strerror(sndfile) << endl;
-	return 1;
+        return 1;
     }
 
     ofstream *out = 0;
     if (outfilename != "") {
         out = new ofstream(outfilename.c_str(), ios::out);

         } else {
             blockSize = stepSize;
         }
         cerr << blockSize << endl;
     }
+    int overlapSize = blockSize - stepSize;
+    sf_count_t currentStep = 0;
+    int finalStepsRemaining = max(1, (blockSize / stepSize) - 1); // at end of file, this many part-silent frames needed after we hit EOF
 
     int channels = sfinfo.channels;
 
     float *filebuf = new float[blockSize * channels];
     float **plugbuf = new float*[channels];

     RealTime rt;
     PluginWrapper *wrapper = 0;
     RealTime adjustment = RealTime::zeroTime;
 
     if (outputs.empty()) {
-	cerr << "ERROR: Plugin has no outputs!" << endl;
+        cerr << "ERROR: Plugin has no outputs!" << endl;
         goto done;
     }
 
     if (outputNo < 0) {
 

         // PluginInputDomainAdapter::getTimestampAdjustment
         PluginInputDomainAdapter *ida =
             wrapper->getWrapper<PluginInputDomainAdapter>();
         if (ida) adjustment = ida->getTimestampAdjustment();
     }
-
-    for (sf_count_t i = 0; i < sfinfo.frames; i += stepSize) {
+    
+    // Here we iterate over the frames, avoiding asking the numframes in case it's streaming input.
+    do {
 
         int count;
 
-        if (sf_seek(sndfile, i, SEEK_SET) < 0) {
-            cerr << "ERROR: sf_seek failed: " << sf_strerror(sndfile) << endl;
-            break;
-        }
-        
-        if ((count = sf_readf_float(sndfile, filebuf, blockSize)) < 0) {
-            cerr << "ERROR: sf_readf_float failed: " << sf_strerror(sndfile) << endl;
-            break;
+        if ((blockSize==stepSize) || (currentStep==0)) {
+            // read a full fresh block
+            if ((count = sf_readf_float(sndfile, filebuf, blockSize)) < 0) {
+                cerr << "ERROR: sf_readf_float failed: " << sf_strerror(sndfile) << endl;
+                break;
+            }
+            if (count != blockSize) --finalStepsRemaining;
+        } else {
+            //  otherwise shunt the existing data down and read the remainder.
+            memmove(filebuf, filebuf + (stepSize * channels), overlapSize * channels * sizeof(float));
+            if ((count = sf_readf_float(sndfile, filebuf + (overlapSize * channels), stepSize)) < 0) {
+                cerr << "ERROR: sf_readf_float failed: " << sf_strerror(sndfile) << endl;
+                break;
+            }
+            if (count != stepSize) --finalStepsRemaining;
+            count += overlapSize;
         }
 
         for (int c = 0; c < channels; ++c) {
             int j = 0;
             while (j < count) {

                 plugbuf[c][j] = 0.0f;
                 ++j;
             }
         }
 
-        rt = RealTime::frame2RealTime(i, sfinfo.samplerate);
+        rt = RealTime::frame2RealTime(currentStep * stepSize, sfinfo.samplerate);
 
         printFeatures
             (RealTime::realTime2Frame(rt + adjustment, sfinfo.samplerate),
              sfinfo.samplerate, outputNo, plugin->process(plugbuf, rt),
              out, useFrames);
 
-        int pp = progress;
-        progress = lrintf((float(i) / sfinfo.frames) * 100.f);
-        if (progress != pp && out) {
-            cerr << "\r" << progress << "%";
-        }
-    }
+        if (sfinfo.frames > 0){
+            int pp = progress;
+            progress = lrintf((float(currentStep * stepSize) / sfinfo.frames) * 100.f);
+            if (progress != pp && out) {
+                cerr << "\r" << progress << "%";
+            }
+        }
+
+        ++currentStep;
+
+    } while (finalStepsRemaining > 0);
+
     if (out) cerr << "\rDone" << endl;
 
-    rt = RealTime::frame2RealTime(sfinfo.frames, sfinfo.samplerate);
+    rt = RealTime::frame2RealTime(currentStep * stepSize, sfinfo.samplerate);
 
     printFeatures(RealTime::realTime2Frame(rt + adjustment, sfinfo.samplerate),
                   sfinfo.samplerate, outputNo,
                   plugin->getRemainingFeatures(), out, useFrames);