From ebda13e738c881cb0909d5ceafcf4190cf9f3e8b Mon Sep 17 00:00:00 2001
From: Mario Fink <mario.fink@record-evolution.de>
Date: Tue, 28 Jul 2020 17:25:41 +0200
Subject: [PATCH] raweat.hpp: get_unit(): check existence of marker

---
 lib/raweat.hpp | 9 ++++++++-
 pyt/example.py | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/lib/raweat.hpp b/lib/raweat.hpp
index 684b036..b460719 100644
--- a/lib/raweat.hpp
+++ b/lib/raweat.hpp
@@ -511,7 +511,14 @@ public:
   {
     // assert ( segments_.size() > 0 );
 
-    return valid_ ? segments_["punit marker"][7] : std::string("None");
+    if ( datasec_["punit"].size() > 0 )
+    {
+      return valid_ ? segments_["punit marker"][7] : std::string("None");
+    }
+    else
+    {
+      return std::string("None");
+    }
   }
 
   // get time offset
diff --git a/pyt/example.py b/pyt/example.py
index c3b7e4e..bbfe5b8 100644
--- a/pyt/example.py
+++ b/pyt/example.py
@@ -1,7 +1,7 @@
 
 import raw_eater
 
-eatraw = raw_eater.raweater(b"sample/pressure_Vacuum.raw")
+eatraw = raw_eater.raweater(b"sample/Rangerover_Evoque_F-RR534_2019-05-07/ABS_A_Port1.raw")
 # eatraw = raw_eater.raweater(b"/home/mario/Downloads/FAMOS_Datenformat_ab61.pdf")
 
 print(str(eatraw.validity()))