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