enum_ordinalize/
lib.rs

1/*!
2# Enum Ordinalize
3
4This library enables enums to not only obtain the ordinal values of their variants but also allows for the construction of enums from an ordinal value.
5
6## Usage
7
8Use `#[derive(Ordinalize)]` to have an enum (which must only has unit variants) implement the `Ordinalize` trait.
9
10```rust
11# #[cfg(all(feature = "derive", feature = "traits"))]
12# {
13use enum_ordinalize::Ordinalize;
14
15#[derive(Debug, PartialEq, Eq, Ordinalize)]
16enum MyEnum {
17    Zero,
18    One,
19    Two,
20}
21
22assert_eq!(3, MyEnum::VARIANT_COUNT);
23assert_eq!([MyEnum::Zero, MyEnum::One, MyEnum::Two], MyEnum::VARIANTS);
24assert_eq!([0i8, 1i8, 2i8], MyEnum::VALUES);
25
26assert_eq!(0i8, MyEnum::Zero.ordinal());
27assert_eq!(1i8, MyEnum::One.ordinal());
28assert_eq!(2i8, MyEnum::Two.ordinal());
29
30assert_eq!(Some(MyEnum::Zero), MyEnum::from_ordinal(0i8));
31assert_eq!(Some(MyEnum::One), MyEnum::from_ordinal(1i8));
32assert_eq!(Some(MyEnum::Two), MyEnum::from_ordinal(2i8));
33
34assert_eq!(MyEnum::Zero, unsafe { MyEnum::from_ordinal_unsafe(0i8) });
35assert_eq!(MyEnum::One, unsafe { MyEnum::from_ordinal_unsafe(1i8) });
36assert_eq!(MyEnum::Two, unsafe { MyEnum::from_ordinal_unsafe(2i8) });
37# }
38```
39
40#### The (Ordinal) Size of an Enum
41
42The ordinal value is an integer whose size is determined by the enum itself. The size of the enum increases with the magnitude of the variants' values, whether larger (or smaller if negative).
43
44For example,
45
46```rust
47# #[cfg(all(feature = "derive", feature = "traits"))]
48# {
49use enum_ordinalize::Ordinalize;
50
51#[derive(Debug, PartialEq, Eq, Ordinalize)]
52enum MyEnum {
53    Zero,
54    One,
55    Two,
56    Thousand = 1000,
57}
58
59assert_eq!(4, MyEnum::VARIANT_COUNT);
60assert_eq!([MyEnum::Zero, MyEnum::One, MyEnum::Two, MyEnum::Thousand], MyEnum::VARIANTS);
61assert_eq!([0i16, 1i16, 2i16, 1000i16], MyEnum::VALUES);
62
63assert_eq!(0i16, MyEnum::Zero.ordinal());
64assert_eq!(1i16, MyEnum::One.ordinal());
65assert_eq!(2i16, MyEnum::Two.ordinal());
66
67assert_eq!(Some(MyEnum::Zero), MyEnum::from_ordinal(0i16));
68assert_eq!(Some(MyEnum::One), MyEnum::from_ordinal(1i16));
69assert_eq!(Some(MyEnum::Two), MyEnum::from_ordinal(2i16));
70
71assert_eq!(MyEnum::Zero, unsafe { MyEnum::from_ordinal_unsafe(0i16) });
72assert_eq!(MyEnum::One, unsafe { MyEnum::from_ordinal_unsafe(1i16) });
73assert_eq!(MyEnum::Two, unsafe { MyEnum::from_ordinal_unsafe(2i16) });
74# }
75```
76
77In order to accommodate the value `1000`, the size of `MyEnum` increases. Consequently, the ordinal is represented in `i16` instead of `i8`.
78
79You can utilize the `#[repr(type)]` attribute to explicitly control the size. For instance,
80
81```rust
82# #[cfg(all(feature = "derive", feature = "traits"))]
83# {
84use enum_ordinalize::Ordinalize;
85
86#[derive(Debug, PartialEq, Eq, Ordinalize)]
87#[repr(usize)]
88enum MyEnum {
89    Zero,
90    One,
91    Two,
92    Thousand = 1000,
93}
94
95assert_eq!(4, MyEnum::VARIANT_COUNT);
96assert_eq!([MyEnum::Zero, MyEnum::One, MyEnum::Two, MyEnum::Thousand], MyEnum::VARIANTS);
97assert_eq!([0usize, 1usize, 2usize, 1000usize], MyEnum::VALUES);
98
99assert_eq!(0usize, MyEnum::Zero.ordinal());
100assert_eq!(1usize, MyEnum::One.ordinal());
101assert_eq!(2usize, MyEnum::Two.ordinal());
102
103assert_eq!(Some(MyEnum::Zero), MyEnum::from_ordinal(0usize));
104assert_eq!(Some(MyEnum::One), MyEnum::from_ordinal(1usize));
105assert_eq!(Some(MyEnum::Two), MyEnum::from_ordinal(2usize));
106
107assert_eq!(MyEnum::Zero, unsafe { MyEnum::from_ordinal_unsafe(0usize) });
108assert_eq!(MyEnum::One, unsafe { MyEnum::from_ordinal_unsafe(1usize) });
109assert_eq!(MyEnum::Two, unsafe { MyEnum::from_ordinal_unsafe(2usize) });
110# }
111```
112
113#### Useful Increment
114
115The integers represented by variants can be extended in successive increments and set explicitly from any value.
116
117```rust
118# #[cfg(all(feature = "derive", feature = "traits"))]
119# {
120use enum_ordinalize::Ordinalize;
121
122#[derive(Debug, PartialEq, Eq, Ordinalize)]
123enum MyEnum {
124    Two   = 2,
125    Three,
126    Four,
127    Eight = 8,
128    Nine,
129    NegativeTen = -10,
130    NegativeNine,
131}
132
133assert_eq!(7, MyEnum::VARIANT_COUNT);
134assert_eq!([MyEnum::Two, MyEnum::Three, MyEnum::Four, MyEnum::Eight, MyEnum::Nine, MyEnum::NegativeTen, MyEnum::NegativeNine], MyEnum::VARIANTS);
135assert_eq!([2i8, 3i8, 4i8, 8i8, 9i8, -10i8, -9i8], MyEnum::VALUES);
136
137assert_eq!(4i8, MyEnum::Four.ordinal());
138assert_eq!(9i8, MyEnum::Nine.ordinal());
139assert_eq!(-9i8, MyEnum::NegativeNine.ordinal());
140
141assert_eq!(Some(MyEnum::Four), MyEnum::from_ordinal(4i8));
142assert_eq!(Some(MyEnum::Nine), MyEnum::from_ordinal(9i8));
143assert_eq!(Some(MyEnum::NegativeNine), MyEnum::from_ordinal(-9i8));
144
145assert_eq!(MyEnum::Four, unsafe { MyEnum::from_ordinal_unsafe(4i8) });
146assert_eq!(MyEnum::Nine, unsafe { MyEnum::from_ordinal_unsafe(9i8) });
147assert_eq!(MyEnum::NegativeNine, unsafe { MyEnum::from_ordinal_unsafe(-9i8) });
148# }
149```
150
151#### Implement Functionality for an enum on Itself
152
153For some reason, if you don't want to implement the `Ordinalize` trait for your enum, you can choose to disable the trait implementation and enable the constants/functions one by one. Functions are `const fn`. Names and visibility can also be defined by you.
154
155```rust
156# #[cfg(feature = "derive")]
157# {
158use enum_ordinalize::Ordinalize;
159
160#[derive(Debug, PartialEq, Eq, Ordinalize)]
161#[ordinalize(impl_trait = false)]
162#[ordinalize(variant_count(pub const VARIANT_COUNT, doc = "The count of variants."))]
163#[ordinalize(variants(pub const VARIANTS, doc = "List of this enum's variants."))]
164#[ordinalize(values(pub const VALUES, doc = "List of values for all variants of this enum."))]
165#[ordinalize(ordinal(pub const fn ordinal, doc = "Retrieve the integer number of this variant."))]
166#[ordinalize(from_ordinal(pub const fn from_ordinal, doc = "Obtain a variant based on an integer number."))]
167#[ordinalize(from_ordinal_unsafe(
168    pub const fn from_ordinal_unsafe,
169    doc = "Obtain a variant based on an integer number.",
170    doc = "# Safety",
171    doc = "You have to ensure that the input integer number can correspond to a variant on your own.",
172))]
173enum MyEnum {
174    A,
175    B,
176}
177
178assert_eq!(2, MyEnum::VARIANT_COUNT);
179assert_eq!([MyEnum::A, MyEnum::B], MyEnum::VARIANTS);
180assert_eq!([0i8, 1i8], MyEnum::VALUES);
181
182assert_eq!(0i8, MyEnum::A.ordinal());
183assert_eq!(1i8, MyEnum::B.ordinal());
184
185assert_eq!(Some(MyEnum::A), MyEnum::from_ordinal(0i8));
186assert_eq!(Some(MyEnum::B), MyEnum::from_ordinal(1i8));
187
188assert_eq!(MyEnum::A, unsafe { MyEnum::from_ordinal_unsafe(0i8) });
189assert_eq!(MyEnum::B, unsafe { MyEnum::from_ordinal_unsafe(1i8) });
190# }
191```
192*/
193
194#![no_std]
195#![cfg_attr(docsrs, feature(doc_auto_cfg))]
196
197#[cfg(feature = "traits")]
198mod traits;
199
200#[cfg(feature = "derive")]
201pub use enum_ordinalize_derive::Ordinalize;
202#[cfg(feature = "traits")]
203pub use traits::Ordinalize;