import numpy as np
from scipy.constants import mu_0, epsilon_0

# useful params
def omega(freq):
    """Angular frequency, omega"""
    return 2.*np.pi*freq

def k(freq, sigma, mu=mu_0, eps=epsilon_0):
    w = omega(freq)
    return np.sqrt(mu * eps * w**2 - 1j * w* mu * sigma)

# Constitutive relations
def e_from_j(prob,j):
    eqLocs = prob._eqLocs
    if eqLocs is 'FE':
        MSigmaI = prob.MeSigmaI
    elif eqLocs is 'EF':
        MSigmaI = prob.MfRho
    return MSigmaI*j

def j_from_e(prob,e):
    eqLocs = prob._eqLocs
    if eqLocs is 'FE':
        MSigma = prob.MeSigma
    elif eqLocs is 'EF':
        MSigma = prob.MfRhoI
    return MSigma*e

def b_from_h(prob,h):
    eqLocs = prob._eqLocs
    if eqLocs is 'FE':
        MMu = prob.MfMuiI
    elif eqLocs is 'EF':
        MMu = prob.MeMu
    return MMu*h

def h_from_b(prob,b):
    eqLocs = prob._eqLocs
    if eqLocs is 'FE':
        MMuI = prob.MfMui
    elif eqLocs is 'EF':
        MMuI = prob.MeMuI
    return MMuI*b