muse2wfdb 1.0.0__tar.gz

muse2wfdb-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 nagyl1999
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

muse2wfdb-1.0.0/PKG-INFO

@@ -0,0 +1,55 @@
+Metadata-Version: 2.4
+Name: muse2wfdb
+Version: 1.0.0
+Summary: Convert GE MUSE ECG XML exports to WFDB format
+Author: Levente Nagy
+License: MIT
+Project-URL: Homepage, https://github.com/nagyl1999/muse2wfdb
+Project-URL: Issues, https://github.com/nagyl1999/muse2wfdb/issues
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: wfdb>=4.1.0
+Requires-Dist: xmltodict>=0.13.0
+Requires-Dist: numpy>=1.22
+Dynamic: license-file
+
+# muse2wfdb
+### Convert GE MUSE XML ECG exports to WFDB format
+
+`muse2wfdb` is a lightweight Python library that converts ECG data exported from the GE MUSE Resting ECG System (in XML format) into WFDB (WaveForm DataBase) format, compatible with PhysioNet tools and the wfdb Python package.
+
+## Installation
+
+Install directly from PyPI:
+
+```bash
+pip install muse2wfdb
+```
+
+Or install locally from source:
+
+```bash
+git clone https://github.com/nagyl1999/muse2wfdb.git
+cd muse2wfdb
+pip install .
+```
+
+## Usage
+
+An example is provided in the `examples` folder.
+
+```python
+from muse2wfdb.converter import muse_to_wfdb
+import wfdb
+
+muse_export = "examples/anonim_pac_xml_export.txt"
+wfdb_filename = "patient001_ecg"
+
+muse_to_wfdb(muse_export, wfdb_filename)
+
+record = wfdb.rdrecord(wfdb_filename)
+annotation = wfdb.rdann(wfdb_filename, 'atr')
+
+wfdb.plot_wfdb(record=record, annotation=annotation, title="MUSE exported ECG")
+```

muse2wfdb-1.0.0/README.md

@@ -0,0 +1,39 @@
+# muse2wfdb
+### Convert GE MUSE XML ECG exports to WFDB format
+
+`muse2wfdb` is a lightweight Python library that converts ECG data exported from the GE MUSE Resting ECG System (in XML format) into WFDB (WaveForm DataBase) format, compatible with PhysioNet tools and the wfdb Python package.
+
+## Installation
+
+Install directly from PyPI:
+
+```bash
+pip install muse2wfdb
+```
+
+Or install locally from source:
+
+```bash
+git clone https://github.com/nagyl1999/muse2wfdb.git
+cd muse2wfdb
+pip install .
+```
+
+## Usage
+
+An example is provided in the `examples` folder.
+
+```python
+from muse2wfdb.converter import muse_to_wfdb
+import wfdb
+
+muse_export = "examples/anonim_pac_xml_export.txt"
+wfdb_filename = "patient001_ecg"
+
+muse_to_wfdb(muse_export, wfdb_filename)
+
+record = wfdb.rdrecord(wfdb_filename)
+annotation = wfdb.rdann(wfdb_filename, 'atr')
+
+wfdb.plot_wfdb(record=record, annotation=annotation, title="MUSE exported ECG")
+```

muse2wfdb-1.0.0/muse2wfdb/converter.py

@@ -0,0 +1,171 @@
+"""
+muse2wfdb.converter
+-------------------
+Convert GE MUSE XML ECG exports to WFDB (WaveForm Database) format.
+
+Author: Levente Nagy
+License: MIT
+"""
+
+import wfdb
+import array
+import base64
+import xmltodict
+import numpy as np
+from pathlib import Path
+from typing import Dict, Any, Optional, List
+
+# Standard 12-lead ECG names
+LEAD_NAMES: List[str] = [
+    'I', 'II', 'III', 'aVR', 'aVL', 'aVF',
+    'V1', 'V2', 'V3', 'V4', 'V5', 'V6'
+]
+
+QRS_TYPE_MAP = {
+    "0": "N",   # Normal
+    "1": "V",   # Ventricular
+    "2": "S",   # Supraventricular
+}
+
+
+def read_muse_file(path: str) -> Dict[str, Any]:
+    """
+    Read and parse a GE MUSE XML file into a Python dictionary.
+
+    Args:
+        path: Path to the XML file exported from MUSE.
+
+    Returns:
+        Parsed XML structure as a Python dictionary.
+    """
+    path = Path(path)
+    if not path.exists():
+        raise FileNotFoundError(f"MUSE file not found: {path}")
+
+    with open(path, "rb") as muse_file:
+        return xmltodict.parse(muse_file.read().decode("utf-8"))
+
+
+def select_waveform(ecg: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Select the 'Rhythm' waveform section from the MUSE ECG XML structure.
+
+    Args:
+        ecg: Parsed MUSE ECG dictionary.
+
+    Returns:
+        The waveform dictionary that corresponds to the 'Rhythm' ECG signal.
+    """
+    for waveform in ecg["RestingECG"]["Waveform"]:
+        if waveform["WaveformType"] == "Rhythm":
+            return waveform
+    raise ValueError("No 'Rhythm' waveform found in MUSE ECG file.")
+
+
+def process_waveforms(waveform: Dict[str, Any]) -> Dict[str, np.ndarray]:
+    """
+    Decode base64 ECG waveforms for each lead and compute derived leads.
+
+    Args:
+        waveform: The selected waveform dictionary from the MUSE file.
+    """
+    lead_waveforms: Dict[str, np.ndarray] = {lead_name: [] for lead_name in LEAD_NAMES}
+
+    for lead in waveform["LeadData"]:
+        lead_data = lead["WaveFormData"]
+        decoded = base64.b64decode(lead_data)
+        samples = np.array(array.array("h", decoded))
+        lead_waveforms[lead["LeadID"]] = samples / 1000  # convert to mV
+
+    # Compute derived leads
+    lead_waveforms["III"] = lead_waveforms["II"] - lead_waveforms["I"]
+    lead_waveforms["aVR"] = -(lead_waveforms["I"] + lead_waveforms["II"]) / 2
+    lead_waveforms["aVL"] = lead_waveforms["I"] - 0.5 * lead_waveforms["II"]
+    lead_waveforms["aVF"] = lead_waveforms["II"] - 0.5 * lead_waveforms["I"]
+
+    return lead_waveforms
+
+
+def save_wfdb(lead_waveforms: Dict[str, np.ndarray], output_name: str = "wfdb_record", fs: int = 500) -> None:
+    """
+    Save the processed ECG signals into WFDB format (.hea and .dat files).
+
+    Args:
+        lead_waveforms: The processed waveforms of the MUSE export.
+        output_name: The base name for the output WFDB files.
+        fs: Sampling frequency in Hz (default: 500).
+    """
+    signals = []
+    for lead in LEAD_NAMES:
+        if lead_waveforms[lead].size == 0:
+            raise ValueError(f"Missing waveform data for lead: {lead}")
+        signals.append(lead_waveforms[lead])
+
+    # Stack all leads into (N, 12) matrix
+    signals = np.column_stack(signals)
+
+    wfdb.wrsamp(
+        record_name=output_name,
+        fs=fs,
+        units=["mV"] * len(LEAD_NAMES),
+        sig_name=LEAD_NAMES,
+        p_signal=signals,
+        fmt=['16'] * signals.shape[1]
+    )
+
+
+def save_qrs_annotations(record_name: str, qrs_data: list, fs: int = 500):
+    """
+    Save QRS annotations from parsed MUSE XML data to a WFDB .qrs file.
+
+    Parameters
+    ----------
+    record_name : str
+        Name of the WFDB record (without extension)
+    qrs_data : list[dict]
+        List of QRS entries, each with keys ['Number', 'Type', 'Time']
+    fs : int
+        Sampling frequency (Hz)
+    """
+    # Convert <Time> (ms) to sample index
+    samples = np.array([int(qrs['Time']) for qrs in qrs_data])
+    
+    # Symbol: all 'N' (normal) by default; can map from 'Type'
+    symbols = np.array([QRS_TYPE_MAP.get(qrs['Type'], 'N') for qrs in qrs_data])
+
+    # Optional aux notes
+    aux_notes = np.array([f"QRS_{qrs['Number']}" for qrs in qrs_data])
+
+    ann = wfdb.wrann(
+        record_name=record_name,
+        sample=samples,
+        symbol=symbols,
+        aux_note=aux_notes,
+        extension='atr'
+    )
+
+
+def muse_to_wfdb(path: str, output_name: str = "wfdb_record") -> None:
+    """
+    Main function to convert a MUSE XML ECG file into WFDB format.
+
+    Args:
+        path: Path to the input MUSE XML file.
+        output_name: Output WFDB record name (without extension).
+    """
+    ecg = read_muse_file(path)
+    waveform = select_waveform(ecg)
+
+    # Process leads
+    frequency = int(ecg["RestingECG"]["Waveform"][0]["SampleBase"])
+    lead_waveforms = process_waveforms(waveform)
+
+    # Save processed data in WFDB format
+    save_wfdb(lead_waveforms, output_name, frequency)
+
+    if "QRS" not in ecg["RestingECG"]["QRSTimesTypes"]:
+        raise ValueError("No 'Rhythm' waveform found in MUSE ECG file, .dat file saved.")
+    
+    # Save complexes in WFDB format
+    complexes = ecg["RestingECG"]["QRSTimesTypes"]["QRS"]
+    save_qrs_annotations(output_name, complexes)

muse2wfdb-1.0.0/muse2wfdb.egg-info/PKG-INFO

@@ -0,0 +1,55 @@
+Metadata-Version: 2.4
+Name: muse2wfdb
+Version: 1.0.0
+Summary: Convert GE MUSE ECG XML exports to WFDB format
+Author: Levente Nagy
+License: MIT
+Project-URL: Homepage, https://github.com/nagyl1999/muse2wfdb
+Project-URL: Issues, https://github.com/nagyl1999/muse2wfdb/issues
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: wfdb>=4.1.0
+Requires-Dist: xmltodict>=0.13.0
+Requires-Dist: numpy>=1.22
+Dynamic: license-file
+
+# muse2wfdb
+### Convert GE MUSE XML ECG exports to WFDB format
+
+`muse2wfdb` is a lightweight Python library that converts ECG data exported from the GE MUSE Resting ECG System (in XML format) into WFDB (WaveForm DataBase) format, compatible with PhysioNet tools and the wfdb Python package.
+
+## Installation
+
+Install directly from PyPI:
+
+```bash
+pip install muse2wfdb
+```
+
+Or install locally from source:
+
+```bash
+git clone https://github.com/nagyl1999/muse2wfdb.git
+cd muse2wfdb
+pip install .
+```
+
+## Usage
+
+An example is provided in the `examples` folder.
+
+```python
+from muse2wfdb.converter import muse_to_wfdb
+import wfdb
+
+muse_export = "examples/anonim_pac_xml_export.txt"
+wfdb_filename = "patient001_ecg"
+
+muse_to_wfdb(muse_export, wfdb_filename)
+
+record = wfdb.rdrecord(wfdb_filename)
+annotation = wfdb.rdann(wfdb_filename, 'atr')
+
+wfdb.plot_wfdb(record=record, annotation=annotation, title="MUSE exported ECG")
+```

muse2wfdb-1.0.0/muse2wfdb.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+LICENSE
+README.md
+pyproject.toml
+muse2wfdb/converter.py
+muse2wfdb.egg-info/PKG-INFO
+muse2wfdb.egg-info/SOURCES.txt
+muse2wfdb.egg-info/dependency_links.txt
+muse2wfdb.egg-info/requires.txt
+muse2wfdb.egg-info/top_level.txt

muse2wfdb-1.0.0/muse2wfdb.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

muse2wfdb-1.0.0/muse2wfdb.egg-info/requires.txt

@@ -0,0 +1,3 @@
+wfdb>=4.1.0
+xmltodict>=0.13.0
+numpy>=1.22

muse2wfdb-1.0.0/muse2wfdb.egg-info/top_level.txt

@@ -0,0 +1 @@
+muse2wfdb

muse2wfdb-1.0.0/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "muse2wfdb"
+version = "1.0.0"
+description = "Convert GE MUSE ECG XML exports to WFDB format"
+authors = [{ name = "Levente Nagy" }]
+license = { text = "MIT" }
+readme = "README.md"
+requires-python = ">=3.8"
+dependencies = [
+    "wfdb>=4.1.0",
+    "xmltodict>=0.13.0",
+    "numpy>=1.22"
+]
+
+[project.urls]
+Homepage = "https://github.com/nagyl1999/muse2wfdb"
+Issues = "https://github.com/nagyl1999/muse2wfdb/issues"

muse2wfdb-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+