Replaced camelCase with undercaps.

This commit is contained in:
Austin Bingham
2014-08-05 19:55:06 +02:00
parent 7014fba323
commit 4cb5cf539a
5 changed files with 96 additions and 96 deletions
+2 -2
View File
@@ -222,14 +222,14 @@ class SimpleTest:
print dialog.get_filename(), 'selected'
filename = dialog.get_filename()
dialog.destroy()
self.segy = segypy.readSegy(filename)
self.segy = segypy.read_segy(filename)
update_segyheader(self)
update_segytraceheader(self)
update_segyplot(self)
def on_new1_activate(self, button):
self.segy = segypy.readSegy('../data_4byteINT.segy')
self.segy = segypy.read_segy('../data_4byteINT.segy')
update_segyheader(self)
update_segytraceheader(self,1)
update_segyplot(self)
+1 -1
View File
@@ -1,4 +1,4 @@
def imageSegy(Data):
def image_segy(Data):
"""
imageSegy(Data)
Image segy Data
+20 -20
View File
@@ -85,7 +85,7 @@ def size_in_bytes(ctype):
return struct.calcsize(ctype) if ctype != 'ibm' else struct.calcsize('f')
def getDefaultSegyHeader(ntraces=100, ns=100):
def get_default_segy_header(ntraces=100, ns=100):
"""
SH = getDefaultSegyHeader()
"""
@@ -107,7 +107,7 @@ def getDefaultSegyHeader(ntraces=100, ns=100):
return SH
def getDefaultSegyTraceHeaders(ntraces=100, ns=100, dt=1000):
def get_default_segy_trace_headers(ntraces=100, ns=100, dt=1000):
"""
SH = getDefaultSegyTraceHeader()
"""
@@ -134,7 +134,7 @@ def read_trace_header(f, reel_header, trace_header_name='cdp', endian='>'):
read_trace_header(reel_header, TraceHeaderName)
"""
bps = getBytePerSample(reel_header)
bps = get_byte_per_sample(reel_header)
# MAKE SOME LOOKUP TABLE THAT HOLDS THE LOCATION OF HEADERS
trace_header_pos = STH_def[trace_header_name]["pos"]
@@ -175,7 +175,7 @@ def file_length(f):
return file_size
def readSegy(f, filename, endian='>'):
def read_segy(f, filename, endian='>'):
"""
Data, SegyHeader, trace_headers = read_reel_header(f)
"""
@@ -193,7 +193,7 @@ def readSegy(f, filename, endian='>'):
# GET TRACE
index = REEL_HEADER_NUM_BYTES
bytes_per_sample = getBytePerSample(reel_header)
bytes_per_sample = get_byte_per_sample(reel_header)
num_data = (file_size - REEL_HEADER_NUM_BYTES) / bytes_per_sample
Data, reel_header, trace_headers = read_traces(f,
@@ -280,7 +280,7 @@ def read_reel_header(f, filename, endian='>'):
"=" + str(reel_header[key]))
# SET NUMBER OF BYTES PER DATA SAMPLE
bps = getBytePerSample(reel_header)
bps = get_byte_per_sample(reel_header)
file_size = file_length(f)
ntraces = (file_size - REEL_HEADER_NUM_BYTES) / \
@@ -292,7 +292,7 @@ def read_reel_header(f, filename, endian='>'):
return reel_header
def writeSegy(filename, Data, dt=1000, STHin={}, SHin={}):
def write_segy(filename, Data, dt=1000, STHin={}, SHin={}):
"""
writeSegy(filename, Data, dt)
@@ -313,8 +313,8 @@ def writeSegy(filename, Data, dt=1000, STHin={}, SHin={}):
ntraces = N[1]
print ntraces, ns
SH = getDefaultSegyHeader(ntraces, ns)
STH = getDefaultSegyTraceHeaders(ntraces, ns, dt)
SH = get_default_segy_header(ntraces, ns)
STH = get_default_segy_trace_headers(ntraces, ns, dt)
# ADD STHin, if exists...
for key in STHin.keys():
@@ -327,14 +327,14 @@ def writeSegy(filename, Data, dt=1000, STHin={}, SHin={}):
print key
SH[key] = SHin[key]
writeSegyStructure(filename, Data, SH, STH)
write_segy_structure(filename, Data, SH, STH)
def writeSegyStructure(filename,
Data,
SH,
STH,
endian='>'): # modified by A Squelch
def write_segy_structure(filename,
Data,
SH,
STH,
endian='>'): # modified by A Squelch
"""
writeSegyStructure(filename, Data, SegyHeader, SegyTraceHeaders)
@@ -374,7 +374,7 @@ def writeSegyStructure(filename,
pos = SH_def[key]["pos"]
format = SH_def[key]["type"]
value = SH[key]
putValue(value, f, pos, format, endian)
put_value(value, f, pos, format, endian)
# SEGY TRACES
ctype = SH_def['DataSampleFormat']['datatype'][revision][dsf]
@@ -396,7 +396,7 @@ def writeSegyStructure(filename,
str(format) +
" Writing " + key +
"=" + str(value))
putValue(value, f, pos, format, endian)
put_value(value, f, pos, format, endian)
# Write Data
cformat = endian + ctype
@@ -410,7 +410,7 @@ def writeSegyStructure(filename,
f.close()
def putValue(value, fileid, index, ctype='l', endian='>', number=1):
def put_value(value, fileid, index, ctype='l', endian='>', number=1):
"""
putValue(data, index, ctype, endian, number)
"""
@@ -490,7 +490,7 @@ def read_binary_value(f, index, ctype='l', endian='>', number=1):
return Value, index_end
def getBytePerSample(SH):
def get_byte_per_sample(SH):
revision = SH["SegyFormatRevisionNumber"]
if revision == 100:
@@ -517,4 +517,4 @@ if __name__ == '__main__':
filename = r'C:\Users\rjs\opendtectroot\Blake_Ridge_Hydrates_3D'\
r'\stack_final_scaled50_int8.sgy'
with open(filename, 'rb') as segy:
Data, SH, SegyTraceHeaders = readSegy(segy, filename)
Data, SH, SegyTraceHeaders = read_segy(segy, filename)
+69 -69
View File
@@ -383,7 +383,7 @@ STH_def["UnassignedInt2"] = {"pos": 236, "type": "int32"} # 'int32'); %236
##############
# FUNCTIONS
def imageSegy(Data):
def image_segy(Data):
"""
imageSegy(Data)
Image segy Data
@@ -418,7 +418,7 @@ def wiggle(Data, SH, skipt=1, maxval=8, lwidth=.1):
pylab.show()
def getDefaultSegyHeader(ntraces=100, ns=100):
def get_default_segy_header(ntraces=100, ns=100):
"""
SH=getDefaultSegyHeader()
"""
@@ -440,7 +440,7 @@ def getDefaultSegyHeader(ntraces=100, ns=100):
return SH
def getDefaultSegyTraceHeaders(ntraces=100, ns=100, dt=1000):
def get_default_segy_trace_headers(ntraces=100, ns=100, dt=1000):
"""
SH=getDefaultSegyTraceHeader()
"""
@@ -466,12 +466,12 @@ def getDefaultSegyTraceHeaders(ntraces=100, ns=100, dt=1000):
return STH
def getSegyTraceHeader(SH, THN='cdp', data=None):
def get_segy_trace_header(SH, THN='cdp', data=None):
"""
getSegyTraceHeader(SH,TraceHeaderName)
"""
logging.info("get trace header {0}".format(THN))
bps = getBytePerSample(SH)
bps = get_byte_per_sample(SH)
if data is None:
with open(SH["filename"]) as f:
@@ -483,7 +483,7 @@ def getSegyTraceHeader(SH, THN='cdp', data=None):
THformat = STH_def[THN]["type"]
ntraces = SH["ntraces"]
thv = zeros(ntraces)
getter = createValueGetter(data, THformat, endian, 1)
getter = create_value_getter(data, THformat, endian, 1)
for itrace in range(1, ntraces + 1, 1):
# i=itrace
@@ -501,10 +501,10 @@ def getSegyTraceHeader(SH, THN='cdp', data=None):
return thv
def getAllSegyTraceHeaders(SH, data='none'):
def get_all_segy_trace_headers(SH, data='none'):
SegyTraceHeaders = {'filename': SH["filename"]}
printverbose(
print_verbose(
'getAllSegyTraceHeaders : trying to get all segy trace headers',
logging.INFO)
@@ -512,42 +512,42 @@ def getAllSegyTraceHeaders(SH, data='none'):
data = open(SH["filename"]).read()
for key in STH_def.keys():
sth = getSegyTraceHeader(SH, key, data)
sth = get_segy_trace_header(SH, key, data)
SegyTraceHeaders[key] = sth
txt = "getAllSegyTraceHeaders : " + key
printverbose(txt, logging.DEBUG)
print_verbose(txt, logging.DEBUG)
return SegyTraceHeaders
def readSegy(filename):
def read_segy(filename):
"""
Data,SegyHeader,SegyTraceHeaders=getSegyHeader(filename)
"""
printverbose("readSegy : Trying to read " + filename, logging.DEBUG)
print_verbose("readSegy : Trying to read " + filename, logging.DEBUG)
with open(filename, 'rb') as raw:
data = raw.read()
filesize = os.path.getsize(filename)
segy_header = getSegyHeader(filename)
segy_header = get_segy_header(filename)
bytes_per_sample = getBytePerSample(segy_header)
bytes_per_sample = get_byte_per_sample(segy_header)
num_traces = (filesize - REEL_HEADER_NUM_BYTES) / \
(segy_header['ns'] * bytes_per_sample + TRACE_HEADER_NUM_BYTES)
printverbose("readSegy : Length of data : " + str(filesize), logging.DEBUG)
print_verbose("readSegy : Length of data : " + str(filesize), logging.DEBUG)
segy_header["ntraces"] = num_traces
ndummy_samples = TRACE_HEADER_NUM_BYTES / bytes_per_sample
printverbose(
print_verbose(
"readSegy : ndummy_samples=" + str(ndummy_samples), logging.DEBUG)
printverbose("readSegy : ntraces=" + str(num_traces) +
print_verbose("readSegy : ntraces=" + str(num_traces) +
" nsamples=" + str(segy_header['ns']), logging.DEBUG)
# GET TRACE
@@ -558,10 +558,10 @@ def readSegy(filename):
# Temporarily removed
# SegyTraceHeaders = getAllSegyTraceHeaders(segy_header,data)
inline_3d_trace_header = getSegyTraceHeader(segy_header, "Inline3D", data)
crossline_3d_trace_header = getSegyTraceHeader(
inline_3d_trace_header = get_segy_trace_header(segy_header, "Inline3D", data)
crossline_3d_trace_header = get_segy_trace_header(
segy_header, "Crossline3D", data)
num_samples_trace_header = getSegyTraceHeader(segy_header, "ns", data)
num_samples_trace_header = get_segy_trace_header(segy_header, "ns", data)
num_inlines = query(inline_3d_trace_header).distinct().count()
num_crosslines = query(crossline_3d_trace_header).distinct().count()
@@ -577,7 +577,7 @@ def readSegy(filename):
!= num_pseudo_samples:
raise RuntimeError("num_inlines * num_crosslines * num_samples != nd")
printverbose("readSegy : reading segy data", logging.DEBUG)
print_verbose("readSegy : reading segy data", logging.DEBUG)
# READ ALL DATA EXCEPT FOR SEGY HEADER
# Data = zeros((SH['ns'],ntraces))
@@ -589,32 +589,32 @@ def readSegy(filename):
DataDescr = SH_def["DataSampleFormat"]["descr"][revision][dsf]
printverbose("readSegy : SEG-Y revision = " + str(revision), logging.INFO)
printverbose("readSegy : DataSampleFormat=" + str(dsf) +
print_verbose("readSegy : SEG-Y revision = " + str(revision), logging.INFO)
print_verbose("readSegy : DataSampleFormat=" + str(dsf) +
"(" + DataDescr + ")", logging.INFO)
if (segy_header["DataSampleFormat"] == 1):
printverbose("readSegy : Assuming DSF=1, IBM FLOATS", logging.INFO)
pseudo_samples = getValue(
print_verbose("readSegy : Assuming DSF=1, IBM FLOATS", logging.INFO)
pseudo_samples = get_value(
data, index, 'ibm', endian, num_pseudo_samples)
elif (segy_header["DataSampleFormat"] == 2):
printverbose("readSegy : Assuming DSF=" +
print_verbose("readSegy : Assuming DSF=" +
str(segy_header["DataSampleFormat"]) + ", 32bit INT",
logging.INFO)
pseudo_samples = getValue(data, index, 'l', endian, num_pseudo_samples)
pseudo_samples = get_value(data, index, 'l', endian, num_pseudo_samples)
elif (segy_header["DataSampleFormat"] == 3):
printverbose("readSegy : Assuming DSF=" +
print_verbose("readSegy : Assuming DSF=" +
str(segy_header["DataSampleFormat"]) + ", 16bit INT",
logging.INFO)
pseudo_samples = getValue(data, index, 'h', endian, num_pseudo_samples)
pseudo_samples = get_value(data, index, 'h', endian, num_pseudo_samples)
elif (segy_header["DataSampleFormat"] == 5):
printverbose("readSegy : Assuming DSF=" +
print_verbose("readSegy : Assuming DSF=" +
str(segy_header["DataSampleFormat"]) + ", IEEE",
logging.INFO)
pseudo_samples = getValue(
pseudo_samples = get_value(
data, index, 'float', endian, num_pseudo_samples)
elif (segy_header["DataSampleFormat"] == 8):
printverbose("readSegy : Assuming DSF=" +
print_verbose("readSegy : Assuming DSF=" +
str(segy_header["DataSampleFormat"]) + ", 8bit CHAR",
logging.INFO)
with open(filename, 'rb') as f:
@@ -623,7 +623,7 @@ def readSegy(filename):
f, numpy.int8, num_pseudo_samples)
# Data1 = getValue(data,index,'B',endian,nd)
else:
printverbose(
print_verbose(
"readSegy : DSF=" +
str(segy_header["DataSampleFormat"]) +
", NOT SUPPORTED",
@@ -667,12 +667,12 @@ def readSegy(filename):
# if Data[i][j]>128:
# Data[i][j]=Data[i][j]-256
printverbose("readSegy : read data", logging.DEBUG)
print_verbose("readSegy : read data", logging.DEBUG)
SegyTraceHeaders = None # Temp
return signed_data_3d, segy_header, SegyTraceHeaders
def getSegyTrace(SH, itrace):
def get_segy_trace(SH, itrace):
"""
SegyTraceHeader,SegyTraceData=getSegyTrace(SegyHeader,itrace)
itrace : trace number to read
@@ -680,7 +680,7 @@ def getSegyTrace(SH, itrace):
"""
data = open(SH["filename"]).read()
bps = getBytePerSample(SH)
bps = get_byte_per_sample(SH)
# GET TRACE HEADER
index = 3200 + (itrace - 1) * (240 + SH['ns'] * bps)
@@ -689,11 +689,11 @@ def getSegyTrace(SH, itrace):
# GET TRACE
index = 3200 + (itrace - 1) * (240 + SH['ns'] * bps) + 240
SegyTraceData = getValue(data, index, 'float', endian, SH['ns'])
SegyTraceData = get_value(data, index, 'float', endian, SH['ns'])
return SegyTraceHeader, SegyTraceData
def getSegyHeader(filename):
def get_segy_header(filename):
"""
SegyHeader=getSegyHeader(filename)
"""
@@ -705,25 +705,25 @@ def getSegyHeader(filename):
pos = SH_def[key]["pos"]
format = SH_def[key]["type"]
SegyHeader[key], index = getValue(data, pos, format, endian)
SegyHeader[key], index = get_value(data, pos, format, endian)
txt = str(pos) + " " + str(format) + " Reading " + \
key + "=" + str(SegyHeader[key])
printverbose(txt, logging.INFO)
print_verbose(txt, logging.INFO)
# SET NUMBER OF BYTES PER DATA SAMPLE
bps = getBytePerSample(SegyHeader)
bps = get_byte_per_sample(SegyHeader)
filesize = len(data)
ntraces = (filesize - 3600) / (SegyHeader['ns'] * bps + 240)
SegyHeader["ntraces"] = ntraces
printverbose('getSegyHeader : succesfully read ' + filename, logging.INFO)
print_verbose('getSegyHeader : succesfully read ' + filename, logging.INFO)
return SegyHeader
def writeSegy(filename, Data, dt=.001, STHin={}):
def write_segy(filename, Data, dt=.001, STHin={}):
"""
writeSegy(filename,Data,dt)
@@ -737,15 +737,15 @@ def writeSegy(filename, Data, dt=.001, STHin={}):
"""
printverbose("writeSegy : Trying to write " + filename, logging.DEBUG)
print_verbose("writeSegy : Trying to write " + filename, logging.DEBUG)
N = Data.shape
ns = N[0]
ntraces = N[1]
print ntraces, ns
SH = getDefaultSegyHeader(ntraces, ns)
STH = getDefaultSegyTraceHeaders(ntraces, ns, dt)
SH = get_default_segy_header(ntraces, ns)
STH = get_default_segy_trace_headers(ntraces, ns, dt)
# ADD STHin, if exists...
for key in STHin.keys():
@@ -753,10 +753,10 @@ def writeSegy(filename, Data, dt=.001, STHin={}):
for a in range(ntraces):
STH[key] = STHin[key][a]
writeSegyStructure(filename, Data, SH, STH)
write_segy_structure(filename, Data, SH, STH)
def writeSegyStructure(filename, Data, SH, STH):
def write_segy_structure(filename, Data, SH, STH):
"""
writeSegyHeader(filename,Data,SegyHeader,SegyTraceHeaders)
@@ -768,7 +768,7 @@ def writeSegyStructure(filename, Data, SH, STH):
"""
printverbose(
print_verbose(
"writeSegyStructure : Trying to write " + filename, logging.DEBUG)
f = open(filename, 'w')
@@ -780,9 +780,9 @@ def writeSegyStructure(filename, Data, SH, STH):
# revision=1
DataDescr = SH_def["DataSampleFormat"]["descr"][revision][dsf]
printverbose(
print_verbose(
"writeSegyStructure : SEG-Y revision = " + str(revision), logging.INFO)
printverbose("writeSegyStructure : DataSampleFormat=" +
print_verbose("writeSegyStructure : DataSampleFormat=" +
str(dsf) + "(" + DataDescr + ")", logging.INFO)
# WRITE SEGY HEADER
@@ -793,7 +793,7 @@ def writeSegyStructure(filename, Data, SH, STH):
value = SH[key]
# SegyHeader[key],index = putValue(value,f,pos,format,endian);
putValue(value, f, pos, format, endian)
put_value(value, f, pos, format, endian)
txt = str(pos) + " " + str(format) + \
" Reading " + key + "=" + str(value)
@@ -809,7 +809,7 @@ def writeSegyStructure(filename, Data, SH, STH):
for itrace in range(SH['ntraces']):
index = 3600 + itrace * sizeT
printverbose(
print_verbose(
'Writing Trace #' +
str(itrace + 1) +
'/' +
@@ -822,8 +822,8 @@ def writeSegyStructure(filename, Data, SH, STH):
value = STH[key][itrace]
txt = str(pos) + " " + str(format) + \
" Writing " + key + "=" + str(value)
printverbose(txt, logging.INFO)
putValue(value, f, pos, format, endian)
print_verbose(txt, logging.INFO)
put_value(value, f, pos, format, endian)
# Write Data
cformat = endian + ctype
@@ -837,7 +837,7 @@ def writeSegyStructure(filename, Data, SH, STH):
# return segybuffer
def putValue(value, fileid, index, ctype='l', endian='>', number=1):
def put_value(value, fileid, index, ctype='l', endian='>', number=1):
"""
putValue(data,index,ctype,endian,number)
"""
@@ -866,11 +866,11 @@ def putValue(value, fileid, index, ctype='l', endian='>', number=1):
# size = l_float
pass
else:
printverbose('Bad Ctype : ' + ctype, logging.ERROR)
print_verbose('Bad Ctype : ' + ctype, logging.ERROR)
cformat = endian + ctype * number
printverbose(
print_verbose(
'putValue : cformat : ' + cformat + ' ctype=' + ctype, logging.DEBUG)
strVal = struct.pack(cformat, value)
@@ -880,7 +880,7 @@ def putValue(value, fileid, index, ctype='l', endian='>', number=1):
return 1
def createValueGetter(data, ctype='l', endian='>', number=1):
def create_value_getter(data, ctype='l', endian='>', number=1):
"""
getValue(data,index,ctype,endian,number)
"""
@@ -908,7 +908,7 @@ def createValueGetter(data, ctype='l', endian='>', number=1):
elif (ctype == 'ibm'):
size = l_float
else:
printverbose('Bad Ctype : ' + ctype, logging.ERROR)
print_verbose('Bad Ctype : ' + ctype, logging.ERROR)
cformat = endian + ctype * number
@@ -933,7 +933,7 @@ def createValueGetter(data, ctype='l', endian='>', number=1):
else:
if (ctype == 'B'):
printverbose(
print_verbose(
'getValue : Inefficient use of 1 byte Integer...',
logging.WARNING)
@@ -950,7 +950,7 @@ def createValueGetter(data, ctype='l', endian='>', number=1):
return other
def getValue(data, index, ctype='l', endian='>', number=1):
def get_value(data, index, ctype='l', endian='>', number=1):
"""
getValue(data,index,ctype,endian,number)
"""
@@ -978,7 +978,7 @@ def getValue(data, index, ctype='l', endian='>', number=1):
elif (ctype == 'ibm'):
size = l_float
else:
printverbose('Bad Ctype : ' + ctype, logging.ERROR)
print_verbose('Bad Ctype : ' + ctype, logging.ERROR)
cformat = endian + ctype * number
@@ -998,7 +998,7 @@ def getValue(data, index, ctype='l', endian='>', number=1):
Value = struct.unpack(cformat, data[index:index_end])
if (ctype == 'B'):
printverbose(
print_verbose(
'getValue : Inefficient use of 1 byte Integer...', logging.WARNING)
# vtxt = 'getValue : '+'start='+str(index)+' size='+str(size)+ '
@@ -1016,7 +1016,7 @@ def print_version():
print 'SegyPY version is ', version
def printverbose(txt, level=logging.DEBUG):
def print_verbose(txt, level=logging.DEBUG):
pass
# log.log(level, txt)
@@ -1058,7 +1058,7 @@ def ibm2ieee2(ibm_float):
return sign * 16 ** (istic - 64) * (mant / dividend)
def getBytePerSample(SH):
def get_byte_per_sample(SH):
revision = SH["SegyFormatRevisionNumber"]
# if revision == 100:
@@ -1074,7 +1074,7 @@ def getBytePerSample(SH):
log.debug("bytes per sample = {0}".format(bps))
printverbose("getBytePerSample : bps=" + str(bps), logging.DEBUG)
print_verbose("getBytePerSample : bps=" + str(bps), logging.DEBUG)
return bps
@@ -1099,7 +1099,7 @@ class SegyHeaderClass:
def cdp(self):
return "Getting CDP trace header"
def InlineX(self):
def inline_x(self):
return "Getting CDP trace header"
@@ -1113,7 +1113,7 @@ class SegyClass:
self.THOMAS = 'Thomas'
if __name__ == '__main__':
signed_data_3d, sh, sth = readSegy(
signed_data_3d, sh, sth = read_segy(
r'C:\Users\rjs\opendtectroot\Blake_Ridge_Hydrates_3D'
r'\stack_final_scaled50_int8.sgy')
unsigned_data_3d = signed_data_3d.view(numpy.uint8)
+4 -4
View File
@@ -9,20 +9,20 @@ import segypy
def test_read(filename):
# Read Segy File
with open(filename, 'rb') as f:
return segypy.readSegy(f, filename)
return segypy.read_segy(f, filename)
def test_write(filename_out, data, sh, sth):
sh['DataSampleFormat'] = 5
sh['SegyFormatRevisionNumber'] = 100
segypy.writeSegyStructure(filename_out, data, sh, sth)
segypy.write_segy_structure(filename_out, data, sh, sth)
segypy.wiggle(data, sh, 2, .1, .1)
f_ieee = 'data_IEEE.segy'
f_ibm = 'data_IBM_REV1.segy'
d_ieee, sh, sth = segypy.readSegy(f_ieee)
d_ibm, sh, sth = segypy.readSegy(f_ibm)
d_ieee, sh, sth = segypy.read_segy(f_ieee)
d_ibm, sh, sth = segypy.read_segy(f_ibm)
return d_ieee, d_ibm, data