LukeIS on behalf of YabinKang: have iPhoneDriver take a screenshot of…

… the entire page, not just what's in view. (Awesome!) Fixes Issue 4529

r17793

iphone/src/objc/WebViewController.h

@@ -28,6 +28,9 @@
   int numPendingPageLoads_;
 
   NSString *lastJSResult_;
+
+    // Get a screenshot of the page we've loaded
+  UIImage *screenshot_;
 
   NSURLRequestCachePolicy cachePolicy_;
 
@@ -90,6 +93,9 @@
 // Set geolocation
 - (void)setLocation:(NSDictionary *)dict;
 
+// get ss
+- (void)getFullPageScreenShot;
+
 // Check if browser connection is alive
 - (NSNumber *)isBrowserOnline;
 

iphone/src/objc/WebViewController.m

@@ -59,6 +59,7 @@ - (void)viewDidLoad {
   }
 
   lastJSResult_ = nil;
+  screenshot_ = nil;
 
   // Creating a new session if auto-create is enabled
   if ([[RootViewController sharedInstance] isAutoCreateSession]) {
@@ -98,6 +99,7 @@ - (void)didReceiveMemoryWarning {
 - (void)dealloc {
   [[self webView] setDelegate:nil];
   [lastJSResult_ release];
+  [screenshot_ release];
   [super dealloc];
 }
 
@@ -473,18 +475,8 @@ - (NSString *)source {
 
 // Takes a screenshot.
 - (UIImage *)screenshot {
-  UIGraphicsBeginImageContext([[self webView] bounds].size);
-  [[self webView].layer renderInContext:UIGraphicsGetCurrentContext()];
-  UIImage *viewImage = UIGraphicsGetImageFromCurrentImageContext();
-  UIGraphicsEndImageContext();
-
-  // dump the screenshot into a file for debugging
-  //NSString *path = [[[NSSearchPathForDirectoriesInDomains
-  //   (NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0]
-  //  stringByAppendingPathComponent:@"screenshot.png"] retain];
-  //[UIImagePNGRepresentation(viewImage) writeToFile:path atomically:YES];
-
-  return viewImage;
+    [self performSelectorOnMainThread:@selector(getFullPageScreenShot) withObject:nil waitUntilDone:YES];
+    return screenshot_;
 }
 
 - (NSString *)URL {
@@ -569,6 +561,40 @@ - (void)setLocation:(NSDictionary *)dict {
   [locStorage setAltitude:[altitude doubleValue]];
 }
 
+// get the full page screenshot.
+- (void)getFullPageScreenShot {
+    // stop the webview
+    [self.webView setDelegate:nil];
+    [self.webView stopLoading];
+
+    // keep the original window size.
+    CGRect oriFrame = self.webView.frame;
+    CGRect oriBounds = self.webView.bounds;
+
+    // get the page render actual size
+    CGRect tmpFrame = self.webView.frame;
+    tmpFrame.size.height = 1;
+    self.webView.frame = tmpFrame;
+    CGSize fittingSize = [self.webView sizeThatFits:CGSizeZero];
+    tmpFrame.size = fittingSize;
+    self.webView.frame = tmpFrame;
+
+    // render the page and take the screenshot
+    UIGraphicsBeginImageContext(fittingSize);
+    CGContextRef resizedContext = UIGraphicsGetCurrentContext();
+    [[self webView].layer renderInContext:resizedContext];
+    UIImage *viewImage = UIGraphicsGetImageFromCurrentImageContext();
+    UIGraphicsEndImageContext();
+
+    //reset webview back to original size.
+    self.webView.bounds = oriBounds;
+    self.webView.frame = oriFrame;
+
+    // retain the screenshot for webdriver
+    [screenshot_ release];
+    screenshot_ = [viewImage retain];
+}
+
 // Finds out if browser connection is alive
 - (NSNumber *)isBrowserOnline {
   BOOL onlineState = [[self jsEval:@"navigator.onLine"] isEqualToString:@"true"];