Intrabeam scattering tracking implementation

Hello Salah,

Nicely done! Let's use this interface to review your code. One major thing is that documentation is missing, but we can fix this later when you solve all the other requests. Here are my requests on your IBS implementation right now.

1. (lines 13-15) You could move these imports inside your IBS class because they are only used by the IBS part of the code.
2. (line 535) The number of partitions cannot be selected without modifying the source code.
3. (line 536) I believe your self.dt is the same as self.ring.T0.
4. (line 538) You can import the classical electron radius from scipy.constants.physical_constants. However, this should be more general. Consider a situation when the code is used to compute the IBS of a hadron/proton accelerator. In that case, would you still use classical electron radius in the IBS formulas?
5. (line 541) There's no way to select the IBS model to compute T_ibs without modifying the source code.
6. (line 551,552) Why not use the same assignment as in line 546 (beta functions).
7. (lines 563-584) All three functions are the same. The only difference is their arguments. It should be a single function f (a, b, q), and you call it in your code f (a, b, q), f (1/a, b/a, q/a), and f (1/b, b/a, q/b). This will reduce the number of lines in the code and make it more readable.
8. (line 604-612) This is tracking. It should be inside the track() function. Functions get_T_ibs() should compute the T_ibs.
9. (line 621, 622) Same as (line 551, 552)
10. (line 666-676) Same consideration as (line 563-584)
11. (line 545, 614, 716, 770) These two functions share some similarities. You probably should create a separate function that does this shared part (like averaging different positions 's'). Then you call get_T_CIMP() or get_T_Piwinski() from it.
12. (line 797-815) Same consideration as (line 563-584, line 666-676)
13. (line 854) All the tracking related functionality (manipulation of x, xp, y, yp, delta, tau) and related 'kicks' should be implemented here. Model selection should be done when the object is initialised.

mbtrack2/tracking/__init__.py

@@ -13,6 +13,7 @@ from mbtrack2.tracking.element import (
     TransverseMap,
     TransverseMapSector,
     transverse_map_sector_generator,
+    IntrabeamScattering,
 )
 from mbtrack2.tracking.excite import Sweep
 from mbtrack2.tracking.feedback import ExponentialDamper, FIRDamper