traj-dist-rs 1.0.0-rc.5

High-performance trajectory distance & similarity measures in Rust with Python bindings
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

"""
To test TTY environment:
    uv run examples/python/test_non_tty_progress.py

To test non-TTY environment:
    uv run examples/python/test_non_tty_progress.py 2>&1 | cat
"""

import sys
import time

import numpy as np
import traj_dist_rs


def main():
    print("Generating test data...", file=sys.stderr)
    # Generate 1000 trajectories, 200 points each
    # 1000 * 999 / 2 = 499,500 pairs, should take > 10s
    np.random.seed(42)
    trajectories = [np.random.rand(200, 2) for _ in range(1000)]

    metric = traj_dist_rs.Metric.dtw(type_d="euclidean")

    print("Starting DTW computation...", file=sys.stderr)
    start_time = time.time()

    distances = traj_dist_rs.pdist(
        trajectories, metric=metric, parallel=True, show_progress=True
    )

    end_time = time.time()

    print(
        f"Computation finished in {end_time - start_time:.2f} seconds.", file=sys.stderr
    )
    print(f"Computed {len(distances)} distances.", file=sys.stderr)


if __name__ == "__main__":
    main()