new_contact.py
3c976f436924ci-fix-static-checks

new_contact.py
View Options

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

			# 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.pressure,
			tm.PolonskyKeerRey.pressure)

			# Solve for target pressure
			p_target = 0.1
			solver.solve(p_target)

			plt.figure()
			plt.imshow(model.getTraction())
			print(model.getTraction().mean())

			# Cleanup threads
			tm.finalize()

			plt.show()