Merge pull request #150 from jiangyi15/dispersion_integral

feat: dispersion integral models

tf_pwa/amp/__init__.py

@@ -12,6 +12,7 @@
 from .amp import AmplitudeModel, create_amplitude
 from .base import *
 from .core import *
+from .dispersion_integral import DispersionIntegralParticle
 from .flatte import ParticleFlatte
 from .Kmatrix import KmatrixSingleChannelParticle
 from .kmatrix_simple import KmatrixSimple

tf_pwa/amp/core.py

@@ -890,11 +890,15 @@ def _get_cg_matrix(
         if out_sym:
             from sympy.physics.quantum.cg import CG
 
-            sint = lambda x: sym.simplify(_spin_int(x * 2)) / 2
+            sint = (
+                lambda x: sym.simplify(sym.sign(x) * _spin_int(abs(x) * 2)) / 2
+            )
             sqrt = lambda x: sym.sqrt(sint(x))
 
-            def my_cg_coef(a, b, c, d, e, f):
-                return CG(*list(map(sint, [a, c, b, d, e, f]))).doit()
+            def my_cg_coef(j1, j2, m1, m2, j3, m3):
+                ret = CG(*list(map(sint, [j1, m1, j2, m2, j3, m3]))).doit()
+                print(j1, j2, m1, m2, j3, m3, ret)
+                return ret
 
             ret = ret.tolist()
         for i, ls_i in enumerate(ls):