complex.lhs

来自「麻省理工的计算光子晶体的程序」· LHS 代码 · 共 20 行

\begin{code}
module Complex ( get_cmp_part, cmp, loop_complex, sqr_complex, for_complex ) where

import StepGen
\end{code}

\begin{code}
get_cmp_part num = ("cmp")|?| ("imag("<<num<<")") |:| ("real("<<num<<")")

cmp = casedefined ["is_real"] (\_-> ("is_real")|?|"0"|:|(("cmp")|?|"1"|:|"0"))
      $ casedefined ["cmp"] (\_-> ("cmp")|?|"1"|:|"0") $ ("cmp"<<"")
loop_complex job =
    ifelse_ "is_real" realjob (for_true_false "cmp" $ docode [job])
        where realjob = declare "cmp" False $ docode [job]
for_complex job = doblock "DOCMP" job

sqr_complex :: Expression -> Expression
sqr_complex e = ("is_real") |?| (e |*| e)
                            |:| (sum_true_false "cmp" $ e |*| e)
\end{code}