From e92241c65af74f673935b5cc8c94d456b6d2f058 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= <benoitsierro@bluewin.ch>
Date: Tue, 19 Oct 2021 14:50:05 +0200
Subject: [PATCH] Operator coded, not implemented

---
 src/scgenerator/physics/properties.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/scgenerator/physics/properties.py b/src/scgenerator/physics/properties.py
index 442b072..4d82dc2 100644
--- a/src/scgenerator/physics/properties.py
+++ b/src/scgenerator/physics/properties.py
@@ -301,6 +301,17 @@ class EnvelopeNonLinearOperator(AbstractNonLinearOperator):
         raman_op: AbstractRaman,
     ):
         self.gamma_op = gamma_op
+        self.ss_op = ss_op
+        self.spm_op = spm_op
+        self.raman_op = raman_op
+
+    def __call__(self, state: CurrentState) -> np.ndarray:
+        return (
+            -1j
+            * self.gamma_op(state)
+            * (1 + self.ss_op(state))
+            * np.fft.fft(state.field * (self.spm_op(state) + self.raman_op(state)))
+        )
 
 
 ##################################################