File: MonomerInfo.cpp

package info (click to toggle)
rdkit 201403-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 62,288 kB
  • ctags: 15,156
  • sloc: cpp: 125,376; python: 55,674; java: 4,831; ansic: 4,178; xml: 2,499; sql: 1,775; yacc: 1,551; lex: 1,051; makefile: 353; fortran: 183; sh: 148; cs: 93
file content (91 lines) | stat: -rw-r--r-- 4,741 bytes parent folder | download
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
// $Id$
//
//  Copyright (C) 2013 Greg Landrum
//
//   @@ All Rights Reserved @@
//  This file is part of the RDKit.
//  The contents are covered by the terms of the BSD license
//  which is included in the file license.txt, found at the root
//  of the RDKit source tree.
//

#define NO_IMPORT_ARRAY
#include <boost/python.hpp>
#include <string>
#include <GraphMol/RDKitBase.h>
#include <RDGeneral/types.h>
#include <GraphMol/MonomerInfo.h>

namespace python = boost::python;

namespace RDKit{
  struct monomerinfo_wrapper {
    static void wrap() {
      std::string classDoc="The class to store monomer information attached to Atoms\n";
      python::class_<AtomMonomerInfo>("AtomMonomerInfo",classDoc.c_str(),python::init<>())
        .def(python::init<AtomMonomerInfo::AtomMonomerType,const std::string &>((python::arg("type"),
                                                                                 python::arg("name")="")))
        .def("GetName", &AtomMonomerInfo::getName,
             python::return_value_policy<python::copy_const_reference>())
        .def("GetMonomerType", &AtomMonomerInfo::getMonomerType)
        .def("SetName", &AtomMonomerInfo::setName)
        .def("SetMonomerType", &AtomMonomerInfo::setMonomerType)
        ;

      python::enum_<AtomMonomerInfo::AtomMonomerType>("AtomMonomerType")
        .value("UNKNOWN",AtomMonomerInfo::UNKNOWN)
        .value("PDBRESIDUE",AtomMonomerInfo::PDBRESIDUE)
        .value("OTHER",AtomMonomerInfo::OTHER)
        ;

      classDoc="The class to store PDB residue information attached to Atoms\n";
      python::class_<AtomPDBResidueInfo,python::bases<AtomMonomerInfo> >("AtomPDBResidueInfo",classDoc.c_str(),python::init<>())
        .def(python::init<std::string,int,std::string,std::string,int,std::string,std::string,
                          double,double,bool,unsigned int,unsigned int>((python::arg("atomName"),
                                                            python::arg("serialNumber")=1,
                                                            python::arg("altLoc")="",
                                                            python::arg("residueName")="",
                                                            python::arg("residueNumber")=0,
                                                            python::arg("chainId")="",
                                                            python::arg("insertionCode")="",
                                                            python::arg("occupancy")=1.0,
                                                            python::arg("tempFactor")=0.0,
                                                            python::arg("isHeteroAtom")=false,
                                                            python::arg("secondaryStructure")=0,
                                                            python::arg("segmentNumber")=0)))
        .def("GetSerialNumber",&AtomPDBResidueInfo::getSerialNumber)
        .def("GetAltLoc",&AtomPDBResidueInfo::getAltLoc,python::return_value_policy<python::copy_const_reference>())
        .def("GetResidueName",&AtomPDBResidueInfo::getResidueName,python::return_value_policy<python::copy_const_reference>())
        .def("GetResidueNumber",&AtomPDBResidueInfo::getResidueNumber)
        .def("GetChainId",&AtomPDBResidueInfo::getChainId,python::return_value_policy<python::copy_const_reference>())
        .def("GetInsertionCode",&AtomPDBResidueInfo::getInsertionCode,python::return_value_policy<python::copy_const_reference>())
        .def("GetOccupancy",&AtomPDBResidueInfo::getOccupancy)
        .def("GetTempFactor",&AtomPDBResidueInfo::getTempFactor)
        .def("GetIsHeteroAtom",&AtomPDBResidueInfo::getIsHeteroAtom)
        .def("GetSecondaryStructure",&AtomPDBResidueInfo::getSecondaryStructure)
        .def("GetSegmentNumber",&AtomPDBResidueInfo::getSegmentNumber)

        .def("SetSerialNumber",&AtomPDBResidueInfo::setSerialNumber)
        .def("SetAltLoc",&AtomPDBResidueInfo::setAltLoc)
        .def("SetResidueName",&AtomPDBResidueInfo::setResidueName)
        .def("SetResidueNumber",&AtomPDBResidueInfo::setResidueNumber)
        .def("SetChainId",&AtomPDBResidueInfo::setChainId)
        .def("SetInsertionCode",&AtomPDBResidueInfo::setInsertionCode)
        .def("SetOccupancy",&AtomPDBResidueInfo::setOccupancy)
        .def("SetTempFactor",&AtomPDBResidueInfo::setTempFactor)
        .def("SetIsHeteroAtom",&AtomPDBResidueInfo::setIsHeteroAtom)
        .def("SetSecondaryStructure",&AtomPDBResidueInfo::setSecondaryStructure)
        .def("SetSegmentNumber",&AtomPDBResidueInfo::setSegmentNumber)

        ;

    };
  };
}

void wrap_monomerinfo() {
  RDKit::monomerinfo_wrapper::wrap();
}