problemreductions/rules/
partition_cosineproductintegration.rs1use crate::models::misc::{CosineProductIntegration, Partition};
14use crate::reduction;
15use crate::rules::traits::{ReduceTo, ReductionResult};
16
17#[derive(Debug, Clone)]
19pub struct ReductionPartitionToCPI {
20 target: CosineProductIntegration,
21}
22
23impl ReductionResult for ReductionPartitionToCPI {
24 type Source = Partition;
25 type Target = CosineProductIntegration;
26
27 fn target_problem(&self) -> &Self::Target {
28 &self.target
29 }
30
31 fn extract_solution(
32 &self,
33 target_solution: &<Self::Target as crate::traits::Problem>::Solution,
34 ) -> crate::rules::ExtractionResult<<Self::Source as crate::traits::Problem>::Solution> {
35 crate::rules::traits::validate_target_solution(self.target_problem(), target_solution)?;
36
37 Ok(target_solution.to_vec())
38 }
39}
40
41#[reduction(
42 transform = exact {
43 num_coefficients = "num_elements",
44 })]
45impl ReduceTo<CosineProductIntegration> for Partition {
46 type Result = ReductionPartitionToCPI;
47
48 fn reduce_to(&self) -> Result<Self::Result, crate::rules::ReductionError> {
49 let coefficients = self.sizes().to_vec();
50 Ok(ReductionPartitionToCPI {
51 target: CosineProductIntegration::new(coefficients),
52 })
53 }
54}
55
56#[cfg(feature = "example-db")]
57pub(crate) fn canonical_rule_example_specs() -> Vec<crate::example_db::specs::RuleExampleSpec> {
58 use crate::export::SolutionPair;
59
60 vec![crate::example_db::specs::RuleExampleSpec {
61 id: "partition_to_cosineproductintegration",
62 build: || {
63 crate::example_db::specs::rule_example_with_witness::<_, CosineProductIntegration>(
70 Partition::new(vec![3, 1, 1, 2, 2, 1]).unwrap(),
71 SolutionPair {
72 source_config: serde_json::json!(vec![true, false, false, true, false, false]),
73 target_config: serde_json::json!(vec![true, false, false, true, false, false]),
74 },
75 )
76 },
77 }]
78}
79
80#[cfg(test)]
81#[path = "../unit_tests/rules/partition_cosineproductintegration.rs"]
82mod tests;