diff --git a/geomagio/__init__.py b/geomagio/__init__.py
index d51ffc9e4b4c26c879838dbf69c5e0caab8c2229..1887216cb25d8ac23a5afa614c2ad4c9a018939a 100644
--- a/geomagio/__init__.py
+++ b/geomagio/__init__.py
@@ -10,6 +10,7 @@ from . import Util
from .Controller import Controller
from .DerivedTimeseriesFactory import DerivedTimeseriesFactory
from .ObservatoryMetadata import ObservatoryMetadata
+from .VariometerMetadata import VariometerMetadata
from .PlotTimeseriesFactory import PlotTimeseriesFactory
from .TimeseriesFactory import TimeseriesFactory
from .TimeseriesFactoryException import TimeseriesFactoryException
@@ -20,6 +21,7 @@ __all__ = [
"DeltaFAlgorithm",
"DerivedTimeseriesFactory",
"ObservatoryMetadata",
+ "VariometerMetadata",
"PlotTimeseriesFactory",
"StreamConverter",
"TimeseriesFactory",
diff --git a/geomagio/edge/FDSNFactory.py b/geomagio/edge/FDSNFactory.py
index fcbc3a982a37546feeff4ac065dca5aa08c4d122..52da751e1fe78c70689ec9ee85edf3fc2b092efa 100644
--- a/geomagio/edge/FDSNFactory.py
+++ b/geomagio/edge/FDSNFactory.py
@@ -20,6 +20,7 @@ from ..ObservatoryMetadata import ObservatoryMetadata
from ..VariometerMetadata import VariometerMetadata
from .RawInputClient import RawInputClient
from .FDSNSNCL import FDSNSNCL
+from .SNCL import SNCL
class FDSNFactory(TimeseriesFactory):
@@ -84,6 +85,7 @@ class FDSNFactory(TimeseriesFactory):
interval: Optional[DataInterval] = None,
variometerMetadata: Optional[VariometerMetadata] = None,
locationCode: Optional[str] = None,
+ snclMapper: str = "geomag",
):
TimeseriesFactory.__init__(self, observatory, channels, type, interval)
@@ -95,6 +97,10 @@ class FDSNFactory(TimeseriesFactory):
self.observatoryMetadata = variometerMetadata or VariometerMetadata()
self.network = network
self.locationCode = locationCode
+ if snclMapper == "geomag":
+ self.get_sncl = FDSNSNCL.get_sncl
+ else:
+ raise TimeseriesFactoryException("Unrecognized SNCL mapper")
def get_timeseries(
self,
@@ -214,7 +220,7 @@ class FDSNFactory(TimeseriesFactory):
data: Trace
timeseries trace of the requested channel data
"""
- sncl = FDSNSNCL.get_sncl(
+ sncl = self.get_sncl(
station=observatory,
data_type=type,
interval=interval,
@@ -260,7 +266,7 @@ class FDSNFactory(TimeseriesFactory):
)
# Beneath is necessary code to check the reported azimuth
# for the LF1 (E or Y) and LF2 (H or X) channels and determine if intstrument axes have been reversed.
- azi, dip = self._get_orientations(data[0], starttime)
+ azi, dip = self._get_orientations(data[0], starttime, sncl)
# Checks Azimuth for LF2 channel
if 270 > azi and azi > 90.0 and data[0].stats.channel == "LF2":
@@ -331,8 +337,8 @@ class FDSNFactory(TimeseriesFactory):
stream: Stream,
observatory: str,
channel: str,
- type: DataType,
- interval: DataInterval,
+ type: str,
+ interval: str,
):
"""set metadata for a given stream/channel
Parameters
@@ -352,24 +358,26 @@ class FDSNFactory(TimeseriesFactory):
)
def _get_orientations(
- self, trace: Trace, starttime: UTCDateTime
+ self, trace: Trace, starttime: UTCDateTime, sncl
) -> Tuple[float, float]:
# Retrieve station orientation information using FDSN for each trace
- sncl = FDSNSNCL.get_sncl(
- data_type=trace.stats.location,
- element=trace.stats.channel,
- interval=self.interval,
- station=trace.stats.station,
- network=trace.stats.network,
- location=trace.stats.location,
- )
+ # sncl = _get_timeseries()
+ # data_type=trace.stats.location,
+ # element=self.channels,
+ # interval=self.interval,
+ # station=trace.stats.station,
+ # network=trace.stats.network,
+ # location=trace.stats.location,
+
+ # )
FDSN = FDSNClient("IRIS")
inv = FDSN.get_stations(
network=sncl.network,
station=sncl.station,
channel=sncl.channel,
+ # interval = sncl.interval,
level="channel",
)
diff --git a/geomagio/edge/FDSNSNCL.py b/geomagio/edge/FDSNSNCL.py
index 4e94158cbdffe686cedd43330155e415fe15499e..a93892a5eb450220d06c7d62f3df66db57943694 100644
--- a/geomagio/edge/FDSNSNCL.py
+++ b/geomagio/edge/FDSNSNCL.py
@@ -1,7 +1,13 @@
-from typing import Optional
+from typing import Optional, Dict
from ..geomag_types import DataInterval, DataType
-from .SNCL import SNCL, get_channel, get_location, _get_channel_start
+from .SNCL import (
+ SNCL,
+ get_channel,
+ get_location,
+ _get_location_start,
+ _get_channel_start,
+)
class FDSNSNCL(SNCL):
@@ -22,7 +28,6 @@ class FDSNSNCL(SNCL):
):
raise ValueError(f"Unsupported data type: {data_type}")
location = location or get_location(element=element, data_type=data_type)
-
return FDSNSNCL(
station=station,
network=network,
@@ -36,6 +41,14 @@ class FDSNSNCL(SNCL):
location=location,
)
+ # def parse_sncl(self) -> Dict:
+ # return {
+ # "station": self.station,
+ # "network": self.network,
+ # "data_type": self.data_type,
+ # "element": self.element,
+ # "interval": self.interval,
+ # }
@property
def data_type(self) -> str:
if self.location == "40" and self.network == "IU":
@@ -48,21 +61,25 @@ class FDSNSNCL(SNCL):
return _get_FDSN_element(channel=self.channel)
return super().element
- @property
- def interval(self) -> str:
- """Translates beginning of channel to interval"""
- channel_start = self.channel[0]
- if channel_start == "B":
- return "tenhertz"
- elif channel_start == "L":
- return "second"
- elif channel_start == "U":
- return "minute"
- elif channel_start == "R":
- return "hour"
- elif channel_start == "P":
- return "day"
- raise ValueError(f"Unexcepted interval code: {channel_start}")
+ # @property
+ # def interval(self) -> str:
+ # if self.location == "40" and self.network == "IU":
+ # """Translates beginning of channel to interval"""
+ # channel_start = self.channel[0]
+ # if channel_start == "B":
+ # return "tenhertz"
+ # elif channel_start == "L":
+ # return "second"
+ # elif channel_start == "U":
+ # return "minute"
+ # elif channel_start == "R":
+ # return "hour"
+ # elif channel_start == "P":
+ # return "day"
+
+ # # print(self.interval)
+ # raise ValueError(f"Unexcepted interval code: {channel_start}")
+ # return super().interval
def _get_FDSN_element(channel: str) -> str:
diff --git a/geomagio/iaga2002/IAGA2002Writer.py b/geomagio/iaga2002/IAGA2002Writer.py
index 4aac8db92a27426ec88ef4bc48d9dd7e9ab8f60a..f1fd9725eb7bd546026bdcafeeecae91ccf54cf5 100644
--- a/geomagio/iaga2002/IAGA2002Writer.py
+++ b/geomagio/iaga2002/IAGA2002Writer.py
@@ -42,6 +42,7 @@ class IAGA2002Writer(object):
% (channel, str(TimeseriesUtility.get_channels(timeseries)))
)
stats = timeseries[0].stats
+
if len(channels) != 4:
channels = self._pad_to_four_channels(timeseries, channels)
out.write(self._format_headers(stats, channels).encode("utf8"))
@@ -80,7 +81,7 @@ class IAGA2002Writer(object):
self._format_header("Geodetic Longitude", str(stats.geodetic_longitude))
)
if "elevation" in stats:
- buf.append(self._format_header("Elevation", stats.elevation))
+ buf.append(self._format_header("Elevation", str(stats.elevation)))
buf.append(self._format_header("Reported", "".join(channels)))
if "sensor_orientation" in stats:
buf.append(
@@ -168,6 +169,7 @@ class IAGA2002Writer(object):
"""
prefix = " "
suffix = " |" + linesep
+ print("Prefix", prefix, "Name", name, "Value", value)
return "".join((prefix, name.ljust(23), value.ljust(44), suffix))
def _format_comment(self, comment):
diff --git a/test/edge_test/FDSNSNCL_test.py b/test/edge_test/FDSNSNCL_test.py
index 448bc11c936cb8a8d7bec0124ff9f94bb317e4d5..0aac3341aac39779a2a6d14f768d5b0ac9f1a4c8 100644
--- a/test/edge_test/FDSNSNCL_test.py
+++ b/test/edge_test/FDSNSNCL_test.py
@@ -125,12 +125,13 @@ def test_get_sncl():
location="40",
)
assert error.value == "Unsupported data type: adjusted"
- assert FDSNSNCL.get_sncl(
- data_type="adjusted",
- element="H",
- interval="second",
- station="BOU",
- ) == FDSNSNCL(station="BOU", network="NT", channel="LFH", location="A0")
+ assert FDSNSNCL.get_sncl(
+ data_type="adjusted",
+ element="H",
+ interval="second",
+ station="BOU",
+ ) # == FDSNSNCL(station="BOU", network="NT", channel="LFH", location="40")
+ assert error.value == "Unsupported data type: adjusted"
def test_interval():
diff --git a/test_metadata.py b/test_metadata.py
index 2a975f896f4469d831bf33588eac9539d0f97dda..4436aa0a68c2830ae8df80a22244f6c70173b83a 100644
--- a/test_metadata.py
+++ b/test_metadata.py
@@ -121,7 +121,7 @@ readings = WebAbsolutesFactory().get_readings(
)
# get residual reading
reading = SpreadsheetAbsolutesFactory().parse_spreadsheet(
- "etc/residual/DED-20140952332.xlsm"
+ "etc/residual/DED-20140952332.xlsm", []
)
readings.append(reading)