File: export_mat2.cc

package info (click to toggle)
openstructure 2.9.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 205,228 kB
  • sloc: cpp: 188,129; python: 35,361; ansic: 34,298; fortran: 3,275; sh: 286; xml: 146; makefile: 29
file content (80 lines) | stat: -rw-r--r-- 2,332 bytes parent folder | download | duplicates (4)
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
//------------------------------------------------------------------------------
// This file is part of the OpenStructure project <www.openstructure.org>
//
// Copyright (C) 2008-2020 by the OpenStructure authors
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the Free
// Software Foundation; either version 3.0 of the License, or (at your option)
// any later version.
// This library 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 Lesser General Public License for more
// details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this library; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//------------------------------------------------------------------------------
#include <boost/python.hpp>
using namespace boost::python;

#include <ost/geom/geom.hh>



const Real Mat2_getitem(const geom::Mat2& m, tuple i) {
  int a = extract<int> (i[0]);
  int b = extract<int> (i[1]);
  return m.At(a, b);
}

void Mat2_setitem(geom::Mat2& m,const  tuple i,const  Real val) {
  int a = extract<int> (i[0]);
  int b = extract<int> (i[1]);
  m.At(a, b) = val;
}

String mat2_repr(const geom::Mat2& m) {
  std::stringstream ss;

  ss << "geom.Mat2(" << m(0,0) << ", " << m(0,1) << ", "
     << m(1,0) << ", " << m(1,1) << ")";
  return ss.str();
}

list mat2_data(const geom::Mat2& m)
{
  list nrvo;
  for(size_t k=0;k<4;++k) {
    nrvo.append(m.Data()[k]);
  }
  return nrvo;
}

void export_Mat2()
{
  using namespace geom;

  class_<Mat2>("Mat2",init<>())
    .def(init<Real,Real,Real,Real>())
    .def(self += self)
    .def(self -= self)
    .def(self + self)
    .def("__repr__", mat2_repr)
    .def(self - self)
    .def(self *= Real())
    .def(self /= Real())
    .def(self * Real())
    .def(self *= self)
    .def(self * self)
    .def(self * Vec2())
    .def(self / Real())
    .def(self == self)
    .def(self != self)
    .def(self_ns::str(self))
    .def("__getitem__",Mat2_getitem)
    .def("__setitem__",Mat2_setitem)
    .add_property("data",mat2_data)
  ;
}