def testVideoGeneration(): samp = loadBVHFile(os.path.join(os.path.dirname(__file__), '../system/walk.bvh'), 0.01) spl = SplinedBodyModel(samp) positionEstimator = ContactTrackingKalmanFilter( SampledBodyModel.structureCopy(samp), initialTime = spl.startTime, initialPosition = spl.position(spl.startTime), initialVelocity = spl.velocity(spl.startTime)) dt = 0.01 sampleTimes = np.arange(spl.startTime + dt, spl.endTime, dt) for t in sampleTimes: data = [{'jointName' : p.name, 'linearAcceleration' : p.acceleration(t)} for p in spl.joints] positionEstimator(data, t) outdir = tempfile.mkdtemp() filename = os.path.join(outdir, 'movie.avi') autoPositionCamera() renderSolenoidCoil( SolenoidMagneticField(200,20,0.05,0.2, AffineTransform())) createVideo(filename, [BodyModelRenderer(spl), VelocityVectorRenderer(spl.getPoint('ltoes_end')), ContactProbabilityRenderer(spl)], spl.startTime, spl.startTime + 1, 3, (320, 200)) assert os.path.exists(filename) shutil.rmtree(outdir)