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 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
|
#!/usr/bin/perl
use strict;
use warnings;
use Chemistry::OpenSMILES::Stereo qw( mark_all_double_bonds );
use Chemistry::OpenSMILES::Writer qw( write_SMILES );
use Graph::Traversal::DFS;
use Graph::Undirected;
use XML::LibXML;
use XML::LibXML::XPathContext;
@ARGV = ( '-' ) unless @ARGV;
for my $filename (@ARGV) {
my $cml = XML::LibXML->load_xml( location => $filename );
my $xp = XML::LibXML::XPathContext->new( $cml );
$xp->registerNs( 'cml', 'http://www.xml-cml.org/schema' );
my @graphs;
for my $molecule ( $xp->findnodes( '/cml:cml/cml:molecule' ) ) {
my $graph = Graph::Undirected->new( refvertexed => 1 );
my( $atomArray ) = $molecule->getChildrenByTagName( 'atomArray' );
next unless $atomArray; # Skip empty molecules
my %vertex_by_atom_id;
my %atom_id_by_vertex;
my $n = 0;
for my $atom ($atomArray->getChildrenByTagName( 'atom' )) {
my $vertex = { number => $n++ };
if( $atom->hasAttribute( 'elementType' ) ) {
$vertex->{symbol} = $atom->getAttribute( 'elementType' );
}
if( $atom->hasAttribute( 'formalCharge' ) ) {
$vertex->{charge} = int $atom->getAttribute( 'formalCharge' );
}
# Hydrogen count will be handled after bonds are added
if( $atom->hasAttribute( 'hydrogenCount' ) ) {
$vertex->{hydrogen_count} = $atom->getAttribute( 'hydrogenCount' );
}
my( $atomParity ) = $atom->getChildrenByTagName( 'atomParity' );
if( $atomParity &&
$atomParity->hasAttribute( 'atomRefs4' ) &&
$atomParity->textContent =~ /^-?1$/ ) {
my $chirality = int $atomParity->textContent;
$vertex->{chirality} = $chirality == -1 ? '@' : '@@';
# IDs will be replaced with real vertices in the second
# pass, as some of vertices might not be created yet.
$vertex->{chirality_neighbours} =
[ split ' ', $atomParity->getAttribute( 'atomRefs4' ) ];
}
$vertex_by_atom_id{$atom->getAttribute( 'id' )} = $vertex;
$atom_id_by_vertex{$vertex} = $atom->getAttribute( 'id' );
$graph->add_vertex( $vertex );
}
my @bonds;
my( $bondArray ) = $molecule->getChildrenByTagName( 'bondArray' );
if( $bondArray ) {
@bonds = $bondArray->getChildrenByTagName( 'bond' );
}
# Pass through bonds to establish connectivity in the graph and
# to collect cis/trans bond settings.
# TODO: Some CMLs have coordinates, thus additional cis/trans
# settings could be derived from them as well.
my %cis_trans_settings;
for my $bond (@bonds) {
my @vertices = map { $vertex_by_atom_id{$_} }
split ' ', $bond->getAttribute( 'atomRefs2' );
$graph->add_edge( @vertices );
my $bond_type = bond_type( $bond->getAttribute( 'order' ) );
if( $bond_type && $bond_type ne '-' ) {
$graph->set_edge_attribute( @vertices, 'bond', $bond_type );
}
my( $stereo ) = $bond->getChildrenByTagName( 'bondStereo' );
if( $stereo &&
$stereo->hasAttribute( 'atomRefs4' ) &&
$stereo->textContent =~ /^[CT]$/ ) {
my $bond_key = join '', sort @vertices;
my @other_vertices = grep { $_ ne $vertices[0] &&
$_ ne $vertices[1] }
map { $vertex_by_atom_id{$_} }
split ' ', $stereo->getAttribute( 'atomRefs4' );
$cis_trans_settings{$bond_key} = {
marker => $stereo->textContent eq 'C' ? 'cis' : 'trans',
other_vertices => \@other_vertices,
};
}
}
mark_all_double_bonds( $graph,
sub {
my $bond_key = join '', sort @_[1..2];
return unless $cis_trans_settings{$bond_key};
my $setting = $cis_trans_settings{$bond_key};
my $enumerated =
grep { $_ eq $setting->{other_vertices}[0] ||
$_ eq $setting->{other_vertices}[1] }
@_;
if( $enumerated % 2 ) {
return $setting->{marker} eq 'cis' ? 'trans' : 'cis';
} else {
return $setting->{marker};
}
} );
# A graph might describe more than one moiety. If so, it should be
# split into components as Chemistry::OpenSMILES does not understand
# not connected graphs.
my @components;
my $next_root =
sub {
my( $self, $unseen ) = @_;
my $atom = next_vertex( $self, $unseen );
return unless $atom;
push @components, Graph::Undirected->new( refvertexed => 1 );
# Adding the first atom here as this is the only way for
# single-atom moieties to be represented.
$components[-1]->add_vertex( $atom );
return $atom;
};
my $register_edge =
sub {
my( $u, $v, $self ) = @_;
my $graph = $self->graph;
$components[-1]->add_edge( $u, $v );
if( $graph->has_edge_attribute( $u, $v, 'bond' ) ) {
$components[-1]->set_edge_attribute( $u, $v, 'bond',
$graph->get_edge_attribute( $u, $v, 'bond' ) );
}
};
my $dfs = Graph::Traversal::DFS->new(
$graph,
first_root => $next_root,
next_root => $next_root,
next_successor => \&next_vertex,
tree_edge => $register_edge,
non_tree_edge => $register_edge,
);
$dfs->dfs;
# Second pass through the atoms to set hydrogen counts and
# chirality neighbours.
# TODO: Some CMLs have coordinates, thus additional chirality
# settings could be derived from them as well.
for my $component (@components) {
for my $atom ($component->vertices) {
if( exists $atom->{hydrogen_count} ) {
my $hydrogen_count = grep { $_->{symbol} eq 'H' }
$component->neighbours( $atom );
if( $hydrogen_count > $atom->{hydrogen_count} ) {
warn 'total number of attached hydrogen atoms is ' .
"less than the number of explicit hydrogen atoms\n";
}
while( $hydrogen_count < $atom->{hydrogen_count} ) {
$component->add_edge( $atom, { symbol => 'H',
number => $n++ } );
$hydrogen_count++;
}
delete $atom->{hydrogen_count};
}
if( exists $atom->{chirality_neighbours} ) {
$atom->{chirality_neighbours} =
[ map { $vertex_by_atom_id{$_} }
@{$atom->{chirality_neighbours}} ];
for (@{$atom->{chirality_neighbours}}) {
next if $component->has_edge( $atom, $_ );
warn "no edge between atoms '" .
$atom_id_by_vertex{$atom} . "' and '" .
$atom_id_by_vertex{$_} . "'\n";
}
}
}
}
push @graphs, @components;
}
print join( '.', map { write_SMILES( $_ ) } @graphs ), "\n";
}
sub bond_type
{
my( $order ) = @_;
return '-' if $order eq 'S' || $order eq '1';
return '=' if $order eq 'D' || $order eq '2';
return ':' if $order eq 'A';
return '#' if $order eq 'T' || $order eq '3';
warn "unknown bond type: '$order'\n";
return;
}
sub next_vertex
{
my( $self, $unseen ) = @_;
return unless %$unseen;
my( $next ) = map { $unseen->{$_} }
sort { $unseen->{$a}{number} <=> $unseen->{$b}{number} }
keys %$unseen;
return $next;
}
|