toucan-plot 0.2.0__py3-none-any.whl

toucan_plot/utils/__init__.py

@@ -0,0 +1,14 @@
+from . import styles
+
+from .loaders import (
+    csv_load_worker,
+    can_log_load_worker,
+    mf4_load_worker,
+)
+
+__all__ = [
+    "styles",
+    "csv_load_worker",
+    "can_log_load_worker",
+    "mf4_load_worker",
+]

toucan_plot/utils/loaders.py

@@ -0,0 +1,219 @@
+import os
+import csv
+import numpy as np
+import can
+import cantools
+from asammdf import MDF
+
+def detect_delimiter(path):
+    with open(path, 'r') as f:
+        header = f.readline()
+        if header.find(";") != -1:
+            return ";"
+        if header.find(",") != -1:
+            return ","
+    return ";"
+
+
+def csv_load_worker(path, queue):
+    """Multiprocessing worker for CSV loading. Sends progress via queue."""
+    try:
+        queue.put(('progress', 5, 'Reading file...'))
+        with open(path, 'r', newline='') as f:
+            reader = csv.reader(f, delimiter=detect_delimiter(path))
+            headers = next(reader)
+            headers = [h.strip() for h in headers]
+            rows = list(reader)
+
+        if not headers or not rows:
+            queue.put(('error', 'File is empty or has no data rows.'))
+            return
+
+        queue.put(('progress', 20, 'Building columns...'))
+        num_cols = len(headers)
+        columns = {}
+        for col_idx in range(num_cols):
+            values = []
+            for row in rows:
+                if col_idx < len(row):
+                    try:
+                        values.append(float(row[col_idx]))
+                    except ValueError:
+                        values.append(float('nan'))
+                else:
+                    values.append(float('nan'))
+            columns[headers[col_idx]] = np.array(values)
+            pct = 20 + int(70 * (col_idx + 1) / num_cols)
+            queue.put(('progress', pct, f'Processing column {col_idx + 1}/{num_cols}...'))
+
+        queue.put(('progress', 95, 'Finalizing...'))
+        x_col = None
+        for candidate in ('Time', 'timestamp'):
+            if candidate in columns:
+                x_col = candidate
+                break
+        if x_col is None:
+            x_col = headers[0]
+
+        queue.put(('result', (x_col, columns)))
+    except Exception as e:
+        queue.put(('error', str(e)))
+
+
+def _can_reader(path):
+    """Return the appropriate python-can reader for the file extension."""
+    ext = os.path.splitext(path)[1].lower()
+    if ext == '.trc':
+        return can.TRCReader(path)
+    if ext == '.asc':
+        return can.ASCReader(path)
+    return can.BLFReader(path)
+
+
+def can_log_load_worker(blf_path, dbc_paths, queue):
+    """Multiprocessing worker for CAN log loading (BLF/TRC). Sends progress via queue."""
+    try:
+        queue.put(('progress', 5, 'Loading DBC files...'))
+        db = cantools.database.Database()
+        for dbc_path in dbc_paths:
+            db.add_dbc_file(dbc_path)
+
+        queue.put(('progress', 10, 'Reading CAN messages...'))
+        signal_data = {}
+        msg_count = 0
+
+        with _can_reader(blf_path) as reader:
+            for msg in reader:
+                try:
+                    db_msg = db.get_message_by_frame_id(msg.arbitration_id)
+                except KeyError:
+                    msg_count += 1
+                    continue
+                try:
+                    decoded = db_msg.decode(msg.data, decode_choices=False)
+                except Exception:
+                    msg_count += 1
+                    continue
+                t = msg.timestamp
+                for signal_name, value in decoded.items():
+                    full_name = f"{db_msg.name}.{signal_name}"
+                    if full_name not in signal_data:
+                        signal_data[full_name] = []
+                    signal_data[full_name].append((t, float(value)))
+                msg_count += 1
+                if msg_count % 5000 == 0:
+                    queue.put(('progress', -1, f'Reading messages... ({msg_count} processed)'))
+
+        if not signal_data:
+            queue.put(('error', 'No CAN messages could be decoded with the provided DBC file(s).'))
+            return
+
+        queue.put(('progress', 60, f'Building time index ({msg_count} messages)...'))
+        all_timestamps = set()
+        for pairs in signal_data.values():
+            for t, _ in pairs:
+                all_timestamps.add(t)
+        all_timestamps = sorted(all_timestamps)
+
+        t0 = all_timestamps[0] if all_timestamps else 0.0
+        x_array = np.array([t - t0 for t in all_timestamps])
+        time_index = {t: i for i, t in enumerate(all_timestamps)}
+
+        queue.put(('progress', 70, 'Building signal arrays...'))
+        all_columns = {'Time': x_array}
+        sorted_names = sorted(signal_data.keys())
+        total_signals = len(sorted_names)
+        for sig_idx, full_name in enumerate(sorted_names):
+            pairs = signal_data[full_name]
+            data = np.full(len(x_array), np.nan)
+            for t, val in pairs:
+                data[time_index[t]] = val
+            # Forward-fill NaNs with last known value
+            last = np.nan
+            for i in range(len(data)):
+                if np.isnan(data[i]):
+                    data[i] = last
+                else:
+                    last = data[i]
+            all_columns[full_name] = data
+            pct = 70 + int(25 * (sig_idx + 1) / total_signals)
+            if (sig_idx + 1) % 10 == 0 or sig_idx == total_signals - 1:
+                queue.put(('progress', pct, f'Processing signal {sig_idx + 1}/{total_signals}...'))
+
+        queue.put(('result', ('Time', all_columns)))
+    except Exception as e:
+        queue.put(('error', str(e)))
+
+
+def mf4_load_worker(path, queue):
+    """Multiprocessing worker for MF4 loading via asammdf. Sends progress via queue."""
+    try:
+        queue.put(('progress', 5, 'Opening MF4 file...'))
+        mdf = MDF(path)
+
+        queue.put(('progress', 15, 'Extracting channels...'))
+        # Collect all non-empty numeric channels
+        channel_names = []
+        for i, group in enumerate(mdf.groups):
+            for j, channel in enumerate(group.channels):
+                name = channel.name
+                if name and name != 't':
+                    channel_names.append((name, i, j))
+
+        if not channel_names:
+            queue.put(('error', 'No channels found in MF4 file.'))
+            mdf.close()
+            return
+
+        queue.put(('progress', 25, f'Processing {len(channel_names)} channels...'))
+        all_columns = {}
+        x_col = None
+        total = len(channel_names)
+
+        for idx, (name, group_idx, ch_idx) in enumerate(channel_names):
+            try:
+                sig = mdf.get(name, group=group_idx, index=ch_idx)
+            except Exception:
+                continue
+
+            # Skip non-numeric or empty signals
+            if sig.samples.size == 0:
+                continue
+            if sig.samples.ndim != 1:
+                continue
+            if not np.issubdtype(sig.samples.dtype, np.number):
+                continue
+
+            timestamps = sig.timestamps
+            try:
+                samples = sig.samples.astype(float)
+            except (ValueError, TypeError):
+                continue
+
+            # Use the first valid channel's timestamps as x axis
+            if x_col is None:
+                x_col = 'Time'
+                all_columns['Time'] = timestamps
+
+            # If this channel's timestamps match the common time axis, use directly
+            common_t = all_columns['Time']
+            if len(timestamps) == len(common_t) and np.allclose(timestamps, common_t):
+                all_columns[name] = samples
+            else:
+                # Interpolate onto the common time axis
+                all_columns[name] = np.interp(common_t, timestamps, samples)
+
+            pct = 25 + int(70 * (idx + 1) / total)
+            if (idx + 1) % 20 == 0 or idx == total - 1:
+                queue.put(('progress', pct, f'Processing channel {idx + 1}/{total}...'))
+
+        mdf.close()
+
+        if x_col is None or len(all_columns) <= 1:
+            queue.put(('error', 'No numeric channels could be extracted from the MF4 file.'))
+            return
+
+        queue.put(('progress', 98, 'Finalizing...'))
+        queue.put(('result', (x_col, all_columns)))
+    except Exception as e:
+        queue.put(('error', str(e)))

toucan_plot/utils/styles.py

@@ -0,0 +1,55 @@
+import matplotlib.pyplot as plt
+from cycler import cycler
+import scienceplots
+
+def set_default_style() -> None:
+    """Configure matplotlib for screen viewing."""
+    plt.style.use(['default','no-latex'])
+    plt.rcParams.update(
+        {
+            "axes.prop_cycle": cycler("color", ["blue", "orange", "green", "red", "purple", "brown"]),
+            "lines.linewidth": 1,
+            "axes.grid": True,
+            "text.usetex": False,
+            "axes3d.grid": True,
+            "figure.constrained_layout.use": True,
+        }
+    )
+
+def set_style1_style() -> None:
+    """Configure matplotlib for screen viewing."""
+    plt.style.use(['default','no-latex'])
+    plt.rcParams.update(
+        {
+            "axes.prop_cycle": cycler("color", ["b", "r", "m", "c"]),
+            "lines.linewidth": 1,
+            "axes.grid": True,
+            "text.usetex": False,
+            "axes3d.grid": True,
+            "figure.constrained_layout.use": True,
+        }
+    )
+
+def set_style2_style() -> None:
+    """Configure matplotlib for Style 2."""
+    plt.style.use(['default','no-latex'])
+    plt.rcParams.update(
+        {
+            "axes.prop_cycle": cycler("color", ["#2962FF", "r", "#AA00FF", "#00B8D4", "#FF6D00", "#00C853"]),
+            "lines.linewidth": 1,
+            "axes.grid": True,
+            "axes3d.grid": True,
+            "figure.constrained_layout.use": True,
+        }
+    )
+
+def set_ieee_style() -> None:
+    """Configure matplotlib for IEEE style."""
+    plt.style.use(['science','ieee','grid','no-latex'])
+    plt.rcParams.update(
+        {
+            "axes3d.grid": True,
+            "figure.dpi": 150,
+            "figure.constrained_layout.use": True,
+        }
+    )

toucan_plot-0.2.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: toucan-plot
+Version: 0.2.0
+Summary: A versatile PyQt6 and Matplotlib-based plotting tool for CSV and SMV data files.
+License-Expression: MIT
+Requires-Python: >=3.11
+License-File: LICENSE
+Requires-Dist: pyqt6>=6.0
+Requires-Dist: matplotlib>=3.0
+Requires-Dist: numpy>=1.20
+Requires-Dist: pyqtdarktheme>=2.1.0
+Requires-Dist: scienceplots>=2.2.1
+Requires-Dist: python-can>=4.0
+Requires-Dist: cantools>=39.0
+Requires-Dist: asammdf>=7.0
+Dynamic: license-file

toucan_plot-0.2.0.dist-info/RECORD

@@ -0,0 +1,34 @@
+toucan_plot/__main__.py,sha256=GyGkAx0WXg25bWW-FqX3yVHGkiOl9wjvrXPZGSpOHA4,75
+toucan_plot/main.py,sha256=CnuM6IczvVnTuRaBHnyh8fZRbL5xp-snRi2hbl1ONNw,93245
+toucan_plot/assets/adjustments-dark.svg,sha256=2pJd9UEOONi8LOvb052rBvAKZdZ85ri9Xo7H4OeW0dw,568
+toucan_plot/assets/adjustments-light.svg,sha256=BvbEEEKHzSAbys96JMZFt7AuNZ20qOjEVoEeJP4WJKU,568
+toucan_plot/assets/arrow-autofit-height-dark.svg,sha256=Osd1UNJ5mzrw1zmBxE4fnqzWSIjcoSYsE1Cv9_XvXlw,468
+toucan_plot/assets/arrow-autofit-height-light.svg,sha256=a8K9kwJuoe7n5dx83PPtVG23RZllfbek2n8LEml2RiY,468
+toucan_plot/assets/arrow-narrow-left-dark.svg,sha256=Gk4YxEBOyGO8Odpq_GszQwVPd9NfheYeJF8JDhtuLXE,378
+toucan_plot/assets/arrow-narrow-left-light.svg,sha256=ipQ2oJuQqLyOnDBoJVIRuzplrJKzJe1mxy3mbYuLeEc,378
+toucan_plot/assets/arrow-narrow-right-dark.svg,sha256=NhCv2etMYOulRsmSB-ldzKnhEt2MMtOcdslczOuxqxs,380
+toucan_plot/assets/arrow-narrow-right-light.svg,sha256=XYMe0SIpLwsEYDRMkyj8jS__2DqoxcHkKKXfTLpf_ic,380
+toucan_plot/assets/device-floppy-dark.svg,sha256=e5r2oT-FIIVyHHWJ_bwYkp_Yi2ACuq1u9Zizb9vkEDM,487
+toucan_plot/assets/device-floppy-light.svg,sha256=Shl-N6sAWhE5mO4ZDO4HgGMF3Lf7Ntx52DqcF0siE4E,487
+toucan_plot/assets/hand-stop-dark.svg,sha256=w3h2ZtVG2hMCN0T0tA642o0CeV1xytw46KFG_keH1zo,652
+toucan_plot/assets/hand-stop-light.svg,sha256=V7iRtr3lL1yg1Wet4EanYnWqcb5lV0P_E-M2eEi5hMA,652
+toucan_plot/assets/home-dark.svg,sha256=JMmXu6C1eMJy9l75LCqlGZ9rj7NOUxKZ4EnbSGbcqZ8,443
+toucan_plot/assets/home-light.svg,sha256=9oi05ncSwTN5jraYkcBclxHJOOn3FAVFmxmN1phnrhs,443
+toucan_plot/assets/layout-grid-add-dark.svg,sha256=_KFoic-KvgFpPkyzxW7TdpDGZBombWuhV8tKsJQJxsg,606
+toucan_plot/assets/layout-grid-add-light.svg,sha256=qKu8zLUsBrEdThqVBgAXjGxhf3ABed0hnh-0GJr54yU,606
+toucan_plot/assets/math-function-dark.svg,sha256=ql8lWpUY_5PFgRj1Kq8i7rFB5KRT8Xqm34a_5bVzZNg,443
+toucan_plot/assets/math-function-light.svg,sha256=A1hcYzYJxjqfH3_eht1ikflduFrAuamOr4WcWwG5a6o,443
+toucan_plot/assets/ruler-measure-dark.svg,sha256=Ujqsex7bqHRXiU1DWCxVVHf-iiNz7QY123TpSFAioT4,629
+toucan_plot/assets/ruler-measure-light.svg,sha256=Uz7AkVpifwUVJAWrA3iNb2iOIBtqNUDbHkF7GK2LgVE,629
+toucan_plot/assets/zoom-dark.svg,sha256=c4uZt1NIb3lwoLgfoqH3Zfj018vDHC57x_MEzOrTYU0,371
+toucan_plot/assets/zoom-light.svg,sha256=RolfeW-FvrmVA0ERV7iNQ7_RQl7KaOyOY-ZEbnCjfpg,371
+toucan_plot/assets/ico/toucan-plot.ico,sha256=m5ZVE4A7G0MkZU9dIRrT-HVrjiT7jsFgxi1jaUetx_4,751300
+toucan_plot/utils/__init__.py,sha256=MZ1gO96ov04y4QHheZG3tCzHgslYNEOG2qeY7as-uv0,228
+toucan_plot/utils/loaders.py,sha256=mRrM0arJQ7IsY9y4y22A2FtplGBrh2_marZtHmAwqWM,8214
+toucan_plot/utils/styles.py,sha256=liC5yb_7uR0h5EoFh8P_wFui62WBa1XH-wWssO6uu_w,1754
+toucan_plot-0.2.0.dist-info/licenses/LICENSE,sha256=ZvZfVEqq1wmh-7AWxXbLxI2z2iKLpt85hdexHL_B3RM,1106
+toucan_plot-0.2.0.dist-info/METADATA,sha256=1mLoRh6pclTiA8g-3E7h02hSVndEhvFZaaWXf7BOB-0,497
+toucan_plot-0.2.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+toucan_plot-0.2.0.dist-info/entry_points.txt,sha256=FaV5CRr3WQ6i68t72m8bfvIYWBxrv0f_XrZPkjDkqPY,54
+toucan_plot-0.2.0.dist-info/top_level.txt,sha256=4zDACL7KXhtQYEFlRrzXMpTwQiG-sTJOR_mLlVXLnkc,12
+toucan_plot-0.2.0.dist-info/RECORD,,

toucan_plot-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

toucan_plot-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+toucan-plot = toucan_plot.main:main

toucan_plot-0.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Claudio Emanoel Barbosa Lima
+
+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.

toucan_plot-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+toucan_plot