Skip to content

Commit 8d4e44c

Browse files
author
Michael Zingale
committed
finish docstringing
1 parent 70cd05d commit 8d4e44c

1 file changed

Lines changed: 51 additions & 15 deletions

File tree

mesh/patch.py

Lines changed: 51 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,18 @@ def get_var(self, name):
488488

489489
def get_aux(self, keyword):
490490
"""
491-
get the auxillary data associated with keyword
491+
Get the auxillary data associated with keyword
492+
493+
Parameters
494+
----------
495+
keyword : str
496+
The name of the auxillary data to access
497+
498+
Returns
499+
-------
500+
out : variable type
501+
The value corresponding to the keyword
502+
492503
"""
493504
if keyword in self.aux.keys():
494505
return self.aux[keyword]
@@ -498,30 +509,42 @@ def get_aux(self, keyword):
498509

499510
def zero(self, name):
500511
"""
501-
zero out the data array associated with variable name
512+
Zero out the data array associated with variable name.
513+
514+
Parameters
515+
----------
516+
name : str
517+
The name of the variable to zero
518+
502519
"""
503520
n = self.vars.index(name)
504521
self.data[n,:,:] = 0.0
505522

506523

507524
def fill_BC_all(self):
508525
"""
509-
fill boundary conditions on all variables
526+
Fill boundary conditions on all variables.
510527
"""
511528
for name in self.vars:
512529
self.fill_BC(name)
513530

514531

515532
def fill_BC(self, name):
516533
"""
517-
fill the boundary conditions. This operates on a single state
518-
variable at a time, to allow for maximum flexibility
534+
Fill the boundary conditions. This operates on a single state
535+
variable at a time, to allow for maximum flexibility.
519536
520-
we do periodic, reflect-even, reflect-odd, and outflow
537+
We do periodic, reflect-even, reflect-odd, and outflow
521538
522-
each variable name has a corresponding BCObject stored in the
539+
Each variable name has a corresponding BCObject stored in the
523540
CellCenterData2d object -- we refer to this to figure out the
524541
action to take at each boundary.
542+
543+
Parameters
544+
----------
545+
name : str
546+
The name of the variable for which to fill the BCs.
547+
525548
"""
526549

527550
# there is only a single grid, so every boundary is on
@@ -682,7 +705,7 @@ def fill_BC(self, name):
682705

683706
def restrict(self, varname):
684707
"""
685-
restrict the variable varname to a coarser grid (factor of 2
708+
Restrict the variable varname to a coarser grid (factor of 2
686709
coarser) and return an array with the resulting data (and same
687710
number of ghostcells)
688711
"""
@@ -720,9 +743,10 @@ def restrict(self, varname):
720743

721744
def prolong(self, varname):
722745
"""
723-
prolong the data in the current (coarse) grid to a finer
746+
Prolong the data in the current (coarse) grid to a finer
724747
(factor of 2 finer) grid. Return an array with the resulting
725-
data (and same number of ghostcells).
748+
data (and same number of ghostcells). Only the data for the
749+
variable varname will be operated upon.
726750
727751
We will reconstruct the data in the zone from the
728752
zone-averaged variables using the same limited slopes as in
@@ -828,7 +852,7 @@ def write(self, filename):
828852

829853
def pretty_print(self, varname):
830854
"""
831-
print out a small dataset to the screen with the ghost cells
855+
Print out a small dataset to the screen with the ghost cells
832856
a different color, to make things stand out
833857
"""
834858

@@ -864,8 +888,8 @@ def pretty_print(self, varname):
864888

865889
def read(filename):
866890
"""
867-
read a CellCenterData object from a file and return it and the grid
868-
info
891+
Read a CellCenterData object from a file and return it and the grid
892+
info and data.
869893
"""
870894

871895
# if we come in with .pyro, we don't need to add it again
@@ -880,8 +904,20 @@ def read(filename):
880904

881905

882906
def cell_center_data_clone(old):
883-
""" create a new CellCenterData2d object that is a copy of an existing
884-
one """
907+
"""
908+
Create a new CellCenterData2d object that is a copy of an existing
909+
one
910+
911+
Parameters
912+
----------
913+
old : CellCenterData2d object
914+
The CellCenterData2d object we wish to copy
915+
916+
Note
917+
----
918+
It may be that this whole thing can be replaced with a copy.deepcopy()
919+
920+
"""
885921

886922
if not isinstance(old, CellCenterData2d):
887923
msg.fail("Can't clone object")

0 commit comments

Comments
 (0)