Changeset 2035 in lmdz_wrf for trunk/tools

Timestamp:

Aug 7, 2018, 3:09:47 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `create_LateX_figs': Function to create a LaTeX from a folder with multiple plots from different values

Location:

trunk/tools

Files:

• generic.py (modified) (4 diffs)
• generic_tools.py (modified) (8 diffs)

trunk/tools/generic.py

@@ -24 +24 @@
 # coincident_CFtimes: Function to make coincident times for two different sets of CFtimes
 # count_cond: Function to count values of a variable which attain a condition
+# create_LateX_figs: Function to create a LaTeX from a folder with multiple plots from different values
 # datetimeStr_conversion: Function to transform a string date to an another date object
 # grid_combinations: Function to provide all the possible grid points combination for a given pair of values
@@ -53 +54 @@
 # List of available operations
 operations=['ASCIIfile_stats', 'coincident_CFtimes', 'count_cond',                   \
-  'datetimeStr_conversion',                                                          \
+  'create_LateX_figs', 'datetimeStr_conversion',                                     \
   'grid_combinations', 'inf_operSlist',                                              \
   'interpolate_locs', 'latex_fig_array', 'list_operations', 'PolyArea',              \
@@ -89 +90 @@
 ## e.g. # generic.py -o ASCIIfile_stats -S times_CDXWRF1.dat,#,R
 ## e.g. # generic.py -o inf_operSlist -S '-98.21:1.2:3.45:100.1:34321.1,threshold,:,3.'
+## e.g. # generic.py -o create_LateX_figs -S '/home/lluis/estudios/FPS_ALPS/additional/IOP/analysis/figs,WindRose_obs_@SkewT-logP_obs_ta-tda@SkewT-logP_obs_evol@WindRose_obs-sim_step@SkewT-logP_obs-sim_step@SkewT-logP_obs-sim_evol,10868@16080@16144@16546,png'
 
 operationnames = "'" + gen.numVector_String(operations, "', '") + "'"
@@ -146 +148 @@
         print gen.count_cond(np.array(vals[0].split(cV), dtype=np.float),           \
           np.float(vals[1]), vals[2])
+
+
+elif oper == 'create_LateX_figs':
+    Nvals = 4
+    vals = opts.values.split(cS)
+    if vals[0] == 'h':
+        print gen.create_LateX_figs.__doc__
+        quit(-1)
+    else:
+        if len(vals) != Nvals:
+            print errormsg
+            print '  ' + main + ": operation '" + oper + "' requires", Nvals, 'and', \
+              len(vals), ' has passed!!'
+            print gen.create_LateX_figs.__doc__
+            quit(-1)
+        vals0 = vals[0]
+        vals1 = vals[1].split('@')
+        vals2 = vals[2].split('@')
+        vals3 = vals[3]
+        
+        print gen.create_LateX_figs(vals0, vals1, vals2, vals3)
 
 elif oper == 'datetimeStr_conversion':

trunk/tools/generic_tools.py

@@ -13486 +13486 @@
     otex.write('\\documentclass{article}\n')
     otex.write('\\usepackage{graphicx}\n')
+    otex.write('\\usepackage[colorlinks=true,urlcolor=blue]{hyperref}\n')
     otex.write('\n')
     otex.write('\\textheight=23cm\n')
@@ -13495 +13496 @@
     otex.write('\n')
     otex.write('\\begin{document}\n')
+    otex.write('\n')
+    otex.write('\\listoffigures\n')
+    otex.write('\\clearpage\n')
 
     for plotn in plots:
@@ -13513 +13517 @@
             if Nvalfig > 0:
                 Ncol = int(np.sqrt(Nvalfig))
+                Nrow = Nvalfig/Ncol
+                if Nrow > Ncol: Ncol = Nrow
                 cs = Ntchar('c', Ncol)
                 frcwth = 0.9/Ncol
@@ -13530 +13536 @@
                     ifig = ifig + 1
 
-                print 'Lluis ifig:', ifig-1, 'Ncol:', Ncol, 'mod:', np.mod(ifig-1,Ncol)
                 if Nvalfig - ifig + 1 != 0:
                     endrow = Ntchar(' & ', Nvalfig - ifig + 1)
@@ -13536 +13541 @@
                 otex.write('\\end{tabular}\n')
                 otex.write('\\end{center}\n')
+                otex.write('\\caption{' + latex_text(plotn) +' '+ latex_text(valn) + \
+                   '}\n')
+                otex.write('\\label{fig:' + plotn + '_' + valn + '}\n')
                 otex.write('\\end{figure}\n')
 
@@ -13541 +13549 @@
             Nunsortfig = len(unsortfigs)
             Ncol = int(np.sqrt(Nunsortfig))
+            Nrow = Nvalfig/Ncol
+            if Nrow > Ncol: Ncol = Nrow
             cs = Ntchar('c', Ncol)
             frcwth = 0.9/Ncol
@@ -13563 +13573 @@
             otex.write('\\end{tabular}\n')
             otex.write('\\end{center}\n')
+            otex.write('\\caption{' + latex_text(plotn) + '}\n')
+            otex.write('\\label{fig:' + plotn + '_unsorted}\n')
             otex.write('\\end{figure}\n')
             otex.write('\n')
@@ -13575 +13587 @@
     sub.call('pdflatex ' + texfn.split('.')[0], shell=True)
     sub.call('pdflatex ' + texfn.split('.')[0], shell=True)
+    sub.call('evince ' + texfn.split('.')[0]+'.pdf &', shell=True)
 
     return