hooke/driver/mfp1dexport.py

   1 # Copyright (C) 2009-2012 Massimo Sandal <devicerandom@gmail.com>
   2 #                         W. Trevor King <wking@drexel.edu>
   3 #
   4 # This file is part of Hooke.
   5 #
   6 # Hooke is free software: you can redistribute it and/or modify it
   7 # under the terms of the GNU Lesser General Public License as
   8 # published by the Free Software Foundation, either version 3 of the
   9 # License, or (at your option) any later version.
  10 #
  11 # Hooke is distributed in the hope that it will be useful, but WITHOUT
  12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
  14 # Public License for more details.
  15 #
  16 # You should have received a copy of the GNU Lesser General Public
  17 # License along with Hooke.  If not, see
  18 # <http://www.gnu.org/licenses/>.
  19
  20 """Driver for text-exported MFP 1D files
  21 """
  22
  23 import os
  24 import os.path
  25
  26 from .. import libhooke as lh
  27 from .. import curve as lhc
  28
  29
  30 __version__='0.0.0.20090923'
  31
  32
  33 class mfp1dexportDriver(lhc.Driver):
  34
  35     def __init__(self, filename):
  36         '''
  37         This is a driver to import Asylum Research MFP 1D data.
  38         Status: experimental
  39         '''
  40         self.filename = filename
  41         self.filedata = open(filename,'rU')
  42         self.lines = list(self.filedata.readlines())
  43         self.filedata.close()
  44
  45     def close_all(self):
  46         self.filedata.close()
  47
  48     def is_me(self):
  49         if os.path.isdir(path):
  50             return False
  51         try:
  52             self.raw_header = self.lines[0:38]
  53         except:
  54             #Not enough lines for a header; not a good file
  55             return False
  56
  57         #FIXME: We want a more reasonable header recognition
  58         if self.raw_header[0].startswith('Wave'):
  59             return True
  60         else:
  61             return False
  62
  63     def _read_columns(self):
  64
  65         self.raw_columns=self.lines[39:]
  66
  67         kline=None
  68         for line in self.lines:
  69             if line[:7]=='SpringC':
  70                 kline=line
  71                 break
  72
  73         kline=kline.split(':')
  74
  75         #self.k=float(self.raw_header[23][8:])
  76         self.k=float(kline[1])
  77
  78         xext=[]
  79         xret=[]
  80         yext=[]
  81         yret=[]
  82         for line in self.raw_columns:
  83             spline=line.split()
  84             xext.append(float(spline[0]))
  85             yext.append(float(spline[1]))
  86             xret.append(float(spline[2]))
  87             yret.append(float(spline[3]))
  88
  89         return [[xext,yext],[xret,yret]]
  90
  91     def deflection(self):
  92         self.data = self._read_columns()
  93         return self.data[0][1], self.data[1][1]
  94
  95     def default_plots(self):
  96         main_plot = lhc.PlotObject()
  97         defl_ext,defl_ret = self.deflection()
  98         yextforce = [i*self.k for i in defl_ext]
  99         yretforce = [i*self.k for i in defl_ret]
 100         main_plot.add_set(self.data[0][0], yextforce)
 101         main_plot.add_set(self.data[1][0], yretforce)
 102         main_plot.normalize_vectors()
 103         #main_plot.units = ['Z','force']  #FIXME: if there's an header saying something about the time count, should be used
 104         main_plot.units = ['m','N']
 105         main_plot.destination = 0
 106         main_plot.filename = self.filename
 107         main_plot.title = os.path.basename(self.filename)
 108         main_plot.colors = ['red','blue']
 109         main_plot.style = ['plot','plot']
 110         return [main_plot]