zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize via njit

Browse Source
This commit is contained in:
usov_i 2020-10-12 17:02:40 +02:00
parent 341e97887c
commit b1ac05e7f7
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyzebra/xtal.py
View File

@ -2,6 +2,7 @@ import math
import numpy as np import numpy as np
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
from numba import njit
from scipy.optimize import curve_fit from scipy.optimize import curve_fit
import pyzebra import pyzebra
@ -28,6 +29,7 @@ def z4frgn(wave, ga, nu):
return z4 return z4
@njit(cache=True)
def phimat(phi): def phimat(phi):
"""BUSING AND LEVY CONVENTION ROTATION MATRIX FOR PHI OR OMEGA """BUSING AND LEVY CONVENTION ROTATION MATRIX FOR PHI OR OMEGA
@ -66,6 +68,7 @@ def z1frnb(wave, ga, nu, om):
return z3 return z3
@njit(cache=True)
def chimat(chi): def chimat(chi):
"""BUSING AND LEVY CONVENTION ROTATION MATRIX FOR CHI """BUSING AND LEVY CONVENTION ROTATION MATRIX FOR CHI
@ -87,6 +90,7 @@ def chimat(chi):
return dum return dum
@njit(cache=True)
def z1frz3(z3, chi, phi): def z1frz3(z3, chi, phi):
"""CALCULATE Z1 = [PHI]T.[CHI]T.Z3 """CALCULATE Z1 = [PHI]T.[CHI]T.Z3
@ -122,6 +126,7 @@ def z1frmd(wave, ga, om, chi, phi, nu):
return z1 return z1
@njit(cache=True)
def det2pol(ddist, gammad, nud, x, y): def det2pol(ddist, gammad, nud, x, y):
"""CONVERTS FROM DETECTOR COORDINATES TO POLAR COORDINATES """CONVERTS FROM DETECTOR COORDINATES TO POLAR COORDINATES