#NEWLINE# def setup_distrib( self ) : #NEWLINE# #INDENT# ' component declares the local sizes and sets initial values\n        for all distributed inputs and outputs. Returns a dict of\n        index arrays keyed to variable names.\n        ' #NEWLINE# comm = self . comm #NEWLINE# rank = comm . rank #NEWLINE# ( self . sizes , self . offsets ) = evenly_distrib_idxs ( comm . size , self . arr_size ) #NEWLINE# start = self . offsets [ rank ] #NEWLINE# end = ( start + self . sizes [ rank ] ) #NEWLINE# self . set_var_indices ( 'invec' , val = np . ones ( self . sizes [ rank ] , float ) , src_indices = np . arange ( start , end , dtype = int ) )