1// 2// ANTLRTreePatternLexer.m 3// ANTLR 4// 5// Created by Alan Condit on 6/18/10. 6// [The "BSD licence"] 7// Copyright (c) 2010 Alan Condit 8// All rights reserved. 9// 10// Redistribution and use in source and binary forms, with or without 11// modification, are permitted provided that the following conditions 12// are met: 13// 1. Redistributions of source code must retain the above copyright 14// notice, this list of conditions and the following disclaimer. 15// 2. Redistributions in binary form must reproduce the above copyright 16// notice, this list of conditions and the following disclaimer in the 17// documentation and/or other materials provided with the distribution. 18// 3. The name of the author may not be used to endorse or promote products 19// derived from this software without specific prior written permission. 20// 21// THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 22// IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 23// OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 24// IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 25// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 26// NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 30// THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 32#import "ANTLRTreePatternLexer.h" 33 34@implementation ANTLRTreePatternLexer 35 36@synthesize pattern; 37@synthesize p; 38@synthesize c; 39@synthesize n; 40@synthesize sval; 41@synthesize data; 42@synthesize error; 43 44+ (ANTLRTreePatternLexer *)newANTLRTreePatternLexer:(NSString *)aPattern 45{ 46 return [[ANTLRTreePatternLexer alloc] initWithPattern:aPattern]; 47} 48 49- (id) init 50{ 51 if ((self = [super init]) != nil ) { 52 p = -1; 53 n = 0; 54 error = NO; 55 sval = [[NSMutableData dataWithLength:1000] retain]; 56 data = [sval mutableBytes]; 57 pattern = @""; 58 n = [pattern length]; 59 if ( pattern ) [pattern retain]; 60 [self consume]; 61 } 62 return self; 63} 64 65- (id) initWithPattern:(NSString *)aPattern 66{ 67 if ((self = [super init]) != nil ) { 68 p = -1; 69 n = 0; 70 error = NO; 71 sval = [[NSMutableData dataWithLength:1000] retain]; 72 data = [sval mutableBytes]; 73 pattern = [aPattern retain]; 74 n = [pattern length]; 75 [self consume]; 76 } 77 return self; 78} 79 80- (void) dealloc 81{ 82#ifdef DEBUG_DEALLOC 83 NSLog( @"called dealloc in ANTLRTreePatternLexer" ); 84#endif 85 if ( pattern ) [pattern release]; 86 if ( sval ) [sval release]; 87 [super dealloc]; 88} 89 90- (NSInteger) nextToken 91{ 92 n = 0; // reset, but reuse buffer 93 while ( c != ANTLRLexerTokenTypeEOF ) { 94 if ( c==' ' || c=='\n' || c=='\r' || c=='\t' ) { 95 [self consume]; 96 continue; 97 } 98 if ( (c>='a' && c<='z') || (c>='A' && c<='Z') || c=='_' ) { 99 data[n++] = (char)c; 100 [self consume]; 101 while ( (c>='a' && c<='z') || (c>='A' && c<='Z') || 102 (c>='0' && c<='9') || c=='_' ) 103 { 104 data[n++] = (char)c; 105 [self consume]; 106 } 107 return ANTLRLexerTokenTypeID; 108 } 109 if ( c == '(' ) { 110 [self consume]; 111 return ANTLRLexerTokenTypeBEGIN; 112 } 113 if ( c==')' ) { 114 [self consume]; 115 return ANTLRLexerTokenTypeEND; 116 } 117 if ( c=='%' ) { 118 [self consume]; 119 return ANTLRLexerTokenTypePERCENT; 120 } 121 if ( c==':' ) { 122 [self consume]; 123 return ANTLRLexerTokenTypeCOLON; 124 } 125 if ( c=='.' ) { 126 [self consume]; 127 return ANTLRLexerTokenTypeDOT; 128 } 129 if ( c=='[' ) { // grab [x] as a string, returning x 130 [self consume]; 131 while ( c!=']' ) { 132 if ( c=='\\' ) { 133 [self consume]; 134 if ( c!=']' ) { 135 data[n++] = (char)'\\'; 136 } 137 data[n++] = (char)c; 138 } 139 else { 140 data[n++] = (char)c; 141 } 142 [self consume]; 143 } 144 [self consume]; 145 return ANTLRLexerTokenTypeARG; 146 } 147 [self consume]; 148 error = true; 149 return ANTLRLexerTokenTypeEOF; 150 } 151 return ANTLRLexerTokenTypeEOF; 152} 153 154- (void) consume 155{ 156 p++; 157 if ( p >= n ) { 158 c = ANTLRLexerTokenTypeEOF; 159 } 160 else { 161 c = [pattern characterAtIndex:p]; 162 } 163} 164 165- (NSString *)toString 166{ 167 char buf[100]; 168 169 NSInteger idx = 0; 170 for( NSInteger i = p; i < n; i++ ){ 171 buf[idx++] = data[i]; 172 } 173 buf[idx] = '\0'; 174 return [NSString stringWithFormat:@"%s", buf]; 175} 176 177- (NSMutableData *)getSval 178{ 179 return sval; 180} 181 182- (void)setSval:(NSMutableData *)aSval 183{ 184 if ( sval != aSval ) { 185 if ( sval ) [sval release]; 186 [aSval retain]; 187 } 188 sval = aSval; 189} 190 191@end 192