From 9b315100ff1cf88758c7160512a1f34466f01bd1 Mon Sep 17 00:00:00 2001
From: Tobias Dammers <tdammers@gmail.com>
Date: Fri, 24 Sep 2021 18:11:05 +0200
Subject: [PATCH] Robustify

---
 addon-main.nas | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/addon-main.nas b/addon-main.nas
index b310da1..cc9fe5f 100644
--- a/addon-main.nas
+++ b/addon-main.nas
@@ -212,7 +212,9 @@ var importFOB = func (ofp) {
 
     var allocate = func(tankNumber, maxAmount = nil) {
         var tankNode = tankNodes[tankNumber];
-        if (tankNode == nil) {
+        var capacityNode = tankNode.getNode('capacity-m3');
+        var densityNode = tankNode.getNode('density-kgpm3');
+        if (tankNode == nil or capacityNode == nil or densityNode == nil) {
             printf("Tank #%i not installed", tankNumber);
             return;
         }
@@ -267,12 +269,12 @@ var importFOB = func (ofp) {
         var totalFuel = unallocated;
         var totalCapacity = 0;
         foreach (var tankNode; tankNodes) {
-            var capacity = tankNode.getValue('capacity-m3');
+            var capacity = tankNode.getValue('capacity-m3') or 0;
             totalCapacity = totalCapacity + capacity;
         }
         for (var i = 0; i < numTanks; i += 1) {
             var tankNode = tankNodes[i];
-            var capacity = tankNode.getValue('capacity-m3');
+            var capacity = tankNode.getValue('capacity-m3') or 0;
             allocate(i, totalFuel * capacity / totalCapacity);
         }
     }