Stellarator module update #4082

jjwalkowiak · 2026-02-03T15:19:22Z

Description

Main changes:

coil minor radius scaling for stellarators
corrected stress equation for vacuum vessel
corrected coil height (this was actually more on the pre-process side)
stellarator module refactoring

Checklist

I confirm that I have completed the following checks:

My changes follow the PROCESS style guide
I have justified any large differences in the regression tests caused by this pull request in the comments.
I have added new tests where appropriate for the changes I have made.
If I have had to change any existing unit or integration tests, I have justified this change in the pull request comments.
If I have made documentation changes, I have checked they render correctly.
I have added documentation for my change, if appropriate.

…coils aspect ratio variation

…onsistant way a torioidal shape scaling

…rator module

…e internal variables to be more readable

This reverts commit 1d1cbca.

Merge

timothy-nunn

This is just a rough first review. Once these changes have been actioned I can do a more detailed review and get @ym1906 involved (and maybe others).

I notice that there are quite a few style violations. Most of these can be solved by using pre-commit https://ukaea.github.io/PROCESS/development/pre-commit/.

documentation/proc-pages/io/vardes.md

.gitignore

documentation/variable_descriptions.json

documentation/iteration variables.md

documentation/figure_of_merit.md

process/stellarator/denisty_limits.py

process/stellarator/divertor.py

process/stellarator/heating.py

process/stellarator/stellarator.py

timothy-nunn

A couple of other things I have noticed.

process/stellarator/heating.py

process/stellarator/neoclassics.py

process/stellarator/stellarator.py

process/stellarator/denisty_limits.py

process/stellarator/coils/quench.py

timothy-nunn · 2026-02-04T08:54:24Z

When you have actioned all of the comments please do click the 🔄 icon in the reviewer section to tell me you are ready for the next review.

You might also want to consider adding in a test case to our tests/regression/input_files... this will allow us to check the file runs to completion.

jjwalkowiak · 2026-02-04T09:26:40Z

process/stellarator/coils/coils.py:70:23: RUF052 Local dummy variable _tmarg is accessed
I have no idea what _tmarg is doing.
process/stellarator/coils/quench.py:101:5: N802 Function name calculate_vv_max_force_density_from_W7X_scaling should be lowercase
Do you really want to have w7x instead of W7X?

timothy-nunn · 2026-02-04T11:19:34Z

process/stellarator/coils/coils.py:70:23: RUF052 Local dummy variable _tmarg is accessed I have no idea what _tmarg is doing. process/stellarator/coils/quench.py:101:5: N802 Function name calculate_vv_max_force_density_from_W7X_scaling should be lowercase Do you really want to have w7x instead of W7X?

Rename _tmarg to tmarg

You can add "W7X" to this list

PROCESS/pyproject.toml

Lines 207 to 229 in ca03a41

    
           ignore-names = [ 
        
               "*PROCESS*", 
        
               "*Bt*", 
        
               "*Bp*", 
        
               "*keV*", 
        
               "*MPa*", 
        
               "*H*", 
        
               "*T*", 
        
               "*D*", 
        
               "*He*", 
        
               "*Be*", 
        
               "C", 
        
               "N", 
        
               "O", 
        
               "Ne", 
        
               "Si", 
        
               "Ar", 
        
               "Fe", 
        
               "Ni", 
        
               "Kr", 
        
               "Xe", 
        
               "W", 
        
           ]

timothy-nunn · 2026-02-04T11:21:03Z

process/stellarator/coils/quench.py

+
+def calculate_vv_max_force_density_from_W7X_scaling(rad_vv: float) -> float:
+    """Actual VV force density from scaling [MN/m^3]
+    Based on reference values from W-7X."""


Suggested change

Based on reference values from W-7X."""

Based on reference values from W-7X.

"""

timothy-nunn · 2026-02-04T11:23:00Z

process/data_structure/stellarator_variables.py


-f_a: float = None
+f_st_i_total: float = None
+"""Actual totail coil current to reference value from stella_config file"""


Suggested change

"""Actual totail coil current to reference value from stella_config file"""

"""Actual total coil current to reference value from stella_config file"""

timothy-nunn · 2026-02-04T11:23:41Z

process/stellarator/build.py

+    # This is the old version, left for now for comparison.
+    # build_variables.available_radial_space = stellarator_variables.f_r * (
+    #     stellarator_configuration.stella_config_derivative_min_lcfs_coils_dist
+    #     * stellarator_configuration.stella_config_rminor_ref
+    #     * (1 / stellarator_variables.f_aspect - 1)
+    #     + stellarator_configuration.stella_config_min_plasma_coil_distance
+    # )


Suggested change

# This is the old version, left for now for comparison.

# build_variables.available_radial_space = stellarator_variables.f_r * (

# stellarator_configuration.stella_config_derivative_min_lcfs_coils_dist

# * stellarator_configuration.stella_config_rminor_ref

# * (1 / stellarator_variables.f_aspect - 1)

# + stellarator_configuration.stella_config_min_plasma_coil_distance

# )

timothy-nunn · 2026-02-04T11:23:54Z

process/stellarator/build.py

+    #     po.write(self.outfile,10)
+    # 10  format(t43,'Thickness (m)',t60,'Radius (m)')


Suggested change

# po.write(self.outfile,10)

# 10 format(t43,'Thickness (m)',t60,'Radius (m)')

timothy-nunn · 2026-02-04T11:26:08Z

process/stellarator/heating.py

+            + current_drive_variables.p_hcd_injected_electrons_mw
+        ) / current_drive_variables.eta_hcd_primary_injector_wall_plug
+    else:
+        logger.error(f"isthtr {stellarator_variables.isthtr} \n")


Suggested change

logger.error(f"isthtr {stellarator_variables.isthtr} \n")

logger.error(f"isthtr {stellarator_variables.isthtr}")

timothy-nunn · 2026-02-04T11:28:37Z

process/stellarator/stellarator.py

+        # stellarator_variables.f_coil_aspect = (
+        #     (physics_variables.rmajor / stellarator_variables.r_coil_minor) /
+        #     (stellarator_configuration.stella_config_rmajor_ref /
+        #      stellarator_configuration.stella_config_coil_rminor)
+        # )


Suggested change

# stellarator_variables.f_coil_aspect = (

# (physics_variables.rmajor / stellarator_variables.r_coil_minor) /

# (stellarator_configuration.stella_config_rmajor_ref /

# stellarator_configuration.stella_config_coil_rminor)

# )

timothy-nunn · 2026-02-04T11:28:47Z

process/stellarator/stellarator.py

+        surfaces with Fourier coefficients')
+        """
+        physics_variables.vol_plasma = (
+            # stellarator_variables.f_r * stellarator_variables.f_a**2 * stellarator_configuration.stella_config_plasma_volume


Suggested change

# stellarator_variables.f_r * stellarator_variables.f_a**2 * stellarator_configuration.stella_config_plasma_volume

timothy-nunn · 2026-02-04T11:29:27Z

process/stellarator/stellarator.py

+        #  heat_transport_variables.ipowerflow
+
+        #  fwbs_variables.blktmodel


Suggested change

# heat_transport_variables.ipowerflow

# fwbs_variables.blktmodel

timothy-nunn · 2026-02-04T11:30:17Z

process/stellarator/stellarator.py

+        physics_variables.p_plasma_rad_mw = (
+            physics_variables.p_plasma_rad_mw + physics_variables.psolradmw
+        )
+        # pden_plasma_rad_mw = physics_variables.p_plasma_rad_mw / physics_variables.vol_plasma # this line OVERWRITES the original definition of pden_plasma_rad_mw, probably shouldn't be defined like that as the core does not lose SOL power.


Suggested change

# pden_plasma_rad_mw = physics_variables.p_plasma_rad_mw / physics_variables.vol_plasma # this line OVERWRITES the original definition of pden_plasma_rad_mw, probably shouldn't be defined like that as the core does not lose SOL power.

timothy-nunn · 2026-02-04T11:30:32Z

process/stellarator/stellarator.py

+        # Calculate physics_variables.beta limit. Does nothing atm so commented out
+        # call stblim(physics_variables.beta_vol_avg_max)


Suggested change

# Calculate physics_variables.beta limit. Does nothing atm so commented out

# call stblim(physics_variables.beta_vol_avg_max)

timothy-nunn · 2026-02-04T11:36:57Z

Please also ensure the unit test failures are fixed and that the regression tests stop failing with an error

Resolve conflict

jjwalkowiak · 2026-02-04T14:19:57Z

Please also ensure the unit test failures are fixed and that the regression tests stop failing with an error

I solved most issues with units test, but with regression tests it will be more complicated.

jjwalkowiak and others added 30 commits March 7, 2025 16:45

fisrt test

d2dd88b

update

70c76c4

stellerator plasma volume fix

e8d59bb

more tests

1da32f6

sprocess test run

4be4f16

add squid

7869d37

Merge branch 'test' of https://github.com/jjwalkowiak/PROCESS into test

7130737

create autostart for stellarator

d09871b

add comments

bbf03f6

add functions list to documentation

733d620

test for stellarators

46e4c6d

remove unused WIP test

81b8498

introduce changes to the stellarator model, to prepare equations for …

e60fcf6

…coils aspect ratio variation

update test to check backward compatibility

b37cbe4

correct equation for the stellerator avaliable radial space

b4e81f1

Merge branch 'main' into modify-plasma-coil-distance

35c7f01

store list of updated variables (temporary)

88ddb3f

add coild scaling and correct vv r_major in stellartor module

4f27883

correct solver constraint equations check

1814c6a

minus missing in inequality check margin

9bd0984

Change coilsurface scaling in stellartor module. Now it followis in c…

1837540

…onsistant way a torioidal shape scaling

update input description

2497f08

add temperature margin variable for SC critical current in the stella…

1d92b8b

…rator module

calculate vv_stress_quench in stellartor module

0573494

stellarator module: correct vv max force denisty, change names of som…

8622aef

…e internal variables to be more readable

change parabolic profile default behaviour

1c8ef0d

test

04bf46c

stellarator module refactoring

838e531

stellarator coils refactoring

cbb0c95

refactoring stellarator coils

dabc8dc

jjwalkowiak and others added 8 commits February 2, 2026 10:15

Delete stellarator_test directory

8b37a9b

Revert "remov old files"

e5866d1

This reverts commit 1d1cbca.

update gitignore

07de173

Merge pull request #6 from jjwalkowiak/merge

375e219

Merge

update gitignore

d1be692

Delete tests/impurity_radiation.py

5f5e751

Delete tests/vv_stress.py

f9273ff

Merge remote-tracking branch 'upstream/HEAD' into merge

75f98f7

jjwalkowiak requested a review from a team as a code owner February 3, 2026 15:19

timothy-nunn self-assigned this Feb 3, 2026

timothy-nunn requested changes Feb 3, 2026

View reviewed changes

jjwalkowiak and others added 5 commits February 4, 2026 08:45

remove unnecesary files from documentation

ec164ca

merge refactoring

578d47f

pre-commit style corrections

1ff1f60

correct ouptut flag and function conflict

39c64af

Update .gitignore

44e0f2c

timothy-nunn reviewed Feb 4, 2026

View reviewed changes

jjwalkowiak added 2 commits February 4, 2026 10:19

more refactoring

9a7e430

Merge branch 'main' of https://github.com/jjwalkowiak/PROCESS

37afb1f

jjwalkowiak requested a review from timothy-nunn February 4, 2026 09:26

timothy-nunn requested changes Feb 4, 2026

View reviewed changes

timothy-nunn requested a review from ym1906 February 4, 2026 11:31

jjwalkowiak and others added 3 commits February 4, 2026 14:36

more refactoring

6b19091

Add Shield import to process/main.py

93276ce

Resolve conflict

Merge branch 'main' into main

91a8ad3

	Based on reference values from W-7X."""
	Based on reference values from W-7X.
	"""

	"""Actual totail coil current to reference value from stella_config file"""
	"""Actual total coil current to reference value from stella_config file"""

		# po.write(self.outfile,10)
		# 10 format(t43,'Thickness (m)',t60,'Radius (m)')

	logger.error(f"isthtr {stellarator_variables.isthtr} \n")
	logger.error(f"isthtr {stellarator_variables.isthtr}")

		# heat_transport_variables.ipowerflow

		# fwbs_variables.blktmodel

		# Calculate physics_variables.beta limit. Does nothing atm so commented out
		# call stblim(physics_variables.beta_vol_avg_max)

Stellarator module update #4082

Are you sure you want to change the base?

Stellarator module update #4082

Conversation

jjwalkowiak commented Feb 3, 2026

Description

Checklist

Uh oh!

timothy-nunn left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

timothy-nunn left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

timothy-nunn commented Feb 4, 2026

Uh oh!

jjwalkowiak commented Feb 4, 2026

Uh oh!

timothy-nunn commented Feb 4, 2026

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

timothy-nunn commented Feb 4, 2026

Uh oh!

jjwalkowiak commented Feb 4, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants