Source/app/HTMLRenderer.m

   1 #import "HTMLRenderer.h"
   2
   3 //#define DEBUG
   4
   5 #ifdef DEBUG
   6         #define Log(fmt, ...) NSLog(fmt, ##__VA_ARGS__)
   7 #else
   8         #define Log(fmt, ...)
   9 #endif
  10
  11 @implementation HTMLRenderer
  12 + (NSAttributedString *)attributedStringWithURL:(NSURL *)url {
  13         HTMLRenderer *htmlRenderer = [[HTMLRenderer alloc] init];
  14         NSAttributedString *attributedString = [htmlRenderer attributedStringWithURL:url];
  15         [htmlRenderer release];
  16         return attributedString;
  17 }
  18
  19 - (id)init {
  20         if((self = [super init]) == nil)
  21                 return nil;
  22
  23         renderLock = [[NSConditionLock alloc] initWithCondition:0];
  24         finishedRendering = 0;
  25         renderWebView = [[WebView alloc] initWithFrame:NSMakeRect(0., 0., 2000., 100.) frameName:nil groupName:nil];
  26         [renderWebView setFrameLoadDelegate:self];
  27
  28         return self;
  29 }
  30
  31 - (NSAttributedString *)attributedStringWithURL:(NSURL *)url {
  32         NSMutableAttributedString *result = nil;
  33
  34         finishedRendering = 0;
  35         [self renderHTMLFragment:url];
  36
  37         if(finishedRendering == 1)
  38                 return [(id <WebDocumentText>)[[[renderWebView mainFrame] frameView] documentView] attributedString];
  39         else
  40                 return nil;
  41 }
  42
  43 - (void)renderHTMLFragment:(NSURL *)url {
  44         NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
  45         NSRunLoop *runLoop = [NSRunLoop currentRunLoop];
  46
  47         [NSThread setThreadPriority:1.0];
  48         [renderLock lock];
  49
  50         [[renderWebView mainFrame] loadRequest:[NSURLRequest requestWithURL:url]];
  51
  52         while(!finishedRendering)
  53                 [[NSRunLoop currentRunLoop] runUntilDate:[NSDate dateWithTimeIntervalSinceNow:0.05]];
  54
  55         [renderLock unlockWithCondition:1];
  56         [pool release];
  57 }
  58
  59 - (void)dealloc {
  60         [renderLock release];
  61         [renderWebView release];
  62         [super dealloc];
  63 }
  64
  65 - (void)webView:(WebView *)sender didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
  66         finishedRendering = -1;
  67 }
  68
  69 - (void)webView:(WebView *)sender didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
  70         finishedRendering = -1;
  71 }
  72
  73 - (void)webView:(WebView *) sender didFinishLoadForFrame:(WebFrame *) frame {
  74         finishedRendering = 1;
  75 }
  76
  77 @end