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 if not "FNAME" in process:
  70         parser.error("process does not contain the required string 'FNAME'")
  71
  72 #
  73 #----------- Function definitions
  74 #
  75 def pri_date(msg = "Time stamp at start of series processing:"):
  76         """Prints time stamp.
  77
  78         Keyword arguments:
  79         msg -- message to be printed before time stamp
  80         """
  81         if options.pri == True:
  82                 print msg
  83                 os.system('date')
  84         elif options.none:
  85                 print msg
  86                 os.system('date')
  87
  88
  89
  90 def count_digits(nr):
  91         """Counts number of digits from a number
  92
  93         Keyword arguments:
  94         nr -- number to be questioned
  95         """
  96         count=0
  97         while nr/10 !=0:
  98                 nr=nr/10
  99                 count=count+1
 100         return count
 101
 102
 103 def pad0(nr):
 104         """Created a string for zero padding
 105
 106         Keyword arguments:
 107         nr -- number of zeros to be padded
 108         """
 109         pd0=""
 110         for i in range(0, nr):
 111                 pd0 = str(pd0)+"0"
 112
 113         return pd0
 114
 115
 116 def compose_name_nr(nr):
 117         """Creates proper name from basename and number.
 118
 119         Keyword arguments:
 120         nr -- number of filename to be processed
 121         """
 122         if options.pad0 > 0:
 123                 ndig=count_digits(nr)
 124                 null_str=pad0(options.pad0 - ndig)
 125                 nr_str=null_str+str(nr)
 126         else:
 127                 nr_str=str(nr)
 128
 129         if options.prefix:
 130                 if options.arg_n:
 131                         name=str(options.basename)
 132                 else:
 133                         name=nr_str+str(options.basename)
 134         else:
 135                 if options.arg_n:
 136                         name=str(options.basename)
 137                 else:
 138                         name=str(options.basename)+nr_str
 139
 140         if str(options.ext) != "None":
 141                 name=str(name)+str(".")+str(options.ext)
 142
 143         return(name)
 144
 145
 146 def compose_cmd(name, nr):
 147         """Creates proper command.
 148
 149         Keyword arguments:
 150         name -- complete filename
 151         """
 152         command=str(process)
 153         # Eventually, substitutes "-f" with: "nr -f"
 154         if options.arg_n:
 155                 if "-f" in command:
 156                         command=re.sub("-f", str(nr)+" -f", command)
 157                 else:
 158                         command=re.sub("FNAME", str(nr)+" FNAME", command)
 159         # Substitutes "FNAME" with the file base name
 160         command=re.sub("FNAME", str(name), str(command))
 161
 162         return command
 163
 164
 165 def proc_series_ser():
 166         """Processes a series on identic numbered files.
 167         """
 168         for i in range(options.first_nr, options.last_nr+1, options.incr_nr):
 169                 name_nr = compose_name_nr(i)
 170                 command = compose_cmd(name_nr, i)
 171
 172                 if options.pri == True: print command
 173                 elif options.none == True: print command
 174                 if options.none == False: os.system(command)
 175
 176
 177 #
 178 #----------- Calling functions
 179 #
 180 if options.pri == True: pri_date()
 181 elif options.none == True: pri_date()
 182 proc_series_ser()
 183 if options.pri == True: pri_date(msg = "Time stamp at end of series processing:")
 184 elif options.none == True: pri_date(msg = "Time stamp at end of series processing:")
 185 #
 186 #----------- That's all folks
 187 #