论坛 › 知识问答 › ios - Swift 3 中的 vImageBuffer_InitWithCGImage 内存泄漏

菜鸟教程小白 发表于 2022-12-12 12:54:37

ios - Swift 3 中的 vImageBuffer_InitWithCGImage 内存泄漏

                                            <p><p>我正在尝试进行直方图计算。一切正常，除了以下方法在 Instruments 中分析时显示出巨大的内存泄漏。</p>

<p>每次调用以下方法时，它使用 200-300 MB 的内存并且从不释放:</p>

<pre><code>       func histogramCalculation(_ imageRef: CGImage) -&gt; (red: , green: , blue: ) {

            var inBuffer = vImage_Buffer()

            vImageBuffer_InitWithCGImage(
                &amp;inBuffer,
                &amp;format,
                nil,
                imageRef,
                UInt32(kvImageNoFlags))

            let alpha = (repeating: 0, count: 256)
            let red = (repeating: 0, count: 256)
            let green = (repeating: 0, count: 256)
            let blue = (repeating: 0, count: 256)

            let alphaPtr = UnsafeMutablePointer&lt;vImagePixelCount&gt;(mutating: alpha) as UnsafeMutablePointer&lt;vImagePixelCount&gt;?
            let redPtr = UnsafeMutablePointer&lt;vImagePixelCount&gt;(mutating: red) as UnsafeMutablePointer&lt;vImagePixelCount&gt;?
            let greenPtr = UnsafeMutablePointer&lt;vImagePixelCount&gt;(mutating: green) as UnsafeMutablePointer&lt;vImagePixelCount&gt;?
            let bluePtr = UnsafeMutablePointer&lt;vImagePixelCount&gt;(mutating: blue) as UnsafeMutablePointer&lt;vImagePixelCount&gt;?

            let rgba =

            let histogram = UnsafeMutablePointer&lt;UnsafeMutablePointer&lt;vImagePixelCount&gt;?&gt;(mutating: rgba)
            let error : vImage_Error = vImageHistogramCalculation_ARGB8888(&amp;inBuffer, histogram, UInt32(kvImageNoFlags))

            if (error == kvImageNoError) {
                return (red, green, blue)
            }
            return (red, green, blue)
}
</code></pre>

<p>这里有什么问题.....</p></p>
                                    <br><hr><h1><strong>Best Answer-推荐答案</ strong></h1><br>
                                            <p><p><code>vImageBuffer_InitWithCGImage</code> 的文档解释:</p>

<blockquote>
<p>You are responsible for returning the memory referenced by buf-&gt;data to the system using free() when you are done with it.</p>
</blockquote>

<p>所以我希望按照以下方式清理内存:</p>

<pre><code>inBuffer.data.deallocate(bytes: inBuffer.rowBytes * Int(inBuffer.height),
                         alignedTo: MemoryLayout&lt;vImage_Buffer&gt;.alignment)
</code></pre>

<p>附带说明，您在此处使用 <code>UnsafeMutablePointer</code> 并不安全。例如，没有保证 <code>alpha</code> 在您使用引用时会存在。允许 Swift 在您创建 <code>alphaPtr</code> 后立即销毁 <code>alpha</code>(因为它再也不会被引用)。您很少想使用 <code>UnsafeMutablePointer.init</code>。相反，您想使用 <code>withUnsafe...</code> 方法来建立有保证的生命周期。例如(未经测试，但可以编译):</p>

<pre><code>var alpha = (repeating: 0, count: 256)
var red = (repeating: 0, count: 256)
var green = (repeating: 0, count: 256)
var blue = (repeating: 0, count: 256)

let error = alpha.withUnsafeMutableBufferPointer { alphaPtr -&gt; vImage_Error in
    return red.withUnsafeMutableBufferPointer { redPtr in
      return green.withUnsafeMutableBufferPointer { greenPtr in
            return blue.withUnsafeMutableBufferPointer { bluePtr in
                var rgba =
                return rgba.withUnsafeMutableBufferPointer { buffer in
                  return vImageHistogramCalculation_ARGB8888(&amp;inBuffer, buffer.baseAddress!, UInt32(kvImageNoFlags))
                }
            }
      }
    }
}
</code></pre></p>
                                   
                                                <p style="font-size: 20px;">关于ios - Swift 3 中的 vImageBuffer_InitWithCGImage 内存泄漏，我们在Stack Overflow上找到一个类似的问题：
                                                        <a href="https://stackoverflow.com/questions/42055549/" rel="noreferrer noopener nofollow" style="color: red;">
                                                                https://stackoverflow.com/questions/42055549/
                                                        </a>
                                                </p>
                                       

查看完整版本: ios - Swift 3 中的 vImageBuffer_InitWithCGImage 内存泄漏