Skip to content

Commit

Permalink
chore: refactor execute_cmd
Browse files Browse the repository at this point in the history
  • Loading branch information
TomAFrench committed Sep 12, 2023
1 parent 9da822f commit 3ec540f
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 24 deletions.
36 changes: 13 additions & 23 deletions tooling/nargo_cli/src/cli/execute_cmd.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
use acvm::acir::circuit::{Opcode, OpcodeLocation};
use acvm::acir::circuit::OpcodeLocation;
use acvm::acir::{circuit::Circuit, native_types::WitnessMap};
use acvm::pwg::{ErrorLocation, OpcodeResolutionError};
use acvm::Language;
use clap::Args;
use fm::FileManager;
use nargo::constants::PROVER_INPUT_FILE;
Expand Down Expand Up @@ -52,49 +51,41 @@ pub(crate) fn run(
if args.workspace { PackageSelection::All } else { PackageSelection::DefaultOrAll };
let selection = args.package.map_or(default_selection, PackageSelection::Selected);
let workspace = resolve_workspace_from_toml(&toml_path, selection)?;
let witness_dir = &workspace.target_directory_path();
let target_dir = &workspace.target_directory_path();

let (np_language, is_opcode_supported) = backend.get_backend_info()?;
for package in &workspace {
let (return_value, solved_witness) = execute_package(
backend,
package,
&args.prover_name,
&args.compile_options,
np_language,
&is_opcode_supported,
)?;
let (file_manager, compiled_program) =
compile_package(package, &args.compile_options, np_language, &is_opcode_supported)?;

let (return_value, solved_witness) =
execute_program_and_decode(compiled_program, file_manager, package, &args.prover_name)?;

println!("[{}] Circuit witness successfully solved", package.name);
if let Some(return_value) = return_value {
println!("[{}] Circuit output: {return_value:?}", package.name);
}
if let Some(witness_name) = &args.witness_name {
let witness_path = save_witness_to_dir(solved_witness, witness_name, witness_dir)?;
let witness_path = save_witness_to_dir(solved_witness, witness_name, target_dir)?;

println!("[{}] Witness saved to {}", package.name, witness_path.display());
}
}
Ok(())
}

fn execute_package(
backend: &Backend,
fn execute_program_and_decode(
program: CompiledProgram,
file_manager: FileManager,
package: &Package,
prover_name: &str,
compile_options: &CompileOptions,
np_language: Language,
is_opcode_supported: &impl Fn(&Opcode) -> bool,
) -> Result<(Option<InputValue>, WitnessMap), CliError> {
let (context, compiled_program) =
compile_package(package, compile_options, np_language, &is_opcode_supported)?;
let CompiledProgram { abi, circuit, debug } = compiled_program;
let CompiledProgram { abi, circuit, debug } = program;

// Parse the initial witness values from Prover.toml
let (inputs_map, _) =
read_inputs_from_file(&package.root_dir, prover_name, Format::Toml, &abi)?;
let solved_witness =
execute_program(backend, circuit, &abi, &inputs_map, Some((debug, context)))?;
let solved_witness = execute_program(circuit, &abi, &inputs_map, Some((debug, file_manager)))?;
let public_abi = abi.public_abi();
let (_, return_value) = public_abi.decode(&solved_witness)?;

Expand Down Expand Up @@ -190,7 +181,6 @@ fn report_error_with_opcode_locations(
}

pub(crate) fn execute_program(
_backend: &Backend,
circuit: Circuit,
abi: &Abi,
inputs_map: &InputMap,
Expand Down
2 changes: 1 addition & 1 deletion tooling/nargo_cli/src/cli/prove_cmd.rs
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ pub(crate) fn prove_package(
let (inputs_map, _) =
read_inputs_from_file(&package.root_dir, prover_name, Format::Toml, &abi)?;

let solved_witness = execute_program(backend, bytecode.clone(), &abi, &inputs_map, debug_data)?;
let solved_witness = execute_program(bytecode.clone(), &abi, &inputs_map, debug_data)?;

// Write public inputs into Verifier.toml
let public_abi = abi.public_abi();
Expand Down

0 comments on commit 3ec540f

Please sign in to comment.