From 0087a6d29433abcdd4c39f26451e2fca196ddad2 Mon Sep 17 00:00:00 2001
From: David Winogradoff <dwinogra@illinois.edu>
Date: Tue, 18 Sep 2018 13:19:10 -0500
Subject: [PATCH] edit draw cylinders function

---
 mrdna/segmentmodel.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/mrdna/segmentmodel.py b/mrdna/segmentmodel.py
index fd78e54..650e5a5 100644
--- a/mrdna/segmentmodel.py
+++ b/mrdna/segmentmodel.py
@@ -2835,3 +2835,16 @@ proc calcforces {} {
             ## material
             tclFile.write("graphics top materials on \n")
             tclFile.write("graphics top material AOEdgy \n")
+            
+            ## iterate through the model segments
+            for s in self.segments:
+                if isinstance(s,DoubleStrandedSegment):
+                    print("## dsDNA! \n")
+                    draw_cylinder(s,10,"cyan")
+                elif isinstance(s,SingleStrandedSegment):
+                    print("## ssDNA! \n")
+                    draw_cylinder(s,3,"orange")
+                else:
+                    raise Exception ("your model includes beads that are neither ssDNA nor dsDNA")
+            ## tclFile complete
+            tclFile.close()
-- 
GitLab