[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 WebView : NSObject
- (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];
}

- (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];

    [text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];

    WKView *view([view_ _viewRef]);

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

        WKViewLockFocus(view); {
            WKViewDisplayRect(view, CGRectMake(0, 0, rect.origin.x, rect.origin.y));
        } WKViewUnlockFocus(view);
    } CGContextRestoreGState(context_);
}

- (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];

    [text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];

    WKView *view([view_ _viewRef]);

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

        WKViewLockFocus(view); {
            WKViewDisplayRect(view, CGRectMake(0, 0, rect.size.width, rect.size.height));
        } WKViewUnlockFocus(view);
    } CGContextRestoreGState(context_);
}

@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
	46	@interface WebView : NSObject
	47	- (id) initWithFrame:(CGRect)frame;
	48	- (WebFrame *) mainFrame;
	49	- (void) setDrawsBackground:(BOOL)value;
	50	- (void) setPreferences:(WebPreferences *)preferences;
	51	- (WKView *) _viewRef;
	52	@end
	53
	54	@interface WAKWindow : NSObject
	55	- (id) initWithFrame:(CGRect)frame;
	56	- (void) setContentView:(WebView *)view;
	57	@end
	58
	59	static WBMarkup *SharedMarkup_;
	60
	61	@implementation WBMarkup
	62
	63	+ (BOOL) isSharedMarkupCreated {
	64	return SharedMarkup_ != nil;
65	}
66
67	+ (WBMarkup *) sharedMarkup {
68	if (SharedMarkup_ == nil)
69	SharedMarkup_ = [[WBMarkup alloc] init];
70	return SharedMarkup_;
71	}
72
73	- (id) init {
74	if ((self = [super init]) != nil) {
75	WebThreadLock();
76
77	SharedMarkup_ = self;
78
79	view_ = [[WebView alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
80	[view_ setDrawsBackground:NO];
81
82	WebPreferences *preferences([[WebPreferences alloc] initWithIdentifier:@"com.apple.webkit.webmarkup"]);
83	[preferences setPlugInsEnabled:NO];
84	[view_ setPreferences:preferences];
85	[preferences release];
86
87	window_ = [[WAKWindow alloc] initWithFrame:CGRectMake(0, 0, 640, 5000)];
88	[window_ setContentView:view_];
89
90	WebFrame *frame([view_ mainFrame]);
91	[[frame frameView] setAllowsScrolling:NO];
92	[frame _setLoadsSynchronously:YES];
93
94	[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];
95	} return self;
96	}
97
98	- (void) dealloc {
99	[window_ release];
100	[view_ release];
101	[super dealloc];
102	}
103
104	- (WebView *) _webView {
105	return view_;
106	}
107
108	- (void) setStringDrawingOrigin:(CGPoint)origin {
109	origin_ = origin;
110	}
111
112	- (void) clearStringDrawingOrigin {
113	origin_ = CGPointZero;
114	}
115
116	- (CGSize) sizeOfMarkup:(NSString *)markup forWidth:(float)width {
117	WebThreadLock();
118
119	if (![self _webPrepareContextForTextDrawing:NO])
120	return CGSizeZero;
121
122	[text_ setInnerHTML:markup];
123	[text_ removeAttribute:@"style"];
124
125	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: 5000px"], width]);
126	[size_ setAttribute:@"style" value:value];
127	[value release];
128
129	[[view_ mainFrame] forceLayoutAdjustingViewSize:YES];
130	return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
131	}
132
133	- (CGSize) sizeOfString:(NSString )string withStyle:(NSString )style forWidth:(float)width {
134	WebThreadLock();
135
136	if (![self _webPrepareContextForTextDrawing:NO])
137	return CGSizeZero;
138
139	[size_ removeChild:[size_ firstChild]];
140
141	WebFrame *frame([view_ mainFrame]);
142
143	[frame forceLayoutAdjustingViewSize:YES];
144	[text_ setInnerText:string];
145	[self _setupWithStyle:style width:width height:5000];
146	[frame forceLayoutAdjustingViewSize:YES];
147
148	return [[view_ mainFrame] renderedSizeOfNode:text_ constrainedToWidth:width];
149	}
150
151	- (NSString ) _styleFormatString:(NSString )style {
152	return style;
153	}
154
155	- (void) _setupWithStyle:(NSString *)style width:(float)width height:(float)height {
156	WebThreadLock();
157
158	if (style != nil && [style length] != 0)
159	[text_ setAttribute:@"style" value:style];
160	else
161	[text_ removeAttribute:@"style"];
162
163	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], width, height]);
164	[size_ setAttribute:@"style" value:value];
165	[value release];
166
167	[size_ appendChild:text_];
168	}
169
170	- (BOOL) _webPrepareContextForTextDrawing:(BOOL)drawing {
171	WebThreadLock();
172
173	if (document_ == nil) {
174	WebFrame *frame([view_ mainFrame]);
175
176	document_ = [[frame DOMDocument] retain];
177	if (document_ == nil) {
178	NSLog(@"*** ERROR: no DOM document in text-drawing webview");
179	return NO;
180	}
181
182	text_ = [[document_ getElementById:@"text"] retain];
183	size_ = [[document_ getElementById:@"size"] retain];
184
185	if (text_ == nil \|\| size_ == nil) {
186	NSLog(@"*** ERROR: cannot find DOM element required for text drawing");
187	return NO;
188	}
189	}
190
191	context_ = NULL;
192
193	if (!drawing)
194	context_ = NULL;
195	else {
196	context_ = WKGetCurrentGraphicsContext();
197	if (context_ == NULL) {
198	NSLog(@"*** ERROR: no CGContext set for drawing");
199	return NO;
200	}
201	}
202
203	return YES;
204	}
205
206	- (void) drawMarkup:(NSString *)markup atPoint:(CGPoint)point {
207	[self drawMarkup:markup inRect:CGRectMake(point.x, point.y, 65535, 65535)];
208	}
209
210	- (void) drawMarkup:(NSString *)markup inRect:(CGRect)rect {
211	WebThreadLock();
212
213	if (![self _webPrepareContextForTextDrawing:YES])
214	return;
215
216	[text_ setInnerHTML:markup];
217	[text_ removeAttribute:@"style"];
218
219	NSString *value([[NSString alloc] initWithFormat:[self _styleFormatString:@"width: %.0fpx; height: %.0fpx"], CGRectGetWidth(rect), CGRectGetHeight(rect)]);
220	[size_ setAttribute:@"style" value:value];
221	[value release];
222
223	[[view_ mainFrame] forceLayoutAdjustingViewSize:YES];
224
225	[text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];
226
227	WKView *view([view_ _viewRef]);
228
229	CGContextSaveGState(context_); {
230	CGContextTranslateCTM(context_, rect.origin.x, rect.origin.y);
231
232	WKViewLockFocus(view); {
233	WKViewDisplayRect(view, CGRectMake(0, 0, rect.origin.x, rect.origin.y));
234	} WKViewUnlockFocus(view);
235	} CGContextRestoreGState(context_);
236	}
237
238	- (void) drawString:(NSString )string atPoint:(CGPoint)point withStyle:(NSString )style {
239	[self drawString:string inRect:CGRectMake(point.x, point.y, 65535, 65535) withStyle:style];
240	}
241
242	- (void) drawString:(NSString )string inRect:(CGRect)rect withStyle:(NSString )style {
243	WebThreadLock();
244
245	if (![self _webPrepareContextForTextDrawing:YES])
246	return;
247
248	[size_ removeChild:[size_ firstChild]];
249
250	WebFrame *frame([view_ mainFrame]);
251
252	[frame forceLayoutAdjustingViewSize:YES];
253	[text_ setInnerText:string];
254	[self _setupWithStyle:style width:CGRectGetWidth(rect) height:CGRectGetHeight(rect)];
255	[frame forceLayoutAdjustingViewSize:YES];
256
257	[text_ setScrollXOffset:origin_.x scrollYOffset:origin_.y];
258
259	WKView *view([view_ _viewRef]);
260
261	CGContextSaveGState(context_); {
262	CGContextTranslateCTM(context_, rect.origin.x, rect.origin.y);
263
264	WKViewLockFocus(view); {
265	WKViewDisplayRect(view, CGRectMake(0, 0, rect.size.width, rect.size.height));
266	} WKViewUnlockFocus(view);
267	} CGContextRestoreGState(context_);
268	}
269
270	@end