})

domain.set_quantity('stage', quantities['stage'])

#---------
# Evolution
t0 = time.time()

try:
    domain.initialise_visualiser(rect=rect)
    #domain.visualiser.coloring['stage'] = True
    #domain.visualiser.scale_z['stage'] = 0.2
except:
    print 'No visualiser'

from norms import linf_norm

yieldstep = 1
finaltime = 200

#yieldstep = 1000
#finaltime = 50000

for t in domain.evolve(yieldstep=yieldstep, finaltime=finaltime):
    if myid == 0:
        domain.write_time()
    print_test_stats(domain, tri_full_flag)

if myid == 0:
    print 'That took %.2f seconds' % (time.time() - t0)
domain.set_quantity('stage', quantities['stage'])

#---------
# Evolution
t0 = time.time()


try:
    domain.initialise_visualiser(rect=rect)
    #domain.visualiser.coloring['stage'] = True
    #domain.visualiser.scale_z['stage'] = 0.2
except:
    print 'No visualiser'


from norms import linf_norm

yieldstep = 1
finaltime = 200

#yieldstep = 1000
#finaltime = 50000

for t in domain.evolve(yieldstep = yieldstep, finaltime = finaltime):
    if myid == 0:
        domain.write_time()
    print_test_stats(domain, tri_full_flag)

if myid == 0:
    print 'That took %.2f seconds' %(time.time()-t0)
visualise = False
if visualise:
    from anuga.visualiser import RealtimeVisualiser
    vis = RealtimeVisualiser(domain)
    vis.render_quantity_height("stage", zScale=5.0, dynamic=True)
    vis.colour_height_quantity('stage', (0.2, 0.2, 0.8))
    vis.start()
    import time
    time.sleep(2.0)

if myid == 0:
    import time
    t0 = time.time()

for t in domain.evolve(yieldstep=5.0, finaltime=30.0):
    if myid == 0:
        domain.write_time()

    if visualise:
        vis.update()

if visualise: vis.evolveFinished()

# Output some computation statistics

if myid == 0:
    print 'That took %.2f seconds' % (time.time() - t0)
    print 'Communication time %.2f seconds'\
          %domain.communication_time
    print 'Reduction Communication time %.2f seconds'\
if visualise:
    from anuga.visualiser import RealtimeVisualiser
    vis = RealtimeVisualiser(domain)
    vis.render_quantity_height("stage", zScale = 5.0, dynamic=True)
    vis.colour_height_quantity('stage', (0.2, 0.2, 0.8))
    vis.start()
    import time
    time.sleep(2.0)
    


if myid == 0:
    import time
    t0 = time.time()

for t in domain.evolve(yieldstep = 5.0, finaltime = 30.0):
    if myid == 0:
        domain.write_time()
        
    if visualise:
        vis.update()						

        

if visualise: vis.evolveFinished()
    

# Output some computation statistics

if myid == 0:
    print 'That took %.2f seconds' %(time.time()-t0)