# -*- coding: utf-8 -*-				# 日本語のコメントに必要
from mumax2 import *				# mumax2のインポート
import random					# randomのインポート

# 正方形+磁気異方性

# セル数の設定
# 2のべき乗がベストです。
Nx = 128
Ny = 128
Nz = 1
setgridsize(Nx, Ny, Nz)

# 試料サイズ(メートル)
sizeX = 500e-9
sizeY = 500e-9
sizeZ = 3.4e-9
setcellsize(sizeX/Nx, sizeY/Ny, sizeZ/Nz)

# モジュールの読み込み
load('micromagnetism')
load('solver/rk12')				# adaptive Euler-Heun solver
load('anisotropy/uniaxial')			# 異方性を設定するために必要

# solverの設定
setv('dt', 1e-15)				# inital time step
setv('m_maxerror', 1./3000)			# maximum error per step

# 物質定数の設定
setv('Msat', 6.6e5)				# 飽和磁化 Msat
setv('Aex', 1e-11)				# 交換定数 Aex
setv('alpha', 1)				# ダンピング定数 α
setv('Ku', 4.1e5)				# 磁気異方性定数 Ku
setv('anisU', [0,0,1])				# 磁化容易軸の方向ベクトル

# 初期磁化をランダムに設定
mv = makearray(3, Nx, Ny, Nz)
for m in range(Nx):
	for n in range(Ny):
		for o in range(Nz):
			mv[0][m][n][o] = random.uniform(-1, 1)
			mv[1][m][n][o] = random.uniform(-1, 1)
			mv[2][m][n][o] = random.uniform(-1, 1)
setarray('m', mv)

saveas("m", "png", [], "initial.png")		# png形式で磁化配列mを保存
saveas("m", "omf", ["Text"], "initial.omf")	# omf形式で磁化配列mを保存

# 安定状態までシミュレーションを走らせる
run_until_smaller('maxtorque', 1e-3 * gets('gamma') * gets('msat'))

saveas("m", "png", [], "finish.png")		# png形式で磁化配列mを保存
saveas("m", "omf", ["Text"], "finish.omf")	# omf形式で磁化配列mを保存

# 終了
sync()