Swiftgram/submodules/Display/Display/UITracingLayerView.swift

import Foundation
import UIKit
import AsyncDisplayKit

#if BUCK
import DisplayPrivate
#endif

open class UITracingLayerView: UIView {
    private var scheduledWithLayout: (() -> Void)?
    
    open func schedule(layout f: @escaping () -> Void) {
        self.scheduledWithLayout = f
        self.setNeedsLayout()
    }
    
    override open var autoresizingMask: UIView.AutoresizingMask {
        get {
            return []
        } set(value) {
        }
    }
    
    override open class var layerClass: AnyClass {
        return CATracingLayer.self
    }
    
    override open func layoutSubviews() {
        super.layoutSubviews()
        
        if let scheduledWithLayout = self.scheduledWithLayout {
            self.scheduledWithLayout = nil
            scheduledWithLayout()
        }
    }
}