Fix bugs

src/extractor/AmrexCastro.jl

@@ -227,9 +227,9 @@ function _run_linear_models_plots_size_step(extractor::AmrexCastro, X::DataFrame
       
       # Yd timesteps cumulative
       if plotIndex == 1
-         push!(Yd_Timesteps, helper_get_directory_size(directory) )
+         push!(Yd_Timesteps, helper_get_directory_size(plotFileDir) )
       else
-         push!(Yd_Timesteps, last(Yd_Timesteps) + helper_get_directory_size(directory) )
+         push!(Yd_Timesteps, last(Yd_Timesteps) + helper_get_directory_size(plotFileDir) )
       end
       
       levelDirs = helper_get_prefix_directories(string(plotFileDir,"/Level_"))
@@ -267,7 +267,7 @@ function _run_linear_models_plots_size_step(extractor::AmrexCastro, X::DataFrame
             
             if isfile( checkFileName )
               rankFileSize = filesize(checkFileName)
-              push!(Yd_Ranks[rankKey], rankFileSize)
+              # bug push!(Yd_Ranks[rankKey], rankFileSize)
               
               if plotIndex == 1
                 push!(Yd_Ranks[rankKey], rankFileSize)  

src/helper/helperPlot.jl

@@ -7,17 +7,13 @@ function plot_CSV_3col( filename::String )
   
   df = CSV.File(filename) |> DataFrames.DataFrame
   
-  x = df[!,1]
-  y = df[!,2]
+  x = df[!,"plots_cells"]
+  y = df[!,"plots_size"]
     
-  yL0 = df[!,3]
-  yL1 = df[!,4]
-  yL2 = df[!,5]
-    
-  z = df[!,6]
+  z = df[!,"caseID"]
   
   fig1 = Plots.scatter( x, y, groups = z, xaxis=:log, yaxis=:log, 
-                      legendfontsize=9,
+                      legendfontsize=7,
                       legend = :outertopleft, 
                       # [shapes], size, alpha
                       marker = ([:x :+ :star4 :vline :square :circle], 7, 1.0),
@@ -31,8 +27,11 @@ function plot_CSV_3col( filename::String )
   Plots.gui(fig1)
   #Plots.png(fig1)
   readline()
-  
-  fig2 = Plots.scatter( x, yL0, groups = z, label="case4-L0",
+
+  yL0 = df[!,3]
+  yL1 = df[!,4]
+  yL2 = df[!,5]  
+  fig2 = Plots.scatter( x, yL0, groups = z, label="L0",
                           marker = ([:x], 10, 1.0),
                           markercolor = [:blue :green :red :orange :pink],
                           legendfontsize=9,
@@ -43,12 +42,12 @@ function plot_CSV_3col( filename::String )
                           size=(1200,850),
                           reuse = true
                           )
-  Plots.scatter!( x, yL1, groups = z, label= "case4-L1", 
+  Plots.scatter!( x, yL1, groups = z, label= "L1", 
                           marker = ([:+], 10, 1.0),
                           markercolor = [:blue :green :red :orange :pink],
                            )
                             
-  Plots.scatter!( x, yL2, groups = z, label="case4-L2", 
+  Plots.scatter!( x, yL2, groups = z, label="L2", 
                           marker = ([:star4], 10, 1.0),
                           markercolor = [:blue :green :red :orange :pink],
                           )
@@ -63,8 +62,8 @@ end
 function main()
 
   # modify this to the file path
-  #filename = "/home/wgodoy/hdd_home/work/proxy_io/AmrexCastro/plot_size.csv"
-  filename = "/home/wgodoy/hdd_home/work/proxy_io/AmrexCastro/plot_size_case4.csv"
+  filename = "/home/wgodoy/hdd_home/work/proxy_io/AmrexCastro/plot_size.csv"
+  # filename = "/home/wgodoy/hdd_home/work/proxy_io/AmrexCastro/plot_size_case4.csv"
   plot_CSV_3col(filename)
 end