;
; $Id$
;
;   Get time series for various parameters from data snapshots
;
;  Author: Anders Johansen (ajohan@astro.ku.dk)
;  $Date: 2007-08-03 09:53:26 $
;  $Revision: 1.5 $
;
;  10-oct-03/anders: coded (coding layout adapted from Tony)
;  13-oct-03/anders: renamed from pc_get_ts.pro
;  
pro pc_read_var_ts,t_arr=t_arr,array=array, $
                   snap_start=snap_start,snap_end=snap_end, $
                   datadir=datadir,proc=proc,type=type,allprocs=allprocs, $
                   HELP=HELP,QUIET=QUIET,SWAP_ENDIAN=SWAP_ENDIAN
COMPILE_OPT IDL2,HIDDEN
  common pc_precision, zero, one, precision, data_type, data_bytes, type_idl
; If no meaningful parameters are given show some help!
  IF ( keyword_set(HELP) ) THEN BEGIN
    print, "Usage: "
    print, ""
    print, "pc_read_var_ts, t_arr=t_arr, array=array, snap_start=snap_start, "
    print, "                snap_end=snap_end, datadir=datadir, proc=proc, "
    print, "                type=type, /HELP, /QUIET"
    print, ""    
    print, "Extracts time series of parameter specified by 'type'."
    print, "The times are stored in 't_arr' and the parameter is stored in 'array'."
    print, "If snap_start and snap_end are specified, the time series is"
    print, "extracted from this range of snapshots. Otherwise all VAR* will"
    print, "be used. Beware that the VAR files must come in numerical order."
    print, ""
    print, "A non-default data directory and a processor number can be"
    print, "specified with 'datadir' (string) and 'proc' (number)."
    print, ""
    print, "/QUIET: Instruction not to print any 'helpful'."
    print, "/HELP: Display this usage information, and exit."
    print, ""
    print, "Types currently implemented:"
    print, "type='uxdmean': Mean dust x velocity (AJ)"
    print, "type='uydmean': Mean dust y velocity (AJ)"
    print, "type='udmean': Mean dust velocity (AJ)"
    print, ""
    print, "Feel free to add more."
    return
  ENDIF
; Default data directory

datadir = pc_get_datadir(datadir)
default,proc,0
default,allprocs,0
default,snap_start,0
default,snap_end,0

default, procdir, '/proc'+strtrim(proc,2)+'/
if (keyword_set (allprocs)) then if (allprocs eq 1) then procdir = '/allprocs/'

if (snap_start eq 0 and snap_end eq 0) then begin
  if (not keyword_set(QUIET)) then begin
    print, 'No start and end index specified.'
    print, 'Taking all snapshots from "'+datadir+procdir'".'
  endif
  snap_start=1
  dummy=file_search(datadir+procdir+'VAR*',count=snap_end)
  if (not keyword_set(QUIET)) then begin
    print, 'Found '+strtrim(snap_end,2)+' snapshots.'
  endif
endif

array=fltarr(snap_end-snap_start+1)
t_arr=fltarr(snap_end-snap_start+1)

for nsnap=snap_start,snap_end do begin
  varfile='VAR'+strcompress(nsnap,/remove_all)
  if (type eq 'udxmean') then begin
    pc_read_var,t=t,uud=uud,varfile=varfile,datadir=datadir,proc=proc, $
                allprocs=allprocs,QUIET=QUIET,SWAP_ENDIAN=SWAP_ENDIAN
    array(nsnap-snap_start)=mean(uud(*,*,*,0))
  end else if (type eq 'udymean') then begin
    pc_read_var,t=t,uud=uud,varfile=varfile,datadir=datadir,proc=proc, $
                allprocs=allprocs,QUIET=QUIET,SWAP_ENDIAN=SWAP_ENDIAN
    array(nsnap-snap_start)=mean(uud(*,*,*,1))
  end else if (type eq 'udmean') then begin
    pc_read_var,t=t,uud=uud,varfile=varfile,datadir=datadir,proc=proc, $
                allprocs=allprocs,QUIET=QUIET,SWAP_ENDIAN=SWAP_ENDIAN
    array(nsnap-snap_start)= $
        mean(sqrt(uud(*,*,*,0)^2+uud(*,*,*,1)^2+uud(*,*,*,2)^2))
  end
  t_arr(nsnap-snap_start)=t
endfor


end