Changeset 3406 for pjproject/trunk/pjmedia/src/pjmedia-audiodev/coreaudio_dev.c

Timestamp:

Jan 20, 2011 3:32:14 AM (14 years ago)

Author:

ming

Message:

Re #3398: Fixes assertion in the resample procedure and callback.

File:

• pjproject/trunk/pjmedia/src/pjmedia-audiodev/coreaudio_dev.c (modified) (6 diffs)

pjproject/trunk/pjmedia/src/pjmedia-audiodev/coreaudio_dev.c

@@ -121 +121 @@
     AudioConverterRef            resample;
     pj_int16_t                  *resample_buf;
+    void                        *resample_buf_ptr;
     unsigned                     resample_buf_count;
     unsigned                     resample_buf_size;
@@ -656 +657 @@
     struct coreaudio_stream *strm = (struct coreaudio_stream*)inUserData;
 
-    pj_assert(*ioNumberDataPackets == strm->resample_buf_count);
+    if (*ioNumberDataPackets > strm->resample_buf_size)
+        *ioNumberDataPackets = strm->resample_buf_size;
 
     ioData->mNumberBuffers = 1;
     ioData->mBuffers[0].mNumberChannels = 1;
-    ioData->mBuffers[0].mData = strm->resample_buf;
-    ioData->mBuffers[0].mDataByteSize = strm->resample_buf_count *
+    ioData->mBuffers[0].mData = strm->resample_buf_ptr;
+    ioData->mBuffers[0].mDataByteSize = *ioNumberDataPackets *
                                         strm->param.bits_per_sample >> 3;
-    return 0;
+
+    return noErr;
 }
 
@@ -680 +683 @@
     pj_int16_t *input;
     UInt32 resampleSize;
-    UInt32 size;
 
     pj_assert(!strm->quit_flag);
@@ -718 +720 @@
     input = (pj_int16_t *)buf->mBuffers[0].mData;
 
-    /* Calculate how many frames we need to fill an entire packet */
-    resampleSize = strm->param.samples_per_frame *
-                   strm->param.bits_per_sample >> 3;
-    size = sizeof(resampleSize);
-    ostatus = AudioConverterGetProperty(
-                  strm->resample,
-                  kAudioConverterPropertyCalculateInputBufferSize,
-                  &size,
-                  &resampleSize);
-    if (ostatus != noErr) {
-        PJ_LOG(5, (THIS_FILE, "Core audio converter measure error %i",
-                   ostatus));
-        goto on_break;
-    }
-    resampleSize /= strm->param.bits_per_sample >> 3;
-
+    resampleSize = strm->resample_buf_size;
     nsamples = inNumberFrames * strm->param.channel_count +
                strm->resample_buf_count;
-    pj_assert(nsamples < strm->resample_buf_size);
 
     if (nsamples >= resampleSize) {
+        pjmedia_frame frame;
         UInt32 resampleOutput = strm->param.samples_per_frame;
-        unsigned chunk_count = 0;
-        pjmedia_frame frame;
         AudioBufferList ab;
 
         frame.type = PJMEDIA_FRAME_TYPE_AUDIO;
+        frame.buf = (void*) strm->rec_buf;
         frame.size = strm->param.samples_per_frame *
                      strm->param.bits_per_sample >> 3;
         frame.bit_info = 0;
+        
+        ab.mNumberBuffers = 1;
+        ab.mBuffers[0].mNumberChannels = 1;
+        ab.mBuffers[0].mData = strm->rec_buf;
+        ab.mBuffers[0].mDataByteSize = frame.size;
 
         /* If buffer is not empty, combine the buffer with the just incoming
          * samples, then call put_frame.
          */
-
-        chunk_count = resampleSize - strm->resample_buf_count;
-        pjmedia_copy_samples(strm->resample_buf + strm->resample_buf_count,
-                             input, chunk_count);
-        strm->resample_buf_count += chunk_count;
-
-        frame.buf = (void*) strm->rec_buf;
-        frame.timestamp.u64 = strm->rec_timestamp.u64;
-
-        ab.mNumberBuffers = 1;
-        ab.mBuffers[0].mNumberChannels = 1;
-        ab.mBuffers[0].mDataByteSize = frame.size;
-        ab.mBuffers[0].mData = frame.buf;
-
-        /* Do the resample */
-        ostatus = AudioConverterFillComplexBuffer(strm->resample,
-                                                  resampleProc,
-                                                  strm,
-                                                  &resampleOutput,
-                                                  &ab,
-                                                  NULL);
-        if (ostatus != noErr) {
-            goto on_break;
-        }
-
-        status = (*strm->rec_cb)(strm->user_data, &frame);
-
-        input = input + chunk_count;
-        nsamples -= resampleSize;
-        strm->resample_buf_count = 0;
-        strm->rec_timestamp.u64 += strm->param.samples_per_frame /
-                                   strm->param.channel_count;
-
-        pj_assert(nsamples < resampleSize);
+        if (strm->resample_buf_count) {
+            unsigned chunk_count = resampleSize - strm->resample_buf_count;
+            pjmedia_copy_samples(strm->resample_buf + strm->resample_buf_count,
+                                 input, chunk_count);
+
+            /* Do the resample */
+
+            strm->resample_buf_ptr = strm->resample_buf;
+            ostatus = AudioConverterFillComplexBuffer(strm->resample,
+                                                      resampleProc,
+                                                      strm,
+                                                      &resampleOutput,
+                                                      &ab,
+                                                      NULL);
+            if (ostatus != noErr) {
+                goto on_break;
+            }
+            frame.timestamp.u64 = strm->rec_timestamp.u64;
+
+            status = (*strm->rec_cb)(strm->user_data, &frame);
+
+            input = input + chunk_count;
+            nsamples -= resampleSize;
+            strm->resample_buf_count = 0;
+            strm->rec_timestamp.u64 += strm->param.samples_per_frame /
+                                       strm->param.channel_count;
+        }
+        
+        
+        /* Give all frames we have */
+        while (nsamples >= resampleSize && status == 0) {
+            frame.timestamp.u64 = strm->rec_timestamp.u64;
+            
+            /* Do the resample */
+            strm->resample_buf_ptr = input;
+            ab.mBuffers[0].mDataByteSize = frame.size;
+            resampleOutput = strm->param.samples_per_frame;
+            ostatus = AudioConverterFillComplexBuffer(strm->resample,
+                                                      resampleProc,
+                                                      strm,
+                                                      &resampleOutput,
+                                                      &ab,
+                                                      NULL);
+            if (ostatus != noErr) {
+                goto on_break;
+            }       
+            
+            status = (*strm->rec_cb)(strm->user_data, &frame);
+            
+            input = (pj_int16_t*) input + resampleSize;
+            nsamples -= resampleSize;
+            strm->rec_timestamp.u64 += strm->param.samples_per_frame /
+                                       strm->param.channel_count;
+        }
 
         /* Store the remaining samples into the buffer */
@@ -862 +873 @@
     nsamples = inNumberFrames * strm->param.channel_count +
                strm->rec_buf_count;
-    if (nsamples >= strm->param.samples_per_frame)
-     {
+    if (nsamples >= strm->param.samples_per_frame) {
         pjmedia_frame frame;
 
@@ -1178 +1188 @@
         return PJMEDIA_AUDIODEV_ERRNO_FROM_COREAUDIO(ostatus);
     }
-
+    
     /*
-     * Allocate the buffer required to hold the enough input data
-     * for two complete frames of audio.
+     * Allocate the buffer required to hold enough input data
      */
-    strm->resample_buf_size = (unsigned)(desc->mSampleRate *
-                              strm->param.samples_per_frame /
-                              strm->param.clock_rate * 2);
+    strm->resample_buf_size =  (unsigned)(desc->mSampleRate *
+                                          strm->param.samples_per_frame /
+                                          strm->param.clock_rate);
     strm->resample_buf = (pj_int16_t*)
                          pj_pool_alloc(strm->pool,