예제 #1
0
def image(path = None):
    name = str(raw_input('Please enter variable name: '))
    session_name = str(raw_input('Please Enter session name: '))
    use_max = int(raw_input('Use max from max input file? (0 for No, 1 for Yes): '))
    use_min = int(raw_input('Use min from min input file? (0 for No, 1 for Yes): '))

    #Get the working directory
    if path == None:
        work_dir = os.getcwd()
    if path != None:
        work_dir = os.chdir(path)
        work_dir = os.getcwd()
        
    #Get end time variable
    i = 0
    path = os.path.exists(work_dir + "/batch/" + name + "_batch_%d.vts" % i)
    while path == True:
        i +=1
        path = os.path.exists(work_dir + "/batch/" + name + "_batch_%d.vts" % i)
    t = i
    
    #Make dir for storing images
    if not os.path.exists("images"):
        os.makedirs("images")

    #Set the image dir
    img_dir = work_dir + "/images"
    
    #Launch VisIt
    sys.path.insert(0,visual.visit_dir)
    import visit
    visit.Launch(vdir=visual.visit_bin)
    
    #Max min
    if use_max == 0:
        max = False
        min = False
    if use_min == 0:
        max = False
        min = False
    if (use_max == 1) or (use_min == 1):
        #Import the max and min values for the data
        mm_array = np.loadtxt('max_min_' + name +'.txt')
    if use_max == 1:
        max = mm_array[0]
    if use_min == 1:
        min = mm_array[1]

#    Launch Visit Session
    session_path = work_dir + '/' + session_name + '.session'
    # Export an image sequence of data
    visual.draw_vtk(session_path,img_dir,name,t,session_name,max,min)
예제 #2
0
def draw(name,path = None):
#    dir of data
    if path == None:
        work_dir = os.getcwd()
    if path != None:
        work_dir = os.chdir(path)
        work_dir = os.getcwd()
    
    #Get end time variable
    i = 0
    path = os.path.exists("batch/" + name + "_batch_%d.vts" % i)
    while path == True:
        i +=1
        path = os.path.exists("batch/" + name + "_batch_%d.vts" % i)
    
    t = i
    #Make dir for storing images
    if not os.path.exists("images"):
            os.makedirs("images")
    
    #Set the image dir
    img_dir = work_dir + "/images"
    
    #Launch VisIt
    sys.path.insert(0,visual.visit_dir)
    import visit
    visit.Launch(vdir=visual.visit_bin)
    
    print "Please orientate the Plot how you would like the image sequence to appear, then save the session file"
        
    # If a max min file exists assign a value if not set Max,Min = False
    if not os.path.exists("max_min_" + name + '.txt'):
        max,min = False, False
    else:
        # Import the max and min values for the data
        mm_array = np.loadtxt('max_min_' + name + '.txt')
        max = mm_array[0]
        min = mm_array[1]
    
    print 'max',max
    print 'min',min    

    #Launch VisIt
    sys.path.insert(0,visual.visit_dir)
    import visit
        
    # View the VTK files so that the user can reorientate the Plot
    session_path,session_name = visual.view_vtk(work_dir,name,max,min)
    print session_path
       
    use_max = int(raw_input('Use max from max input file? (0 for No, 1 for Yes): '))
    use_min = int(raw_input('Use min from min input file? (0 for No, 1 for Yes): '))
    
    # Max min
    if use_max == 0:
        max = False
        min = False
    if use_min == 0:
        max = False
        min = False
    if (use_max == 1) or (use_min == 1):
        # Import the max and min values for the data
        mm_array = np.loadtxt('max_min_' + name +'.txt')
    if use_max == 1:
        max = mm_array[0]
    if use_min == 1:
        min = mm_array[1]  
    
    # Export an image sequence of the orientated data
    visual.draw_vtk(session_path,img_dir,name,t,session_name,max,min)