Source code for pygimli.physics.em.io

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Import functions for several Em data formats (to be extended)."""
import numpy as np


[docs] def readusffile(filename, data=None): """Read data from single USF (universal sounding file) file. data = readusffile( filename ) data = readusffile( filename, data ) will append to data """ if not data: data = [] columns = [] nr = 0 sounding = {} sounding['FILENAME'] = filename isdata = False fid = open(filename) for line in fid: zeile = line.rstrip('\n').replace(',', '') # commas useless here if zeile: # anything at all if zeile[0] == '/': # comment-like if zeile[1:4] == 'END': # end of a sounding if isdata: # have already read some data sounding['data'] = columns for i, cn in enumerate(sounding['column_names']): sounding[cn] = columns[:, i] data.append(sounding) sounding = {} isdata = not isdata # turn off data mode elif zeile.find(':') > 0: # key-value pair key, value = zeile[1:].split(':') try: val = float(value) sounding[key] = val except: sounding[key] = value else: if isdata: values = zeile.split() try: for i, v in enumerate(values): columns[nr, i] = float(v) nr += 1 except: sounding['column_names'] = values columns = np.zeros((int(sounding['POINTS']), len(values))) nr = 0 fid.close() return data
def readusffiles(filenames): """Read all soundings data from a list of usf files. Example ------- DATA = readusffiles(filenames) """ DATA = [] for onefile in filenames: DATA = readusffile(onefile, DATA) return DATA
[docs] def importMaxminData(filename, verbose=False): """Import function reading in positions, data, frequencies, geometry.""" delim = None fid = open(filename) coilspacing = 0. freq = [] aline = "" i = 0 for i, aline in enumerate(fid): if aline.split()[0][0].isdigit(): # number found break elif aline.upper().find('COIL') > 0: # [:6] == '/ COIL': coilspacing = float(aline.split()[-2]) elif aline.upper().find('FREQ') > 0: # [:6] == '/ FREQ': freq = np.array([float(aa) for aa in aline[aline.find( ':') + 1:].replace(',', ' ').split() if aa[0].isdigit()]) fid.close() if verbose: print("CS=", coilspacing, "F=", freq) if aline.find(',') > 0: delim = ',' nf = len(freq) if verbose: print("delim=", delim, "nf=", nf) A = np.loadtxt(filename, skiprows=i, delimiter=delim).T x, IP, OP = A[0], A[2:nf * 2 + 2:2].T, A[3:nf * 2 + 2:2].T return x, freq, coilspacing, IP, OP
if __name__ == '__main__': print("do some tests here")