ReverbConvolver.h (3686B)
1 /* 2 * Copyright (C) 2010 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of 14 * its contributors may be used to endorse or promote products derived 15 * from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #ifndef ReverbConvolver_h 30 #define ReverbConvolver_h 31 32 #include "ReverbAccumulationBuffer.h" 33 #include "ReverbInputBuffer.h" 34 #include "mozilla/MemoryReporting.h" 35 #include "mozilla/Monitor.h" 36 #include "mozilla/UniquePtr.h" 37 #ifdef LOG 38 # undef LOG 39 #endif 40 #include <atomic> 41 42 #include "base/thread.h" 43 44 namespace WebCore { 45 46 using mozilla::UniquePtr; 47 48 class ReverbConvolverStage; 49 50 class ReverbConvolver { 51 public: 52 // maxFFTSize can be adjusted (from say 2048 to 32768) depending on how much 53 // precision is necessary. For certain tweaky de-convolving applications the 54 // phase errors add up quickly and lead to non-sensical results with larger 55 // FFT sizes and single-precision floats. In these cases 2048 is a good size. 56 // If not doing multi-threaded convolution, then should not go > 8192. 57 ReverbConvolver(const float* impulseResponseData, 58 size_t impulseResponseLength, size_t maxFFTSize, 59 size_t convolverRenderPhase, bool useBackgroundThreads, 60 bool* aAllocationFailure); 61 ~ReverbConvolver(); 62 63 void process(const float* sourceChannelData, float* destinationChannelData); 64 65 size_t impulseResponseLength() const { return m_impulseResponseLength; } 66 67 ReverbInputBuffer* inputBuffer() { return &m_inputBuffer; } 68 69 bool useBackgroundThreads() const { return m_useBackgroundThreads; } 70 void backgroundThreadEntry(); 71 72 size_t sizeOfIncludingThis(mozilla::MallocSizeOf aMallocSizeOf) const; 73 74 private: 75 nsTArray<UniquePtr<ReverbConvolverStage> > m_stages; 76 nsTArray<UniquePtr<ReverbConvolverStage> > m_backgroundStages; 77 size_t m_impulseResponseLength; 78 79 ReverbAccumulationBuffer m_accumulationBuffer; 80 81 // One or more background threads read from this input buffer which is fed 82 // from the realtime thread. 83 ReverbInputBuffer m_inputBuffer; 84 85 // Background thread and synchronization 86 base::Thread m_backgroundThread; 87 mozilla::Monitor m_backgroundThreadMonitor MOZ_UNANNOTATED; 88 bool m_useBackgroundThreads; 89 std::atomic<bool> m_wantsToExit; 90 std::atomic<bool> m_moreInputBuffered; 91 }; 92 93 } // namespace WebCore 94 95 #endif // ReverbConvolver_h