myprojects-hg-reborn

Fork

#! /usr/bin/env python
from fipy import *

nx = 200
ny = nx
dx = 1./nx
dy = dx
L = dx * nx
mesh = Grid2D(dx=dx, dy=dy, nx=nx, ny=ny)
phi = CellVariable(name = "solution variable",\
                   mesh = mesh,\
                   value = 0.)

D = 1.

valueTopLeft = 0.2
valueBottomRight = 1


valueTopRight = 0.2
valueBottomLeft = 1



x, y = mesh.getFaceCenters()
facesTopLeft = ((mesh.getFacesLeft() )  \
                | (mesh.getFacesTop()))
facesBottomRight = ((mesh.getFacesRight())\
                    | (mesh.getFacesBottom()))


facesTopRight = ((mesh.getFacesLeft() )  \
                | (mesh.getFacesTop() ))


facesBottomLeft = ((mesh.getFacesRight() )\
                    | (mesh.getFacesBottom() ))



BCs = (FixedValue(faces=facesTopLeft, value=valueTopLeft),\
       FixedValue(faces=facesTopRight, value=valueTopRight),\
       FixedValue(faces=facesBottomRight, value=valueBottomRight),\
       FixedValue(faces=facesBottomLeft, value=valueBottomLeft))


if __name__ == '__main__':
    viewer = Viewer(vars=phi, datamin=0., datamax=1.)




DiffusionTerm().solve(var=phi, \
                      boundaryConditions = BCs)
if __name__ == '__main__':
    viewer.plot()


# print numerix.allclose(phi(((L,), (0,))), valueBottomRight, atol = 1e-2)

if __name__ == '__main__':
    raw_input("Implicit steady-state diffusion. Press <return> to proceed...")

print "So far so good"