Changeset 1118 in lmdz_wrf

Timestamp:

Sep 29, 2016, 1:40:36 PM (8 years ago)

Author:

lfita

Message:

Adding the `main' variable

File:

• trunk/tools/drawing.py (modified) (2 diffs)

trunk/tools/drawing.py

@@ -4473 +4473 @@
       'fixsignc,[colname]': fixed color [colname], >0 stright, <0 dashed  line
 """
-
-#print string_operation
-
-parser = OptionParser()
-parser.add_option("-f", "--netCDF_file", dest="ncfile", 
-                  help="file to use", metavar="FILE")
-parser.add_option("-o", "--operation", type='choice', dest="operation", 
-       choices=namegraphics, 
-                  help="operation to make: " + ngraphics, metavar="OPER")
-parser.add_option("-S", "--valueS", dest="values", 
-                  help="[WHEN APPLICABLE] values to use according to the operation", metavar="VALUES")
-parser.add_option("-v", "--variable", dest="varname",
-                  help="[WHEN APPLICABLE] variable to check", metavar="VAR")
-
-(opts, args) = parser.parse_args()
-
-#######    #######
-## MAIN
-    #######
+####### ###### ##### #### ### ## #
 
 # Not checking file operation
@@ -4500 +4482 @@
   'draw_vals_trajectories', 'variable_values']
 
-####### ###### ##### #### ### ## #
 errormsg='ERROR -- error -- ERROR -- error'
 
-varn=opts.varname
-oper=opts.operation
-
-if opts.ncfile is not None and not os.path.isfile(opts.ncfile) and                   \
-  not drw.searchInlist(Notcheckingfile, oper):
-    print errormsg
-    print '  ' + main + ': File ' + opts.ncfile + ' does not exist !!'
-    quit(-1)
-
-if oper == 'create_movie':
-    create_movie(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad':
-    draw_2D_shad(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad_time':
-    draw_2D_shad_time(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad_cont':
-    draw_2D_shad_cont(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad_cont_time':
-    draw_2D_shad_cont_time(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad_line':
-    draw_2D_shad_line(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2D_shad_line_time':
-    draw_2D_shad_line_time(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_barbs':
-    draw_barbs(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_basins':
-    draw_basins(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_Neighbourghood_evol':
-    draw_Neighbourghood_evol(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2lines':
-    draw_2lines(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_2lines_time':
-    draw_2lines_time(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_lines':
-    draw_lines(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_lines_time':
-    draw_lines_time(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_points':
-    draw_points(opts.ncfile, opts.values)
-elif oper == 'draw_points_lonlat':
-    draw_points_lonlat(opts.ncfile, opts.values)
-elif oper == 'draw_ptZvals':
-    draw_ptZvals(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_river_desc':
-    draw_river_desc(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_subbasin':
-    draw_subbasin(opts.ncfile, opts.values)
-elif oper == 'draw_timeSeries':
-    draw_timeSeries(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_topo_geogrid':
-    draw_topo_geogrid(opts.ncfile, opts.values)
-elif oper == 'draw_topo_geogrid_boxes':
-    draw_topo_geogrid_boxes(opts.ncfile, opts.values)
-elif oper == 'draw_trajectories':
-    draw_trajectories(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_vals_trajectories':
-    draw_vals_trajectories(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_vectors':
-    draw_vectors(opts.ncfile, opts.values, opts.varname)
-elif oper == 'draw_vertical_levels':
-    draw_vertical_levels(opts.ncfile, opts.values, opts.varname)
-elif oper == 'list_graphics':
-# From: http://www.diveintopython.net/power_of_introspection/all_together.html
-    import drawing as myself
-    object = myself
-    for opern in namegraphics:
-        if  opern != 'list_graphics': 
-            print opern + '_______ ______ _____ ____ ___ __ _'
-            print getattr(object, opern).__doc__
-elif oper == 'variable_values':
-    variable_values(opts.values)
-else:
-    print errormsg
-    print '  ' + main + ": the graphic '" + oper + "' is not ready !!"
-    print errormsg
-    quit()
+# From: http://stackoverflow.com/questions/4041238/why-use-def-main
+def main():
+#######    #######
+## MAIN
+    #######
+
+    parser = OptionParser()
+    parser.add_option("-f", "--netCDF_file", dest="ncfile", 
+     help="file to use", metavar="FILE")
+    parser.add_option("-o", "--operation", type='choice', dest="operation", 
+      choices=namegraphics, help="operation to make: " + ngraphics, metavar="OPER")
+    parser.add_option("-S", "--valueS", dest="values", 
+      help="[WHEN APPLICABLE] values to use according to the operation", 
+      metavar="VALUES")
+    parser.add_option("-v", "--variable", dest="varname",
+      help="[WHEN APPLICABLE] variable to check", metavar="VAR")
+
+    (opts, args) = parser.parse_args()
+
+    varn=opts.varname
+    oper=opts.operation
+
+    if opts.ncfile is not None and not os.path.isfile(opts.ncfile) and                   \
+      not drw.searchInlist(Notcheckingfile, oper):
+        print errormsg
+        print '  ' + main + ': File ' + opts.ncfile + ' does not exist !!'
+        quit(-1)
+
+    if oper == 'create_movie':
+        create_movie(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad':
+        draw_2D_shad(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad_time':
+        draw_2D_shad_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad_cont':
+        draw_2D_shad_cont(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad_cont_time':
+        draw_2D_shad_cont_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad_line':
+        draw_2D_shad_line(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2D_shad_line_time':
+        draw_2D_shad_line_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_barbs':
+        draw_barbs(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_basins':
+        draw_basins(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_Neighbourghood_evol':
+        draw_Neighbourghood_evol(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2lines':
+        draw_2lines(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_2lines_time':
+        draw_2lines_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_lines':
+        draw_lines(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_lines_time':
+        draw_lines_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_points':
+        draw_points(opts.ncfile, opts.values)
+    elif oper == 'draw_points_lonlat':
+        draw_points_lonlat(opts.ncfile, opts.values)
+    elif oper == 'draw_ptZvals':
+        draw_ptZvals(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_river_desc':
+        draw_river_desc(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_subbasin':
+        draw_subbasin(opts.ncfile, opts.values)
+    elif oper == 'draw_timeSeries':
+        draw_timeSeries(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_topo_geogrid':
+        draw_topo_geogrid(opts.ncfile, opts.values)
+    elif oper == 'draw_topo_geogrid_boxes':
+        draw_topo_geogrid_boxes(opts.ncfile, opts.values)
+    elif oper == 'draw_trajectories':
+        draw_trajectories(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_vals_trajectories':
+        draw_vals_trajectories(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_vectors':
+        draw_vectors(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_vertical_levels':
+        draw_vertical_levels(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'list_graphics':
+    # From: http://www.diveintopython.net/power_of_introspection/all_together.html
+        import drawing as myself
+        object = myself
+        for opern in namegraphics:
+            if  opern != 'list_graphics': 
+                print opern + '_______ ______ _____ ____ ___ __ _'
+                print getattr(object, opern).__doc__
+    elif oper == 'variable_values':
+        variable_values(opts.values)
+    else:
+        print errormsg
+        print '  ' + main + ": the graphic '" + oper + "' is not ready !!"
+        print errormsg
+        quit()
+
+if __name__ == '__main__':
+    main()
+