From 0fe774511ccef33cc2a1a2e00a1caab5d14a72af Mon Sep 17 00:00:00 2001
From: Huy Nguyen <huy@pinterest.com>
Date: Thu, 25 Feb 2016 12:19:14 -0800
Subject: [PATCH] ASCellNode notifies its layout delegate during layout
 transition

---
 AsyncDisplayKit/ASCellNode.m | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/AsyncDisplayKit/ASCellNode.m b/AsyncDisplayKit/ASCellNode.m
index 5083edf12c..a8a8b6f97f 100644
--- a/AsyncDisplayKit/ASCellNode.m
+++ b/AsyncDisplayKit/ASCellNode.m
@@ -11,10 +11,12 @@
 #import "ASInternalHelpers.h"
 #import <AsyncDisplayKit/_ASDisplayView.h>
 #import <AsyncDisplayKit/ASDisplayNode+Subclasses.h>
+#import <AsyncDisplayKit/ASDisplayNode+Beta.h>
 #import <AsyncDisplayKit/ASTextNode.h>
 
 #import <AsyncDisplayKit/ASViewController.h>
 #import <AsyncDisplayKit/ASInsetLayoutSpec.h>
+#import <AsyncDisplayKit/ASLayout.h>
 
 #pragma mark -
 #pragma mark ASCellNode
@@ -119,10 +121,30 @@
 {
   CGSize oldSize = self.calculatedSize;
   [super setNeedsLayout];
+  [self didRelayoutFromOldSize:oldSize toNewSize:self.calculatedSize];
+}
 
+- (ASLayout *)transitionLayoutWithAnimation:(BOOL)animated
+{
+  CGSize oldSize = self.calculatedSize;
+  ASLayout *layout = [super transitionLayoutWithAnimation:animated];
+  [self didRelayoutFromOldSize:oldSize toNewSize:layout.size];
+  return layout;
+}
+
+- (ASLayout *)transitionLayoutWithSizeRange:(ASSizeRange)constrainedSize animated:(BOOL)animated
+{
+  CGSize oldSize = self.calculatedSize;
+  ASLayout *layout = [super transitionLayoutWithSizeRange:constrainedSize animated:animated];
+  [self didRelayoutFromOldSize:oldSize toNewSize:layout.size];
+  return layout;
+}
+
+- (void)didRelayoutFromOldSize:(CGSize)oldSize toNewSize:(CGSize)newSize
+{
   if (_layoutDelegate != nil && self.isNodeLoaded) {
     ASPerformBlockOnMainThread(^{
-      BOOL sizeChanged = !CGSizeEqualToSize(oldSize, self.calculatedSize);
+      BOOL sizeChanged = !CGSizeEqualToSize(oldSize, newSize);
       [_layoutDelegate nodeDidRelayout:self sizeChanged:sizeChanged];
     });
   }