use quantr::{
states::{ProductState, SuperPosition},
Circuit, Gate, Measurement, Printer, QuantrError,
};
fn main() -> Result<(), QuantrError> {
let mut qc: Circuit = Circuit::new(3)?;
qc.add_repeating_gate(Gate::X, &[1, 2])?
.add_gate(Gate::Custom(qft, vec![0, 1], "QFT".to_string()), 2)?;
let mut printer = Printer::new(&qc);
printer.print_diagram();
qc.set_print_progress(true);
let simulated_circuit = qc.simulate();
if let Measurement::NonObservable(final_sup) = simulated_circuit.get_state() {
println!("\nThe final superposition is:");
for (state, amplitude) in final_sup.into_iter() {
println!("|{}> : {}", state, amplitude);
}
}
Ok(())
}
fn qft(input_state: ProductState) -> Option<SuperPosition> {
let qubit_num = input_state.num_qubits();
let mut mini_circuit: Circuit = Circuit::new(qubit_num).unwrap();
for pos in 0..qubit_num {
mini_circuit.add_gate(Gate::H, pos).unwrap();
for k in 2..=(qubit_num - pos) {
mini_circuit
.add_gate(Gate::CRk(k as i32, pos + k - 1), pos)
.unwrap();
}
}
mini_circuit
.change_register(SuperPosition::from(input_state))
.unwrap();
Some(mini_circuit.simulate().take_state().take())
}