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

// Copyright 2024 Ulvetanna Inc.

use super::{
	packed::{impl_broadcast, impl_ops_for_zero_height, PackedPrimitiveType},
	packed_arithmetic::{alphas, impl_tower_constants},
};
use crate::{
	arch::{PackedStrategy, PairwiseRecursiveStrategy, PairwiseStrategy},
	arithmetic_traits::{
		impl_invert_with, impl_mul_alpha_with, impl_mul_with, impl_square_with,
		impl_transformation_with_strategy,
	},
	BinaryField16b, BinaryField1b, BinaryField2b, BinaryField32b, BinaryField4b, BinaryField64b,
	BinaryField8b,
};
use cfg_if::cfg_if;

// Define 64 bit packed field types
pub type PackedBinaryField64x1b = PackedPrimitiveType<u64, BinaryField1b>;
pub type PackedBinaryField32x2b = PackedPrimitiveType<u64, BinaryField2b>;
pub type PackedBinaryField16x4b = PackedPrimitiveType<u64, BinaryField4b>;
pub type PackedBinaryField8x8b = PackedPrimitiveType<u64, BinaryField8b>;
pub type PackedBinaryField4x16b = PackedPrimitiveType<u64, BinaryField16b>;
pub type PackedBinaryField2x32b = PackedPrimitiveType<u64, BinaryField32b>;
pub type PackedBinaryField1x64b = PackedPrimitiveType<u64, BinaryField64b>;

// Define broadcast
impl_broadcast!(u64, BinaryField1b);
impl_broadcast!(u64, BinaryField2b);
impl_broadcast!(u64, BinaryField4b);
impl_broadcast!(u64, BinaryField8b);
impl_broadcast!(u64, BinaryField16b);
impl_broadcast!(u64, BinaryField32b);
impl_broadcast!(u64, BinaryField64b);

// Define operations for height 0
impl_ops_for_zero_height!(PackedBinaryField64x1b);

// Define constants
impl_tower_constants!(BinaryField1b, u64, { alphas!(u64, 0) });
impl_tower_constants!(BinaryField2b, u64, { alphas!(u64, 1) });
impl_tower_constants!(BinaryField4b, u64, { alphas!(u64, 2) });
impl_tower_constants!(BinaryField8b, u64, { alphas!(u64, 3) });
impl_tower_constants!(BinaryField16b, u64, { alphas!(u64, 4) });
impl_tower_constants!(BinaryField32b, u64, { alphas!(u64, 5) });

// Define multiplication
impl_mul_with!(PackedBinaryField32x2b @ PackedStrategy);
impl_mul_with!(PackedBinaryField16x4b @ PackedStrategy);
cfg_if! {
	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
		impl_mul_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
		impl_mul_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
		impl_mul_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
		impl_mul_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
	} else {
		impl_mul_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
		impl_mul_with!(PackedBinaryField4x16b @ PairwiseRecursiveStrategy);
		impl_mul_with!(PackedBinaryField2x32b @ PairwiseRecursiveStrategy);
		impl_mul_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);
	}
}

// Define square
impl_square_with!(PackedBinaryField32x2b @ PackedStrategy);
impl_square_with!(PackedBinaryField16x4b @ PackedStrategy);
cfg_if! {
	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
		impl_square_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
		impl_square_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
		impl_square_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
		impl_square_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
	} else {
		impl_square_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
		impl_square_with!(PackedBinaryField4x16b @ PairwiseStrategy);
		impl_square_with!(PackedBinaryField2x32b @ PairwiseRecursiveStrategy);
		impl_square_with!(PackedBinaryField1x64b @ crate::arch::HybridRecursiveStrategy);
	}
}

// Define invert
impl_invert_with!(PackedBinaryField32x2b @ PackedStrategy);
impl_invert_with!(PackedBinaryField16x4b @ PackedStrategy);
cfg_if! {
	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
		impl_invert_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
		impl_invert_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
		impl_invert_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
		impl_invert_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
	} else {
		impl_invert_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
		impl_invert_with!(PackedBinaryField4x16b @ PairwiseStrategy);
		impl_invert_with!(PackedBinaryField2x32b @ PairwiseStrategy);
		impl_invert_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);
	}
}

// Define multiply by alpha
impl_mul_alpha_with!(PackedBinaryField32x2b @ PackedStrategy);
impl_mul_alpha_with!(PackedBinaryField16x4b @ PackedStrategy);
impl_mul_alpha_with!(PackedBinaryField8x8b @ PackedStrategy);
impl_mul_alpha_with!(PackedBinaryField4x16b @ PackedStrategy);
impl_mul_alpha_with!(PackedBinaryField2x32b @ PackedStrategy);
impl_mul_alpha_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);

// Define linear transformations
impl_transformation_with_strategy!(PackedBinaryField64x1b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField32x2b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField16x4b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField8x8b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField4x16b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField2x32b, PackedStrategy);
impl_transformation_with_strategy!(PackedBinaryField1x64b, PairwiseStrategy);