windtunnelsimulator.py 文件源码

def nextFrame(self, arg):
        # (arg is the frame number, which we don't need)
        self.frame_c += 1
        print("computed frame:" + str(self.frame_c))
        if WindSim.performanceData and (arg % 30 == 0) and (arg > 0):
            endTime = clock()
            print ("Took {0} seconds".format(endTime - self.startTime))
            print ("%1.1f" % (30 / (endTime - self.startTime)),
                   'frames per second')
            self.startTime = endTime
            if (WindSim.save):
                frameName = 'output/frame%04d.png' % arg
                plt.savefig(frameName)
                self.frameList.write(frameName + '\n')
        for step in range(WindSim.step_range):
            self.stream()
            self.collide()
        self.fluidImage.set_array(self.curl(WindSim.ux, WindSim.uy))
        return (self.fluidImage, self.barrierImage)