support heap code to declare global variables used by the benchmark

1 files changed, 9 insertions, 0 deletions

diff --git a/lib/Kratos/DFADriver/Model.pm b/lib/Kratos/DFADriver/Model.pm
index e6f8142..ef821e7 100644
--- a/lib/Kratos/DFADriver/Model.pm
+++ b/lib/Kratos/DFADriver/Model.pm
@@ -156,6 +156,9 @@ sub parse_xml {
 	if ( my ($node) = $xml->findnodes('/data/startup/code') ) {
 		$self->{startup}{code} = $node->textContent;
 	}
+	if ( my ($node) = $xml->findnodes('/data/heap/code') ) {
+		$self->{heap}{code} = $node->textContent;
+	}
 	if ( my ($node) = $xml->findnodes('/data/after-transition/code') ) {
 		$self->{after_transition}{code} = $node->textContent;
 	}
@@ -453,6 +456,12 @@ sub startup_code {
 	return $self->{startup}{code} // q{};
 }
 
+sub heap_code {
+	my ($self) = @_;
+
+	return $self->{heap}{code} // q{};
+}
+
 sub after_transition_code {
 	my ($self) = @_;