From 16ebf14376b8897d431e44ec6c7f0f70142432f9 Mon Sep 17 00:00:00 2001
From: nicolasgoutaland <nicolas.goutaland@gmail.com>
Date: Fri, 22 May 2015 16:50:54 +0200
Subject: [PATCH 1/2] Offset generated layer, to reduce PSD file size

---
 MMLayershots/SFPSDWriter+MMLayershots.m | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/MMLayershots/SFPSDWriter+MMLayershots.m b/MMLayershots/SFPSDWriter+MMLayershots.m
index 5ccfaff..6879d5d 100644
--- a/MMLayershots/SFPSDWriter+MMLayershots.m
+++ b/MMLayershots/SFPSDWriter+MMLayershots.m
@@ -22,14 +22,14 @@ - (void)addImagesForLayer:(CALayer *)layer renderedToRootLayer:(CALayer *)rootLa
 
         if (layer.sublayers.count>0 && self.currentGroupDepth<PSD_MAX_GROUP_DEPTH) {
             // add self
-            UIImage *image = [rootLayer imageRepresentation];
+            UIImage *image = [layer imageRepresentation];
 
             // Compute layer name
             NSString *layerName = [self computeNameForLayer:layer];
             [self addLayerWithCGImage:image.CGImage
                               andName:[layerName stringByAppendingString:@"-Content"]
                            andOpacity:1.0
-                            andOffset:CGPointZero];
+                            andOffset:[layer convertPoint:CGPointZero toLayer:nil]];
 
             // hide own layer visuals while rendering children
             NSMutableDictionary *layerProperties = [NSMutableDictionary new];
@@ -76,10 +76,10 @@ - (void)addImagesForLayer:(CALayer *)layer renderedToRootLayer:(CALayer *)rootLa
                 [self showLayersInSubtree:layer];
             }
             NSString *layerName = [self computeNameForLayer:layer];
-            [self addLayerWithCGImage:[rootLayer imageRepresentation].CGImage
+            [self addLayerWithCGImage:[layer imageRepresentation].CGImage
                               andName:layerName
                            andOpacity:1.0
-                             andOffset:CGPointZero];
+                             andOffset:[layer convertPoint:CGPointZero toLayer:nil]];
         }
 
         layer.hidden = YES;

From 3a886897e9b5915e15066e8059328f4c0e80c67c Mon Sep 17 00:00:00 2001
From: nicolasgoutaland <nicolas.goutaland@gmail.com>
Date: Fri, 22 May 2015 17:11:09 +0200
Subject: [PATCH 2/2] Fixed groups generation bug

---
 MMLayershots/SFPSDWriter+MMLayershots.m | 33 ++++++++++++++-----------
 1 file changed, 18 insertions(+), 15 deletions(-)

diff --git a/MMLayershots/SFPSDWriter+MMLayershots.m b/MMLayershots/SFPSDWriter+MMLayershots.m
index 6879d5d..8088f9d 100644
--- a/MMLayershots/SFPSDWriter+MMLayershots.m
+++ b/MMLayershots/SFPSDWriter+MMLayershots.m
@@ -20,7 +20,7 @@ - (void)addImagesForLayer:(CALayer *)layer renderedToRootLayer:(CALayer *)rootLa
     if (layer.hiddenBeforeHidingSublayers == NO) {
         layer.hidden = NO;
 
-        if (layer.sublayers.count>0 && self.currentGroupDepth<PSD_MAX_GROUP_DEPTH) {
+        if (layer.sublayers.count > 0) {
             // add self
             UIImage *image = [layer imageRepresentation];
 
@@ -48,8 +48,14 @@ - (void)addImagesForLayer:(CALayer *)layer renderedToRootLayer:(CALayer *)rootLa
             }
 
             // create layer group
-            [self incrementCurrentGroupDepth];
-            [self openGroupLayerWithName:layerName];
+            BOOL newGroupAdded = NO;
+            if (self.currentGroupDepth<PSD_MAX_GROUP_DEPTH) {
+                // A New group was added
+                newGroupAdded = YES;
+
+                [self incrementCurrentGroupDepth];
+                [self openGroupLayerWithName:layerName];
+            }
 
             // render children
             [[layer.sublayers copy] enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
@@ -61,20 +67,17 @@ - (void)addImagesForLayer:(CALayer *)layer renderedToRootLayer:(CALayer *)rootLa
                 [layer setValue:layerProperties[layerProperty] forKey:layerProperty];
             }
 
-            // Close layer group
-            NSError *error = nil;
-            [self closeCurrentGroupLayerWithError:&error];
-            [self decrementCurrentGroupDepth];
-            if (error) {
-                NSLog(@"%@ - %@", error.localizedDescription, error.localizedRecoveryOptions);
+            // Close layer group if needed
+            if (newGroupAdded) {
+                NSError *error = nil;
+                [self closeCurrentGroupLayerWithError:&error];
+                [self decrementCurrentGroupDepth];
+                if (error) {
+                    NSLog(@"%@ - %@", error.localizedDescription, error.localizedRecoveryOptions);
+                }
             }
         } else {
-            // base case
-            
-            if (layer.sublayers>0) {
-                // reshow sublayers before taking a snapshot
-                [self showLayersInSubtree:layer];
-            }
+            // Base case
             NSString *layerName = [self computeNameForLayer:layer];
             [self addLayerWithCGImage:[layer imageRepresentation].CGImage
                               andName:layerName