File: vector.py

package info (click to toggle)
ezdxf 1.4.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 104,528 kB
  • sloc: python: 182,341; makefile: 116; lisp: 20; ansic: 4
file content (122 lines) | stat: -rw-r--r-- 3,075 bytes parent folder | download | duplicates (2)
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
# Copyright (c) 2020, Manfred Moitzi
# License: MIT License
import sys
import time
from datetime import datetime
from pathlib import Path
from ezdxf.acc import USE_C_EXT

if USE_C_EXT is False:
    print("C-extension disabled or not available.")
    sys.exit(1)

from ezdxf.math._vector import Vec3, Vec2  # Python implementation
from ezdxf.acc.vector import (
    Vec3 as CVec3,
    Vec2 as CVec2,
)  # Cython Implementation
from ezdxf.version import __version__


def open_log(name: str):
    parent = Path(__file__).parent
    p = parent / "logs" / Path(name + ".csv")
    if not p.exists():
        with open(p, mode="wt") as fp:
            fp.write(
                '"timestamp"; "pytime"; "cytime"; '
                '"python_version"; "ezdxf_version"\n'
            )
    log_file = open(p, mode="at")
    return log_file


def log(name: str, pytime: float, cytime: float):
    log_file = open_log(name)
    timestamp = datetime.now().isoformat()
    log_file.write(
        f'{timestamp}; {pytime}; {cytime}; "{sys.version}"; "{__version__}"\n'
    )
    log_file.close()


def vec2_add(VType, count):
    v = VType(0, 0)
    inc = VType(1, 2)
    for _ in range(count):
        v = v + inc


def vec2_subtract(VType, count):
    v = VType(0, 0)
    dec = VType(1, 2)
    for _ in range(count):
        v = v - dec


def vec2_multiply(VType, count):
    v = VType(1, 2)
    for _ in range(count):
        v = v * 123.0


def vec2_normalize(VType, count):
    v = VType(10, 20)
    for _ in range(count):
        v.normalize(3.0)


def vec3_add(VType, count):
    v = VType()
    inc = VType(1, 2, 3)
    for _ in range(count):
        v = v + inc


def vec3_subtract(VType, count):
    v = VType()
    dec = VType(1, 2, 3)
    for _ in range(count):
        v = v - dec


def vec3_multiply(VType, count):
    v = VType(1, 2, 3)
    for _ in range(count):
        v = v * 123.0


def vec3_normalize(VType, count):
    v = VType(10, 20, 30)
    for _ in range(count):
        v.normalize(3.0)


def profile1(func, *args) -> float:
    t0 = time.perf_counter()
    func(*args)
    t1 = time.perf_counter()
    return t1 - t0


def profile(text, func, pytype, cytype, *args):
    pytime = profile1(func, pytype, *args)
    cytime = profile1(func, cytype, *args)
    ratio = pytime / cytime
    print(f"Python - {text} {pytime:.3f}s")
    print(f"Cython - {text} {cytime:.3f}s")
    print(f"Ratio {ratio:.1f}x")
    log(func.__name__, pytime, cytime)


RUNS = 1_000_000

print(f"Profiling Vec2/Vec3 Python and Cython implementation:")
profile(f"add {RUNS} Vec2: ", vec2_add, Vec2, CVec2, RUNS)
profile(f"subtract {RUNS} Vec2: ", vec2_subtract, Vec2, CVec2, RUNS)
profile(f"multiply {RUNS} Vec2: ", vec2_multiply, Vec2, CVec2, RUNS)
profile(f"normalize {RUNS} Vec2: ", vec2_normalize, Vec2, CVec2, RUNS)
profile(f"add {RUNS} Vec3: ", vec3_add, Vec3, CVec3, RUNS)
profile(f"subtract {RUNS} Vec3: ", vec3_subtract, Vec3, CVec3, RUNS)
profile(f"multiply {RUNS} Vec3: ", vec3_multiply, Vec3, CVec3, RUNS)
profile(f"normalize {RUNS} Vec3: ", vec3_normalize, Vec3, CVec3, RUNS)