load module

Define functions to load and preprocess the TraceWin files.

load_dat_file(dat_path, *, keep='none', instructions_to_insert=())

Load the dat file and convert it into a list of lines.

Parameters:

• dat_path (pathlib.Path) – Filepath to the .dat file, as understood by TraceWin.

• keep (Literal["none", "comments", "empty lines", "all"], optional) – To determine which un-necessary lines in the dat file should be kept. The default is 'none'.

• instructions_to_insert (Collection[Instruction], optional) – Some elements or commands that are not present in the .dat file but that you want to add. The default is an empty tuple.

Returns:

dat_filecontent – List containing all the lines of dat_path.

Return type:

list[list[str]]

_strip_comments(line)

Remove comments from a line.

Parameters:

line (str)

Return type:

str

_split_named_elements(line)

Split named elements from a line.

Parameters:

line (str)

Return type:

list[str]

_split_weighted_elements(line)

Split elements with parentheses into separate parts.

Parameters:

line (str)

Return type:

list[str]

slice_dat_line(line)

Slice a .dat line into its components.

Parameters:

line (str)

Return type:

list[str]

table_structure_file(path)

Load the file produced by Data Save table to file.

Parameters:

path (Path)

Return type:

list[list[str]]

results(path, prop)

Load a property from TraceWin’s “Data” table.

Parameters:

• path (pathlib.Path) – Path to results file. It must be saved from TraceWin: Data > Save table to file.

• prop (str) – Name of the desired property. Must be in d_property.

Returns:

data_ref – Array containing the desired property.

Return type:

numpy.ndarray

transfer_matrices(path)

Load the transfer matrix as calculated by TraceWin.

Parameters:

path (Path)

Return type:

tuple[ndarray, ndarray, ndarray]

_transfer_matrix(lines)

Load a single element transfer matrix.

Parameters:

lines (list[str])

Return type:

ndarray