save voltage(s) in model files

2 files changed, 18 insertions, 0 deletions

diff --git a/lib/Kratos/DFADriver.pm b/lib/Kratos/DFADriver.pm
index 321860d..83a572f 100644
--- a/lib/Kratos/DFADriver.pm
+++ b/lib/Kratos/DFADriver.pm
@@ -644,6 +644,8 @@ sub update_model {
 		}
 	}
 
+	$self->model->set_voltage($self->{log}{aggregate}{min_voltage}, $self->{log}{aggregate}{max_voltage});
+
 	$self->model->save;
 }
 
diff --git a/lib/Kratos/DFADriver/Model.pm b/lib/Kratos/DFADriver/Model.pm
index ef821e7..18bccfe 100644
--- a/lib/Kratos/DFADriver/Model.pm
+++ b/lib/Kratos/DFADriver/Model.pm
@@ -412,6 +412,22 @@ sub set_transition_params {
 	}
 }
 
+sub set_voltage {
+	my ($self, $min_voltage, $max_voltage) = @_;
+
+	my ($data_node) = $self->xml->findnodes('/data');
+
+	for my $voltage_node ($data_node->findnodes('./voltage')) {
+		$data_node->removeChild($voltage_node);
+	}
+
+	my $voltage_node = XML::LibXML::Element->new('voltage');
+	$voltage_node->setAttribute('min', $min_voltage);
+	$voltage_node->setAttribute('max', $max_voltage);
+
+	$data_node->appendChild($voltage_node);
+}
+
 sub save {
 	my ($self) = @_;