Add Bmad-X Drift Dipole and Transverse Deflecting Cavity

[jp-ga]

Description

Adds Bmad-X tracking routines to Drift and Dipole, and adds a new Element: a TransverseDeflectingCavity

Motivation and Context

• I have raised an issue to propose this change (required for new features and bug fixes)

Types of changes

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)
• Documentation (update in the documentation)

Checklist

• I have updated the changelog accordingly (required).
• My change requires a change to the documentation.
• I have updated the tests accordingly (required for a bug fix or a new feature).
• I have updated the documentation accordingly.
• I have reformatted the code and checked that formatting passes (required).
• I have have fixed all issues found by flake8 (required).
• I have ensured that all pytest tests pass (required).
• I have run pytest on a machine with a CUDA GPU and made sure all tests pass (required).
• I have checked that the documentation builds (required).

Note: We are using a maximum length of 88 characters per line.

[jank324]

So as far as I'm concerned, this PR is fine, except for the docstring and type annotation that's missing. @jp-ga, I think you are in the best position to add those.

I'm not sure if the plot function is an issue. @jp-ga, @cr-xu what are your thoughts?

Aside from that, I fixed an issue that always cast everything to torch.float64. This was rooted in a general issue in Cheetah, where constant floating point numbers were cast to tensors when they didn't have to be.

[jp-ga]

So as far as I'm concerned, this PR is fine, except for the docstring and type annotation that's missing. @jp-ga, I think you are in the best position to add those.

I'm not sure if the plot function is an issue. @jp-ga, @cr-xu what are your thoughts?

Aside from that, I fixed an issue that always cast everything to torch.float64. This was rooted in a general issue in Cheetah, where constant floating point numbers were cast to tensors when they didn't have to be.

Sounds good, I will work on the changes. I can also change the color of the TDC.