Skip to content
Snippets Groups Projects

Plot horizon distance from ranking statistics

Merged ChiWai Chan requested to merge plot_psd_horizon into master
1 unresolved thread
1 file
+ 10
8
Compare changes
  • Side-by-side
  • Inline
+ 10
8
@@ -49,8 +49,10 @@ class DataFileMixin:
extension = self.extension
return T050017_filename(ifos, self.description(svd_bin, subtype), span, extension)
def file_pattern(self, svd_bin=None, subtype=None):
return f"*-{self.description(svd_bin, subtype)}-*-*{self.extension}"
def file_pattern(self, svd_bin=None, subtype=None, extension=None):
if not extension:
extension = self.extension
return f"*-{self.description(svd_bin, subtype)}-*-*{extension}"
def directory(self, root=None, start=None):
path = self.name.lower()
@@ -217,7 +219,7 @@ class DataCache:
return cls(name, [CacheEntry.from_T050017(entry) for entry in cache])
@classmethod
def find(cls, name, start=None, end=None, root=None, segments=None, svd_bins=None, subtype=None):
def find(cls, name, start=None, end=None, root=None, segments=None, svd_bins=None, extension=None, subtype=None):
cache = []
if svd_bins:
svd_bins = set([svd_bins]) if isinstance(svd_bins, str) else set(svd_bins)
@@ -227,8 +229,8 @@ class DataCache:
subtype = [subtype]
for svd_bin in svd_bins:
for stype in subtype:
cache.extend(glob.glob(cls._glob_path(name, root, svd_bin, stype)))
cache.extend(glob.glob(cls._glob_path(name, root, svd_bin, stype, gps_dir=False)))
cache.extend(glob.glob(cls._glob_path(name, root, svd_bin, stype, extension=extension)))
cache.extend(glob.glob(cls._glob_path(name, root, svd_bin, stype, extension=extension, gps_dir=False)))
cache = [CacheEntry.from_T050017(entry) for entry in cache]
if segments:
@@ -249,11 +251,11 @@ class DataCache:
return path
@staticmethod
def _glob_path(name, root=None, svd_bin=None, subtype=None, gps_dir=True):
def _glob_path(name, root=None, svd_bin=None, subtype=None, extension=None, gps_dir=True):
if gps_dir:
glob_path = os.path.join(str(name).lower(), "*", name.file_pattern(svd_bin, subtype))
glob_path = os.path.join(str(name).lower(), "*", name.file_pattern(svd_bin, subtype, extension=extension))
else:
glob_path = os.path.join(str(name).lower(), name.file_pattern(svd_bin, subtype))
glob_path = os.path.join(str(name).lower(), name.file_pattern(svd_bin, subtype, extension=extension))
if root:
glob_path = os.path.join(root, glob_path)
return glob_path
Loading