Change special axe digSpeed depending on tree size

Hi,

i am currently creating a broad axe for my mod. I am using my algorithm to get the log blocks of the tree the player tries to fell. This method stores all log coords in the axe´s nbt.

Now I would like to change the time it takes to break the log that is currently being harvested depending on the size of the array in my items nbt.

I already found the getDigSpeed() method can do this, but my problem is, that this method is fired/checked before my tree algorithm is called in onBlockStartBreak() .

Is there a way to use this methods or other ways to change the time it takes to break the block depending on the size of the tree?