Merge pull request #47 from malcolmw/dev/tags_attribute

Add tags attribute to ASDFDataSet class.

Author: krischer

pyasdf/asdf_data_set.py

@@ -492,6 +492,14 @@ def events(self, event):
         self.__file["QuakeML"].resize(data.shape)
         self.__file["QuakeML"][:] = data
 
+    @property
+    def waveform_tags(self):
+        """
+        Returns a set with all tags in the dataset.
+        """
+        return set(itertools.chain.from_iterable(i.get_waveform_tags()
+                                                 for i in self.waveforms))
+
     def add_auxiliary_data_file(
             self, filename_or_object, path, parameters=None,
             provenance_id=None):

pyasdf/tests/test_asdf_data_set.py

@@ -72,6 +72,21 @@ def example_data_set(tmpdir):
     return Namespace(filename=asdf_filename, tmpdir=tmpdir.strpath)
 
 
+def test_waveform_tags_attribute(tmpdir):
+    asdf_filename = os.path.join(tmpdir.strpath, "test.h5")
+    data_path = os.path.join(data_dir, "small_sample_data_set")
+
+    data_set = ASDFDataSet(asdf_filename)
+
+    itag = 1
+    for filename in glob.glob(os.path.join(data_path, "*.mseed")):
+        data_set.add_waveforms(filename, tag="tag%d" % itag)
+        itag += 1
+
+    expected = set([u"tag1", u"tag2", u"tag3", u"tag4", u"tag5", u"tag6"])
+    assert data_set.waveform_tags == expected
+
+
 def test_data_set_creation(tmpdir):
     """
     Test data set creation with a small test dataset.

setup.py

@@ -13,6 +13,11 @@
 ---------
 
 ::
+    Version 0.4.x
+    -------------
+    * New .waveform_tags property for the dataset object return a set of all
+      available waveform tags (see #46, #47)
+
     Version 0.4.0 (March 12, 2018)
     -------------------
     * Support for ASDF version 1.0.2. Allows writing traces that are less than