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

// Copyright (c) 2021-2024 Espresso Systems (espressosys.com)
// This file is part of the HotShot repository.

// You should have received a copy of the MIT License
// along with the HotShot repository. If not, see <https://mit-license.org/>.

//! Utilities to help building a stake table.

use ark_ff::{Field, PrimeField};
use primitive_types::U256;

/// A trait that converts into a field element.
pub trait ToFields<F: Field> {
    /// The number of field elements needed to represent the given struct.
    const SIZE: usize;

    /// Convert the given struct into a list of field elements.
    fn to_fields(&self) -> Vec<F>;
}

/// convert a U256 to a field element.
pub(crate) fn u256_to_field<F: PrimeField>(v: &U256) -> F {
    let mut bytes = vec![0u8; 32];
    v.to_little_endian(&mut bytes);
    F::from_le_bytes_mod_order(&bytes)
}