1 files changed, 6 insertions, 0 deletions
diff --git a/dpd/src/Model.py b/dpd/src/Model.py
index 67feeb6..b2c303f 100644
--- a/dpd/src/Model.py
+++ b/dpd/src/Model.py
@@ -2,6 +2,12 @@
 from src.Model_Poly import Poly
 from src.Model_Lut import Lut
 
+def select_model_from_dpddata(dpddata):
+    if dpddata[0] == 'lut':
+        return Lut
+    elif dpddata[0] == 'poly':
+        return Poly
+
 # The MIT License (MIT)
 #
 # Copyright (c) 2017 Andreas Steger