src/figures/cantilever-data/get_loading_rates.py

   1 #!/usr/bin/python
   2
   3 DATA="./data/raw"
   4 UNFOLD_FORCE_COL=0
   5 UNFOLD_RATE_COL=1
   6 SPRING_CONST_COL=2
   7 LOAD_RATE_COL=3
   8
   9 from avg_data import KCUTS
  10
  11 Ks = [24.33, None, 131.98]   # HACK!
  12 OFILEs = [None, None, None]
  13
  14 for i,K in enumerate(Ks):
  15     if K == None:
  16         continue
  17     OFILEs[i] = file('loading-rate.d/loading_rate_%.2f.dat' % K, 'w')
  18
  19 for line in file(DATA, 'r'):
  20     if line[0] == '#':
  21         continue
  22     fields = line.strip().split()
  23     F = float(fields[UNFOLD_FORCE_COL])
  24     V = float(fields[UNFOLD_RATE_COL])
  25     K = float(fields[SPRING_CONST_COL])
  26     L = float(fields[LOAD_RATE_COL])
  27
  28     ispring = 0
  29     while ispring < len(KCUTS) and K > KCUTS[ispring]:
  30         ispring += 1
  31     if ispring == 1:
  32         continue # drop middle spring constants
  33
  34     print >> OFILEs[ispring], "%g\t%g" % (L, F)
  35
  36 for i,K in enumerate(Ks):
  37     if K == None:
  38         continue
  39     OFILEs[i].close()
  40     OFILEs[i] = None