1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
|
/**************************************************************************
* *
* Regina - A Normal Surface Theory Calculator *
* Python Interface *
* *
* Copyright (c) 1999-2008, Ben Burton *
* For further details contact Ben Burton (bab@debian.org). *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License as *
* published by the Free Software Foundation; either version 2 of the *
* License, or (at your option) any later version. *
* *
* This program 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 this program; if not, write to the Free *
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, *
* MA 02110-1301, USA. *
* *
**************************************************************************/
/* end stub */
#include "maths/nmatrixint.h"
#include <boost/python.hpp>
#include <boost/python/detail/api_placeholder.hpp> // For len().
using namespace boost::python;
using regina::NMatrixInt;
namespace {
regina::NLargeInteger& (NMatrixInt::*entry_non_const)(unsigned long,
unsigned long) = &NMatrixInt::entry;
void (NMatrixInt::*addRow_triple)(unsigned long, unsigned long,
regina::NLargeInteger) = &NMatrixInt::addRow;
void (NMatrixInt::*addCol_triple)(unsigned long, unsigned long,
regina::NLargeInteger) = &NMatrixInt::addCol;
std::auto_ptr<NMatrixInt> multiply(const NMatrixInt& m1,
const NMatrixInt& m2) {
return m1.multiplyAs<NMatrixInt>(m2);
}
void setEntry(NMatrixInt& matrix, unsigned long row, unsigned long column,
const regina::NLargeInteger& value) {
matrix.entry(row, column) = value;
}
void initialise_list(NMatrixInt& matrix, boost::python::list values) {
if (boost::python::len(values) != matrix.rows() * matrix.columns()) {
PyErr_SetString(PyExc_IndexError,
"Initialisation list does not contain the "
"expected number of elements.");
boost::python::throw_error_already_set();
} else {
unsigned long r, c;
unsigned i = 0;
for (r = 0; r < matrix.rows(); ++r)
for (c = 0; c < matrix.columns(); ++c) {
// Accept any type that we know how to convert to a
// large integer.
extract<regina::NLargeInteger&> x_large(values[i]);
if (x_large.check()) {
matrix.entry(r, c) = x_large();
++i;
continue;
}
extract<long> x_long(values[i]);
if (x_long.check()) {
matrix.entry(r, c) = x_long();
++i;
continue;
}
extract<const char*> x_str(values[i]);
if (x_str.check()) {
matrix.entry(r, c) = x_str();
++i;
continue;
}
// Throw an exception.
x_large();
}
}
}
BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(OL_addRow,
NMatrixInt::addRow, 2, 3);
BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(OL_addCol,
NMatrixInt::addCol, 2, 3);
}
void addNMatrixInt() {
scope s = class_<NMatrixInt, bases<regina::ShareableObject>,
std::auto_ptr<NMatrixInt>, boost::noncopyable>("NMatrixInt",
init<unsigned long, unsigned long>())
.def(init<const NMatrixInt&>())
.def("initialise", &NMatrixInt::initialise)
.def("initialise", initialise_list)
.def("rows", &NMatrixInt::rows)
.def("columns", &NMatrixInt::columns)
.def("entry", entry_non_const, return_internal_reference<>())
.def("set", setEntry)
.def("isIdentity", &NMatrixInt::isIdentity)
.def("swapRows", &NMatrixInt::swapRows)
.def("swapColumns", &NMatrixInt::swapColumns)
.def("makeIdentity", &NMatrixInt::makeIdentity)
.def("addRow", addRow_triple, OL_addRow())
.def("addCol", addCol_triple, OL_addCol())
.def("multRow", &NMatrixInt::multRow)
.def("multCol", &NMatrixInt::multCol)
.def("det", &NMatrixInt::det)
.def(self == self)
.def(self != self)
.def("__mul__", multiply)
;
s.attr("zero") = NMatrixInt::zero;
s.attr("one") = NMatrixInt::one;
}
|