matterviz 0.3.7 → 0.4.1

package/dist/trajectory/TrajectoryInfoPane.svelte.d.ts

@@ -1,3 +1,4 @@
+import type { PaneProps, PaneToggleProps } from '../overlays';
 import DraggablePane from '../overlays/DraggablePane.svelte';
 import type { ComponentProps } from 'svelte';
 import type { TrajectoryType } from './index';
@@ -9,8 +10,8 @@ type $$ComponentProps = Omit<ComponentProps<typeof DraggablePane>, `children`> &
     file_size?: number | null;
     file_object?: File | null;
     pane_open?: boolean;
-    toggle_props?: ComponentProps<typeof DraggablePane>[`toggle_props`];
-    pane_props?: ComponentProps<typeof DraggablePane>[`pane_props`];
+    toggle_props?: PaneToggleProps;
+    pane_props?: PaneProps;
 };
 declare const TrajectoryInfoPane: import("svelte").Component<$$ComponentProps, {}, "pane_open">;
 type TrajectoryInfoPane = ReturnType<typeof TrajectoryInfoPane>;

package/dist/trajectory/constants.js

@@ -1,7 +1,11 @@
 // Constants for trajectory parsing and large file handling
+import { DEFAULTS } from '../settings';
 export const MAX_SAFE_STRING_LENGTH = 0x1fffffe8 * 0.5; // 50% of JS max string length as safety
 export const MAX_METADATA_SIZE = 50 * 1024 * 1024; // 50MB limit for metadata
 export const LARGE_FILE_THRESHOLD = 400 * 1024 * 1024; // 400MB
 export const INDEX_SAMPLE_RATE = 100; // Default sample rate for frame indexing
-export const MAX_BIN_FILE_SIZE = 100 * 1024 * 1024; // 100MB default for ArrayBuffer files
-export const MAX_TEXT_FILE_SIZE = 50 * 1024 * 1024; // 50MB default for string files
+// Fallback thresholds for component usage without loading_options, derived from the
+// settings schema so settings-driven contexts (e.g. the VSCode extension) and direct
+// component use agree on when large-file/indexed loading kicks in.
+export const MAX_BIN_FILE_SIZE = DEFAULTS.trajectory.bin_file_threshold; // 50MB
+export const MAX_TEXT_FILE_SIZE = DEFAULTS.trajectory.text_file_threshold; // 25MB

package/dist/trajectory/extract.js

@@ -1,5 +1,6 @@
 // Data extraction functions for trajectory analysis and plotting
 import { get_density } from '../structure/index';
+import { calc_force_stats, copy_numeric_fields } from './helpers';
 // Common data extractor that extracts energy and structural properties
 export const energy_data_extractor = (frame) => {
     const data = {
@@ -7,18 +8,13 @@ export const energy_data_extractor = (frame) => {
     };
     if (frame.metadata) {
         // Extract energy-related properties
-        const energy_fields = [
+        copy_numeric_fields(data, frame.metadata, [
             `energy`,
             `energy_per_atom`,
             `potential_energy`,
             `kinetic_energy`,
             `total_energy`,
-        ];
-        for (const field of energy_fields) {
-            if (field in frame.metadata && typeof frame.metadata[field] === `number`) {
-                data[field] = frame.metadata[field];
-            }
-        }
+        ]);
     }
     return data;
 };
@@ -30,13 +26,8 @@ export const force_stress_data_extractor = (frame) => {
     if (frame.metadata) {
         // Calculate force properties from forces array if available (preferred)
         if (frame.metadata.forces && Array.isArray(frame.metadata.forces)) {
-            const forces = frame.metadata.forces;
-            if (forces.length > 0) {
-                const force_magnitudes = forces.map((force) => Math.hypot(...force));
-                data.force_max = Math.max(...force_magnitudes);
-                // Calculate RMS (root mean square) of force magnitudes
-                data.force_norm = Math.sqrt(force_magnitudes.reduce((sum, f) => sum + f ** 2, 0) / force_magnitudes.length);
-            }
+            // Object.assign ignores the null calc_force_stats returns for empty forces
+            Object.assign(data, calc_force_stats(frame.metadata.forces));
         }
         else {
             // Fallback to metadata values if forces array not available
@@ -52,12 +43,12 @@ export const force_stress_data_extractor = (frame) => {
             }
         }
         // Extract other stress and pressure properties (no duplicates expected)
-        const other_stress_fields = [`stress_max`, `stress_frobenius`, `stress_trace`, `pressure`];
-        for (const field of other_stress_fields) {
-            if (field in frame.metadata && typeof frame.metadata[field] === `number`) {
-                data[field] = frame.metadata[field];
-            }
-        }
+        copy_numeric_fields(data, frame.metadata, [
+            `stress_max`,
+            `stress_frobenius`,
+            `stress_trace`,
+            `pressure`,
+        ]);
     }
     return data;
 };
@@ -79,31 +70,20 @@ export const structural_data_extractor = (frame) => {
     }
     if (frame.metadata) {
         // Extract other structural properties, avoiding volume duplicate
-        const structural_fields = [`temperature`];
-        for (const field of structural_fields) {
-            if (field in frame.metadata && typeof frame.metadata[field] === `number`)
-                data[field] = frame.metadata[field];
-        }
-        // Handle density separately - prefer metadata, but calculate if not available
-        if (frame.metadata.density && typeof frame.metadata.density === `number`) {
+        copy_numeric_fields(data, frame.metadata, [`temperature`]);
+        // Prefer metadata density (fall back to calculating from structure below).
+        // Finite-number check (not truthiness) so a legitimate density of 0 is kept.
+        if (typeof frame.metadata.density === `number` &&
+            Number.isFinite(frame.metadata.density)) {
             data.density = frame.metadata.density;
         }
-        else if (`lattice` in frame.structure) {
-            try {
-                data.density = get_density(frame.structure);
-            }
-            catch (error) {
-                console.warn(`Failed to calculate density for frame ${frame.step}:`, error);
-            }
-        }
         // Only use metadata volume if lattice volume is not available
         if (!data.volume && frame.metadata.volume && typeof frame.metadata.volume === `number`) {
             data.volume = frame.metadata.volume;
         }
         // Note: pressure is handled by force_stress_data_extractor to avoid duplication
     }
-    else if (`lattice` in frame.structure) {
-        // Calculate density even when no metadata is available
+    if (data.density === undefined && `lattice` in frame.structure) {
         try {
             data.density = get_density(frame.structure);
         }

package/dist/trajectory/format-detect.d.ts

@@ -1,4 +1,4 @@
-export declare function strip_compression_extensions(filename: string): string;
+export declare function ext_hint(filename: string | undefined, ext_regex: RegExp): boolean | null;
 export declare const FORMAT_PATTERNS: {
     readonly ase: (data: unknown, filename?: string) => boolean;
     readonly hdf5: (data: unknown, filename?: string) => boolean;

package/dist/trajectory/format-detect.js

@@ -1,33 +1,43 @@
 // Format detection for trajectory files
-import { COMPRESSION_EXTENSIONS_REGEX, CONFIG_DIRS_REGEX, MD_SIM_EXCLUDE_REGEX, TRAJ_EXTENSIONS_REGEX, TRAJ_FALLBACK_EXTENSIONS_REGEX, TRAJ_KEYWORDS_SIMPLE_REGEX, XDATCAR_REGEX, } from '../constants';
+import { CONFIG_DIRS_REGEX, MD_SIM_EXCLUDE_REGEX, TRAJ_EXTENSIONS_REGEX, TRAJ_FALLBACK_EXTENSIONS_REGEX, TRAJ_KEYWORDS_SIMPLE_REGEX, XDATCAR_REGEX, } from '../constants';
+import { strip_compression_extensions } from '../io';
 import { count_xyz_frames } from './helpers';
-export function strip_compression_extensions(filename) {
-    let base_name = filename.toLowerCase();
-    while (COMPRESSION_EXTENSIONS_REGEX.test(base_name)) {
-        base_name = base_name.replace(COMPRESSION_EXTENSIONS_REGEX, ``);
-    }
-    return base_name;
+// Extensions that explicitly identify a format — when present, format detection trusts
+// the extension instead of sniffing content
+const KNOWN_FORMAT_EXT_REGEX = /\.(xyz|extxyz|traj|h5|hdf5|lammpstrj|json|cif|poscar|vasp|yaml|yml|xml|csv)$/;
+// Classify the filename hint for a format whose extensions match ext_regex:
+// true = filename matches, false = filename names a different known format,
+// null = no usable hint (missing filename or unrecognized extension, e.g. the UUID
+// basenames of blob: object URLs) — callers should fall back to content detection
+export function ext_hint(filename, ext_regex) {
+    if (!filename)
+        return null;
+    const base = strip_compression_extensions(filename);
+    if (ext_regex.test(base))
+        return true;
+    return KNOWN_FORMAT_EXT_REGEX.test(base) ? false : null;
 }
-// Unified format detection
+// Unified format detection. Each pattern trusts a matching file extension when present
+// but falls back to content/magic-byte detection when the filename gives no hint
+// (e.g. blob: object URLs, extensionless API endpoints).
 export const FORMAT_PATTERNS = {
     ase: (data, filename) => {
-        const base_name = filename ? strip_compression_extensions(filename) : undefined;
-        if (!base_name?.endsWith(`.traj`) || !(data instanceof ArrayBuffer)) {
+        if (ext_hint(filename, /\.traj$/) === false || !(data instanceof ArrayBuffer)) {
             return false;
         }
         const view = new Uint8Array(data.slice(0, 24));
         return [0x2d, 0x20, 0x6f, 0x66, 0x20, 0x55, 0x6c, 0x6d].every((byte, idx) => view[idx] === byte);
     },
     hdf5: (data, filename) => {
-        const base_name = filename ? strip_compression_extensions(filename) : undefined;
-        const has_ext = base_name?.match(/\.(h5|hdf5)$/);
-        if (!has_ext || !(data instanceof ArrayBuffer) || data.byteLength < 8)
+        if (ext_hint(filename, /\.(h5|hdf5)$/) === false)
+            return false;
+        if (!(data instanceof ArrayBuffer) || data.byteLength < 8)
             return false;
         const signature = new Uint8Array(data.slice(0, 8));
-        return [0x89, 0x48, 0x44, 0x46, 0x0d, 0x0a, 0x1a, 0x0a].every((b, idx) => signature[idx] === b);
+        return [0x89, 0x48, 0x44, 0x46, 0x0d, 0x0a, 0x1a, 0x0a].every((byte, idx) => signature[idx] === byte);
     },
     vasp: (data, filename) => {
-        const basename = filename?.toLowerCase().split(`/`).pop() || ``;
+        const basename = filename?.toLowerCase().split(`/`).pop() ?? ``;
         if (basename === `xdatcar` || basename.startsWith(`xdatcar`))
             return true;
         const lines = data.trim().split(/\r?\n/);
@@ -37,14 +47,12 @@ export const FORMAT_PATTERNS = {
             lines.slice(2, 5).every((line) => line.trim().split(/\s+/).length === 3));
     },
     xyz_multi: (data, filename) => {
-        const base = filename ? strip_compression_extensions(filename) : ``;
-        if (!/\.(xyz|extxyz)$/.test(base))
+        if (ext_hint(filename, /\.(xyz|extxyz)$/) === false)
             return false;
         return count_xyz_frames(data) >= 2;
     },
     lammpstrj: (data, filename) => {
-        const base = filename ? strip_compression_extensions(filename) : ``;
-        if (!base.endsWith(`.lammpstrj`))
+        if (ext_hint(filename, /\.lammpstrj$/) === false)
             return false;
         return data.includes(`ITEM: TIMESTEP`) && data.includes(`ITEM: ATOMS`);
     },

package/dist/trajectory/frame-reader.d.ts

@@ -12,6 +12,5 @@ export declare class TrajFrameReader implements FrameLoader {
     }, on_progress?: (progress: ParseProgress) => void): Promise<TrajectoryMetadata[]>;
     private load_xyz_frame;
     private load_ase_frame;
-    private parse_xyz_metadata;
     private parse_ase_metadata;
 }

package/dist/trajectory/frame-reader.js

@@ -1,7 +1,16 @@
+// Unified frame loader for XYZ and ASE trajectories (large file indexing)
 import * as math from '../math';
 import { MAX_METADATA_SIZE } from './constants';
-import { coerce_element_symbol, convert_atomic_numbers, count_xyz_frames, create_trajectory_frame, read_ndarray_from_view, validate_3x3_matrix, } from './helpers';
-import { strip_compression_extensions } from './format-detect';
+import { copy_numeric_fields, count_xyz_frames, iter_xyz_frames, validate_3x3_matrix, } from './helpers';
+import { strip_compression_extensions } from '../io';
+import { decode_ase_frame, read_ase_header } from './parse/ase';
+import { build_xyz_frame, parse_xyz_comment_metadata } from './parse/xyz';
+// Restrict frame metadata to the requested property keys (no-op when unset)
+const filter_properties = (metadata, properties) => {
+    if (!properties)
+        return;
+    metadata.properties = Object.fromEntries(Object.entries(metadata.properties).filter(([key]) => properties.includes(key)));
+};
 export class TrajFrameReader {
     format;
     global_numbers;
@@ -17,8 +26,7 @@ export class TrajFrameReader {
         }
         if (!(data instanceof ArrayBuffer))
             throw new Error(`ASE loader requires binary data`);
-        const view = new DataView(data);
-        return Number(view.getBigInt64(32, true));
+        return read_ase_header(new DataView(data)).n_items;
     }
     async build_frame_index(data, sample_rate, on_progress) {
         const total_frames = await this.get_total_frames(data);
@@ -29,35 +37,25 @@ export class TrajFrameReader {
             const encoder = new TextEncoder();
             const newline_sequence = data_str.includes(`\r\n`) ? `\r\n` : `\n`;
             const newline_byte_len = encoder.encode(newline_sequence).length;
-            let [current_frame, line_idx, byte_offset] = [0, 0, 0];
-            while (line_idx < lines.length && current_frame < total_frames) {
-                if (!lines[line_idx]?.trim()) {
-                    byte_offset += encoder.encode(lines[line_idx]).length + newline_byte_len;
-                    line_idx++;
-                    continue;
-                }
-                const num_atoms = parseInt(lines[line_idx].trim(), 10);
-                if (isNaN(num_atoms) || num_atoms <= 0 || line_idx + num_atoms + 1 >= lines.length) {
-                    byte_offset += encoder.encode(lines[line_idx]).length + newline_byte_len;
-                    line_idx++;
-                    continue;
-                }
+            const line_bytes = (idx) => encoder.encode(lines[idx]).length + newline_byte_len;
+            // cursor = next line whose bytes haven't been added to byte_offset yet
+            let [current_frame, cursor, byte_offset] = [0, 0, 0];
+            for (const { start, num_atoms } of iter_xyz_frames(lines)) {
+                if (current_frame >= total_frames)
+                    break;
+                // Accumulate bytes of blank/invalid lines skipped before this frame
+                for (; cursor < start; cursor++)
+                    byte_offset += line_bytes(cursor);
+                let frame_size = 0;
+                for (; cursor < start + num_atoms + 2; cursor++)
+                    frame_size += line_bytes(cursor);
                 if (current_frame % sample_rate === 0) {
                     frame_index.push({
                         frame_number: current_frame,
                         byte_offset,
-                        estimated_size: 0,
+                        estimated_size: frame_size,
                     });
                 }
-                const frame_start = line_idx;
-                line_idx += 2 + num_atoms;
-                let frame_size = 0;
-                for (let idx = frame_start; idx < line_idx; idx++) {
-                    frame_size += encoder.encode(lines[idx]).length + newline_byte_len;
-                }
-                if (current_frame % sample_rate === 0) {
-                    frame_index[frame_index.length - 1].estimated_size = frame_size;
-                }
                 byte_offset += frame_size;
                 current_frame++;
                 if (on_progress && current_frame % 1000 === 0) {
@@ -71,7 +69,7 @@ export class TrajFrameReader {
         }
         else {
             const view = new DataView(data);
-            const offsets_pos = Number(view.getBigInt64(40, true));
+            const { offsets_pos } = read_ase_header(view);
             for (let idx = 0; idx < total_frames; idx += sample_rate) {
                 const frame_offset = Number(view.getBigInt64(offsets_pos + idx * 8, true));
                 frame_index.push({
@@ -109,34 +107,28 @@ export class TrajFrameReader {
         const total_frames = await this.get_total_frames(data);
         if (this.format === `xyz`) {
             const lines = data.trim().split(/\r?\n/);
-            let [current_frame, line_idx] = [0, 0];
-            while (line_idx < lines.length && current_frame < total_frames) {
-                if (!lines[line_idx]?.trim()) {
-                    line_idx++;
-                    continue;
-                }
-                const num_atoms = parseInt(lines[line_idx].trim(), 10);
-                if (isNaN(num_atoms) || num_atoms <= 0 || line_idx + num_atoms + 1 >= lines.length) {
-                    line_idx++;
-                    continue;
-                }
+            let current_frame = 0;
+            for (const { start, comment } of iter_xyz_frames(lines)) {
+                if (current_frame >= total_frames)
+                    break;
                 if (current_frame % sample_rate === 0) {
-                    const comment = lines[line_idx + 1] || ``;
                     let frame_metadata = null;
                     try {
-                        frame_metadata = this.parse_xyz_metadata(comment, current_frame);
+                        const { step, properties: props } = parse_xyz_comment_metadata(comment);
+                        frame_metadata = {
+                            frame_number: current_frame,
+                            step: step ?? current_frame,
+                            properties: props,
+                        };
                     }
                     catch (error) {
-                        console.warn(`Failed to parse XYZ metadata for frame ${current_frame} at line ${line_idx + 1}:`, error);
-                    }
-                    if (frame_metadata && properties) {
-                        const filtered = Object.fromEntries(Object.entries(frame_metadata.properties).filter(([key]) => properties.includes(key)));
-                        frame_metadata.properties = filtered;
+                        console.warn(`Failed to parse XYZ metadata for frame ${current_frame} at line ${start + 1}:`, error);
                     }
-                    if (frame_metadata)
+                    if (frame_metadata) {
+                        filter_properties(frame_metadata, properties);
                         metadata_list.push(frame_metadata);
+                    }
                 }
-                line_idx += 2 + num_atoms;
                 current_frame++;
                 if (on_progress && current_frame % 5000 === 0) {
                     on_progress({
@@ -149,8 +141,7 @@ export class TrajFrameReader {
         }
         else if (this.format === `ase`) {
             const view = new DataView(data);
-            const n_items = Number(view.getBigInt64(32, true));
-            const offsets_pos = Number(view.getBigInt64(40, true));
+            const { n_items, offsets_pos } = read_ase_header(view);
             for (let idx = 0; idx < n_items; idx += sample_rate) {
                 try {
                     const frame_offset = Number(view.getBigInt64(offsets_pos + idx * 8, true));
@@ -161,10 +152,7 @@ export class TrajFrameReader {
                     }
                     const frame_data = JSON.parse(new TextDecoder().decode(new Uint8Array(data, frame_offset + 8, json_length)));
                     const frame_metadata = this.parse_ase_metadata(frame_data, idx);
-                    if (properties) {
-                        const filtered = Object.fromEntries(Object.entries(frame_metadata.properties).filter(([key]) => properties.includes(key)));
-                        frame_metadata.properties = filtered;
-                    }
+                    filter_properties(frame_metadata, properties);
                     metadata_list.push(frame_metadata);
                     if (on_progress && idx % 5000 === 0) {
                         on_progress({
@@ -183,140 +171,53 @@ export class TrajFrameReader {
     }
     load_xyz_frame(data, frame_number) {
         const lines = data.trim().split(/\r?\n/);
-        let [current_frame, line_idx] = [0, 0];
-        while (line_idx < lines.length && current_frame < frame_number) {
-            if (!lines[line_idx]?.trim()) {
-                line_idx++;
-                continue;
-            }
-            const num_atoms = parseInt(lines[line_idx].trim(), 10);
-            if (isNaN(num_atoms) || num_atoms <= 0) {
-                line_idx++;
-                continue;
-            }
-            line_idx += 2 + num_atoms;
-            current_frame++;
-        }
-        if (line_idx >= lines.length)
-            return null;
-        const num_atoms = parseInt(lines[line_idx].trim(), 10);
-        if (isNaN(num_atoms) || line_idx + num_atoms + 1 >= lines.length)
-            return null;
-        const comment = lines[line_idx + 1] || ``;
-        const positions = [];
-        const elements = [];
-        for (let idx = 0; idx < num_atoms; idx++) {
-            const parts = lines[line_idx + 2 + idx]?.trim().split(/\s+/);
-            if (parts?.length >= 4) {
-                const x_coord = parseFloat(parts[1]);
-                const y_coord = parseFloat(parts[2]);
-                const z_coord = parseFloat(parts[3]);
-                if (!Number.isFinite(x_coord) ||
-                    !Number.isFinite(y_coord) ||
-                    !Number.isFinite(z_coord)) {
-                    console.warn(`Skipping XYZ atom with invalid coordinates in indexed frame ${frame_number} at line ${line_idx + 2 + idx}`);
-                    continue;
-                }
-                const raw_symbol = parts[0];
-                const element_symbol = coerce_element_symbol(raw_symbol);
-                if (!element_symbol) {
-                    console.warn(`Skipping XYZ atom with unknown element symbol "${raw_symbol}" in indexed frame ${frame_number}`);
-                    continue;
-                }
-                elements.push(element_symbol);
-                positions.push([x_coord, y_coord, z_coord]);
-            }
+        let current_frame = 0;
+        for (const frame of iter_xyz_frames(lines)) {
+            if (current_frame++ < frame_number)
+                continue; // skip frames before the target
+            return build_xyz_frame(lines, frame, {
+                frame_label: `indexed frame ${frame_number}`,
+                default_step: frame_number,
+            });
         }
-        const metadata = this.parse_xyz_metadata(comment, frame_number);
-        return create_trajectory_frame(positions, elements, undefined, undefined, metadata.step, metadata.properties);
+        return null;
     }
     load_ase_frame(data, frame_number) {
         try {
             const view = new DataView(data);
-            const n_items = Number(view.getBigInt64(32, true));
-            const offsets_pos = Number(view.getBigInt64(40, true));
+            const { n_items, offsets_pos } = read_ase_header(view);
             if (frame_number >= n_items)
                 return null;
             const frame_offset = Number(view.getBigInt64(offsets_pos + frame_number * 8, true));
-            const json_length = Number(view.getBigInt64(frame_offset, true));
-            const frame_data = JSON.parse(new TextDecoder().decode(new Uint8Array(data, frame_offset + 8, json_length)));
-            const positions_ref = frame_data[`positions.`] ?? frame_data.positions;
-            const positions = positions_ref?.ndarray
-                ? read_ndarray_from_view(view, positions_ref)
-                : positions_ref;
-            const numbers_ref = frame_data[`numbers.`] ?? frame_data.numbers ?? this.global_numbers;
-            const numbers = numbers_ref?.ndarray
-                ? read_ndarray_from_view(view, numbers_ref).flat()
-                : numbers_ref;
-            if (numbers)
-                this.global_numbers = numbers;
-            if (!numbers || !positions)
-                throw new Error(`Missing atomic numbers or positions`);
-            const cell = frame_data.cell ? validate_3x3_matrix(frame_data.cell) : undefined;
-            const metadata = {
-                step: frame_number,
-                ...frame_data.calculator,
-                ...frame_data.info,
-            };
-            if (cell) {
-                try {
-                    metadata.volume = Math.abs(math.det_3x3(cell));
-                }
-                catch (error) {
-                    console.warn(`Failed to calculate volume for frame ${frame_number}:`, error);
-                }
-            }
-            return create_trajectory_frame(positions, convert_atomic_numbers(numbers), cell, frame_data.pbc ?? [true, true, true], frame_number, metadata);
+            const { frame, numbers } = decode_ase_frame(view, data, frame_offset, frame_number, this.global_numbers);
+            this.global_numbers = numbers;
+            return frame;
         }
         catch (error) {
             console.warn(`Failed to load ASE frame ${frame_number}:`, error);
             return null;
         }
     }
-    parse_xyz_metadata(comment, frame_number) {
-        const properties = {};
-        const patterns = {
-            energy: /(?:energy|E|etot)\s*[=:]?\s*([-+]?\d*\.?\d+(?:[eE][-+]?\d+)?)/i,
-            volume: /(?:volume|vol|V)\s*[=:]?\s*([-+]?\d*\.?\d+(?:[eE][-+]?\d+)?)/i,
-            pressure: /(?:pressure|press|P)\s*[=:]?\s*([-+]?\d*\.?\d+(?:[eE][-+]?\d+)?)/i,
-            force_max: /(?:max_force|fmax)\s*[=:]?\s*([-+]?\d*\.?\d+(?:[eE][-+]?\d+)?)/i,
-        };
-        Object.entries(patterns).forEach(([key, pattern]) => {
-            const match = pattern.exec(comment);
-            if (match)
-                properties[key] = parseFloat(match[1]);
-        });
-        const step_match = /(?:step|frame)\s*[=:]?\s*(\d+)/i.exec(comment);
-        const step = step_match ? parseInt(step_match[1]) : frame_number;
-        return { frame_number, step, properties };
-    }
     parse_ase_metadata(frame_data, frame_number) {
         const properties = {};
         const step = frame_number;
         if (frame_data.calculator && typeof frame_data.calculator === `object`) {
-            const calculator = frame_data.calculator;
-            const calc_properties = [`energy`, `potential_energy`, `kinetic_energy`, `total_energy`];
-            for (const prop of calc_properties) {
-                if (prop in calculator && typeof calculator[prop] === `number`) {
-                    properties[prop] = calculator[prop];
-                }
-            }
+            copy_numeric_fields(properties, frame_data.calculator, [
+                `energy`,
+                `potential_energy`,
+                `kinetic_energy`,
+                `total_energy`,
+            ]);
         }
         if (frame_data.info && typeof frame_data.info === `object`) {
-            const info = frame_data.info;
-            const info_properties = [
+            copy_numeric_fields(properties, frame_data.info, [
                 `force_max`,
                 `force_norm`,
                 `stress_max`,
                 `stress_frobenius`,
                 `pressure`,
                 `temperature`,
-            ];
-            for (const prop of info_properties) {
-                if (prop in info && typeof info[prop] === `number`) {
-                    properties[prop] = info[prop];
-                }
-            }
+            ]);
         }
         if (frame_data.cell && Array.isArray(frame_data.cell)) {
             try {

package/dist/trajectory/helpers.d.ts

@@ -3,8 +3,6 @@ import * as math from '../math';
 import type { AnyStructure } from '../structure/index';
 import type { Pbc } from '../structure/pbc';
 import type { TrajectoryFrame } from './index';
-export declare const is_valid_element_symbol: (symbol: string) => symbol is ElementSymbol;
-export declare const coerce_element_symbol: (symbol: string) => ElementSymbol | undefined;
 export declare function validate_3x3_matrix(data: unknown): math.Matrix3x3;
 export declare const convert_atomic_numbers: (numbers: number[]) => ElementSymbol[];
 export declare const create_structure: (positions: number[][], elements: ElementSymbol[], lattice_matrix?: math.Matrix3x3, pbc?: Pbc, force_data?: number[][]) => AnyStructure;
@@ -12,4 +10,14 @@ export declare const create_trajectory_frame: (positions: number[][], elements:
 export declare const read_ndarray_from_view: (view: DataView, ref: {
     ndarray: unknown[];
 }) => number[][];
+export declare const copy_numeric_fields: (target: Record<string, number>, source: Record<string, unknown>, fields: readonly string[]) => void;
+export declare function calc_force_stats(forces: number[][]): {
+    force_max: number;
+    force_norm: number;
+} | null;
+export declare function iter_xyz_frames(lines: string[]): Generator<{
+    start: number;
+    num_atoms: number;
+    comment: string;
+}>;
 export declare function count_xyz_frames(data: string): number;