wepy.reporter.wexplore.image module

class wepy.reporter.wexplore.image.WExploreAtomImageReporter(init_image=None, image_atom_idxs=None, json_topology=None, **kwargs)[source]

Bases: wepy.reporter.reporter.ProgressiveFileReporter

Reporter for generating 3D molecular structures from WExplore region images.

This will only be meaningful for WExplore simulations where the region images are actually 3D coordinates.

Constructor for the WExploreAtomImageReporter.

Parameters
  • init_image (numpy.array, optional) – The initial region image. Used for generating the topology as well. If not given will be eventually generated. (Default = None)

  • image_atom_idxs (list of int) – The indices of the atoms that are part of the topology subset that comprises the image.

  • json_topology (str) – JSON format topology for the whole system. A subset of the atoms will be taken using the image_atom_idxs.

FILE_ORDER = ('init_state_path', 'image_path')

Specify an ordering of file paths. Should be customized.

SUGGESTED_EXTENSIONS = ('image_top.pdb', 'wexplore_images.dcd')

Suggested extensions for file paths for use with the automatic reparametrization feature. Should be customized.

init(**kwargs)[source]

Construct a ProgressiveFileReporter.

This is exactly the same as the FileReporter.

Parameters
  • file_paths (list of str) – The list of file paths (in order) to use.

  • modes (list of str) – The list of mode specs (in order) to use.

  • file_path (str) – If ‘file_paths’ not specified, the single file path to use.

  • mode (str) – If ‘file_path’ option used, this is the mode for that file.

DEFAULT_MODE = 'x'

The default mode to set for opening files if none is specified (create if doesn’t exist, fail if it does.)

DEFAULT_SUGGESTED_EXTENSION = 'report'

The default file extension used for files during dynamic reparametrization, if none is specified

MODES = ('x', 'w', 'w-', 'r', 'r+')

Valid modes accepted for files.

SUGGESTED_FILENAME_TEMPLATE = '{config}{narration}{reporter_class}.{ext}'

Template to use for dynamic reparametrization of file path names.

The fields in the template are:

config : indicator of the runtime configuration used

narration : freeform description of the instance

reporter_classthe name of the class that produced the

output. When no specific name is given for a file report generated from a reporter this is used to disambiguate, along with the extension.

ext : The file extension, for multiple files produced from one reporter this should be sufficient to disambiguate the files.

The ‘config’ and ‘narration’ should be the same across all reporters in the same simulation manager, and the ‘narration’ is considered optional.

_validate_mode(mode)

Check if the mode spec is a valid one.

Parameters

mode (str) –

Returns

valid

Return type

bool

cleanup(**kwargs)

Teardown routines for the reporter at the end of the simulation.

Use to cleanly and safely close I/O connections or other cleanup I/O.

Use to close file descriptors, database connections etc.

Reporters can expect to have the following key word arguments passed to them during a simulation by the sim_manager.

Parameters
  • runner (Runner object) – The runner at the end of the simulation

  • work_mapper (WorkeMapper object) – The work mapper at the end of the simulation

  • resampler (Resampler object) – The resampler at the end of the simulation

  • boundary_conditions (BoundaryConditions object) – The boundary conditions at the end of the simulation

  • reporters (list of Reporter objects) – The list of reporters at the end of the simulation

property file_path

For single file path reporters the file path to that file spec.

property file_paths

The file paths for this reporter, in order.

property mode

For single file path reporters the mode of that file.

property modes

The modes for the files, in order.

reparametrize(file_paths, modes)

Set the file paths and modes for all files in the reporter.

Parameters
  • file_paths (list of str) – New file paths for each file, in order.

  • modes (list of str) – New modes for each file, in order.

report(cycle_idx=None, resampler_data=None, **kwargs)[source]

Given data concerning the main simulation components state, perform I/O operations to persist that data.

Void method for reporter base class.

Reporters can expect to have the following key word arguments passed to them during a simulation by the sim_manager.

Parameters
  • cycle_idx (int) –

  • new_walkers (list of Walker objects) – List of walkers that were produced from running their dynamics by the runner.

  • warp_data (list of dict of str : value) – List of dict-like records for each warping event from the last cycle.

  • bc_data (list of dict of str : value) – List of dict-like records specifying the changes to the state of the boundary conditions in the last cycle.

  • progress_data (dict str : list) – A record indicating the progress values for each walker in the last cycle.

  • resampling_data (list of dict of str : value) – List of records specifying the resampling to occur at this cycle.

  • resampler_data (list of dict of str : value) – List of records specifying the changes to the state of the resampler in the last cycle.

  • n_segment_steps (int) – The number of dynamics steps that were completed in the last cycle

  • worker_segment_times (dict of int : list of float) – Mapping worker index to the times they took for each segment they processed.

  • cycle_runner_time (float) – Total time runner took in last cycle.

  • cycle_bc_time (float) – Total time boundary conditions took in last cycle.

  • cycle_resampling_time (float) – Total time resampler took in last cycle.

  • resampled_walkers (list of Walker objects) – List of walkers that were produced from the new_walkers from applying resampling and boundary conditions.

set_mode(file_idx, mode)

Set the mode for a single indexed file.

Parameters
  • file_idx (int) – Index in the listing of files.

  • mode (str) – The new mode spec.

set_path(file_idx, path)

Set the path for a single indexed file.

Parameters
  • file_idx (int) – Index in the listing of files.

  • path (str) – The new path to set for this file