aboutsummaryrefslogtreecommitdiff
path: root/test/sca/perf_combine.py
blob: 8ac274c9fe7074d8bd7a1e84237c5b8390f84ec1 (plain)
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
#!/usr/bin/env python
import click

from importlib_resources import files, as_file
from test.utils import Profiler
import test.data.sca
from pyecsca.sca import (
    InspectorTraceSet,
    average,
    variance,
    standard_deviation,
    add,
    subtract,
    conditional_average,
)


@click.command()
@click.option(
    "-p",
    "--profiler",
    type=click.Choice(("py", "c", "raw")),
    default="py",
    envvar="PROF",
)
@click.option("-o", "--operations", type=click.INT, default=100)
@click.option(
    "-d",
    "--directory",
    type=click.Path(file_okay=False, dir_okay=True),
    default=None,
    envvar="DIR",
)
def main(profiler, operations, directory):
    with as_file(files(test.data.sca).joinpath("example.trs")) as path:
        traces = InspectorTraceSet.read(path)
        with Profiler(profiler, directory, f"combine_average_example_{operations}"):
            for _ in range(operations):
                average(*traces)
        with Profiler(profiler, directory, f"combine_condavg_example_{operations}"):
            for _ in range(operations):
                conditional_average(*traces, condition=lambda trace: trace[0] > 0)
        with Profiler(profiler, directory, f"combine_variance_example_{operations}"):
            for _ in range(operations):
                variance(*traces)
        with Profiler(profiler, directory, f"combine_stddev_example_{operations}"):
            for _ in range(operations):
                standard_deviation(*traces)
        with Profiler(profiler, directory, f"combine_add_example_{operations}"):
            for _ in range(operations):
                add(*traces)
        with Profiler(profiler, directory, f"combine_subtract_example_{operations}"):
            for _ in range(operations):
                subtract(traces[0], traces[1])


if __name__ == "__main__":
    main()