コード例 #1
0
ファイル: test.py プロジェクト: ottbot/Image-deformation
    def test_dS_reshape(self):
        im = Immersion(50,50)

        U = np.zeros(im.mat_shape)

        dSv = im.calc_dS(U)

        dSm = np.reshape(dSv, im.mat_shape)
        
        for n in xrange(50):
            npt.assert_allclose(dSm[:,n],im.dS[n].vector().array(), 
                                rtol=1e-10, atol=1e-12)
コード例 #2
0
ファイル: test.py プロジェクト: ottbot/Image-deformation
    def test_template_target_does_not_change(self):
        im = Immersion(100,10)
        
        qA = 1.0*im.qA.vector().array()
        qB = 1.0*im.qA.vector().array()

        U = 0.1*np.ones(im.mat_shape)

        im.calc_S(U)

        npt.assert_allclose(qA, im.qA.vector().array(), rtol=1e-10, atol=1e-12, 
                            err_msg="qA changes after calculating S!!")
        npt.assert_allclose(qB, im.qA.vector().array(), rtol=1e-10, atol=1e-12,
                                err_msg="qA changes after calculating S!!")
コード例 #3
0
ファイル: test.py プロジェクト: ottbot/Image-deformation
    def test_matrix_to_coeff_conversion(self):
        im = Immersion(50,50)
        
        U = np.random.rand(*im.mat_shape)

        Uc = im.matrix_to_coeffs(U)

        U2 = im.coeffs_to_matrix(Uc)


        for n in xrange(50):
            npt.assert_allclose(np.sum(U[:,n]), np.sum(Uc[n].vector().array()), 
                                rtol=1e-10, atol=1e-12)
            

        npt.assert_allclose(U, U2, rtol=1e-10, atol=1e-12)
コード例 #4
0
ファイル: test.py プロジェクト: ottbot/Image-deformation
    def test_derivative(self):
        N = 100
        M = 10

        im = Immersion(M,N)
        
        u = dol.Expression(('cos(x[0])/10.0','cos(x[0])/10.0'))
        #u = dol.Expression(('0.1','0.1'))
        u = dol.interpolate(u, im.V)

        U = np.zeros(im.mat_shape)

        u_scaler = 0.0

        for n in xrange(im.N):
            U[:,n] = u_scaler * u.vector().array()
            
        S  = im.calc_S(U)
        
        dSarr = np.reshape(im.calc_dS(U),im.mat_shape)

        vdS = 0

        v = dol.Expression(('cos(x[0])/2.0','cos(x[0])/2.0'))
        #v = dol.Expression(('pow(x[0],2)/3.0','pow(x[0],2)/3.0'))
        #v = dol.Expression(('x[0]','x[0]'))
        #v = dol.Expression(('x[0]*0.001','x[0]*0.001'))
        v = dol.interpolate(v, im.V)


        s_vdS = 0
        s_v = 0

        for dS in im.matrix_to_coeffs(dSarr):
            #print "--->", dS.vector().array(), v.vector().array()
            #print "--> ", np.sum(dS.vector().array()), np.sum(v.vector().array())

            vdS += dol.assemble(dol.inner(v,dS)*dol.dx)

        #print s_vdS * im.dt
        vdS *= im.dt



        lims = []
        Ss = []
        Sps = []
        eps = 10.**(-sp.arange(20))
        

        for ep in eps:
            im = Immersion(M,N)
            Up = np.zeros(im.mat_shape)
            for n in xrange(im.N):
                Up[:,n] = u_scaler * u.vector().array() + ep*v.vector().array()

                
            Sp = im.calc_S(Up)
            Ss.append(S)
            Sps.append(Sp)
            lims.append((Sp - S)/ep)

        print "%s %15s %12s %12s %15s %15s" % ("Eps","LHS","RHS","S","Sp", "Sp - S")
        for n in xrange(len(eps)):
            print "%.0e  %15.6f  %12.6f  %12.7f  %12.7f %12.7f" % \
                (eps[n], vdS, lims[n], Ss[n], Sps[n], Sps[n] - Ss[n])