contrib/mfp_igor_scripts/FMjoin.py

   1 # Copyright (C) 2010-2012 Alberto Gomez-Casado <a.gomezcasado@tnw.utwente.nl>
   2 #                         W. Trevor King <wking@tremily.us>
   3 #
   4 # This file is part of Hooke.
   5 #
   6 # Hooke is free software: you can redistribute it and/or modify it under the
   7 # terms of the GNU Lesser General Public License as published by the Free
   8 # Software Foundation, either version 3 of the License, or (at your option) any
   9 # later version.
  10 #
  11 # Hooke is distributed in the hope that it will be useful, but WITHOUT ANY
  12 # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  13 # A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
  14 # details.
  15 #
  16 # You should have received a copy of the GNU Lesser General Public License
  17 # along with Hooke.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 '''
  20 FMjoin.py
  21 Copies all .ibw files contained in a folder and its subfolders into a single folder. Useful for force maps.
  22
  23 Usage:
  24 python FMjoin.py origindir destdir
  25 '''
  26
  27 import os
  28 import shutil
  29 import sys
  30
  31 def main(*args):
  32         if len(sys.argv) < 2:
  33                 print 'You must at least specify origin and destination folders.'
  34                 return 0
  35         origin=sys.argv[1]
  36         dest=sys.argv[2]
  37
  38         if os.path.exists(origin):
  39                 if os.path.exists(dest):
  40                         if os.listdir(dest)!=[]:
  41                                 print 'Destination folder is not empty! Use another folder.'
  42                                 return 0
  43                 else:
  44                         print 'Destination folder does not exist, will create it'
  45                         os.mkdir(dest)
  46         else:
  47                 print 'You provided a wrong origin folder name, try again.'
  48
  49         origin=os.path.abspath(origin)
  50         dest=os.path.abspath(dest)
  51
  52         for root, dirs, files in os.walk(origin):
  53                 for filename in files:
  54                         if filename.split('.')[1]!="ibw":
  55                                 continue
  56                         filepath=os.path.join(root,filename)
  57                         #to avoid overwriting, we collapse unique paths into filenames
  58                         rawdest=filepath.split(os.path.commonprefix([origin, filepath]))[1]
  59                         rawdest=rawdest.replace('/','') #for linux
  60                         rawdest=rawdest.replace('\\','') #for windows
  61                         destfile=os.path.join(dest,rawdest)
  62                         print 'Copying '+rawdest
  63                         shutil.copy(filepath,destfile)
  64
  65         return 0
  66
  67 if __name__ == '__main__':
  68     sys.exit(main(*sys.argv))
  69
  70