logos-storage/plonky2
1
0
Fork 0
mirror of https://github.com/logos-storage/plonky2.git synced 2026-01-04 06:43:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace indices: HashMap with indices: Fn(T)->usize

Browse Source
This commit is contained in:
wborgeaud 2021-07-02 14:37:07 +02:00
parent d93cf693ba
commit b6554ba2ec
2 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/circuit_builder.rs
View File

@ -359,7 +359,13 @@ impl<F: Extendable<D>, const D: usize> CircuitBuilder<F, D> {
fn sigma_vecs(&self, k_is: &[F], subgroup: &[F]) -> Vec<PolynomialValues<F>> {
let degree = self.gate_instances.len();
let degree_log = log2_strict(degree);
let mut target_partition = TargetPartition::default();
let mut target_partition = TargetPartition::new(|t| match t {
Target::Wire(Wire { gate, input }) => gate * self.config.num_routed_wires + input,
Target::PublicInput { index } => degree * self.config.num_routed_wires + index,
Target::VirtualTarget { index } => {
degree * self.config.num_routed_wires + self.public_input_index + index
}
});
for gate in 0..degree {
for input in 0..self.config.num_routed_wires {

20
src/permutation_argument.rs
View File

@ -18,21 +18,18 @@ pub struct ForestNode<T: Debug + Copy + Eq + PartialEq> {
}
#[derive(Debug, Clone)]
pub struct TargetPartition<T: Debug + Copy + Eq + PartialEq + Hash> {
pub struct TargetPartition<T: Debug + Copy + Eq + PartialEq + Hash, F: Fn(T) -> usize> {
forest: Vec<ForestNode<T>>,
indices: HashMap<T, usize>,
indices: F,
}
impl<T: Debug + Copy + Eq + PartialEq + Hash> Default for TargetPartition<T> {
fn default() -> Self {
impl<T: Debug + Copy + Eq + PartialEq + Hash, F: Fn(T) -> usize> TargetPartition<T, F> {
pub fn new(f: F) -> Self {
Self {
forest: Vec::new(),
indices: Default::default(),
indices: f,
}
}
}
impl<T: Debug + Copy + Eq + PartialEq + Hash> TargetPartition<T> {
/// Add a new partition with a single member.
pub fn add(&mut self, t: T) {
let index = self.forest.len();
@ -42,7 +39,6 @@ impl<T: Debug + Copy + Eq + PartialEq + Hash> TargetPartition<T> {
size: 1,
index,
});
self.indices.insert(t, index);
}
/// Path halving
@ -58,8 +54,8 @@ impl<T: Debug + Copy + Eq + PartialEq + Hash> TargetPartition<T> {
/// Merge the two partitions containing the two given targets. Does nothing if the targets are
/// already members of the same partition.
pub fn merge(&mut self, tx: T, ty: T) {
let index_x = self.indices[&tx];
let index_y = self.indices[&ty];
let index_x = (self.indices)(tx);
let index_y = (self.indices)(ty);
let mut x = self.forest[index_x];
let mut y = self.forest[index_y];
@ -81,7 +77,7 @@ impl<T: Debug + Copy + Eq + PartialEq + Hash> TargetPartition<T> {
self.forest[index_y] = y;
}
}
impl TargetPartition<Target> {
impl<F: Fn(Target) -> usize> TargetPartition<Target, F> {
pub fn wire_partitions(&mut self) -> WirePartitions {
let mut partition = HashMap::<_, Vec<_>>::new();
let nodes = self.forest.clone();