From a1c21722dde4ece79cac2a6aad7eafe76126a262 Mon Sep 17 00:00:00 2001 From: andri lim Date: Wed, 4 Sep 2019 13:36:48 +0700 Subject: [PATCH] more flexible calcDifficulty --- nimbus/utils/difficulty.nim | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/nimbus/utils/difficulty.nim b/nimbus/utils/difficulty.nim index 025a3cb0a..78ec3ebb4 100644 --- a/nimbus/utils/difficulty.nim +++ b/nimbus/utils/difficulty.nim @@ -149,3 +149,14 @@ func calcDifficulty*(timeStamp: EthTime, parent: BlockHeader): DifficultyInt = result = calcDifficultyHomestead(timeStamp, parent) else: result = calcDifficultyFrontier(timeStamp, parent) + +func calcDifficulty*(timeStamp: EthTime, parent: BlockHeader, fork: Fork): DifficultyInt = + case fork + of FkConstantinople: + result = calcDifficultyConstantinople(timeStamp, parent) + of FkByzantium: + result = calcDifficultyByzantium(timeStamp, parent) + of FkHomestead..FkSpurious: + result = calcDifficultyHomestead(timeStamp, parent) + else: + result = calcDifficultyFrontier(timeStamp, parent) \ No newline at end of file