new_adhesion.py
99e8b8cff2f4maxwell_viscoelastic…

new_adhesion.py
View Options

			import tamaas as tm
			import sys
			import matplotlib.pyplot as plt
			import numpy as np

			# Initialize threads and fftw
			tm.initialize()

			# Surface size
			n = 512

			# 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 = tm.ExponentialAdhesionFunctional(model.getBEEngine(), surface)

			adhesion_params = {
			"rho":2e-5,
			"surface_energy":2e-5
			}

			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)
			zones[tractions > 0] = 1
			zones[tractions < 0] = -1
			plt.imshow(zones, cmap='Greys')

			# Cleanup threads
			tm.finalize()

			plt.show()