Auto integrate maxwell probe
@@ -2,6 +2,7 @@ import neo
from .neobaseextractor import NeoBaseRecordingExtractor
+import probeinterface as pi
class MaxwellRecordingExtractor(NeoBaseRecordingExtractor):
"""
@@ -29,6 +30,13 @@ class MaxwellRecordingExtractor(NeoBaseRecordingExtractor):
def __init__(self, file_path, stream_id=None, rec_name=None):
neo_kwargs = {'filename': str(file_path)}
NeoBaseRecordingExtractor.__init__(self, stream_id=stream_id, rec_name=rec_name, **neo_kwargs)
+
+ # well_name is stream_id
+ well_name = self.stream_id
+ # rec_name auto set by neo
+ rec_name=self.neo_reader.rec_name
+ probe = pi.read_maxwell(file_path, well_name=well_name, rec_name=rec_name)
+ self.set_probe(probe, in_place=True)
self._kwargs = dict(file_path=file_path, stream_id=stream_id, rec_name = rec_name)
