blob: a71b8275d4064db618580c8bdcf9c553ccacc230 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
//
// ANTLRReaderStream.m
// ANTLR
//
// Created by Alan Condit on 2/21/11.
// Copyright 2011 Alan's MachineWorks. All rights reserved.
//
#import "ANTLRReaderStream.h"
@implementation ANTLRReaderStream
@synthesize fh;
@synthesize size;
@synthesize rbSize;
static NSInteger READ_BUFFER_SIZE = 1024;
static NSInteger INITIAL_BUFFER_SIZE = 1024;
+ (NSInteger) READ_BUFFER_SIZE
{
return READ_BUFFER_SIZE;
}
+ (NSInteger) INITIAL_BUFFER_SIZE
{
return INITIAL_BUFFER_SIZE;
}
+ (id) newANTLRReaderStream
{
return [[ANTLRReaderStream alloc] init];
}
+ (id) newANTLRReaderStream:(NSFileHandle *)r
{
return [[ANTLRReaderStream alloc] initWithReader:r size:INITIAL_BUFFER_SIZE readBufferSize:READ_BUFFER_SIZE];
}
+ (id) newANTLRReaderStream:(NSFileHandle *)r size:(NSInteger)aSize
{
return [[ANTLRReaderStream alloc] initWithReader:r size:aSize readBufferSize:READ_BUFFER_SIZE];
}
+ (id) newANTLRReaderStream:(NSFileHandle *)r size:(NSInteger)aSize readBufferSize:(NSInteger)aReadChunkSize
{
// load(r, aSize, aReadChunkSize);
return [[ANTLRReaderStream alloc] initWithReader:r size:aSize readBufferSize:aReadChunkSize];
}
- (id) init
{
self = [super init];
if ( self != nil ) {
fh = nil;
rbSize = READ_BUFFER_SIZE;
size = INITIAL_BUFFER_SIZE;
}
return self;
}
- (id) initWithReader:(NSFileHandle *)r size:(NSInteger)aSize readBufferSize:(NSInteger)aReadChunkSize
{
self = [super init];
if ( self != nil ) {
fh = r;
rbSize = aSize;
size = aReadChunkSize;
[self load:aSize readBufferSize:aReadChunkSize];
}
return self;
}
- (void) load:(NSInteger)aSize readBufferSize:(NSInteger)aReadChunkSize
{
NSData *retData = nil;
if ( fh==nil ) {
return;
}
if ( aSize<=0 ) {
aSize = INITIAL_BUFFER_SIZE;
}
if ( aReadChunkSize<=0 ) {
aReadChunkSize = READ_BUFFER_SIZE;
}
#pragma mark fix these NSLog calls
@try {
int numRead=0;
int p1 = 0;
retData = [fh readDataToEndOfFile];
numRead = [retData length];
NSLog( @"read %d chars; p was %d is now %d", n, p1, (p1+numRead) );
p1 += numRead;
n = p1;
data = [[NSString alloc] initWithData:retData encoding:NSASCIIStringEncoding];
NSLog( @"n=%d", n );
}
@finally {
[fh closeFile];
}
}
- (void)setUpStreamForFile:(NSString *)path {
// iStream is NSInputStream instance variable
NSInputStream *iStream = [[NSInputStream alloc] initWithFileAtPath:path];
// [iStream setDelegate:self];
[iStream scheduleInRunLoop:[NSRunLoop currentRunLoop]
forMode:NSDefaultRunLoopMode];
[iStream open];
}
- (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)eventCode
{
NSMutableData *myData = nil;
NSNumber *bytesRead = [NSNumber numberWithInteger:0];
switch(eventCode) {
case NSStreamEventHasBytesAvailable:
{
if(!myData) {
myData = [[NSMutableData data] retain];
}
uint8_t buf[1024];
unsigned int len = 0;
len = [(NSInputStream *)stream read:buf maxLength:1024];
if(len) {
[myData appendBytes:(const void *)buf length:len];
// bytesRead is an instance variable of type NSNumber.
bytesRead = [NSNumber numberWithInteger:[bytesRead intValue]+len];
} else {
NSLog(@"no buffer!");
}
break;
}
case NSStreamEventEndEncountered:
{
[stream close];
[stream removeFromRunLoop:[NSRunLoop currentRunLoop]
forMode:NSDefaultRunLoopMode];
[stream release];
stream = nil; // stream is ivar, so reinit it
break;
}
// continued
}
}
- (void) close
{
[fh closeFile];
}
@end
|
