uff-relax

A high-performance, parallelized molecular structure optimizer for Rust, powered by the Universal Force Field (UFF) and the FIRE algorithm.

Features

🚀 High Performance: Optimized force evaluations with Cell Lists for efficient neighbor searching.
🧵 Parallel Processing: Scalable multi-threading via Rayon, automatically enabled for large systems (>1000 atoms).
💠 PBC Support: Periodic boundary conditions for Orthorhombic and Triclinic systems.
🧪 Smart Type Assignment: Automatically infers UFF atom types from atomic numbers and connectivity.
🦀 Pure Rust: Fast, safe, and easy to integrate.

Installation

Add this to your Cargo.toml:

[dependencies]
uff-relax = "1.0.0"
glam = "0.31"

Quick Start

use uff_relax::{System, Atom, Bond, UnitCell, UffOptimizer};
use glam::DVec3;

fn main() {
    // 1. Define Atoms
    let atoms = vec![
        Atom::new(6, DVec3::new(0.0, 0.0, 0.0)),
        Atom::new(6, DVec3::new(1.3, 0.0, 0.0)),
    ];

    // 2. Define Bonds
    let bonds = vec![
        Bond { atom_indices: (0, 1), order: 1.0 },
    ];

    // 3. Create System
    let mut system = System::new(atoms, bonds, UnitCell::new_none());

    // 4. Run Optimizer
    let optimizer = UffOptimizer::new(1000, 1e-3);
    optimizer.optimize(&mut system);

    println!("Final Energy: {:.4} kcal/mol", system.compute_forces().total);
}

Running Examples

Try the included examples to see the optimizer in action:

cargo run --example benzene

Benchmarks

This crate includes specialized benchmarks to measure scaling performance and handle large-scale systems. These are standalone binaries (harness = false) to ensure minimal overhead.

1. Scaling Threshold

Measures the efficiency of parallelization as the number of atoms increases.

cargo bench --bench threshold

2. Large System Stress Test

Simulates a system with 100,000 atoms to verify stability and memory efficiency in large-scale optimizations.

cargo bench --bench large_system_bench

License

Licensed under either of:

Apache License, Version 2.0 (LICENSE-APACHE or http://www.apache.org/licenses/LICENSE-2.0)
MIT license (LICENSE-MIT or http://opensource.org/licenses/MIT)

at your option.

Author

Forblaze Project
Website: https://forblaze-works.com/

uff-relax 1.0.4