Python ParticleSystemData2の例

コード例 #1

ファイル: test_particle_system_data.py プロジェクト: utilForever/CubbyFlow

def test_add_vector_data2():
    ps = pyCubbyFlow.ParticleSystemData2()
    ps.Resize(12)

    a0 = ps.AddVectorData((2.0, 4.0))
    a1 = ps.AddVectorData((9.0, -2.0))
    assert ps.numberOfParticles == 12
    assert a0 == 3
    assert a1 == 4

    as0 = np.array(ps.VectorDataAt(a0))
    for val in as0:
        assert val == [2.0, 4.0]

    as1 = np.array(ps.VectorDataAt(a1))
    for val in as1:
        assert val == [9.0, -2.0]

コード例 #2

ファイル: ParticleSystemDataTests.py プロジェクト: transpicuous/CubbyFlow-v0

    def testAddVectorData(self):
        ps = pyCubbyFlow.ParticleSystemData2()
        ps.Resize(12)

        a0 = ps.AddVectorData((2.0, 4.0))
        a1 = ps.AddVectorData((9.0, -2.0))
        self.assertEqual(ps.numberOfParticles, 12)
        self.assertEqual(a0, 3)
        self.assertEqual(a1, 4)

        as0 = np.array(ps.VectorDataAt(a0))
        for val in as0:
            self.assertEqual(val.tolist(), [2.0, 4.0])

        as1 = np.array(ps.VectorDataAt(a1))
        for val in as1:
            self.assertEqual(val.tolist(), [9.0, -2.0])

コード例 #3

ファイル: test_particle_system_data.py プロジェクト: utilForever/CubbyFlow

def test_add_scalar_data2():
    ps = pyCubbyFlow.ParticleSystemData2()
    ps.Resize(12)

    a0 = ps.AddScalarData(2.0)
    a1 = ps.AddScalarData(9.0)
    assert ps.numberOfParticles == 12
    assert a0 == 0
    assert a1 == 1

    as0 = np.array(ps.ScalarDataAt(a0))
    for val in as0:
        assert val == 2.0

    as1 = np.array(ps.ScalarDataAt(a1))
    for val in as1:
        assert val == 9.0

コード例 #4

ファイル: ParticleSystemDataTests.py プロジェクト: transpicuous/CubbyFlow-v0

    def testAddScalarData(self):
        ps = pyCubbyFlow.ParticleSystemData2()
        ps.Resize(12)

        a0 = ps.AddScalarData(2.0)
        a1 = ps.AddScalarData(9.0)
        self.assertEqual(ps.numberOfParticles, 12)
        self.assertEqual(a0, 0)
        self.assertEqual(a1, 1)

        as0 = np.array(ps.ScalarDataAt(a0))
        for val in as0:
            self.assertEqual(val, 2.0)

        as1 = np.array(ps.ScalarDataAt(a1))
        for val in as1:
            self.assertEqual(val, 9.0)

コード例 #5

ファイル: test_particle_system_data.py プロジェクト: utilForever/CubbyFlow

def test_add_particles2():
    ps = pyCubbyFlow.ParticleSystemData2()
    ps.Resize(12)

    ps.AddParticles([(1.0, 2.0), (4.0, 5.0)], [(7.0, 8.0), (8.0, 7.0)],
                    [(5.0, 4.0), (2.0, 1.0)])

    assert ps.numberOfParticles == 14
    p = np.array(ps.positions)
    v = np.array(ps.velocities)
    f = np.array(ps.forces)

    assert [1.0, 2.0] == p[12]
    assert [4.0, 5.0] == p[13]
    assert [7.0, 8.0] == v[12]
    assert [8.0, 7.0] == v[13]
    assert [5.0, 4.0] == f[12]
    assert [2.0, 1.0] == f[13]

コード例 #6

ファイル: ParticleSystemDataTests.py プロジェクト: transpicuous/CubbyFlow-v0

    def testAddParticles(self):
        ps = pyCubbyFlow.ParticleSystemData2()
        ps.Resize(12)

        ps.AddParticles([(1.0, 2.0), (4.0, 5.0)], [(7.0, 8.0), (8.0, 7.0)],
                        [(5.0, 4.0), (2.0, 1.0)])

        self.assertEqual(ps.numberOfParticles, 14)
        p = np.array(ps.positions)
        v = np.array(ps.velocities)
        f = np.array(ps.forces)

        self.assertEqual([1.0, 2.0], p[12].tolist())
        self.assertEqual([4.0, 5.0], p[13].tolist())
        self.assertEqual([7.0, 8.0], v[12].tolist())
        self.assertEqual([8.0, 7.0], v[13].tolist())
        self.assertEqual([5.0, 4.0], f[12].tolist())
        self.assertEqual([2.0, 1.0], f[13].tolist())

コード例 #7

ファイル: test_particle_system_data.py プロジェクト: utilForever/CubbyFlow

def test_init2():
    ps = pyCubbyFlow.ParticleSystemData2()
    assert ps.numberOfParticles == 0

    ps2 = pyCubbyFlow.ParticleSystemData2(100)
    assert ps2.numberOfParticles == 100

コード例 #8

ファイル: test_particle_system_data.py プロジェクト: utilForever/CubbyFlow

def test_resize2():
    ps = pyCubbyFlow.ParticleSystemData2()
    ps.Resize(12)
    assert ps.numberOfParticles == 12

コード例 #9

ファイル: ParticleSystemDataTests.py プロジェクト: transpicuous/CubbyFlow-v0

    def testInit(self):
        ps = pyCubbyFlow.ParticleSystemData2()
        self.assertEqual(ps.numberOfParticles, 0)

        ps2 = pyCubbyFlow.ParticleSystemData2(100)
        self.assertEqual(ps2.numberOfParticles, 100)

コード例 #10

ファイル: ParticleSystemDataTests.py プロジェクト: transpicuous/CubbyFlow-v0

 def testResize(self):
     ps = pyCubbyFlow.ParticleSystemData2()
     ps.Resize(12)
     self.assertEqual(ps.numberOfParticles, 12)