File: example2.cpp

package info (click to toggle)
rdkit 202503.6-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 222,000 kB
  • sloc: cpp: 411,111; python: 78,482; ansic: 26,181; java: 8,285; javascript: 4,404; sql: 2,393; yacc: 1,626; lex: 1,267; cs: 1,090; makefile: 581; xml: 229; fortran: 183; sh: 121
file content (55 lines) | stat: -rw-r--r-- 1,881 bytes parent folder | download | duplicates (2)
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
//
// Reading molecules - example2.cpp

#include <fstream>
#include <iostream>
#include <string>
#include <GraphMol/GraphMol.h>
#include <GraphMol/FileParsers/MolSupplier.h>

#include <boost/iostreams/device/file.hpp>
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/filter/gzip.hpp>
#include <boost/iostreams/stream.hpp>

int main(int argc, char **argv) {
  std::string file_root = getenv("RDBASE");
  file_root += "/Docs/Book";

  std::unique_ptr<RDKit::ROMol> mol;
  std::string sdf_file = file_root + "/data/5ht3ligs.sdf";
  bool takeOwnership = true;
  RDKit::SDMolSupplier mol_supplier(sdf_file, takeOwnership);
  while (!mol_supplier.atEnd()) {
    mol.reset(mol_supplier.next());
    std::cout << mol->getProp<std::string>("_Name") << " has "
              << mol->getNumAtoms() << " atoms." << std::endl;
  }

  for (int i = int(mol_supplier.length()) - 1; i >= 0; --i) {
    std::unique_ptr<RDKit::ROMol> mol(mol_supplier[i]);
    if (mol) {
      std::cout << mol->getProp<std::string>("_Name") << " has "
                << mol->getNumAtoms() << " atoms." << std::endl;
    }
  }

  boost::iostreams::filtering_istream ins;
  ins.push(boost::iostreams::gzip_decompressor());
  std::string comp_sdf_file = file_root + "/data/actives_5ht3.sdf.gz";
  ins.push(boost::iostreams::file_source(comp_sdf_file));
  // takeOwnership must be false for this, as we don't want the SDWriter trying
  // to delete the boost::iostream
  takeOwnership = false;
  RDKit::ForwardSDMolSupplier forward_supplier(&ins, takeOwnership);
  while (!forward_supplier.atEnd()) {
    mol.reset(forward_supplier.next());
    if (mol) {
      std::cout << mol->getProp<std::string>("_Name") << " has "
                << mol->getNumAtoms() << " atoms." << std::endl;
    }
  }

  // This is not allowed, and will give a compiler error:
  // mol.reset(forward_supplier[1]);
}