The R Project SVN R-packages

#import "Tools/RTooltip.h"

		[self performSelector:@selector(doSyntaxHighlighting) withObject:nil afterDelay:0.05f];

			[(NoodleLineNumberView*)[[self enclosingScrollView] verticalRulerView] setLineWrappingMode:[Preferences flagForKey:enableLineWrappingKey withDefault: YES]];

	NSSize contentSize = [scrollView contentSize];

	NSTextContainer *textContainer = [self textContainer];

		[(NoodleLineNumberView*)[[self enclosingScrollView] verticalRulerView] setLineWrappingMode:YES];

	// NSSize layoutSize;

	// 

	// lineWrappingEnabled = [Preferences flagForKey:enableLineWrappingKey withDefault: YES];

	// [self setHorizontallyResizable:YES];

	// if (!lineWrappingEnabled) {

	// 	layoutSize = NSMakeSize(10e6,10e6);

	// 	[scrollView setHasHorizontalScroller:YES];

	// 	[self setMaxSize:layoutSize];

	// 	[[self textContainer] setContainerSize:layoutSize];

	// 	[[self textContainer] setWidthTracksTextView:NO];

	// } else {

	// 	[scrollView setHasHorizontalScroller:NO];

	// 	layoutSize = [self maxSize];

	// 	[self setMaxSize:layoutSize];

	// 	[[self textContainer] setContainerSize:layoutSize];

	// 	[[self textContainer] setWidthTracksTextView:YES];

	// 	// Enforce view to be re-layouted correctly

	// 	[[self undoManager] disableUndoRegistration];

	// 	[self selectAll:nil];

	// 	[self cut:nil];

	// 	[self paste:nil];

	// 	[[self undoManager] enableUndoRegistration];

	// }

	// [[self textContainer] setHeightTracksTextView:NO];

		if(lineNumberingEnabled) {

			[NSObject cancelPreviousPerformRequestsWithTarget:[[self enclosingScrollView] verticalRulerView] 

										selector:@selector(refresh) 

										object:nil];

			[[[self enclosingScrollView] verticalRulerView] performSelector:@selector(refresh) withObject:nil afterDelay:0.001f];

		}

/*

The idea of code folding is to replace the folded range while layouting by NSControlGlyphs

which will be rendered as zero width glyphs. This has the advantage that all text actions like 

copying, line numbering, etc. works without additional stuff.

For now it is only possible to fold a block if:

- the starting { is placed at the end of a line (or with a following comment)

- there's only ONE starting { in that line and NO }

- the ending } is placed at the of a line (or with a following comment)

- there's only ONE ending } in that line and NO {

Due to speed issues all folded ranges will be stored in a 3d C array of the size

R_MAX_FOLDED_ITEMS = 1024 defined in RScriptEditorTextStorage.h whereby index:

0 - range location

1 - range length

2 - location + length (pre-calculate for speed)

If the user locates the cursor inside a folded range or an action will locate the cursor

inside such a range (go to line, find something) the folded range will be unfolded.

... for more info ask Hans-J. Bibiko

*/

- (IBAction)unfoldCurrentBlock:(id)sender

{

	NSUInteger caretPosition = [self selectedRange].location;

	NSRange r = [[self string] lineRangeForRange:NSMakeRange(caretPosition, 0)];

	if([theTextStorage foldedAtIndex:NSMaxRange(r)] > -1) {

		[self unfoldLinesContainingCharacterAtIndex:NSMaxRange(r)];

		return;

	}

	if([theTextStorage foldedAtIndex:r.location] > -1) {

		[self unfoldLinesContainingCharacterAtIndex:r.location];

		return;

	}

}

	NSUInteger caretPosition = [self selectedRange].location;

	NSInteger foldItem = 0;

	unichar c;

	NSRange r = [[self string] lineRangeForRange:NSMakeRange(caretPosition, 0)];

	foldItem = [self foldStatusAtIndex:NSMaxRange(r)-2];

	if(foldItem == 1) { // is current line set to ▼ set caret to end of line

		caretPosition = NSMaxRange(r)-1;

	} else { // otherwise set caret to begin of line

		caretPosition = r.location;

	}

	NSUInteger stringLength = [[self string] length];

	if(caretPosition == 0 || caretPosition >= [[self string] length]) return;

	CFStringRef parserStringRef = (CFStringRef)[self string];

	unichar co = '{'; // opening char

	unichar cc = '}'; // closing char

	NSInteger start = -1;

	NSInteger end = -1;

	NSInteger bracketCounter = 0;

	c = CFStringGetCharacterAtIndex(parserStringRef, caretPosition);

	if(c == cc)

		bracketCounter--;

	if(c == co)

		bracketCounter++;

	for(NSInteger i=caretPosition; i>=0; i--) {

		if([self parserContextForPosition:i] != pcExpression) continue;

		c = CFStringGetCharacterAtIndex(parserStringRef, i);

		if(c == co) {

			if(!bracketCounter) {

				start = i;

				break;

			}

			bracketCounter--;

		}

		if(c == cc) {

			bracketCounter++;

		}

	}

	if(start < 0 ) return;

	// go up for lines like "} else {"

	if(start && [self foldStatusAtIndex:start-1] == 0) {

		for(NSInteger i=start-1; i>=0; i--) {

			c = CFStringGetCharacterAtIndex(parserStringRef, i);

			if(c == '\n' || c == '\r') break;

			if([self parserContextForPosition:i] != pcExpression) continue;

			if(c == cc && i > 0) {

				bracketCounter = 0;

				for(NSInteger j=i-1; j>=0; j--) {

					if([self parserContextForPosition:j] != pcExpression) continue;

					c = CFStringGetCharacterAtIndex(parserStringRef, j);

					if(c == co) {

						if(!bracketCounter) {

							start = j;

							break;

						}

						bracketCounter--;

					}

					if(c == cc) {

						bracketCounter++;

					}

				}

				break;

			}

		}

	}		

	bracketCounter = 0;

	for(NSUInteger i=caretPosition; i<stringLength; i++) {

		if([self parserContextForPosition:i] != pcExpression) continue;

		c = CFStringGetCharacterAtIndex(parserStringRef, i);

		if(c == co) {

			bracketCounter++;

		}

		if(c == cc) {

			if(!bracketCounter) {

				end = i+1;

				BOOL goAhead = NO;

				//go ahead for lines a la  "} else {"

				for(NSUInteger j=end; j<stringLength; j++) {

					c = CFStringGetCharacterAtIndex(parserStringRef, j);

					if(c == '\n' || c == '\r') {

						break;

					}

					if(c == '\t' || c == ' ') {

						continue;

					}

					if([self parserContextForPosition:j] != pcExpression) continue;

					if(c == co) {

						goAhead = YES;

						break;

					}

				}

				if(!goAhead) break;

			}

			bracketCounter--;

		}

	}

	if(end < 0 || bracketCounter || end-start < 1) return;

	NSRange foldRange = NSMakeRange(start, end-start);

	if(![theTextStorage existsFoldedRange:foldRange]) {

		// set caret for ▲ line inside {} for scrolling

		if(foldItem == 2)

			[self setSelectedRange:NSMakeRange(r.location, 0)];

		[self foldLinesInRange:foldRange];

	}

}

- (IBAction)foldBlockAtLevel:(id)sender

{

	NSInteger level = [sender tag];

	NSInteger bracketCounter = 0;

	NSInteger start = 0;

	NSInteger end = 0;

	CFStringRef str = (CFStringRef)[self string];

	unichar c;

	// [self unFoldAllBlocks:self];

	for(NSInteger i=0; i<[[self string] length]; i++) {

		c = CFStringGetCharacterAtIndex(str, i);

		if([self parserContextForPosition:i] != pcExpression) continue;

		if(c == '{') {

			bracketCounter++;

			if([self foldStatusAtIndex:i] == 1) {

				if(bracketCounter == level+1) {

					start = i;

				}

			}

			continue;

		}

		if(c == '}') {

			bracketCounter--;

			if([self foldStatusAtIndex:i] == 2) {

				if(bracketCounter == level) {

					end = i;

					NSRange r = NSMakeRange(start, end - start+1);

					if(![theTextStorage existsFoldedRange:r])

						[self foldLinesInRange:r];

				}

				if(bracketCounter < 0) {

					NSBeep();

					return;

				}

			}

		}

	}

}

- (IBAction)unFoldAllBlocks:(id)sender

{

	[theTextStorage removeAllFoldedRanges];

	[self didChangeText];

	NSRange r = NSMakeRange(0, [[self string] length]);

	[theTextStorage fixAttributesInRange:r];

	[theTextStorage fixAttachmentAttributeInRange:r];

	[theTextStorage ensureAttributesAreFixedInRange:r];

	if(lineNumberingEnabled)

		[[[self enclosingScrollView] verticalRulerView] performSelector:@selector(refresh) withObject:nil afterDelay:0.0f];

	[NSObject cancelPreviousPerformRequestsWithTarget:self 

							selector:@selector(doSyntaxHighlighting) 

							object:nil];

	[self performSelector:@selector(doSyntaxHighlighting) withObject:nil afterDelay:0.0f];

}

- (void)refoldLinesInRange:(NSRange)range

{

		[RTooltip showWithObject:NLS(@"Maximum number of folded code fragments is reached.") atLocation:[NSEvent mouseLocation]];

	range.location++;

	range.length -= 2;

	if(!range.length) return;

	NSString *tooltip = nil;

	if(range.length < 300)

	else

		tooltip = [[[self string] substringWithRange:NSMakeRange(range.location, 300)] stringByAppendingString:@"\n…"];

	[theTextStorage beginEditing];

	[theTextStorage addAttribute:NSCursorAttributeName value:[NSCursor arrowCursor] range:range];

	[theTextStorage addAttribute:NSToolTipAttributeName value:tooltip range:range];

	[theTextStorage endEditing];

}

{

		return NO;

	BOOL caretWasInsideFoldedRange = NO;

	// Check for valid folding range

	NSRange startLineRange = [selfStr lineRangeForRange:NSMakeRange(range.location, 0)];

	NSRange endLineRange   = [selfStr lineRangeForRange:NSMakeRange(NSMaxRange(range), 0)];

	if(startLineRange.location == endLineRange.location) {

		return NO;

	}

	status = [self foldStatusAtIndex:NSMaxRange(startLineRange)-2];	

		return NO;

	}

	unichar c = CFStringGetCharacterAtIndex((CFStringRef)selfStr, NSMaxRange(endLineRange)-1);

	if(c == '\n' || c == '\r')

		status = [self foldStatusAtIndex:NSMaxRange(endLineRange)-2];

	else

		status = [self foldStatusAtIndex:NSMaxRange(endLineRange)-1];	

	if(status != 2) {

		return NO;

	}

	if(caretPosition >= range.location && caretPosition < NSMaxRange(range)) {

		[self setSelectedRange:NSMakeRange(NSMaxRange(range), 0)];

		caretWasInsideFoldedRange = YES;

	}

	NSInteger foldId = [theTextStorage registerFoldedRange:range];

	if(foldId < 0) {

		[RTooltip showWithObject:NLS(@"Maximum number of folded code fragments is reached.") atLocation:[NSEvent mouseLocation]];

		return NO;

	}

	range.location++;

	range.length -= 2;

	if(!range.length) return NO;

	[[self undoManager] disableUndoRegistration];

	if(![self shouldChangeTextInRange:range replacementString:nil]) {

		[[self undoManager] enableUndoRegistration];

		return NO;

	}

	[[self undoManager] enableUndoRegistration];

	NSRange r = [[self layoutManager] characterRangeForGlyphRange:[[self layoutManager] 

										glyphRangeForBoundingRect:[scrollView documentVisibleRect] 

												  inTextContainer:[self textContainer]] actualGlyphRange:NULL];

	[theTextStorage ensureAttributesAreFixedInRange:r];

	[self performSelector:@selector(doSyntaxHighlighting) withObject:nil afterDelay:0.02f];

	return YES;

}

- (NSInteger)foldStatusAtIndex:(NSInteger)index

{

	if(index < 0 || index >= [[self string] length]) return 0;

	NSInteger status = 0; // 0 = no; 1 = ▼; 2 = ▲

	NSInteger i = index;

	NSInteger type;

	NSString *selfStr = [self string];

	unichar c;

	BOOL isRd = [[self delegate] isRdDocument];

	unichar commentSign = (isRd) ? '%' : '#';

	// start checking from the end of a line

	while(i >= 0) {

		c = CFStringGetCharacterAtIndex((CFStringRef)selfStr, i);

		// Check only one line

		if(c=='\n' || c=='\r') break;

		// Ignore white spaces and comment sign

		if(c==' ' || c=='\t' || c==commentSign) {

			i--;

			continue;

		}

		type = [self parserContextForPosition:i];

		// Ignore comments

		if(type == pcComment || c==commentSign) {

			i--;

			continue;

		}

		// ======= Check for ▼ 

		if(c=='{' && type == pcExpression) {

			status = 1;

			// look for lines a la "} else {" - if so do not draw folding marker

			i--;

			while(i>=0) {

				c = CFStringGetCharacterAtIndex((CFStringRef)selfStr, i);

				if(c=='\n' || c=='\r') break;

				if((c=='}' || c=='{')&& [self parserContextForPosition:i] == pcExpression) {

					status = 0;

					break;

				}

				i--;

			}

			break;

		}

		// ======= Check for ▲

		if(c=='}') {

			status = 2;

			i--;

			while(i>=0) {

				c = CFStringGetCharacterAtIndex((CFStringRef)selfStr, i);

				if(c=='\n' || c=='\r') break;

				if((c=='}' || c=='{')&& [self parserContextForPosition:i] == pcExpression) {

					status = 0;

					break;

				}

				if(isRd) {

					if(c != '\t' || c != ' ') {

						status = 0;

						break;

					}

				}

				i--;

			}

		}

		break;

	}

	return status;

		NSRange range = [theTextStorage foldedRangeAtIndex:foldIndex];

		[[self undoManager] disableUndoRegistration];

		if(![self shouldChangeTextInRange:range replacementString:nil]) {

			[[self undoManager] enableUndoRegistration];

			return NO;

		}

		[[self undoManager] enableUndoRegistration];

		NSRange r = [[self layoutManager] characterRangeForGlyphRange:[[self layoutManager] 

											glyphRangeForBoundingRect:[scrollView documentVisibleRect] 

		[theTextStorage ensureAttributesAreFixedInRange:r];

		if(lineNumberingEnabled)

			[[[self enclosingScrollView] verticalRulerView] performSelector:@selector(refresh) withObject:nil afterDelay:0.0f];

		[self performSelector:@selector(doSyntaxHighlighting) withObject:nil afterDelay:0.02f];

	if(![theTextStorage isKindOfClass:[RScriptEditorTextStorage class]]) {

		[super mouseDown:event];

		return;

	}

	RScriptEditorLayoutManager *layoutManager = (RScriptEditorLayoutManager*)[self layoutManager];

	NSUInteger glyphIndex = [layoutManager glyphIndexForPoint:[self convertPointFromBase:[event locationInWindow]] inTextContainer:[self textContainer]];

	if (glyphIndex < [layoutManager numberOfGlyphs]) {

		NSUInteger charIndex = [layoutManager characterIndexForGlyphAtIndex:glyphIndex];

		NSInteger foldIndex  = [theTextStorage foldedForIndicatorAtIndex:charIndex];

		if (foldIndex > -1) {

			NSInteger foldStart = [theTextStorage foldedRangeAtIndex:foldIndex].location+1;

			NSTextAttachment *attachment = [theTextStorage attribute:NSAttachmentAttributeName atIndex:foldStart effectiveRange:NULL];

			if (attachment) {

				NSTextAttachmentCell *cell = (NSTextAttachmentCell *)[attachment attachmentCell];

				NSRect cellFrame;

				NSPoint delta;

				glyphIndex = [layoutManager glyphIndexForCharacterAtIndex:foldStart];

				cellFrame.origin = [self textContainerOrigin];

				cellFrame.size = [layoutManager attachmentSizeForGlyphAtIndex:glyphIndex];

				delta = [layoutManager lineFragmentRectForGlyphAtIndex:glyphIndex effectiveRange:NULL].origin;

				cellFrame.origin.x += delta.x;

				cellFrame.origin.y += delta.y;

				cellFrame.origin.x += [layoutManager locationForGlyphAtIndex:glyphIndex].x;

				if ([cell wantsToTrackMouseForEvent:event inRect:cellFrame ofView:self atCharacterIndex:foldStart] 

						&& [cell trackMouse:event inRect:cellFrame ofView:self atCharacterIndex:foldStart untilMouseUp:YES]) return;

			}

		}

	}