Android View Kotlin extension for a one-shot callback after layout

LayoutExtensions.kt

fun View.afterLayout(block: View.() -> Unit) {
  val originalVto = viewTreeObserver
  originalVto.addOnPreDrawListener(object : ViewTreeObserver.OnPreDrawListener {
    override fun onPreDraw(): Boolean {
      if (originalVto.isAlive) originalVto.removeOnPreDrawListener(this)
      viewTreeObserver.removeOnPreDrawListener(this)
      block(this@afterLayout)
      return true
    }
  })
}

MainActivity.kt

// usage
image_view.afterLayout { Log.d("demo", "image width $width height $height") }