在mayavi锁定相机

时间:2013-06-12 10:24:04

标签: python mayavi

我正在尝试使用mayavi中的一系列数据文件制作动画。不幸的是我注意到相机没有锁定(它是缩放和缩小)。我认为它正在发生,因为我的网格的Z组件正在改变,mayavi正在尝试重新计算比例。

我该如何解决? enter image description here enter image description here enter image description here

import numpy
from mayavi import mlab

mlab.figure(size = (1024,768),bgcolor = (1,1,1))
mlab.view(azimuth=45, elevation=60, distance=0.01, focalpoint=(0,0,0))
#mlab.move(forward=23, right=32, up=12)

for i in range(8240,8243):
    n=numpy.arange(10,400,20)
    k=numpy.arange(10,400,20)
    [x,y] = numpy.meshgrid(k,n)
    z=numpy.zeros((20,20))
    z[:] = 5

    M = numpy.loadtxt('B:\\Dropbox\\Master.Diploma\\presentation\\movie\\1disk_j9.5xyz\\'+'{0:05}'.format(i)+'.txt')
    Mx = M[:,0]; My = M[:,1]; Mz = M[:,2]
    Mx = Mx.reshape(20,20); My = My.reshape(20,20); Mz = Mz.reshape(20,20);


    s = mlab.quiver3d(x,y,z,Mx, My, -Mz, mode="cone",resolution=40,scale_factor=0.016,color = (0.8,0.8,0.01))

    Mz = numpy.loadtxt('B:\\Dropbox\\Master.Diploma\\presentation\\movie\\Mzi\\' + '{0:05}'.format(i) + '.txt')
    n=numpy.arange(2.5,400,2)
    k=numpy.arange(2.5,400,2)
    [x,y] = numpy.meshgrid(k,n)

    f = mlab.mesh(x, y, -Mz/1.5,representation = 'wireframe',opacity=0.3,line_width=1)

    mlab.savefig('B:\\Dropbox\\Master.Diploma\\presentation\\movie\\figs\\'+'{0:05}'.format(i)+'.png')
    mlab.clf()
    #mlab.savefig('B:\\Dropbox\\Master.Diploma\\figures\\vortex.png')
    print(i)

mlab.show()

3 个答案:

答案 0 :(得分:4)

对于仍然对此感兴趣的任何人,您可以尝试在此上下文中包装您正在执行的任何工作,这将禁用渲染并在上下文退出后将disable_render值和摄像机视图返回到其原始状态。

with constant_camera_view():
    do_stuff()

这是班级:

class constant_camera_view(object):
def __init__(self):
    pass

def __enter__(self):
    self.orig_no_render = mlab.gcf().scene.disable_render
    if not self.orig_no_render:
        mlab.gcf().scene.disable_render = True
    cc = mlab.gcf().scene.camera
    self.orig_pos = cc.position
    self.orig_fp = cc.focal_point
    self.orig_view_angle = cc.view_angle
    self.orig_view_up = cc.view_up
    self.orig_clipping_range = cc.clipping_range

def __exit__(self, t, val, trace):
    cc = mlab.gcf().scene.camera
    cc.position = self.orig_pos
    cc.focal_point = self.orig_fp
    cc.view_angle =  self.orig_view_angle 
    cc.view_up = self.orig_view_up
    cc.clipping_range = self.orig_clipping_range

    if not self.orig_no_render:
        mlab.gcf().scene.disable_render = False
    if t != None:
        print t, val, trace
        ipdb.post_mortem(trace)

答案 1 :(得分:1)

我没有真正看到你的情节中的问题,但是在每个绘图实例插入你的观点后重置视图:

mlab.view(azimuth=45, elevation=60, distance=0.01, focalpoint=(0,0,0))

直接位于您{for循环中mlab.savefig来电之上。

答案 2 :(得分:1)

你可以在你的网格命令中使用vmin和vmax功能,如果你这样做,比例不会随你的数据而改变,你的相机应该保持原样。 像这样:

f = mlab.mesh(x, y, -Mz/1.5,representation = 'wireframe',vmin='''some value''',vmax='''some value''',opacity=0.3,line_width=1)