research/transmission_spectroscopy/26-1-12/data_files1/process.py

   1 import os\r
   2 import sys\r
   3 import re\r
   4 import subprocess\r
   5 \r
   6 \r
   7 \r
   8 \r
   9 \r
  10 dataFiles = os.listdir()\r
  11 for dataFile in dataFiles:\r
  12     if re.match(".*\.dat", dataFile) != None and re.match(".*\.reference\.dat", dataFile) == None and re.match(".*\.dark\.dat", dataFile) == None:\r
  13         if os.path.exists("tmp\\r-d.dat"):\r
  14             os.remove("tmp\\r-d.dat")\r
  15         if os.path.exists("tmp\\s-d.dat"):\r
  16             os.remove("tmp\\s-d.dat")\r
  17                                                                                                                \r
  18         sys.stdout.write("Processing file \""+dataFile+"\"... ");\r
  19         name = re.split("\.dat", dataFile)[0]\r
  20 \r
  21         os.system("dataX.exe " + name + ".reference.dat" + " - " + name + ".dark.dat" + " >> tmp\\r-d.dat")\r
  22         os.system("dataX.exe " + name + ".dat" + " - " + name + ".dark.dat" + " >> tmp\\s-d.dat")\r
  23         os.system("dataX.exe tmp\\s-d.dat / tmp\\r-d.dat >> processed\\"+name+"_(on_glass).dat")\r
  24         sys.stdout.write("Done!\n")\r
  25 \r
  26 dataFiles = os.listdir("processed")\r
  27 for dataFile in dataFiles:\r
  28     if re.match("control.*", dataFile) == None:\r
  29         name = re.split("_\(on_glass\)\.dat", dataFile)[0]\r
  30         sys.stdout.write("Calculate transmission for " + name + "... ")\r
  31         os.system("dataX.exe processed\\" + dataFile + " / " + "processed\\control_(on_glass).dat >> processed\\" + name + ".dat")\r
  32         sys.stdout.write(" Done!\n")\r
  33                                                                                                                \r
  34                         \r
  35         \r