[winterboard.git] / WBMarkup.mm

#include "WBMarkup.h"

@class WKView;

extern "C" {
    void WebThreadLock();
    CGContextRef WKGetCurrentGraphicsContext();
    void WKViewLockFocus(WKView *);
    void WKViewUnlockFocus(WKView *);
    void WKViewDisplayRect(WKView *, CGRect);
}

@interface DOMElement : NSObject
- (void) setInnerHTML:(NSString *)value;
- (void) setInnerText:(NSString *)value;
- (void) setAttribute:(NSString *)name value:(NSString *)value;
- (void) removeAttribute:(NSString *)name;
- (DOMElement *) firstChild;
- (void) appendChild:(DOMElement *)child;
- (void) removeChild:(DOMElement *)child;
- (void) setScrollXOffset:(float)x scrollYOffset:(float)y;
@end

@interface DOMDocument : NSObject
- (DOMElement *) getElementById:(NSString *)id;
@end

@interface WebPreferences : NSObject
- (id) initWithIdentifier:(NSString *)identifier;
- (void) setPlugInsEnabled:(BOOL)value;
@end

@interface WebFrameView : NSObject
- (void) setAllowsScrolling:(BOOL)value;
@end

@interface WebFrame : NSObject
- (WebFrameView *) frameView;
- (void) _setLoadsSynchronously:(BOOL)value;
- (void) loadHTMLString:(NSString *)string baseURL:(id)url;
- (void) forceLayoutAdjustingViewSize:(BOOL)adjust;
- (CGSize) renderedSizeOfNode:(DOMElement *)node constrainedToWidth:(float)width;
- (DOMDocument *) DOMDocument;
@end

@interface WAKView : NSObject
- (void) _drawRect:(CGRect)rect context:(CGContext *)context lockFocus:(bool)focus;
@end

@interface WebView : WAKView
- (id) initWithFrame:(CGRect)frame;
- (WebFrame *) mainFrame;
- (void) setDrawsBackground:(BOOL)value;
- (void) setPreferences:(WebPreferences *)preferences;
- (WKView *) _viewRef;
@end

@interface WAKWindow : NSObject
- (id) initWithFrame:(CGRect)frame;
- (void) setContentView:(WebView *)view;
@end

static WBMarkup *SharedMarkup_;

@implementation WBMarkup

+ (BOOL) isSharedMarkupCreated {
    return SharedMarkup_ != nil;
}

+ (WBMarkup *) sharedMarkup {
    if (SharedMarkup_ == nil)
        SharedMarkup_ = [[WBMarkup alloc] init];
    return SharedMarkup_;
}

- (id) init {
    if ((self = [super init]) != nil) {
        WebThreadLock();

        SharedMarkup_ = self;

        view_ = [[WebView alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
        [view_ setDrawsBackground:NO];

        WebPreferences *preferences([[WebPreferences alloc] initWithIdentifier:@"com.apple.webkit.webmarkup"]);
        [preferences setPlugInsEnabled:NO];
        [view_ setPreferences:preferences];
        [preferences release];

        window_ = [[WAKWindow alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
        [window_ setContentView:view_];

        WebFrame *frame([view_ mainFrame]);
        [[frame frameView] setAllowsScrolling:NO];
        [frame _setLoadsSynchronously:YES];

        [frame loadHTMLString:@"<html><body style='margin: 0px; word-wrap: break-word; -khtml-nbsp-mode: space; -khtml-line-break: after-white-space'><div id='size'><div id='text'></div></div></body></html>" baseURL:nil];
    } return self;
}

- (void) dealloc {
    [window_ release];
    [view_ release];
    [super dealloc];
}

- (void) drawRect:(CGRect)rect {
    [text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];

    CGRect draw(CGRectMake(0, 0, rect.size.width - rect.origin.x, rect.size.height - rect.origin.y));

    CGContextSaveGState(context_); {
        CGContextTranslateCTM(context_, rect.origin.x, rect.origin.y);

        if (kCFCoreFoundationVersionNumber > 700)
            [view_ _drawRect:draw context:context_ lockFocus:YES];
        else {
            WKView *view([view_ _viewRef]);
            WKViewLockFocus(view); {
                WKViewDisplayRect(view, draw);
            } WKViewUnlockFocus(view);
        }
    } CGContextRestoreGState(context_);
}

- (WebView *) _webView {
    return view_;
}

- (void) setStringDrawingOrigin:(CGPoint)origin {
    origin_ = origin;
}

- (void) clearStringDrawingOrigin {
    origin_ = CGPointZero;
}

- (CGSize) sizeOfMarkup:(NSString *)markup forWidth:(float)width {
    WebThreadLock();

    if (![self _webPrepareContextForTextDrawing:NO])
        return CGSizeZero;

    [text_ setInnerHTML:markup];
    [text_ removeAttribute:@"style"];

    NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: 5000px"], width]);
    [size_ setAttribute:@"style" value:value];
    [value release];

    [[view_ mainFrame] forceLayoutAdjustingViewSize:YES];
    return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
}

- (CGSize) sizeOfString:(NSString *)string withStyle:(NSString *)style forWidth:(float)width {
    WebThreadLock();

    if (![self _webPrepareContextForTextDrawing:NO])
        return CGSizeZero;

    [size_ removeChild:[size_ firstChild]];

    WebFrame *frame([view_ mainFrame]);

    [frame forceLayoutAdjustingViewSize:YES];
    [text_ setInnerText:string];
    [self _setupWithStyle:style width:width height:5000];
    [frame forceLayoutAdjustingViewSize:YES];

    return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
}

- (NSString *) _styleFormatString:(NSString *)style {
    return style;
}

- (void) _setupWithStyle:(NSString *)style width:(float)width height:(float)height {
    WebThreadLock();

    if (style != nil && [style length] != 0)
        [text_ setAttribute:@"style" value:style];
    else
        [text_ removeAttribute:@"style"];

    NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], width, height]);
    [size_ setAttribute:@"style" value:value];
    [value release];

    [size_ appendChild:text_];
}

- (BOOL) _webPrepareContextForTextDrawing:(BOOL)drawing {
    WebThreadLock();

    if (document_ == nil) {
        WebFrame *frame([view_ mainFrame]);

        document_ = [[frame DOMDocument] retain];
        if (document_ == nil) {
            NSLog(@"*** ERROR: no DOM document in text-drawing webview");
            return NO;
        }

        text_ = [[document_ getElementById:@"text"] retain];
        size_ = [[document_ getElementById:@"size"] retain];

        if (text_ == nil || size_ == nil) {
            NSLog(@"*** ERROR: cannot find DOM element required for text drawing");
            return NO;
        }
    }

    context_ = NULL;

    if (!drawing)
        context_ = NULL;
    else {
        context_ = WKGetCurrentGraphicsContext();
        if (context_ == NULL) {
            NSLog(@"*** ERROR: no CGContext set for drawing");
            return NO;
        }
    }

    return YES;
}

- (void) drawMarkup:(NSString *)markup atPoint:(CGPoint)point {
    [self drawMarkup:markup inRect:CGRectMake(point.x, point.y, 65535, 65535)];
}

- (void) drawMarkup:(NSString *)markup inRect:(CGRect)rect {
    WebThreadLock();

    if (![self _webPrepareContextForTextDrawing:YES])
        return;

    [text_ setInnerHTML:markup];
    [text_ removeAttribute:@"style"];

    NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], CGRectGetWidth(rect), CGRectGetHeight(rect)]);
    [size_ setAttribute:@"style" value:value];
    [value release];

    [[view_ mainFrame] forceLayoutAdjustingViewSize:YES];

    [self drawRect:rect];
}

- (void) drawString:(NSString *)string atPoint:(CGPoint)point withStyle:(NSString *)style {
    [self drawString:string inRect:CGRectMake(point.x, point.y, 65535, 65535) withStyle:style];
}

- (void) drawString:(NSString *)string inRect:(CGRect)rect withStyle:(NSString *)style {
    WebThreadLock();

    if (![self _webPrepareContextForTextDrawing:YES])
        return;

    [size_ removeChild:[size_ firstChild]];

    WebFrame *frame([view_ mainFrame]);

    [frame forceLayoutAdjustingViewSize:YES];
    [text_ setInnerText:string];
    [self _setupWithStyle:style width:CGRectGetWidth(rect) height:CGRectGetHeight(rect)];
    [frame forceLayoutAdjustingViewSize:YES];

    [self drawRect:rect];
}

@end
Commit	Line	Data
2de63e63 JF	1	#include "WBMarkup.h"
	2
	3	@class WKView;
	4
	5	extern "C" {
	6	void WebThreadLock();
	7	CGContextRef WKGetCurrentGraphicsContext();
	8	void WKViewLockFocus(WKView *);
	9	void WKViewUnlockFocus(WKView *);
	10	void WKViewDisplayRect(WKView *, CGRect);
	11	}
	12
	13	@interface DOMElement : NSObject
	14	- (void) setInnerHTML:(NSString *)value;
	15	- (void) setInnerText:(NSString *)value;
	16	- (void) setAttribute:(NSString )name value:(NSString )value;
	17	- (void) removeAttribute:(NSString *)name;
	18	- (DOMElement *) firstChild;
	19	- (void) appendChild:(DOMElement *)child;
	20	- (void) removeChild:(DOMElement *)child;
	21	- (void) setScrollXOffset:(float)x scrollYOffset:(float)y;
	22	@end
	23
	24	@interface DOMDocument : NSObject
	25	- (DOMElement ) getElementById:(NSString )id;
	26	@end
	27
	28	@interface WebPreferences : NSObject
	29	- (id) initWithIdentifier:(NSString *)identifier;
	30	- (void) setPlugInsEnabled:(BOOL)value;
	31	@end
	32
	33	@interface WebFrameView : NSObject
	34	- (void) setAllowsScrolling:(BOOL)value;
	35	@end
	36
	37	@interface WebFrame : NSObject
	38	- (WebFrameView *) frameView;
	39	- (void) _setLoadsSynchronously:(BOOL)value;
	40	- (void) loadHTMLString:(NSString *)string baseURL:(id)url;
	41	- (void) forceLayoutAdjustingViewSize:(BOOL)adjust;
	42	- (CGSize) renderedSizeOfNode:(DOMElement *)node constrainedToWidth:(float)width;
	43	- (DOMDocument *) DOMDocument;
	44	@end
	45
09ea4166 JF	46	@interface WAKView : NSObject
	47	- (void) _drawRect:(CGRect)rect context:(CGContext *)context lockFocus:(bool)focus;
	48	@end
	49
	50	@interface WebView : WAKView
2de63e63 JF	51	- (id) initWithFrame:(CGRect)frame;
	52	- (WebFrame *) mainFrame;
	53	- (void) setDrawsBackground:(BOOL)value;
	54	- (void) setPreferences:(WebPreferences *)preferences;
	55	- (WKView *) _viewRef;
	56	@end
	57
	58	@interface WAKWindow : NSObject
	59	- (id) initWithFrame:(CGRect)frame;
	60	- (void) setContentView:(WebView *)view;
	61	@end
	62
	63	static WBMarkup *SharedMarkup_;
	64
	65	@implementation WBMarkup
	66
	67	+ (BOOL) isSharedMarkupCreated {
	68	return SharedMarkup_ != nil;
	69	}
	70
	71	+ (WBMarkup *) sharedMarkup {
	72	if (SharedMarkup_ == nil)
	73	SharedMarkup_ = [[WBMarkup alloc] init];
	74	return SharedMarkup_;
	75	}
	76
	77	- (id) init {
	78	if ((self = [super init]) != nil) {
	79	WebThreadLock();
	80
	81	SharedMarkup_ = self;
	82
	83	view_ = [[WebView alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
	84	[view_ setDrawsBackground:NO];
	85
	86	WebPreferences *preferences([[WebPreferences alloc] initWithIdentifier:@"com.apple.webkit.webmarkup"]);
	87	[preferences setPlugInsEnabled:NO];
	88	[view_ setPreferences:preferences];
	89	[preferences release];
	90
	91	window_ = [[WAKWindow alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
	92	[window_ setContentView:view_];
	93
	94	WebFrame *frame([view_ mainFrame]);
	95	[[frame frameView] setAllowsScrolling:NO];
	96	[frame _setLoadsSynchronously:YES];
	97
	98	[frame loadHTMLString:@"<html><body style='margin: 0px; word-wrap: break-word; -khtml-nbsp-mode: space; -khtml-line-break: after-white-space'><div id='size'><div id='text'></div></div></body></html>" baseURL:nil];
	99	} return self;
	100	}
	101
	102	- (void) dealloc {
	103	[window_ release];
	104	[view_ release];
	105	[super dealloc];
	106	}
	107
09ea4166 JF	108	- (void) drawRect:(CGRect)rect {
	109	[text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];
	110
	111	CGRect draw(CGRectMake(0, 0, rect.size.width - rect.origin.x, rect.size.height - rect.origin.y));
	112
	113	CGContextSaveGState(context_); {
	114	CGContextTranslateCTM(context_, rect.origin.x, rect.origin.y);
	115
	116	if (kCFCoreFoundationVersionNumber > 700)
	117	[view_ _drawRect:draw context:context_ lockFocus:YES];
	118	else {
	119	WKView *view([view_ _viewRef]);
	120	WKViewLockFocus(view); {
	121	WKViewDisplayRect(view, draw);
	122	} WKViewUnlockFocus(view);
	123	}
	124	} CGContextRestoreGState(context_);
	125	}
	126
2de63e63 JF	127	- (WebView *) _webView {
	128	return view_;
	129	}
	130
	131	- (void) setStringDrawingOrigin:(CGPoint)origin {
	132	origin_ = origin;
	133	}
	134
	135	- (void) clearStringDrawingOrigin {
	136	origin_ = CGPointZero;
	137	}
	138
	139	- (CGSize) sizeOfMarkup:(NSString *)markup forWidth:(float)width {
	140	WebThreadLock();
	141
	142	if (![self _webPrepareContextForTextDrawing:NO])
	143	return CGSizeZero;
	144
	145	[text_ setInnerHTML:markup];
	146	[text_ removeAttribute:@"style"];
	147
	148	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: 5000px"], width]);
	149	[size_ setAttribute:@"style" value:value];
	150	[value release];
	151
	152	[[view_ mainFrame] forceLayoutAdjustingViewSize:YES];
	153	return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
	154	}
	155
	156	- (CGSize) sizeOfString:(NSString )string withStyle:(NSString )style forWidth:(float)width {
	157	WebThreadLock();
	158
	159	if (![self _webPrepareContextForTextDrawing:NO])
	160	return CGSizeZero;
	161
	162	[size_ removeChild:[size_ firstChild]];
	163
	164	WebFrame *frame([view_ mainFrame]);
	165
	166	[frame forceLayoutAdjustingViewSize:YES];
	167	[text_ setInnerText:string];
	168	[self _setupWithStyle:style width:width height:5000];
	169	[frame forceLayoutAdjustingViewSize:YES];
	170
	171	return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
	172	}
	173
	174	- (NSString ) _styleFormatString:(NSString )style {
	175	return style;
	176	}
	177
	178	- (void) _setupWithStyle:(NSString *)style width:(float)width height:(float)height {
	179	WebThreadLock();
	180
	181	if (style != nil && [style length] != 0)
	182	[text_ setAttribute:@"style" value:style];
	183	else
	184	[text_ removeAttribute:@"style"];
	185
	186	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], width, height]);
	187	[size_ setAttribute:@"style" value:value];
	188	[value release];
	189
	190	[size_ appendChild:text_];
191	}
192
193	- (BOOL) _webPrepareContextForTextDrawing:(BOOL)drawing {
194	WebThreadLock();
195
196	if (document_ == nil) {
197	WebFrame *frame([view_ mainFrame]);
198
199	document_ = [[frame DOMDocument] retain];
200	if (document_ == nil) {
201	NSLog(@"*** ERROR: no DOM document in text-drawing webview");
202	return NO;
203	}
204
205	text_ = [[document_ getElementById:@"text"] retain];
206	size_ = [[document_ getElementById:@"size"] retain];
207
208	if (text_ == nil \|\| size_ == nil) {
209	NSLog(@"*** ERROR: cannot find DOM element required for text drawing");
210	return NO;
211	}
212	}
213
214	context_ = NULL;
215
216	if (!drawing)
217	context_ = NULL;
218	else {
219	context_ = WKGetCurrentGraphicsContext();
220	if (context_ == NULL) {
221	NSLog(@"*** ERROR: no CGContext set for drawing");
222	return NO;
223	}
224	}
225
226	return YES;
227	}
228
229	- (void) drawMarkup:(NSString *)markup atPoint:(CGPoint)point {
230	[self drawMarkup:markup inRect:CGRectMake(point.x, point.y, 65535, 65535)];
231	}
232
233	- (void) drawMarkup:(NSString *)markup inRect:(CGRect)rect {
234	WebThreadLock();
235
236	if (![self _webPrepareContextForTextDrawing:YES])
237	return;
238
239	[text_ setInnerHTML:markup];
240	[text_ removeAttribute:@"style"];
241
242	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], CGRectGetWidth(rect), CGRectGetHeight(rect)]);
243	[size_ setAttribute:@"style" value:value];
244	[value release];
245
246	[[view_ mainFrame] forceLayoutAdjustingViewSize:YES];
247
09ea4166	248	[self drawRect:rect];
2de63e63 JF	249	}
	250
	251	- (void) drawString:(NSString )string atPoint:(CGPoint)point withStyle:(NSString )style {
	252	[self drawString:string inRect:CGRectMake(point.x, point.y, 65535, 65535) withStyle:style];
	253	}
	254
	255	- (void) drawString:(NSString )string inRect:(CGRect)rect withStyle:(NSString )style {
	256	WebThreadLock();
	257
	258	if (![self _webPrepareContextForTextDrawing:YES])
	259	return;
	260
	261	[size_ removeChild:[size_ firstChild]];
	262
	263	WebFrame *frame([view_ mainFrame]);
	264
	265	[frame forceLayoutAdjustingViewSize:YES];
	266	[text_ setInnerText:string];
	267	[self _setupWithStyle:style width:CGRectGetWidth(rect) height:CGRectGetHeight(rect)];
	268	[frame forceLayoutAdjustingViewSize:YES];
	269
09ea4166	270	[self drawRect:rect];
2de63e63 JF	271	}
	272
	273	@end