Changeset 531 for SHX/trunk/SeismicHandler/basics/tools.py

Timestamp:

02/02/2012 05:51:18 PM (11 years ago)

Author:

marcus

Message:

• ...
• method for expanding trace lists

File:

• SHX/trunk/SeismicHandler/basics/tools.py (modified) (2 diffs)

SHX/trunk/SeismicHandler/basics/tools.py

@@ -9 +9 @@
 def Property(func):
     return property(**func())
+
 
 class Singleton(type):
@@ -24 +25 @@
 
         return cls.instance
+
+
+def expandTraceList(trcs, selection):
+    """
+    Expand comma separated list of traces into sorted internal trace numbers
+    (reverse order).
+    """
+
+    if not len(trcs):
+        return None
+
+    selection = [i.lower() for i in selection.split(",")]
+
+    if "all" in selection:
+        selected = range(0, len(trcs))
+        selected.reverse()
+        return selected
+
+    selected = []
+    for s in selection:
+        # range
+        if "-" in s:
+            try:
+                start, stop = map(int, s.split("-"))
+            except:
+                print "  skipping '%s'" % s
+                continue
+
+            if stop < start:
+                start, stop = stop, start
+
+            selected.extend(range(start - 1, stop))
+
+        else:
+            if s.isdigit():
+                selected.append(int(s) - 1)
+            else:
+                print "  skipping '%s'" % s
+
+    # unique
+    selected = list(set(selected))
+    selected.sort()
+    selected.reverse()
+
+    return selected