https://github.com/geodynamics/citcoms
Revision 8fd3a731b733a5dc2e98d02e37878c8545e317c2 authored by Vlad Manea on 11 March 2005, 08:24:46 UTC, committed by Vlad Manea on 11 March 2005, 08:24:46 UTC
1 parent 570c900
Tip revision: 8fd3a731b733a5dc2e98d02e37878c8545e317c2 authored by Vlad Manea on 11 March 2005, 08:24:46 UTC
*** empty log message ***
*** empty log message ***
Tip revision: 8fd3a73
exchange.py
#!/usr/bin/env python
#
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#
# <LicenseText>
#
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#
from mpi.Application import Application
import journal
class TestExchanger(Application):
def __init__(self, name="citcom"):
Application.__init__(self, name)
self.exchanger = None
return
def run(self):
layout = self.inventory.layout
layout.initialize(self)
self.findLayout(layout)
exchanger = self.exchanger
if exchanger:
self.test(exchanger)
return
def test(self, exchanger):
# testing exchanger creation
exchanger.selectModule()
exchanger.createExchanger(self)
#print exchanger.name, exchanger.exchanger
# testing boundary creation and exchange
exchanger.findBoundary()
print exchanger.name, ": boundary found"
# testing applyBoundaryConditions
exchanger.applyBoundaryConditions()
print exchanger.name, ": applyBoundaryConditions worked"
# testing initTemperature
#exchanger.initTemperature()
#print exchanger.name, ": temperature transferred"
# select dt
try:
# success if exchanger is a FGE
exchanger.fge_t
dt = 0.4
except:
# exception if exchanger is a CGE
dt = 1
# testing dt exchanging
print "%s - old dt = %g exchanged dt = %g old dt = %g" % (
exchanger.name, dt,
exchanger.module.exchangeTimestep(exchanger.exchanger, dt),
dt)
# testing exchangeSignal
steps = 2*3 + 1
for step in range(steps):
time = exchanger.stableTimestep(dt)
print "%s - step %d: %g" % (exchanger.name, step, time)
exchanger.applyBoundaryConditions()
if step == steps-1:
done = True
else:
done = False
done = exchanger.endTimestep(done)
if done:
break
return
def findLayout(self, layout):
import CitcomS.Exchanger
if layout.coarse:
self.exchanger = self.inventory.coarse
self.communicator = layout.coarse
self.all_variables = CitcomS.Exchanger.CoarsereturnE()
elif layout.fine:
self.exchanger = self.inventory.fine
self.communicator = layout.fine
self.all_variables = CitcomS.Exchanger.FinereturnE()
else:
import journal
journal.warning(self.name).log("node '%d' is an orphan" % layout.rank)
self.intercomm = layout.intercomm
self.rank = layout.rank
self.nodes = layout.nodes
self.leader = layout.leader
self.remoteLeader = layout.remoteLeader
print "%s exchanger: rank=%d leader=%d remoteLeader=%d" % (
self.exchanger.name, self.rank, self.leader, self.remoteLeader)
return
class Inventory(Application.Inventory):
import pyre.facilities
import CitcomS.Components.Exchanger as Exchanger
import CitcomS.Controller as Controller
import CitcomS.Layout as Layout
inventory = [
pyre.facilities.facility("controller", default=Controller.controller()),
pyre.facilities.facility("layout", default=Layout.layout()),
pyre.facilities.facility("coarse", default=Exchanger.coarsegridexchanger("coarse")),
pyre.facilities.facility("fine", default=Exchanger.finegridexchanger("fine")),
]
# main
if __name__ == "__main__":
import mpi
# testing Exchangermodule.so
import CitcomS.Exchanger
if not mpi.world().rank:
print CitcomS.Exchanger.copyright()
print dir(CitcomS.Exchanger)
import journal
#journal.debug("Array2D").activate()
journal.debug("Exchanger").activate()
journal.info(" X").activate()
journal.info(" proc").activate()
journal.info(" bid").activate()
app = TestExchanger("test")
app.main()
# version
__id__ = "$Id: exchange.py,v 1.15 2003/10/24 05:23:36 tan2 Exp $"
# End of file
Computing file changes ...