readcol.pro
10.6 KB
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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
; -----------------------------------------------------------------------------
;
; Copyright (C) 1997-2008 Krzysztof M. Gorski, Eric Hivon, Anthony J. Banday
;
;
;
;
;
; This file is part of HEALPix.
;
; HEALPix is free software; you can redistribute it and/or modify
; it under the terms of the GNU General Public License as published by
; the Free Software Foundation; either version 2 of the License, or
; (at your option) any later version.
;
; HEALPix is distributed in the hope that it will be useful,
; but WITHOUT ANY WARRANTY; without even the implied warranty of
; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
; GNU General Public License for more details.
;
; You should have received a copy of the GNU General Public License
; along with HEALPix; if not, write to the Free Software
; Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
;
; For more information about HEALPix see http://healpix.jpl.nasa.gov
;
; -----------------------------------------------------------------------------
pro readcol,name,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15, $
v16,v17,v18,v19,v20,v21,v22,v23,v24,v25, $
FORMAT = fmt, DEBUG=debug, SILENT=silent, SKIPLINE = skipline, $
NUMLINE = numline, DELIMITER = delimiter, COMMENT = comment
;+
; NAME:
; READCOL
; PURPOSE:
; Read a free-format ASCII file with columns of data into IDL vectors
; EXPLANATION:
; Lines of data not meeting the specified format (e.g. comments) are
; ignored. Columns may be separated by commas or spaces.
;
; Use READFMT to read a fixed-format ASCII file. Use RDFLOAT for
; much faster I/O (but less flexibility). Use FORPRINT to write
; columns of data (inverse of READCOL).
;
; CALLING SEQUENCE:
; READCOL, name, v1, [ v2, v3, v4, v5, ... v25 ,
; DELIMITER= ,FORMAT = , /DEBUG , /SILENT , SKIPLINE = , NUMLINE =,
; COMMENT= ]
;
; INPUTS:
; NAME - Name of ASCII data file, scalar string. In VMS, an extension of
; .DAT is assumed, if not supplied.
;
; OPTIONAL INPUT KEYWORDS:
; FORMAT - scalar string containing a letter specifying an IDL type
; for each column of data to be read. Allowed letters are
; A - string data, B - byte, D - double precision, F- floating
; point, I - integer, L - longword, Z - longword hexadecimal,
; and X - skip a column.
;
; Columns without a specified format are assumed to be floating
; point. Examples of valid values of FMT are
;
; 'A,B,I' ;First column to read as a character string, then
; 1 column of byte data, 1 column integer data
; 'L,L,L,L' ;Four columns will be read as longword arrays.
; ' ' ;All columns are floating point
;
; If a FORMAT keyword string is not supplied, then all columns are
; assumed to be floating point.
;
; /SILENT - Normally, READCOL will display each line that it skips over.
; If SILENT is set and non-zero then these messages will be
; suppressed.
; /DEBUG - If this keyword is non-zero, then additional information is
; printed as READCOL attempts to read and interpret the file.
; DELIMITER - single character specifying delimiter used to separate
; columns. Default is either a comma or a blank.
; SKIPLINE - Scalar specifying number of lines to skip at the top of file
; before reading. Default is to start at the first line.
; NUMLINE - Scalar specifying number of lines in the file to read.
; Default is to read the entire file
; COMMENT - Scalar string : lines beginning with this string are ignored
; (case sensitive)
;
; OUTPUTS:
; V1,V2,V3,...V25 - IDL vectors to contain columns of data.
; Up to 25 columns may be read. The type of the output vectors
; are as specified by FORMAT.
;
; EXAMPLES:
; Each row in a file position.dat contains a star name and 6 columns
; of data giving an RA and Dec in sexigesimal format. Read into IDL
; variables. (NOTE: The star names must not contain internal spaces.)
;
; IDL> FMT = 'A,I,I,F,I,I,F'
; IDL> READCOL,'position.dat',F=FMT,name,hr,min,sec,deg,dmin,dsec
;
; The HR,MIN,DEG, and DMIN variables will be integer vectors.
;
; Alternatively, all except the first column could be specified as
; floating point.
;
; IDL> READCOL,'position.dat',F='A',name,hr,min,sec,deg,dmin,dsec
;
; To read just the variables HR,MIN,SEC
; IDL> READCOL,'position.dat',F='X,I,I,F',HR,MIN,SEC
;
; RESTRICTIONS:
; This procedure is designed for generality and not for speed.
; If a large ASCII file is to be read repeatedly, it may be worth
; writing a specialized reader.
;
; Columns to be read as strings must not contain the delimiter character
; (i.e. commas or spaces by default). Either change the default
; delimiter with the DELIMITER keyword, or use READFMT to read such files.
;
; Numeric values are converted to specified format. For example,
; the value 0.13 read with an 'I' format will be converted to 0.
;
; PROCEDURES CALLED
; GETTOK(), NUMLINES(), REPCHR(), STRNUMBER(), ZPARCHECK
;
; REVISION HISTORY:
; Written W. Landsman November, 1988
; Modified J. Bloch June, 1991
; (Fixed problem with over allocation of logical units.)
; Added SKIPLINE and NUMLINE keywords W. Landsman March 92
; Read a maximum of 25 cols. Joan Isensee, Hughes STX Corp., 15-SEP-93.
; Call NUMLINES() function W. Landsman Feb. 1996
; Added DELIMITER keyword W. Landsman Nov. 1999
; Fix indexing typos (i for k) that mysteriously appeared W. L. Mar. 2000
; Hexadecimal support added. MRG, RITSS, 15 March 2000.
;
; added keyword comment
;-
On_error,2 ;Return to caller
if N_params() lt 2 then begin
print,'Syntax - readcol, name, v1, [ v2, v3,...v25, '
print,' FORMAT= ,/SILENT ,SKIPLINE =, NUMLINE = , /DEBUG]'
return
endif
; Get number of lines in file
nlines = NUMLINES( name )
if nlines LT 0 then return
if keyword_set(DEBUG) then $
message,strupcase(name)+' contains ' + strtrim(nlines,2) + ' lines',/INF
if not keyword_set( SKIPLINE ) then skipline = 0
nlines = nlines - skipline
if keyword_set( NUMLINE) then nlines = numline < nlines
ncol = N_params() - 1 ;Number of columns of data expected
vv = 'v' + strtrim( indgen(ncol)+1, 2)
nskip = 0
if N_elements(fmt) GT 0 then begin ;FORMAT string supplied?
zparcheck, 'READCOL', fmt, 2, 7, 0, 'FORMAT string'
; Remove blanks from format string
frmt = strupcase(strcompress(fmt,/REMOVE))
remchar, frmt, '(' ;Remove parenthesis from format
remchar, frmt, ')'
; Determine number of columns to skip ('X' format)
pos = strpos(frmt, 'X', 0)
while pos NE -1 do begin
pos = strpos( frmt, 'X', pos+1)
nskip = nskip + 1
endwhile
endif else begin ;Read everything as floating point
frmt = 'F'
if ncol GT 1 then for i = 1,ncol-1 do frmt = frmt + ',F'
if not keyword_set( SILENT ) then message, $
'Format keyword not supplied - All columns assumed floating point',/INF
endelse
nfmt = ncol + nskip
idltype = intarr(nfmt)
; Create output arrays according to specified formats
k = 0L ;Loop over output columns
hex = bytarr(nfmt)
for i = 0L, nfmt-1 do begin
fmt1 = gettok( frmt, ',' )
if fmt1 EQ '' then fmt1 = 'F' ;Default is F format
case strmid(fmt1,0,1) of
'A': idltype[i] = 7
'D': idltype[i] = 5
'F': idltype[i] = 4
'I': idltype[i] = 2
'B': idltype[i] = 1
'L': idltype[i] = 3
'Z': begin
idltype[i] = 3 ;Hexadecimal
hex[i] = 1b
end
'X': idltype[i] = 0 ;IDL type of 0 ==> to skip column
ELSE: message,'Illegal format ' + fmt1 + ' in field ' + strtrim(i,2)
endcase
; Define output arrays
if idltype[i] NE 0 then begin
st = vv[k] + '= make_array(nlines,TYPE = idltype[i] )'
tst = execute(st)
k = k+1
endif
endfor
openr, lun, name, /get_lun
ngood = 0L
temp = ' '
if skipline GT 0 then $
for i = 0, skipline-1 do readf, lun, temp ;Skip any lines
if n_elements(comment) eq 0 then comment=''
lencom = strlen(comment)
for j = 0L, nlines-1 do begin
readf, lun, temp
;; if strlen(temp) LT ncol then begin ;Need at least 1 chr per output line
if (strlen(temp) LT ncol) or (lencom gt 0 and strmid(temp,0,lencom) eq comment) then begin ; modified by EH
ngood = ngood-1
if not keyword_set(SILENT) then $
message,'Skipping Line ' + strtrim(skipline+j+1,2),/INF
goto, BADLINE
endif
if not keyword_set(delimiter) then begin
temp = repchr(temp,',',' ') ;Replace comma delimiters by spaces
delimiter = ' '
endif
k = 0
for i = 0L,nfmt-1 do begin
temp = strtrim(temp,1) ;Remove leading spaces
var = gettok(temp,delimiter) ;Read next field
if ( idltype[i] NE 0 ) then begin ;Expecting data?
if ( idltype[i] NE 7 ) then begin ;Check for valid numeric data
tst = strnumber(var,val,hex=hex[i]) ;Valid number?
if tst EQ 0 then begin ;If not, skip this line
if not keyword_set(SILENT) then $
message,'Skipping Line ' + strtrim(skipline+j+1,2),/INF
ngood = ngood-1
goto, BADLINE
endif
st = vv[k] + '[ngood] = val'
endif else $
st = vv[k] + '[ngood] = strtrim(var,2)'
tst = execute(st)
k = k+1
endif
endfor
BADLINE: ngood = ngood+1
endfor
free_lun,lun
if ngood EQ 0 then begin
message,'ERROR - No valid lines found for specified format',/INFORM
return
endif
if not keyword_set(SILENT) then $
message,strtrim(ngood,2) + ' valid lines read', /INFORM
; Compress arrays to match actual number of valid lines
for i = 0,ncol-1 do begin
tst = execute(vv[i] + '='+ vv[i]+ '[0:ngood-1]')
endfor
return
end