Homec4science
Diffusion tamaas (features/netcdf-mpi) Browse

tamaas/tests/test_fftransform.py_no
4e3edbeeee28features/netcdf-mpi

Last Change
Raw File

test_fftransform.py_no
View Options

#!/usr/bin/env python
# coding: utf-8
# -----------------------------------------------------------------------------
# @author Lucas Frérot <lucas.frerot@epfl.ch>
#
# @section LICENSE
#
# Copyright (©) 2016 EPFL (Ecole Polytechnique Fédérale de
# Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
# Solides)
#
# Tamaas is free software: you can redistribute it and/or modify it under the
# terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
# -----------------------------------------------------------------------------
import sys
import numpy as np
import matplotlib.pyplot as plt
from util_test import import_tamaas
def main():
tm = import_tamaas()
size = 512
# Generate random rough surface
SG = tm.SurfaceGeneratorFilterFFT()
SG.getGridSize().assign(size)
SG.getHurst().assign(0.8)
SG.getRMS().assign(1.);
SG.getQ0().assign(4);
SG.getQ1().assign(4);
SG.getQ2().assign(32);
SG.getRandomSeed().assign(156);
SG.Init()
surface = SG.buildSurface()
# Compute fourier transform through numpy
numpy_spectral = np.fft.fft2(surface)
# Compute fourier tranform through tamaas
tamaas_spectral = np.zeros((size, size))
#transform = tm.FFTransformReal(size, surface, transform)
return 0
if __name__ == '__main__':
sys.exit(main())
c4science · Help