// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.

// OpenEthereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// OpenEthereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with OpenEthereum.  If not, see <http://www.gnu.org/licenses/>.

//! benchmarking for EVM
//! should be started with:
//! ```bash
//! cargo bench
//! ```

#[macro_use]
extern crate criterion;
extern crate ethcore;
extern crate ethereum_types;
extern crate evm;
extern crate rustc_hex;
extern crate vm;

use criterion::{black_box, Criterion};
use std::sync::Arc;

use ethereum_types::U256;
use evm::Factory;
use rustc_hex::FromHex;
use vm::{tests::FakeExt, ActionParams, Ext};

criterion_group!(
    evmbin,
    bench_simple_loop_usize,
    bench_simple_loop_u256,
    bench_rng_usize,
    bench_rng_u256
);
criterion_main!(evmbin);

fn bench_simple_loop_usize(c: &mut Criterion) {
    simple_loop(U256::from(::std::usize::MAX), c, "simple_loop_usize")
}

fn bench_simple_loop_u256(c: &mut Criterion) {
    simple_loop(!U256::zero(), c, "simple_loop_u256")
}

fn simple_loop(gas: U256, c: &mut Criterion, bench_id: &str) {
    let code = black_box(
		"606060405260005b620042408112156019575b6001016007565b600081905550600680602b6000396000f3606060405200".from_hex().unwrap()
	);

    c.bench_function(bench_id, move |b| {
        b.iter(|| {
            let mut params = ActionParams::default();
            params.gas = gas;
            params.code = Some(Arc::new(code.clone()));

            let mut ext = FakeExt::new();
            let evm = Factory::default().create(params, ext.schedule(), ext.depth());
            let _ = evm.exec(&mut ext);
        })
    });
}

fn bench_rng_usize(c: &mut Criterion) {
    rng(U256::from(::std::usize::MAX), c, "rng_usize")
}

fn bench_rng_u256(c: &mut Criterion) {
    rng(!U256::zero(), c, "rng_u256")
}

fn rng(gas: U256, c: &mut Criterion, bench_id: &str) {
    let code = black_box(
		"6060604052600360056007600b60005b62004240811215607f5767ffe7649d5eca84179490940267f47ed85c4b9a6379019367f8e5dd9a5c994bba9390930267f91d87e4b8b74e55019267ff97f6f3b29cda529290920267f393ada8dd75c938019167fe8d437c45bb3735830267f47d9a7b5428ffec019150600101600f565b838518831882186000555050505050600680609a6000396000f3606060405200".from_hex().unwrap()
	);

    c.bench_function(bench_id, move |b| {
        b.iter(|| {
            let mut params = ActionParams::default();
            params.gas = gas;
            params.code = Some(Arc::new(code.clone()));

            let mut ext = FakeExt::new();
            let evm = Factory::default().create(params, ext.schedule(), ext.depth());
            let _ = evm.exec(&mut ext);
        })
    });
}