Add bytecode segments to compiled classes

NethermindEth · Feb 24, 2024 · 0b4a74a · 0b4a74a
1 parent 633a2a1
commit 0b4a74a
Show file tree

Hide file tree

Showing 10 changed files with 47,698 additions and 33 deletions.
diff --git a/adapters/core2sn/class.go b/adapters/core2sn/class.go
@@ -7,13 +7,21 @@ import (
 	"github.com/NethermindEth/juno/utils"
 )
 
+func AdaptSegmentLengths(l core.SegmentLengths) starknet.SegmentLengths {
+	return starknet.SegmentLengths{
+		Length:   l.Length,
+		Children: utils.Map(l.Children, AdaptSegmentLengths),
+	}
+}
+
 func AdaptCompiledClass(coreCompiledClass *core.CompiledClass) starknet.CompiledClass {
 	var feederCompiledClass starknet.CompiledClass
 	feederCompiledClass.Bytecode = coreCompiledClass.Bytecode
 	feederCompiledClass.PythonicHints = coreCompiledClass.PythonicHints
 	feederCompiledClass.CompilerVersion = coreCompiledClass.CompilerVersion
 	feederCompiledClass.Hints = coreCompiledClass.Hints
 	feederCompiledClass.Prime = "0x" + coreCompiledClass.Prime.Text(felt.Base16)
+	feederCompiledClass.BytecodeSegmentLengths = AdaptSegmentLengths(coreCompiledClass.BytecodeSegmentLengths)
 
 	adapt := func(ep core.CompiledEntryPoint) starknet.CompiledEntryPoint {
 		return starknet.CompiledEntryPoint{

diff --git a/adapters/sn2core/sn2core.go b/adapters/sn2core/sn2core.go
@@ -296,6 +296,7 @@ func AdaptCompiledClass(compiledClass *starknet.CompiledClass) (*core.CompiledCl
 	compiled.PythonicHints = compiledClass.PythonicHints
 	compiled.CompilerVersion = compiledClass.CompilerVersion
 	compiled.Hints = compiledClass.Hints
+	compiled.BytecodeSegmentLengths = AdaptSegmentLengths(compiledClass.BytecodeSegmentLengths)
 
 	var ok bool
 	compiled.Prime, ok = new(big.Int).SetString(compiledClass.Prime, 0)
@@ -311,6 +312,13 @@ func AdaptCompiledClass(compiledClass *starknet.CompiledClass) (*core.CompiledCl
 	return &compiled, nil
 }
 
+func AdaptSegmentLengths(l starknet.SegmentLengths) core.SegmentLengths {
+	return core.SegmentLengths{
+		Length:   l.Length,
+		Children: utils.Map(l.Children, AdaptSegmentLengths),
+	}
+}
+
 func AdaptCairo0Class(response *starknet.Cairo0Definition) (core.Class, error) {
 	class := new(core.Cairo0Class)
 	class.Abi = response.Abi