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()