update energy handling in stf VHEE generator and tests

Author: wahln

matRad/steering/matRad_StfGeneratorParticleVHEE.m

@@ -49,6 +49,7 @@
             else
                 beam.VHEEenergy = this.energy;
             end
+            this.energy = beam.VHEEenergy;
             % Optional: check if that energy is in the machine data
             if isfield(this.machine.data,'energies') && ~isempty(this.machine.data.energies)
                 if ~ismember(beam.VHEEenergy, this.machine.data.energies)

test/steering/test_stfGeneratorVHEE.m

@@ -10,9 +10,9 @@ function test_basic_construct()
 
     function test_pln_construct()
         load VHEE_testData.mat
+        pln.propStf.energy = 150;
         stfGen = matRad_StfGeneratorParticleVHEE(pln);
         stfGen.isAvailable(pln);
-        pln.propStf.energy = 150;
         assertTrue(isa(stfGen, 'matRad_StfGeneratorParticleVHEE'));
         assertEqual(stfGen.gantryAngles, pln.propStf.gantryAngles);
         assertEqual(stfGen.couchAngles, pln.propStf.couchAngles);
@@ -25,9 +25,9 @@ function test_pln_construct()
     function test_pln_construct_focused()
         load VHEE_testData.mat
         pln.machine = 'Focused';
+        pln.propStf.energy = 150;
         stfGen = matRad_StfGeneratorParticleVHEE(pln);
         stfGen.isAvailable(pln);
-        pln.propStf.energy = 150;
         assertTrue(isa(stfGen, 'matRad_StfGeneratorParticleVHEE'));
         assertEqual(stfGen.gantryAngles, pln.propStf.gantryAngles);
         assertEqual(stfGen.couchAngles, pln.propStf.couchAngles);