48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
|
|
from backend.data_types import DataField
|
|
|
|
|
|
extensions = frozenset({".sxm"})
|
|
calibrated = True
|
|
|
|
|
|
def load(path: Path) -> list[DataField]:
|
|
import nanonispy as nap
|
|
sxm = nap.read.Scan(str(path))
|
|
signals = sxm.signals
|
|
if not signals:
|
|
raise ValueError(f"No signals found in {path.name}")
|
|
|
|
scan_range = sxm.header.get("scan_range", [1e-6, 1e-6])
|
|
|
|
fields = []
|
|
for sig in signals.values():
|
|
data = sig.get("forward", list(sig.values())[0])
|
|
data = np.asarray(data, dtype=np.float64)
|
|
if data.ndim != 2:
|
|
data = data.reshape(data.shape[-2], data.shape[-1])
|
|
fields.append(DataField(
|
|
data=data,
|
|
xreal=float(scan_range[0]),
|
|
yreal=float(scan_range[1]),
|
|
si_unit_xy="m",
|
|
si_unit_z="m",
|
|
))
|
|
return fields
|
|
|
|
|
|
def channel_names(path: Path) -> list[str]:
|
|
import nanonispy as nap
|
|
try:
|
|
sxm = nap.read.Scan(str(path))
|
|
if sxm.signals:
|
|
return list(sxm.signals.keys())
|
|
except Exception:
|
|
pass
|
|
return []
|