Added scaling of transverse coordinates in update_p0c

giadarol · giadarol · commit e14a8e345e7e · 2021-08-11T15:57:37.000+02:00
diff --git a/xtrack/particles/particles.py b/xtrack/particles/particles.py
@@ -386,6 +386,9 @@ def gen_local_particle_api(mode='no_local_copy'):
 
     LocalParticle_update_delta(part, new_delta);
 
+    LocalParticle_scale_px(part, old_p0c/new_p0c_value);
+    LocalParticle_scale_py(part, old_p0c/new_p0c_value);
+
 }
 '''
 

Original file line number	Diff line number	Diff line change
`@@ -386,6 +386,9 @@ def gen_local_particle_api(mode='no_local_copy'):`
`386`	`386`
`387`	`387`	`LocalParticle_update_delta(part, new_delta);`
`388`	`388`
	`389`	`+ LocalParticle_scale_px(part, old_p0c/new_p0c_value);`
	`390`	`+ LocalParticle_scale_py(part, old_p0c/new_p0c_value);`
	`391`	`+`
`389`	`392`	`}`
`390`	`393`	`'''`
`391`	`394`