# Source code for cvxpy.atoms.affine.bmat

"""

This file is part of CVXPY.

CVXPY is free software: you can redistribute it and/or modify
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

CVXPY is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with CVXPY.  If not, see <http://www.gnu.org/licenses/>.
"""

from cvxpy.atoms.affine.vstack import vstack
from cvxpy.atoms.affine.hstack import hstack

[docs]def bmat(block_lists):
"""Constructs a block matrix.

Takes a list of lists. Each internal list is stacked horizontally.
The internal lists are stacked vertically.

Parameters
----------
block_lists : list of lists
The blocks of the block matrix.

Return
------
CVXPY expression
The CVXPY expression representing the block matrix.
"""
row_blocks = [hstack(blocks) for blocks in block_lists]
return vstack(row_blocks)