def __init__(self, paths):
visit.Launch()
if isinstance(paths, list):
try: #Preventing crashing of scripts because only 1/100 files didn't open
for path in paths:
visit.OpenDatabase(path)
except:
print("Invalid file path")
else:
visit.OpenDatabase(paths)
def test_visit():
import visit
VISIT_ARGS = os.environ.get('VISIT_ARGS', ["-nosplash"])
if isinstance(VISIT_ARGS, str):
VISIT_ARGS = VISIT_ARGS.split()
for arg in VISIT_ARGS:
visit.AddArgument(arg)
visit.Launch()
print "visit is installed in", os.path.dirname(visit.__file__)
print "visit version", visit.Version()
visit.ResetView()
v3D = visit.GetView3D()
v3D.SetViewUp(0, 0, 1)
v3D.SetViewNormal(-0.5, -0.8, 0.4)
v3D.SetImageZoom(1.0)
visit.SetView3D(v3D)
aa = visit.AnnotationAttributes()
aa.SetAxesType(2) # outside edges
aa.SetDatabaseInfoFlag(False)
aa.SetUserInfoFlag(False)
visit.SetAnnotationAttributes(aa)
t = visit.CreateAnnotationObject("Text2D")
t.SetText("Hello VisIt!")
t.SetPosition(0.4, 0.9) # (0,0) is lower left corner
t.SetFontFamily(0) # 0: Arial, 1: Courier, 2: Times
t.SetWidth(0.25) # 25%
t.SetTextColor((0, 0, 0))
t.SetUseForegroundForTextColor(False)
t.SetVisible(True)
visit.OpenDatabase("tmp_.vtk")
visit.AddPlot("Mesh", "scalars")
ma = visit.MeshAttributes()
visit.SetPlotOptions(ma)
visit.AddPlot("Pseudocolor", "scalars")
pa = visit.PseudocolorAttributes()
visit.SetPlotOptions(pa)
visit.RedrawWindow()
visit.DrawPlots()
wait()
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from MeshClass import *
from DataClass import *
from CaseClass import *
Loc = "OpenFoamCases/"
Files = ['elbow/', 'Coutte_Flow_40x40-FixedTemperature/', 'forwardStep/']
Loc = Loc + Files[0]
# Reading Mesh Data and Evaluating the initail points
case = Case()
case.loadOpenFoamFile(Loc)
case.print_output_vtk()
# visit
sys.path.append("/home/neo/visit/2.11.0/linux-x86_64/lib/site-packages")
import visit as vs
vs.Launch()
vs.OpenDatabase("Results/results-time-1.vtk")
vs.AddPlot("Pseudocolor", 'X-Velocity')
vs.AddPlot("Pseudocolor", "Pressure")
vs.SetActivePlots(0)
vs.HideActivePlots()
vs.DrawPlots()
d = input('Press anything to quit')
import os
import visit as Vi
try:
Vi.Launch() # Here to allow import of other modules.
except Exception:
pass
def GrabImagesFromSessions():
"""
A function that loads session files
and grabs images from it.
"""
Directory = str(os.getcwd()) + "/../Sessions/XML_Edited"
for file in sorted(os.listdir(Directory)):
if file.endswith(".session"):
# Find each session file and restore it in VisIt
Vi.RestoreSession(os.path.join(Directory, file), 0)
# Save directory defined by session.
Vi.SaveWindow()
# need to run with Python 2 (e.g., python2)
import sys
sys.path.append(
"/home/robertsj/opt/visit2_12_3.linux-x86_64/2.12.3/linux-x86_64/lib/site-packages/"
)
import visit
visit.Launch()
visit.ShowAllWindows()
visit.OpenDatabase("test.vtk")
visit.AddPlot("Pseudocolor", "scalars")
visit.DrawPlots()
visit.SaveWindow()
View2DAtts = visit.View2DAttributes()
View2DAtts.fullFrameActivationMode = View2DAtts.On
# View2DAtts.fullFrameAutoThreshold = 100
visit.SetView2D(View2DAtts)
visit.ResetView()
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.axes2D.yAxis.grid = 0
AnnotationAtts.axes2D.xAxis.grid = 0
AnnotationAtts.axes2D.xAxis.title.userTitle = 1
AnnotationAtts.axes2D.xAxis.title.userUnits = 1
AnnotationAtts.axes2D.xAxis.title.title = "ct-z"
AnnotationAtts.axes2D.xAxis.title.units = "m"
AnnotationAtts.axes2D.yAxis.title.userTitle = 1
AnnotationAtts.axes2D.yAxis.title.userUnits = 1
AnnotationAtts.axes2D.yAxis.title.title = "gamma"
AnnotationAtts.axes2D.yAxis.title.units = ""
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
# AnnotationAtts.legendInfoFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
visit.DrawPlots()
if __name__ == '__main__':
pBaseName = sys.argv[1]
eDB, fDB, iDB, localPowerAllDB = getDBNames.getDBNames(pBaseName)
visit.Launch(vdir=localVisItDir)
binPhase(eDB)
visit.OpenGUI()
import numpy as np
import re
import glob
import sys
sys.path.insert(
0,
"/home/dventuri/visit/visit2_13_3.linux-x86_64/2.13.3/linux-x86_64/lib/site-packages/"
)
import visit
visit.Launch(vdir="/home/dventuri/visit/visit2_13_3.linux-x86_64/bin")
numbers = re.compile(r'(\d+)')
def numericalSort(value):
parts = numbers.split(value)
parts[1::2] = map(int, parts[1::2])
return parts
input_folder = "/mnt/c/Users/Pichau/Documents/run/CVaras2017_riser/"
output_folder = input_folder + "results/png/comp/"
cases = [
# 'case01',
# 'case02',
# 'case03',
'case04'
]
def queryAndPlot():
SOS = SSOS()
print "Initializing SOS..."
SOS.init()
print "DONE init SOS..."
sql_string = """
SELECT MAX(frame) FROM viewCombined WHERE viewCombined.value_name LIKE "lulesh.time"
;
"""
results, col_names = SOS.query(sql_string, "localhost",
os.environ.get("SOS_CMD_PORT"))
max_cycle = int(results[0][0])
print "Max cycle: " + str(max_cycle)
#####
#
# Get the list of field names for non-string values.
#
sql_string = """
SELECT
DISTINCT value_name
FROM viewCombined
WHERE value_type != 3
AND frame = """ + str(max_cycle) + """
;
"""
results, col_names = SOS.query(sql_string, "localhost",
os.environ.get("SOS_CMD_PORT"))
numeric_fields = dict()
numeric_fields['name'] = [el[0] for el in results]
name_count = len(numeric_fields['name'])
print str(name_count) + " unique names."
#
#####
filenames = []
for c in range(0, (max_cycle + 1)):
print "******* CYCLE " + str(c) + " *********"
#####
#
# Compose a query with the unique numeric fields as columns:
#
sql_string = """ """
sql_string += """ SELECT """
sql_string += """ comm_rank """
# sql_string += """,frame """
for field_name in numeric_fields['name']:
sql_string += """,GROUP_CONCAT( CASE WHEN """
sql_string += ' value_name LIKE "' + field_name + '" '
sql_string += ' THEN value END) AS "' + field_name + '" '
sql_string += """, GROUP_CONCAT( CASE WHEN """
sql_string += ' value_name LIKE "lulesh.coords" '
sql_string += ' THEN value END) AS "lulesh.coords" '
sql_string += """ FROM viewCombined """
sql_string += " WHERE frame = " + str(c) + " "
sql_string += """ GROUP BY """
sql_string += """ comm_rank """
# sql_string += """,frame """
sql_string += """;"""
#print "Composite SQL statement: "
#print sql_string
#print ""
#print "Running composite query..."
results, col_names = SOS.query(sql_string, "localhost",
os.environ.get("SOS_CMD_PORT"))
#print ""
#
# Print out the results:
#
#print "=========="
#for col in col_names:
# print str(col) + " "
#print "=========="
#for row in results:
# for col_index in range(len(row)):
# print str(col_names[col_index]) + ": " + str(row[col_index])
# print "----------"
#print "=========="
#
#####
#####
#
# Build an attribute dictionary of the values.
#
attr = dict()
attr['comm_rank'] = [el[0] for el in results]
position = 1
for field_name in numeric_fields['name']:
attr[field_name] = [el[position] for el in results]
#print str(field_name) + " in position " + str(position) + " = " + str(attr[field_name])
position += 1
res_coords = [el[position] for el in results]
#print "lulesh.coords in position " + str(position) + " = " + str(res_coords)
for field_name in numeric_fields['name']:
rank = 0
for this_ranks_value in attr[field_name]:
print "comm_rank(" + str(
rank) + ")." + field_name + " = " + this_ranks_value
rank += 1
rank_max = len(attr['comm_rank'])
coords = list()
coords = [el.split() for el in res_coords]
#print attr
dset = vtk_writer.vtk_hex_data_set()
dset.clear()
dset.set_cycle(c)
for rank in range(rank_max):
fields = {}
for field_name in numeric_fields['name']:
fields[field_name] = attr[field_name][rank]
fields["rank"] = rank
hex_coords = [None] * 24
xpt = [None] * 8
ypt = [None] * 8
zpt = [None] * 8
cpt = [None] * 8
for i in range(24):
hex_coords[i] = float(coords[rank][i])
dset.add_hex(hex_coords, fields, rank)
dset.write_vtk_file()
filenames.append(dset.get_file_name())
vtk_writer.write_visit_file(filenames)
visit.AddArgument("-par")
visit.Launch()
OpenDatabase("dataset.visit")
AddPlot("Pseudocolor", "rank")
AddPlot("Mesh", "mesh")
DrawPlots()
# loop through times
tsNames = GetWindowInformation().timeSliders
for ts in tsNames:
SetActiveTimeSlider(ts)
for state in list(range(TimeSliderGetNStates()))[::10] + [0]:
SetTimeSliderState(state)
print "Setting share_power permissions on the newly created VTK files..."
subprocess.call("$PROJECT_BASE/share_power .", shell=True)
print ""
print "Sleeping for 100 seconds..."
print ""
time.sleep(100)
SOS.finalize()
print " ...DONE!"
print
def launch_visit():
# This function launches the visit window. Used only via
# a cli version. Does not launch the entire GUI
visit.Launch(vdir='/usr/local/visit/bin/')
import time
import sys
sys.path.append("/home/tbent/packages/visit2_7_0/2.7.0/linux-x86_64/lib/site-packages/")
import visit
visit.Launch('/home/tbent/packages/visit2_7_0/bin/')
filename_prefix = 'solution_'
filename_prefix = 'init_refinement_'
for i in range(12):
time.sleep(0.5)
visit.DeleteAllPlots()
visit.OpenDatabase('./data/test/' + filename_prefix + str(i) + '.0000.pvtu')
visit.DefineScalarExpression('logabsoldszx', 'log(abs(old_szx))')
# visit.AddPlot('Pseudocolor', 'logabsoldszx')
visit.AddPlot('Pseudocolor', 'vel')
# visit.AddPlot('Mesh', 'mesh')
visit.DrawPlots()
