src/misc/series_ser.py

   1 #!/usr/bin/env python
   2
   3 #   gpiv_series - Processes a set of numbered input data
   4
   5 #   Copyright (C) 2008 Gerber van der Graaf
   6
   7 #   This program is free software; you can redistribute it and/or modify
   8 #   it under the terms of the GNU General Public License as published by
   9 #   the Free Software Foundation; either version 2, or (at your option)
  10 #   any later version.
  11
  12 #   This program is distributed in the hope that it will be useful,
  13 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #   GNU General Public License for more details.
  16
  17 #   You should have received a copy of the GNU General Public License
  18 #   along with this program; if not, write to the Free Software Foundation,
  19 #   Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  20
  21
  22 #--------------------------------------------------------------------
  23
  24 #
  25 # This version is for serial processing
  26 #
  27 import os, re
  28
  29 #
  30 #----------- Command line arguments parser
  31 #
  32 from optparse import OptionParser
  33
  34 usage = "%prog [options] \"process\""
  35 parser = OptionParser(usage)
  36 parser.add_option("-a", "--arg_n",
  37                   action="store_true", dest="arg_n", default=False,
  38                   help="if the process needs the current number in its \
  39                   argument list instead of prepending/appending it to the \
  40                   filebase name, the number will be put before (-f) \
  41                   \"filename\" in the \"process\" string.")
  42 parser.add_option("-b", "--basename", type='string', dest="basename",
  43                   help="File basename for reading", metavar="FILE")
  44 parser.add_option("-e", "--ext", type='string', dest="ext", metavar="EXT",
  45                   help="add an extension after the file basename + number (without leading \".\")")
  46 parser.add_option("-f", "--first", type='int', dest="first_nr", default=0,
  47                   help="first numbered file (default: 0)", metavar="N")
  48 parser.add_option("-l", "--last", type='int', dest="last_nr", default=0,
  49                   help="last numbered file(default: 0)", metavar="N")
  50 parser.add_option("-i", "--incr", type='int', dest="incr_nr", default=1,
  51                   help="increment file number (default: 1)", metavar="N")
  52 parser.add_option("-p", "--print",
  53                   action="store_true", dest="pri", default=False,
  54                   help="prints process parameters/variables to stdout")
  55 parser.add_option("--pad", type='int', dest="pad0", default=0,
  56                   help="padding number with zero's (default: 0)", metavar="N")
  57 parser.add_option("-n", "--none",
  58                   action="store_true", dest="none", default=False,
  59                   help="suppresses real execution")
  60 parser.add_option("-x", "--prefix", action="store_true", dest="prefix", default=False,
  61                   help="prefix numbering to file basename")
  62
  63 (options, args) = parser.parse_args()
  64 if len(args) != 1:
  65         parser.error("incorrect number of arguments")
  66 else:
  67         process = args[0]
  68
  69
  70 #
  71 #----------- Function definitions
  72 #
  73 def pri_date(msg = "Time stamp at start of series processing:"):
  74         """Prints time stamp.
  75
  76         Keyword arguments:
  77         msg -- message to be printed before time stamp
  78         """
  79         if options.pri == True:
  80                 print msg
  81                 os.system('date')
  82         elif options.none:
  83                 print msg
  84                 os.system('date')
  85
  86
  87
  88 def count_digits(nr):
  89         """Counts number of digits from a number
  90
  91         Keyword arguments:
  92         nr -- number to be questioned
  93         """
  94         count=0
  95         while nr/10 !=0:
  96                 nr=nr/10
  97                 count=count+1
  98         return count
  99
 100
 101 def pad0(nr):
 102         """Created a string for zero padding
 103
 104         Keyword arguments:
 105         nr -- number of zeros to be padded
 106         """
 107         pd0=""
 108         for i in range(0, nr):
 109                 pd0 = str(pd0)+"0"
 110
 111         return pd0
 112
 113
 114 def compose_name_nr(nr):
 115         """Creates proper name from basename and number.
 116
 117         Keyword arguments:
 118         nr -- number of filename to be processed
 119         """
 120         if options.pad0 > 0:
 121                 ndig=count_digits(nr)
 122                 null_str=pad0(options.pad0 - ndig)
 123                 nr_str=null_str+str(nr)
 124         else:
 125                 nr_str=str(nr)
 126
 127         if options.prefix:
 128                 if options.arg_n:
 129                         name=str(options.basename)
 130                 else:
 131                         name=nr_str+str(options.basename)
 132         else:
 133                 if options.arg_n:
 134                         name=str(options.basename)
 135                 else:
 136                         name=str(options.basename)+nr_str
 137
 138         if str(options.ext) != "None":
 139                 name=str(name)+str(".")+str(options.ext)
 140
 141         return(name)
 142
 143
 144 def compose_cmd(name, nr):
 145         """Creates proper command.
 146
 147         Keyword arguments:
 148         name -- complete filename
 149         """
 150         command=str(process)
 151         if options.arg_n:
 152                 # Eventually, substitutes "-f" with: "nr -f"
 153                 command_tmp=re.sub("-f", str(nr)+" -f", command)
 154                 if command_tmp != command:
 155                         command = str(command_tmp)+" "+str(name)
 156                 else:
 157                         command = str(command_tmp)+" "+str(nr)+" "+str(name)
 158         else:
 159                 command=str(command)+" "+str(name)
 160
 161         return command
 162
 163
 164 def proc_series_ser():
 165         """Processes a series on identic numbered files.
 166         """
 167         for i in range(options.first_nr, options.last_nr+1, options.incr_nr):
 168                 name_nr = compose_name_nr(i)
 169                 command = compose_cmd(name_nr, i)
 170                 if options.pri == True: print command
 171                 elif options.none == True: print command
 172                 if options.none == False: os.system(command)
 173
 174
 175 #
 176 #----------- Calling functions
 177 #
 178 if options.pri == True: pri_date()
 179 elif options.none == True: pri_date()
 180 proc_series_ser()
 181 if options.pri == True: pri_date(msg = "Time stamp at end of series processing:")
 182 elif options.none == True: pri_date(msg = "Time stamp at end of series processing:")
 183 #
 184 #----------- That's all folks
 185 #