model
index
/home/gcallah/TandonDevOps/IndraABM/lib/model.py

This module contains the code for the base class of all Indra models.

 
Modules
       
lib.actions
json
sys
lib.user

 
Classes
       
builtins.object
Model

 
class Model(builtins.object)
    Model(model_nm='model', props=None, grp_struct={'def_grp': {'mbr_creator': <function create_agent at 0x7d0184d59430>, 'grp_action': None, 'mbr_action': <function def_action at 0x7d0184d59550>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'blue'}, 'red_grp': {'mbr_creator': <function create_agent at 0x7d0184d59430>, 'grp_action': None, 'mbr_action': <function def_action at 0x7d0184d59550>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'red'}}, env_action=None, random_placing=True, serial_obj=None, exec_key=None, create_for_test=False)
 
This class is the base class for all Indra models.
It will have all of the basic methods a model needs, as
well as a `run()` method that will kick of the model,
display the menu (if on a terminal), and register all
methods necessary to be registered for the API server
to work properly.
It should also make the notebook generator much simpler,
since the class methods will necessarily be present.
 
  Methods defined here:
__init__(self, model_nm='model', props=None, grp_struct={'def_grp': {'mbr_creator': <function create_agent at 0x7d0184d59430>, 'grp_action': None, 'mbr_action': <function def_action at 0x7d0184d59550>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'blue'}, 'red_grp': {'mbr_creator': <function create_agent at 0x7d0184d59430>, 'grp_action': None, 'mbr_action': <function def_action at 0x7d0184d59550>, 'num_mbrs': 1, 'num_mbrs_prop': None, 'color': 'red'}}, env_action=None, random_placing=True, serial_obj=None, exec_key=None, create_for_test=False)
Initialize self.  See help(type(self)) for accurate signature.
__repr__(self)
This returns a JSON representation of the model.
__str__(self)
Return str(self).
add_child(self, group)
Put a child agent in the womb.
group: which group will add new agent
The womb should move up into model eventually.
add_switch(self, agent_nm, from_grp_nm, to_grp_nm)
Switch agent from 1 group to another.
The agent and groups should be passed by name.
bar_graph(self)
collect_stats(self)
create_anew(self, model_nm, props, grp_struct, exec_key, env_action, random_placing, create_for_test=False)
Create the model for the first time.
create_env(self, env_action=None, random_placing=True)
Override this method to create a unique env...
but this one will already set the model name and add
the groups.
create_from_serial_obj(self, serial_obj)
Restore the model from its serialized version.
create_groups(self)
Override this method in your model to create all of your groups.
In general, you shouldn't need to: fill in the grp_struct instead.
create_pop_hist(self)
There are several methods that still (like in V2) reside in
Env, but which we mean to move to Model. So we provide an interface to
them here so when we move them other code won't break.
`create_pop_hist()` is such a method.
create_user(self)
This will create a user of the correct type.
from_json(self, jrep)
This method restores a model from its JSON rep.
get_locations(self)
get_periods(self)
get_pop_hist(self)
get_prop(self, prop_nm, default=None)
Have a way to get a prop through the model to hide props structure.
get_user_msgs(self)
handle_args(self)
handle_props(self, props, model_dir=None)
A generic parameter handling method.
We get height and width here, since so many models use them.
handle_switches(self)
This will actually process the pending switches.
handle_womb(self)
This method adds new agents from the womb.
The womb should move up into model eventually.
is_api_user(self)
is_test_user(self)
line_graph(self)
pending_switches(self)
How many switches are there to execute?
rpt_census(self, acts, moves)
This is the default census report.
Right now, `acts` is not used: do we need it?
Return: a string saying what happened in a period.
rpt_stats(self)
This is a "wrap up" report on the results of a model run.
Each model can do what it wants here.
perhaps will take an iterator object?
a file?
Function takes in a CSV formatted string from function
collect_stats() and writes it to a csv file.
 
Note: added logic so func will not write to stdout
rpt_switches(self)
Generate a string to report our switches.
run(self, periods=None)
This method runs the model. If `periods` is not None,
it will run it for that many periods. Otherwise, on
a terminal, it will display the menu.
Return: 0 if run was fine.
runN(self, periods=10)
Run our model for N periods.
Return the total number of actions taken.
run_batch(self, runs, steps)
Run our model for N periods X steps.
Writes the period specific model statistics to a CSV file.
Files are saved as input filename-[integer-counter].csv
Returns the total number of actions taken.
scatter_plot(self)
to_json(self)
This method generates the JSON representation for this model.
update_pop_hist(self)
This method records our populations each period.

Data descriptors defined here:
__dict__
dictionary for instance variables (if defined)
__weakref__
list of weak references to the object (if defined)

 
Functions
       
grp_val(grp, key)
Let's have a function that fill in defaults if a model
fails to specify any of the above group properties.
main()

 
Data
        BLUE_GRP = {'color': 'blue', 'grp_action': None, 'mbr_action': <function def_action>, 'mbr_creator': <function create_agent>, 'num_mbrs': 1, 'num_mbrs_prop': None}
BLUE_GRP_NM = 'def_grp'
COLOR = 'color'
DEBUG = <lib.utils.Debug object>
DEF_GRP = {'color': 'blue', 'grp_action': None, 'mbr_action': <function def_action>, 'mbr_creator': <function create_agent>, 'num_mbrs': 1, 'num_mbrs_prop': None}
DEF_GRP_NM = 'def_grp'
DEF_GRP_STRUCT = {'def_grp': {'color': 'blue', 'grp_action': None, 'mbr_action': <function def_action>, 'mbr_creator': <function create_agent>, 'num_mbrs': 1, 'num_mbrs_prop': None}, 'red_grp': {'color': 'red', 'grp_action': None, 'mbr_action': <function def_action>, 'mbr_creator': <function create_agent>, 'num_mbrs': 1, 'num_mbrs_prop': None}}
DEF_NUM_MEMBERS = 1
DEF_TIME = 10
GRID_HEIGHT = 'grid_height'
GRID_WIDTH = 'grid_width'
GRP_ACTION = 'grp_action'
MBR_ACTION = 'mbr_action'
MBR_CREATOR = 'mbr_creator'
NUM_MBRS = 'num_mbrs'
NUM_MBRS_PROP = 'num_mbrs_prop'
PROPS_PATH = './props'
RED_GRP = {'color': 'red', 'grp_action': None, 'mbr_action': <function def_action>, 'mbr_creator': <function create_agent>, 'num_mbrs': 1, 'num_mbrs_prop': None}
RED_GRP_NM = 'red_grp'