Homec4science
Diffusion tamaas (maxwell_viscoelastic-branch) Browse

tamaas/examples/new_adhesion.py
d39acbc1cd9emaxwell_viscoelastic…

Last Change
Raw File

new_adhesion.py
View Options

import tamaas as tm
import sys
import matplotlib.pyplot as plt
import numpy as np
import argparse
class AdhesionPython(tm.Functional):
def __init__(self, engine, rho, gamma):
tm.Functional.__init__(self, engine)
self.rho = rho
self.gamma = gamma
def computeF(self, gap, pressure):
return -self.gamma * np.sum(np.exp(-gap / self.rho))
def computeGradF(self, gap, gradient):
gradient += self.gamma * np.exp(-gap / self.rho) / self.rho
parser = argparse.ArgumentParser()
parser.add_argument('--local-functional', dest="py_adh", action="store_true",
help="use the adhesion functional written in python")
args = parser.parse_args()
# Initialize threads and fftw
tm.initialize()
# Surface size
n = 1024
# Surface generator
sg = tm.SurfaceGeneratorFilter2D()
sg.setSizes([n, n])
sg.setRandomSeed(0)
# Spectrum
spectrum = tm.Isopowerlaw2D()
sg.setFilter(spectrum)
# Parameters
params = {
"Q0": 16,
"Q1": 16,
"Q2": 64,
"Hurst": 0.8
}
helper = tm.ParamHelper(spectrum)
helper.set(params)
# Generating surface
surface = sg.buildSurface()
#surface /= tm.SurfaceStatistics.computeSpectralRMSSlope(surface)
surface /= n
#print(spectrum.rmsSlopes())
#print(tm.SurfaceStatistics.computeRMSSlope(surface))
plt.imshow(surface)
# Creating model
model = tm.ModelFactory.createModel(tm.model_type_basic_2d, [1., 1.], [n, n])
# Solver
solver = tm.PolonskyKeerRey(model, surface, 1e-12,
tm.PolonskyKeerRey.gap,
tm.PolonskyKeerRey.gap)
adhesion_params = {
"rho":2e-3,
"surface_energy":2e-5
}
if args.py_adh:
adhesion = AdhesionPython(model.getBEEngine(),
adhesion_params["rho"],
adhesion_params["surface_energy"])
else:
adhesion = tm.ExponentialAdhesionFunctional(model.getBEEngine(), surface)
adhesion.setParameters(adhesion_params)
solver.addFunctionalTerm(adhesion)
# Solve for target pressure
g_target = 5e-2
solver.solve(g_target)
tractions = model.getTraction()
plt.figure()
plt.imshow(tractions)
plt.colorbar()
plt.figure()
zones = np.zeros_like(tractions)
tol = 1e-6
zones[tractions > tol] = 1
zones[tractions < -tol] = -1
plt.imshow(zones, cmap='Greys')
# Cleanup threads
tm.finalize()
plt.show()
c4science · Help