crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

used forwards ref to avoid circlar imports

Browse source
This commit is contained in:
Jonathan Shimwell 2022-10-21 19:04:45 +01:00
parent 17a6552027
commit 4116f6cda0
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
openmc/source.py
View file

@ -3,7 +3,7 @@ from enum import Enum
from numbers import Real
import warnings
import typing # imported separately as py3.8 requires typing.Iterable
from typing import Optional, Union
from typing import Optional, Union, Tuple
from xml.etree import ElementTree as ET
import numpy as np
@ -84,7 +84,7 @@ class Source:
parameters: Optional[str] = None,
strength: float = 1.0,
particle: str = 'neutron',
domains: Optional[Union[openmc.Cell, openmc.Material, openmc.Universe]] = None
domains: Optional[Union[openmc.Cell, openmc.Material, 'openmc.Universe']] = None
):
self._space = None
self._angle = None
@ -257,7 +257,7 @@ class Source:
return element
@classmethod
def from_xml_element(cls, elem: ET.Element) -> openmc.Source:
def from_xml_element(cls, elem: ET.Element) -> 'openmc.Source':
"""Generate source from an XML element
Parameters
@ -364,8 +364,8 @@ class SourceParticle:
"""
def __init__(
self,
r: typing.Iterable[float, float, float] = (0., 0., 0.),
u: typing.Iterable[float, float, float] = (0., 0., 1.),
r: Tuple[float, float, float] = (0., 0., 0.),
u: Tuple[float, float, float] = (0., 0., 1.),
E: float = 1.0e6,
time: float = 0.0,
wgt: float = 1.0,
@ -401,7 +401,7 @@ class SourceParticle:
def write_source_file(
source_particles: typing.Iterable[openmc.SourceParticle],
source_particles: 'typing.Iterable[openmc.SourceParticle]',
filename: PathLike, **kwargs
):
"""Write a source file using a collection of source particles