Author: bugman
Date: Thu Dec 18 16:08:32 2014
New Revision: 27134
URL: http://svn.gna.org/viewcvs/relax?rev=27134&view=rev
Log:
Significantly expanded the gnuplot script from via the correlation_matrix()
plotting API function.
This is for the structure.atomic_fluctuations user function. The output
terminal is now set to EPS,
the colour map changed from the default to a blue-red map, labels have been
added, the plot is now
square, and comments are now included throughout the script to help a user hand
modify it after
creation.
Modified:
trunk/lib/plotting/gnuplot.py
Modified: trunk/lib/plotting/gnuplot.py
URL:
http://svn.gna.org/viewcvs/relax/trunk/lib/plotting/gnuplot.py?rev=27134&r1=27133&r2=27134&view=diff
==============================================================================
--- trunk/lib/plotting/gnuplot.py (original)
+++ trunk/lib/plotting/gnuplot.py Thu Dec 18 16:08:32 2014
@@ -23,7 +23,7 @@
"""Module for data plotting using gnuplot."""
# relax module imports.
-from lib.io import open_write_file, swap_extension
+from lib.io import file_root, open_write_file, swap_extension
from lib.plotting import text
@@ -40,20 +40,83 @@
@type dir: str or None
"""
+ # The dimensions.
+ n = len(matrix)
+
# Generate the text file for loading into gnuplot.
text.correlation_matrix(matrix=matrix, labels=labels, file=file, dir=dir,
force=force)
- # The script file name, with a swapped extension.
+ # The script file name with the extension swapped.
file_name = swap_extension(file=file, ext='gnu')
# Open the script file for writing.
output = open_write_file(file_name, dir=dir, force=force)
# Set the plot type.
+ output.write("\n# Set the plot type.\n")
output.write("set pm3d map\n")
+ # Set up the terminal type and make the plot square.
+ output.write("\n# Make the plot square.\n")
+ output.write("set terminal postscript eps size 10,10 enhanced color font
'Helvetica,20' linewidth 0.1\n")
+
+ # The blue-red colour map.
+ output.write("\n# Blue-red colour map.\n")
+ colours = [
+ "#000090",
+ "#000fff",
+ "#0090ff",
+ "#0fffee",
+ "#90ff70",
+ "#ffee00",
+ "#ff7000",
+ "#ee0000",
+ "#7f0000"
+ ]
+ output.write("set palette defined (")
+ for i in range(len(colours)):
+ if i != 0:
+ output.write(", ")
+ output.write("%s \"%s\"" % (i, colours[i]))
+ output.write(")\n")
+
+ # The labels.
+ if labels != None:
+ output.write("\n# Labels.\n")
+ for axis in ['x', 'y']:
+ output.write("set %stics out " % axis)
+ if axis == 'x':
+ output.write("rotate ")
+ output.write("font \",8\" (")
+ for i in range(n):
+ if i != 0:
+ output.write(", ")
+ output.write("\"%s\" %s" % (format_enhanced(labels[i]), i))
+ output.write(")\n")
+
+ # Output to EPS by default.
+ output.write("\n# Output to EPS by default.\n")
+ output.write("set output \"%s.eps\"\n" % file_root(file))
+
# Load and show the text data.
+ output.write("\n# Load and show the text data\n")
output.write("splot \"%s\" matrix\n" % file)
# Close the file.
output.close()
+
+
+def format_enhanced(text):
+ """Convert and return the text to handle enhanced postscript.
+
+ @param text: The text to convert to enhanced mode.
+ @type text: str
+ @return: The formatted text for enhanced postscript mode.
+ @rtype: str
+ """
+
+ # Handle the '@' character.
+ text = text.replace('@', '\\\\@')
+
+ # Return the text.
+ return text
_______________________________________________
relax (http://www.nmr-relax.com)
This is the relax-commits mailing list
[email protected]
To unsubscribe from this list, get a password
reminder, or change your subscription options,
visit the list information page at
https://mail.gna.org/listinfo/relax-commits
