snipplrCallback({"id":40186,"title":"create-snapshot-images-from-webpages-in-cocoa","source":"
  1. \/\/  MWWebSnapshot<\/span><\/div><\/li>
  2. \/\/<\/span><\/div><\/li>
  3. \/\/  Created by Jim McGowan on 08\/09\/2010.<\/span><\/div><\/li>
  4. \/\/  Copyright 2010 Jim McGowan. All rights reserved.<\/span><\/div><\/li>
  5. \/\/<\/span><\/div><\/li>
  6. \/\/  This code is made available under the BSD license.  <\/span><\/div><\/li>
  7. \/\/  Please see the accompanying license.txt file<\/span><\/div><\/li>
  8. \/\/  or view the license online at http:\/\/www.malkinware.com\/developer\/License.txt<\/span><\/div><\/li>
  9. \/\/<\/span><\/div><\/li>
  10.  <\/div><\/li>
  11. #import <Cocoa\/Cocoa.h><\/span><\/div><\/li>
  12. #import <WebKit\/WebKit.h><\/span><\/div><\/li>
  13.  <\/div><\/li>
  14.  <\/div><\/li>
  15. @interface<\/span> MWWebSnapshot :<\/span> NSObject<\/span><\/a> <\/div><\/li>
  16. {<\/span><\/div><\/li>
  17. \tvoid<\/span> (<\/span>^<\/span>completionBlock)<\/span>(<\/span>NSImage<\/span><\/a> *<\/span>image)<\/span>;<\/div><\/li>
  18. \tWebView *<\/span>webView;<\/div><\/li>
  19. }<\/span><\/div><\/li>
  20.  <\/div><\/li>
  21. +<\/span> (<\/span>void<\/span>)<\/span>takeSnapshotOfWebPageAtURL:<\/span>(<\/span>NSURL<\/span><\/a> *<\/span>)<\/span>url completionBlock:<\/span>(<\/span>void<\/span> (<\/span>^<\/span>)<\/span>(<\/span>NSImage<\/span><\/a> *<\/span>)<\/span>)<\/span>block;<\/div><\/li>
  22.  <\/div><\/li>
  23. @end<\/span><\/div><\/li>
  24.  <\/div><\/li>
  25.  <\/div><\/li>
  26.  <\/div><\/li>
  27. @interface<\/span> MWWebSnapshot(<\/span>)<\/span><\/div><\/li>
  28. -<\/span> (<\/span>id<\/span>)<\/span>_initWithCompletionBlock:<\/span>(<\/span>void<\/span> (<\/span>^<\/span>)<\/span>(<\/span>NSImage<\/span><\/a> *<\/span>)<\/span>)<\/span>block;<\/div><\/li>
  29. -<\/span> (<\/span>void<\/span>)<\/span>_beginDownloadFromURL:<\/span>(<\/span>NSURL<\/span><\/a> *<\/span>)<\/span>url;<\/div><\/li>
  30. @end<\/span><\/div><\/li>
  31.  <\/div><\/li>
  32.  <\/div><\/li>
  33.  <\/div><\/li>
  34. @implementation<\/span> MWWebSnapshot<\/div><\/li>
  35.  <\/div><\/li>
  36.  <\/div><\/li>
  37. +<\/span> (<\/span>void<\/span>)<\/span>takeSnapshotOfWebPageAtURL:<\/span>(<\/span>NSURL<\/span><\/a> *<\/span>)<\/span>url completionBlock:<\/span>(<\/span>void<\/span> (<\/span>^<\/span>)<\/span>(<\/span>NSImage<\/span><\/a> *<\/span>)<\/span>)<\/span>block;<\/div><\/li>
  38. {<\/span><\/div><\/li>
  39. \tMWWebSnapshot *<\/span>instance =<\/span> [<\/span>[<\/span>self alloc]<\/span> _initWithCompletionBlock:<\/span>block]<\/span>;<\/div><\/li>
  40. \t[<\/span>instance _beginDownloadFromURL:<\/span>url]<\/span>;<\/div><\/li>
  41. \t[<\/span>instance autorelease]<\/span>;<\/div><\/li>
  42. }<\/span><\/div><\/li>
  43.  <\/div><\/li>
  44.  <\/div><\/li>
  45. -<\/span> (<\/span>id<\/span>)<\/span>_initWithCompletionBlock:<\/span>(<\/span>void<\/span> (<\/span>^<\/span>)<\/span>(<\/span>NSImage<\/span><\/a> *<\/span>)<\/span>)<\/span>block;<\/div><\/li>
  46. {<\/span><\/div><\/li>
  47. \tself =<\/span> [<\/span>super init]<\/span>;<\/div><\/li>
  48. \tif<\/span> (<\/span>self !=<\/span> nil<\/span>)<\/span><\/div><\/li>
  49. \t{<\/span><\/div><\/li>
  50. \t\tcompletionBlock =<\/span> [<\/span>block copy]<\/span>;<\/div><\/li>
  51.  <\/div><\/li>
  52. \t\twebView =<\/span> [<\/span>[<\/span>WebView alloc]<\/span> initWithFrame:<\/span>NSMakeRect(<\/span>0.0<\/span>, 0.0<\/span>, 1000.0<\/span>, 1000.0<\/span>)<\/span> <\/div><\/li>
  53.                                                frameName:<\/span>nil<\/span> <\/div><\/li>
  54.                                                groupName:<\/span>nil<\/span>]<\/span>;<\/div><\/li>
  55. \t\t[<\/span>webView setFrameLoadDelegate:<\/span>self]<\/span>;<\/div><\/li>
  56. \t}<\/span><\/div><\/li>
  57. \treturn<\/span> self;<\/div><\/li>
  58. }<\/span><\/div><\/li>
  59.  <\/div><\/li>
  60.  <\/div><\/li>
  61. -<\/span> (<\/span>void<\/span>)<\/span>_beginDownloadFromURL:<\/span>(<\/span>NSURL<\/span><\/a> *<\/span>)<\/span>url;<\/div><\/li>
  62. {<\/span><\/div><\/li>
  63. \t[<\/span>self retain]<\/span>;<\/div><\/li>
  64. \t[<\/span>webView setMainFrameURL:<\/span>[<\/span>url absoluteString]<\/span>]<\/span>;<\/div><\/li>
  65. }<\/span><\/div><\/li>
  66.  <\/div><\/li>
  67.  <\/div><\/li>
  68. -<\/span> (<\/span>void<\/span>)<\/span>webView:<\/span>(<\/span>WebView *<\/span>)<\/span>sender didFinishLoadForFrame:<\/span>(<\/span>WebFrame *<\/span>)<\/span>frame<\/div><\/li>
  69. {<\/span><\/div><\/li>
  70. \tif<\/span> (<\/span>frame !=<\/span> [<\/span>webView mainFrame]<\/span>)<\/span><\/div><\/li>
  71. \t{<\/span><\/div><\/li>
  72. \t\treturn<\/span>;<\/div><\/li>
  73. \t}<\/span><\/div><\/li>
  74.  <\/div><\/li>
  75. \tNSView<\/span><\/a> *<\/span>webFrameViewDocView =<\/span> [<\/span>[<\/span>[<\/span>webView mainFrame]<\/span> frameView]<\/span> documentView]<\/span>;<\/div><\/li>
  76. \tNSRect<\/span> cacheRect =<\/span> [<\/span>webFrameViewDocView bounds]<\/span>;<\/div><\/li>
  77.  <\/div><\/li>
  78. \tNSBitmapImageRep<\/span><\/a> *<\/span>bitmapRep =<\/span> [<\/span>webFrameViewDocView bitmapImageRepForCachingDisplayInRect:<\/span>cacheRect]<\/span>;<\/div><\/li>
  79. \t[<\/span>webFrameViewDocView cacheDisplayInRect:<\/span>cacheRect toBitmapImageRep:<\/span>bitmapRep]<\/span>;<\/div><\/li>
  80.  <\/div><\/li>
  81. \tNSSize imgSize =<\/span> cacheRect.size;<\/div><\/li>
  82. \tif<\/span> (<\/span>imgSize.height > imgSize.width)<\/span><\/div><\/li>
  83. \t{<\/span><\/div><\/li>
  84. \t\timgSize.height =<\/span> imgSize.width;<\/div><\/li>
  85. \t}<\/span><\/div><\/li>
  86.  <\/div><\/li>
  87. \tNSRect<\/span> srcRect =<\/span> NSZeroRect;<\/div><\/li>
  88. \tsrcRect.size =<\/span> imgSize;<\/div><\/li>
  89. \tsrcRect.origin.y =<\/span> cacheRect.size.height -<\/span> imgSize.height;<\/div><\/li>
  90.  <\/div><\/li>
  91. \tNSRect<\/span> destRect =<\/span> NSZeroRect;<\/div><\/li>
  92. \tdestRect.size =<\/span> imgSize;<\/div><\/li>
  93.  <\/div><\/li>
  94. \tNSImage<\/span><\/a> *<\/span>webImage =<\/span> [<\/span>[<\/span>[<\/span>NSImage<\/span><\/a> alloc]<\/span> initWithSize:<\/span>imgSize]<\/span> autorelease]<\/span>;<\/div><\/li>
  95. \t[<\/span>webImage lockFocus]<\/span>;<\/div><\/li>
  96. \t[<\/span>bitmapRep drawInRect:<\/span>destRect <\/div><\/li>
  97.                      fromRect:<\/span>srcRect <\/div><\/li>
  98.                     operation:<\/span>NSCompositeCopy <\/div><\/li>
  99.                      fraction:<\/span>1.0<\/span> <\/div><\/li>
  100.                respectFlipped:<\/span>YES<\/span> <\/div><\/li>
  101.                         hints:<\/span>nil<\/span>]<\/span>;<\/div><\/li>
  102. \t[<\/span>webImage unlockFocus]<\/span>;<\/div><\/li>
  103.  <\/div><\/li>
  104. \tNSSize defaultDisplaySize;<\/div><\/li>
  105. \tdefaultDisplaySize.height =<\/span> 64.0<\/span> *<\/span> (<\/span>imgSize.height \/<\/span> imgSize.width)<\/span>;<\/div><\/li>
  106. \tdefaultDisplaySize.width =<\/span> 64.0<\/span>;<\/div><\/li>
  107. \t[<\/span>webImage setSize:<\/span>defaultDisplaySize]<\/span>;<\/div><\/li>
  108.  <\/div><\/li>
  109. \tcompletionBlock(<\/span>webImage)<\/span>;<\/div><\/li>
  110.  <\/div><\/li>
  111. \t[<\/span>self autorelease]<\/span>;<\/div><\/li>
  112. }<\/span><\/div><\/li>
  113.  <\/div><\/li>
  114.  <\/div><\/li>
  115. -<\/span> (<\/span>void<\/span>)<\/span>dealloc<\/div><\/li>
  116. {<\/span><\/div><\/li>
  117. \t[<\/span>completionBlock release]<\/span>;<\/div><\/li>
  118. \t[<\/span>webView release]<\/span>;<\/div><\/li>
  119.  <\/div><\/li>
  120. \t[<\/span>super dealloc]<\/span>;<\/div><\/li>
  121. }<\/span><\/div><\/li>
  122.  <\/div><\/li>
  123. @end<\/span><\/div><\/li><\/ol><\/pre>","link":"https:\/\/snipplr.com\/view\/40186\/create-snapshot-images-from-webpages-in-cocoa"});