From f562a84316eaf408df7ed633437cdf8cfce21711 Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@rwth-aachen.de>
Date: Fri, 26 Feb 2016 18:54:20 +0100
Subject: [PATCH] getRestriction added to DFT interface

Former-commit-id: ba114e4478f80c73353310368a0d2c13838a39a7
---
 src/storage/dft/DFT.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/storage/dft/DFT.h b/src/storage/dft/DFT.h
index 22041da85..32e807197 100644
--- a/src/storage/dft/DFT.h
+++ b/src/storage/dft/DFT.h
@@ -165,6 +165,11 @@ namespace storm {
                 assert(isDependency(index));
                 return std::static_pointer_cast<DFTDependency<ValueType> const>(mElements[index]);
             }
+            
+            std::shared_ptr<DFTRestriction<ValueType> const> getRestriction(size_t index) const {
+                assert(isRestriction(index));
+                return std::static_pointer_cast<DFTRestriction<ValueType> const>(mElements[index]);
+            }
 
             std::vector<std::shared_ptr<DFTBE<ValueType>>> getBasicElements() const {
                 std::vector<std::shared_ptr<DFTBE<ValueType>>> elements;