From 2adb068258ccaa7dc189a68037693e8bd2eb5f9a Mon Sep 17 00:00:00 2001
From: Vytautas Jancauskas <unaudio@gmail.com>
Date: Tue, 27 Apr 2021 13:06:53 +0200
Subject: [PATCH] add get_distribution to SCAnalysisResults

---
 easyvvuq/analysis/sc_analysis.py | 42 ++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/easyvvuq/analysis/sc_analysis.py b/easyvvuq/analysis/sc_analysis.py
index 4962a122f..574565528 100644
--- a/easyvvuq/analysis/sc_analysis.py
+++ b/easyvvuq/analysis/sc_analysis.py
@@ -34,6 +34,32 @@
 __license__ = "LGPL"
 
 
+class SCDistribution():
+    def __init__(self):
+        pass
+
+    def pdf(self, x_data):
+        pass
+
+    def cdf(self, x_data):
+        pass
+
+    def fwd(self, x_data):
+        pass
+
+    def inv(self, q_data):
+        pass
+
+    def sample(self):
+        pass
+
+    def mom(self):
+        pass
+
+    def ttr(self, kloc):
+        pass
+
+
 class SCAnalysisResults(AnalysisResults):
     def _get_sobols_first(self, qoi, input_):
         raw_dict = AnalysisResults._keys_to_tuples(self.raw_data['sobols_first'])
@@ -58,6 +84,22 @@ def _describe(self, qoi, statistic):
         else:
             raise NotImplementedError
 
+    def get_distribution(self, qoi):
+        """Returns a distribution for the given qoi.
+
+        Parameters
+        ----------
+        qoi: str
+            QoI name
+        
+        Returns
+        -------
+        A ChaosPy PDF
+        """
+        if qoi not in self.qois:
+            raise RuntimeError('no such quantity of interest - {}'.format(qoi))
+        return SCDistribution(self.raw_data)
+
 
 class SCAnalysis(BaseAnalysisElement):