preparations for issue #39

maxscheurer · Apr 17, 2017 · 46242eb · 46242eb
1 parent f42b5bb
commit 46242eb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/PyContact/gui/Plotters.py b/PyContact/gui/Plotters.py
@@ -56,6 +56,20 @@ def plot_all_contacts_figure(self, contacts):
         self.axes.set_xlabel("frame")
         self.axes.set_ylabel("score")
 
+    def plot_hbondNumber(self, contacts):
+        values = []
+        for c in contacts:
+            c.hbondFramesScan()
+
+        for frame in range(len(contacts[0].scoreArray)):
+            current = 0
+            for c in contacts:
+                current += c.hbondFrames[frame]
+            values.append(current)
+        self.axes.plot(values)
+        self.axes.set_xlabel("frame")
+        self.axes.set_ylabel("hbond number")
+
 
 class ContactPlotParameters:
     """Parameter for the contact plotter."""

diff --git a/PyContact/gui/Statistics.py b/PyContact/gui/Statistics.py
@@ -33,4 +33,5 @@ def __init__(self, data, parent=None):
 
         self.contactPlotter = ContactPlotter(None, width=4, height=2, dpi=80)
         self.contactPlotter.plot_all_contacts_figure(self.contacts)
+        # self.contactPlotter.plot_hbondNumber(self.contacts)
         self.plotGridLayout.addWidget(self.contactPlotter)