[85] | 1 | function getget2d, name, charvar, count=count, offset=offset, stride=stride, coordmean=coordmean, anomaly=anomaly |
---|
| 2 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
---|
| 3 | ; name: char |
---|
| 4 | ; charvar: char |
---|
| 5 | ; count: 4 elements array ; The dimensions of the sub-image |
---|
| 6 | ; offset: 4 elements array ; A variable that contains the offset for the sub-image |
---|
| 7 | ; stride: 4 elements array ; Create a variable to be used as a value for the STRIDE keyword |
---|
| 8 | ; coordmean: 1 to 4 elements array (dimension 1 is 1) |
---|
| 9 | ; anomaly: a non-null element that would be filled with mean |
---|
| 10 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
---|
| 11 | |
---|
| 12 | |
---|
| 13 | nosave = 1 ;; comment to get an IDL data file |
---|
| 14 | notime = 1 ;; comment to get mention of time taken |
---|
| 15 | if (n_elements(notime) eq 0) then timetime = SYSTIME(1) |
---|
| 16 | |
---|
| 17 | ; |
---|
| 18 | ; MINIMAL ARGUMENTS |
---|
| 19 | ; |
---|
| 20 | if (n_elements(name) eq 0) then stop |
---|
| 21 | if (n_elements(charvar) eq 0) then stop |
---|
| 22 | |
---|
| 23 | ; |
---|
| 24 | ; IDL DATA DIRECTORY |
---|
| 25 | ; |
---|
| 26 | name_idl = charvar+name+'.idl' |
---|
| 27 | if (n_elements(coordmean) ne 0) then begin |
---|
| 28 | nc = n_elements(coordmean) |
---|
| 29 | for i=0,nc-1 do begin |
---|
| 30 | name_idl = string(coordmean(i),'(I0)')+name_idl |
---|
| 31 | endfor |
---|
| 32 | endif |
---|
| 33 | name_idl = './idl_dat/'+name_idl |
---|
| 34 | |
---|
| 35 | ; |
---|
| 36 | ; TEST IF FILE EXISTS |
---|
| 37 | ; |
---|
| 38 | openr,unit,name_idl,/get_lun,error=err |
---|
| 39 | |
---|
| 40 | ; |
---|
| 41 | ; IF IDL FILE EXISTS, THAT IS RATHER SIMPLE |
---|
| 42 | ; |
---|
| 43 | IF (err EQ 0) THEN BEGIN |
---|
| 44 | |
---|
| 45 | print, 'read from file' |
---|
| 46 | free_lun,unit |
---|
| 47 | restore, filename=name_idl |
---|
| 48 | |
---|
| 49 | ; |
---|
| 50 | ; IF IDL FILE DOES NOT EXIST, WELL, READ NETCDF |
---|
| 51 | ; |
---|
| 52 | ENDIF ELSE BEGIN |
---|
| 53 | |
---|
| 54 | ; |
---|
| 55 | ; check save directory |
---|
| 56 | ; |
---|
| 57 | if (n_elements(nosave) eq 0) then SPAWN, 'mkdir -p idl_dat' |
---|
| 58 | |
---|
| 59 | ; |
---|
| 60 | ; open file |
---|
| 61 | ; |
---|
| 62 | id=ncdf_open(name) |
---|
| 63 | |
---|
| 64 | ; |
---|
| 65 | ; get dimensions |
---|
| 66 | ; |
---|
| 67 | var = intarr(3) |
---|
| 68 | NCDF_DIMINQ, id, NCDF_DIMID(id, 'west_east' ), toto, titi & var(0) = titi |
---|
| 69 | NCDF_DIMINQ, id, NCDF_DIMID(id, 'south_north' ), toto, titi & var(1) = titi |
---|
| 70 | NCDF_DIMINQ, id, NCDF_DIMID(id, 'Time' ), toto, titi & var(2) = titi |
---|
| 71 | ;print, var |
---|
| 72 | |
---|
| 73 | ; |
---|
| 74 | ; define default data windows |
---|
| 75 | ; |
---|
| 76 | if (n_elements(offset) le 3) then offset=[0,0,0] |
---|
| 77 | if (n_elements(stride) le 3) then stride=[1,1,1] |
---|
| 78 | if (n_elements(count) le 3) then count=intarr(3) |
---|
| 79 | w = where(count eq 0) & if (w(0) ne -1) then count[w] = var[w] / stride[w] |
---|
| 80 | |
---|
| 81 | ; |
---|
| 82 | ; read variable |
---|
| 83 | ; |
---|
| 84 | varid=ncdf_varid(id,charvar) ;& print, 'get '+charvar |
---|
| 85 | if (n_elements(notime) eq 0) then timetime = SYSTIME(1) ;& help, /memory |
---|
| 86 | ncdf_varget, id, varid, invar, count=count, offset=offset, stride=stride & if (n_elements(notime) eq 0) then print, SYSTIME(1) - timetime, ' s' ;& help, /memory |
---|
| 87 | ncdf_close, id |
---|
| 88 | |
---|
| 89 | ; |
---|
| 90 | ; OPTIONAL: ANOMALY (value is returned in the keyword) |
---|
| 91 | ; |
---|
| 92 | if (n_elements(anomaly) ne 0) then begin |
---|
| 93 | anomaly=TOTAL(TOTAL(invar,1),1)/float(var[0])/float(var[1]) |
---|
| 94 | for i=0,var[0]-1 do for j=0,var[1]-1 do invar(i,j,*) = TEMPORARY(invar(i,j,*)) - anomaly ; coute cher |
---|
| 95 | endif |
---|
| 96 | |
---|
| 97 | ; |
---|
| 98 | ; OPTIONAL: MEAN |
---|
| 99 | ; |
---|
| 100 | if (n_elements(coordmean) ne 0) then begin |
---|
| 101 | for i=0,nc-1 do begin |
---|
| 102 | print, 'mean over dimension '+string(coordmean(i),'(I0)')+' - '+string(var(coordmean(i)),'(I0)')+' elements' |
---|
| 103 | invar=total(TEMPORARY(invar),coordmean(i))/var(coordmean(i)) |
---|
| 104 | endfor |
---|
| 105 | endif |
---|
| 106 | |
---|
| 107 | ; |
---|
| 108 | ; SAVE |
---|
| 109 | ; |
---|
| 110 | if (n_elements(nosave) eq 0) then save, invar, filename=name_idl |
---|
| 111 | |
---|
| 112 | ENDELSE |
---|
| 113 | |
---|
| 114 | if (n_elements(notime) eq 0) then print, SYSTIME(1) - timetime, ' s' |
---|
| 115 | |
---|
| 116 | ; |
---|
| 117 | ; RETURN FIELD |
---|
| 118 | ; |
---|
| 119 | return, invar |
---|
| 120 | |
---|
| 121 | end |
---|