fix: behavior for non-existent xvals.csv

examples/batch_analysis.jl

@@ -107,17 +107,16 @@ function batch_scintillation_index(dirname)
 
     # gather all hdf5 files in the given directory `dirname`
     files = readdir(dirname, join=true)
-    csvidx = findfirst(contains("xvals.csv"), files)
     h5files = filter(x -> splitext(x)[2] == ".hdf5", files)
 
     Nf = length(h5files)
 
-    # gather xvals from corresponding csv if possible - otherwise 1:N
-    df = if isnothing(csvidx)
-        DataFrame(:File => range(1, Nf))
-    else
-        CSV.read(files[csvidx], DataFrame)
+    # gather xvals from corresponding csv if possible - otherwise generate xvals.csv and fill with 1:N
+    xvals_fn = joinpath(dirname, "xvals.csv")
+    !isfile(xvals_fn) && open(xvals_fn, "w") do fid
+        CSV.write(fid, DataFrame(:file => range(1, Nf)))
     end
+    df = CSV.read(xvals_fn, DataFrame)
 
     SI_est = zeros(Nf)
     SI_pw = zeros(Nf)
@@ -133,7 +132,9 @@ function batch_scintillation_index(dirname)
 
         # make sure the order the files are read matches the xvals in the DataFrame
         xname = first(names(df))
-        @assert df[k, xname] == metadata[string(xname)]
+        if xname != "file"
+            @assert df[k, xname] == metadata[string(xname)]
+        end
 
         Cn² = metadata["Cn2"]
         λ = metadata["lambda"]