aligning SGML to text

[Steven Bethard]

I have some plain text data and some SGML markup for that text that I
need to align. (The SGML doesn't maintain the original whitespace, so I
have to do some alignment; I can't just calculate the indices directly.)
For example, some of my text looks like:

TNF binding induces release of AIP1 (DAB2IP) from TNFR1, resulting in
cytoplasmic translocation and concomitant formation of an intracellular
signaling complex comprised of TRADD, RIP1, TRAF2, and AIPl.

And the corresponding SGML looks like:

<PROTEIN> TNF </PROTEIN> binding induces release of <PROTEIN> AIP1
</PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from <PROTEIN> TNFR1
</PROTEIN> , resulting in cytoplasmic translocation and concomitant
formation of an <PROTEIN> intracellular signaling complex </PROTEIN>
comprised of <PROTEIN> TRADD </PROTEIN> , <PROTEIN> RIP1 </PROTEIN> ,
<PROTEIN> TRAF2 </PROTEIN> , and AIPl .

Note that the SGML inserts spaces not only within the SGML elements, but
also around punctuation.


I need to determine the indices in the original text that each SGML
element corresponds to. Here's some working code to do this, based on a
suggestion for a related problem by Fredrik Lundh[1]::

def align(text, sgml):
sgml = sgml.replace('& ', '&amp;')
tree = etree.fromstrin g('<xml>%s</xml>' % sgml)
words = []
if tree.text is not None:
words.extend(tr ee.text.split() )
word_indices = []
for elem in tree:
elem_words = elem.text.split ()
start = len(words)
end = start + len(elem_words)
word_indices.ap pend((start, end, elem.tag))
words.extend(el em_words)
if elem.tail is not None:
words.extend(el em.tail.split() )
expr = '\s*'.join('(%s )' % re.escape(word) for word in words)
match = re.match(expr, text)
assert match is not None
for word_start, word_end, label in word_indices:
start = match.start(wor d_start + 1)
end = match.end(word_ end)
yield label, start, end

[color=blue][color=green][color=darkred]
>>> text = '''TNF binding induces release of AIP1 (DAB2IP) from[/color][/color][/color]
TNFR1, resulting in cytoplasmic translocation and concomitant
formation of an intracellular signaling complex comprised of TRADD,
RIP1, TRAF2, and AIPl.'''[color=blue][color=green][color=darkred]
>>> sgml = '''<PROTEIN> TNF </PROTEIN> binding induces release of[/color][/color][/color]
<PROTEIN> AIP1 </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from
<PROTEIN> TNFR1 </PROTEIN> , resulting in cytoplasmic translocation
and concomitant formation of an <PROTEIN> intracellular signaling
complex </PROTEIN> comprised of <PROTEIN> TRADD </PROTEIN> ,
<PROTEIN> RIP1 </PROTEIN> , <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
'''[color=blue][color=green][color=darkred]
>>> list(align(text , sgml))[/color][/color][/color]
[('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 37, 43),
('PROTEIN', 50, 55), ('PROTEIN', 128, 159), ('PROTEIN', 173, 178),
('PROTEIN', 180, 184), ('PROTEIN', 186, 191)]

The problem is, this doesn't work when my text is long (which it is)
because regular expressions are limited to 100 groups. I get an error
like::

Traceback (most recent call last):
...
AssertionError: sorry, but this version only supports 100 named
groups

I also played around with difflib.Sequenc eMatcher for a while, but
couldn't get a solution based on that working. Any suggestions?


[1]http://mail.python.org/pipermail/python-list/2005-December/313388.html

Thanks,

STeVe

[Gerard Flanagan]

Re: aligning SGML to text

Steven Bethard wrote:[color=blue]
> I have some plain text data and some SGML markup for that text that I
> need to align. (The SGML doesn't maintain the original whitespace, so I
> have to do some alignment; I can't just calculate the indices directly.)
> For example, some of my text looks like:
>
> TNF binding induces release of AIP1 (DAB2IP) from TNFR1, resulting in
> cytoplasmic translocation and concomitant formation of an intracellular
> signaling complex comprised of TRADD, RIP1, TRAF2, and AIPl.
>
> And the corresponding SGML looks like:
>
> <PROTEIN> TNF </PROTEIN> binding induces release of <PROTEIN> AIP1
> </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from <PROTEIN> TNFR1
> </PROTEIN> , resulting in cytoplasmic translocation and concomitant
> formation of an <PROTEIN> intracellular signaling complex </PROTEIN>
> comprised of <PROTEIN> TRADD </PROTEIN> , <PROTEIN> RIP1 </PROTEIN> ,
> <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
>
> Note that the SGML inserts spaces not only within the SGML elements, but
> also around punctuation.
>
>
> I need to determine the indices in the original text that each SGML
> element corresponds to. Here's some working code to do this, based on a
> suggestion for a related problem by Fredrik Lundh[1]::
>
> def align(text, sgml):
> sgml = sgml.replace('& ', '&amp;')
> tree = etree.fromstrin g('<xml>%s</xml>' % sgml)
> words = []
> if tree.text is not None:
> words.extend(tr ee.text.split() )
> word_indices = []
> for elem in tree:
> elem_words = elem.text.split ()
> start = len(words)
> end = start + len(elem_words)
> word_indices.ap pend((start, end, elem.tag))
> words.extend(el em_words)
> if elem.tail is not None:
> words.extend(el em.tail.split() )
> expr = '\s*'.join('(%s )' % re.escape(word) for word in words)
> match = re.match(expr, text)
> assert match is not None
> for word_start, word_end, label in word_indices:
> start = match.start(wor d_start + 1)
> end = match.end(word_ end)
> yield label, start, end
>[/color]
[...][color=blue][color=green][color=darkred]
> >>> list(align(text , sgml))[/color][/color]
> [('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 37, 43),
> ('PROTEIN', 50, 55), ('PROTEIN', 128, 159), ('PROTEIN', 173, 178),
> ('PROTEIN', 180, 184), ('PROTEIN', 186, 191)]
>
> The problem is, this doesn't work when my text is long (which it is)
> because regular expressions are limited to 100 groups. I get an error
> like::[/color]
[...]

Steve

This is probably an abuse of itertools...

---8<---
text = '''TNF binding induces release of AIP1 (DAB2IP) from
TNFR1, resulting in cytoplasmic translocation and concomitant
formation of an intracellular signaling complex comprised of TRADD,
RIP1, TRAF2, and AIPl.'''

sgml = '''<PROTEIN> TNF </PROTEIN> binding induces release of
<PROTEIN> AIP1 </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from
<PROTEIN> TNFR1 </PROTEIN> , resulting in cytoplasmic translocation
and concomitant formation of an <PROTEIN> intracellular signaling
complex </PROTEIN> comprised of <PROTEIN> TRADD </PROTEIN> ,
<PROTEIN> RIP1 </PROTEIN> , <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
'''

import itertools as it
import string

def scan(line):
if not line: return
line = line.strip()
parts = string.split(li ne, '>', maxsplit=1)
return parts[0]

def align(txt,sml):
i = 0
for k,g in it.groupby(sml. split('<'),scan ):
g = list(g)
if not g[0]: continue
text = g[0].split('>')[1]#.replace('\n', '')
if k.startswith('/'):
i += len(text)
else:
offset = len(text.strip( ))
yield k, i, i+offset
i += offset

print list(align(text ,sgml))

------------

[('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 38, 44),
('PROTEIN', 52, 57), ('PROTEIN', 131, 162), ('PROTEIN', 176, 181),
('PROTEIN', 184, 188), ('PROTEIN', 191, 196)]

It's off because of the punctuation possibly, can't figure it out.
maybe you can tweak it?

hth

Gerard

[Steven Bethard]

Re: aligning SGML to text

Gerard Flanagan wrote:[color=blue]
> Steven Bethard wrote:[color=green]
>> I have some plain text data and some SGML markup for that text that I
>> need to align. (The SGML doesn't maintain the original whitespace, so I
>> have to do some alignment; I can't just calculate the indices directly.)
>> For example, some of my text looks like:
>>
>> TNF binding induces release of AIP1 (DAB2IP) from TNFR1, resulting in
>> cytoplasmic translocation and concomitant formation of an intracellular
>> signaling complex comprised of TRADD, RIP1, TRAF2, and AIPl.
>>
>> And the corresponding SGML looks like:
>>
>> <PROTEIN> TNF </PROTEIN> binding induces release of <PROTEIN> AIP1
>> </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from <PROTEIN> TNFR1
>> </PROTEIN> , resulting in cytoplasmic translocation and concomitant
>> formation of an <PROTEIN> intracellular signaling complex </PROTEIN>
>> comprised of <PROTEIN> TRADD </PROTEIN> , <PROTEIN> RIP1 </PROTEIN> ,
>> <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
>>
>> Note that the SGML inserts spaces not only within the SGML elements, but
>> also around punctuation.
>>
>>
>> I need to determine the indices in the original text that each SGML
>> element corresponds to. Here's some working code to do this, based on a
>> suggestion for a related problem by Fredrik Lundh[1]::
>>
>> def align(text, sgml):
>> sgml = sgml.replace('& ', '&amp;')
>> tree = etree.fromstrin g('<xml>%s</xml>' % sgml)
>> words = []
>> if tree.text is not None:
>> words.extend(tr ee.text.split() )
>> word_indices = []
>> for elem in tree:
>> elem_words = elem.text.split ()
>> start = len(words)
>> end = start + len(elem_words)
>> word_indices.ap pend((start, end, elem.tag))
>> words.extend(el em_words)
>> if elem.tail is not None:
>> words.extend(el em.tail.split() )
>> expr = '\s*'.join('(%s )' % re.escape(word) for word in words)
>> match = re.match(expr, text)
>> assert match is not None
>> for word_start, word_end, label in word_indices:
>> start = match.start(wor d_start + 1)
>> end = match.end(word_ end)
>> yield label, start, end
>>[/color]
> [...][color=green][color=darkred]
>> >>> list(align(text , sgml))[/color]
>> [('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 37, 43),
>> ('PROTEIN', 50, 55), ('PROTEIN', 128, 159), ('PROTEIN', 173, 178),
>> ('PROTEIN', 180, 184), ('PROTEIN', 186, 191)]
>>
>> The problem is, this doesn't work when my text is long (which it is)
>> because regular expressions are limited to 100 groups. I get an error
>> like::[/color]
> [...]
>
> Steve
>
> This is probably an abuse of itertools...
>
> ---8<---
> text = '''TNF binding induces release of AIP1 (DAB2IP) from
> TNFR1, resulting in cytoplasmic translocation and concomitant
> formation of an intracellular signaling complex comprised of TRADD,
> RIP1, TRAF2, and AIPl.'''
>
> sgml = '''<PROTEIN> TNF </PROTEIN> binding induces release of
> <PROTEIN> AIP1 </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from
> <PROTEIN> TNFR1 </PROTEIN> , resulting in cytoplasmic translocation
> and concomitant formation of an <PROTEIN> intracellular signaling
> complex </PROTEIN> comprised of <PROTEIN> TRADD </PROTEIN> ,
> <PROTEIN> RIP1 </PROTEIN> , <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
> '''
>
> import itertools as it
> import string
>
> def scan(line):
> if not line: return
> line = line.strip()
> parts = string.split(li ne, '>', maxsplit=1)
> return parts[0]
>
> def align(txt,sml):
> i = 0
> for k,g in it.groupby(sml. split('<'),scan ):
> g = list(g)
> if not g[0]: continue
> text = g[0].split('>')[1]#.replace('\n', '')
> if k.startswith('/'):
> i += len(text)
> else:
> offset = len(text.strip( ))
> yield k, i, i+offset
> i += offset
>
> print list(align(text ,sgml))
>
> ------------
>
> [('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 38, 44),
> ('PROTEIN', 52, 57), ('PROTEIN', 131, 162), ('PROTEIN', 176, 181),
> ('PROTEIN', 184, 188), ('PROTEIN', 191, 196)]
>
> It's off because of the punctuation possibly, can't figure it out.[/color]

Thanks for taking a look. Yeah, the alignment's a big part of the
problem. It'd be really nice if the thing that gives me SGML didn't add
whitespace haphazardly. ;-)

STeVe

[Steven Bethard]

Re: aligning SGML to text

Steven Bethard wrote:[color=blue]
> I have some plain text data and some SGML markup for that text that I
> need to align. (The SGML doesn't maintain the original whitespace, so I
> have to do some alignment; I can't just calculate the indices directly.)[/color]
[snip][color=blue]
> Note that the SGML inserts spaces not only within the SGML elements, but
> also around punctuation.[/color]
[snip][color=blue]
> I need to determine the indices in the original text that each SGML
> element corresponds to.[/color]

Ok, below is a working version that doesn't use regular expressions.
It's far from concise, but at least it doesn't fail like re does when I
have more than 100 words. =)
[color=blue][color=green][color=darkred]
>>> import elementtree.Ele mentTree as etree
>>> def align(text, sgml):[/color][/color][/color]
.... # convert SGML tree to words, and assemble a list of the
.... # start word index and end word index for each SGML element
.... sgml = sgml.replace('& ', '&amp;')
.... tree = etree.fromstrin g('<xml>%s</xml>' % sgml)
.... words = []
.... if tree.text is not None:
.... words.extend(tr ee.text.split() )
.... word_spans = []
.... for elem in tree:
.... elem_words = elem.text.split ()
.... start = len(words)
.... end = start + len(elem_words)
.... word_spans.appe nd((start, end, elem.tag))
.... words.extend(el em_words)
.... if elem.tail is not None:
.... words.extend(el em.tail.split() )
.... # determine the start character index and end character index
.... # for each word from the SGML
.... char_spans = []
.... start = 0
.... for word in words:
.... while text[start:start + 1].isspace():
.... start += 1
.... end = start + len(word)
.... assert text[start:end] == word, (text[start:end], word)
.... char_spans.appe nd((start, end))
.... start = end
.... # convert the word indices for each SGML element to
.... # character indices
.... for word_start, word_end, label in word_spans:
.... start, _ = char_spans[word_start]
.... _, end = char_spans[word_end - 1]
.... yield label, start, end
....[color=blue][color=green][color=darkred]
>>> text = '''TNF binding induces release of AIP1 (DAB2IP) from TNFR1,[/color][/color][/color]
resulting in cytoplasmic translocation and concomitant formation of an
intracellular signaling complex comprised of TRADD, RIP1, TRAF2, and
AIPl.'''[color=blue][color=green][color=darkred]
>>> sgml = '''<PROTEIN> TNF </PROTEIN> binding induces release of[/color][/color][/color]
<PROTEIN> AIP1 </PROTEIN> ( <PROTEIN> DAB2IP </PROTEIN> ) from <PROTEIN>
TNFR1 </PROTEIN> , resulting in cytoplasmic translocation and
concomitant formation of an <PROTEIN> intracellular signaling complex
</PROTEIN> comprised of <PROTEIN> TRADD </PROTEIN> , <PROTEIN> RIP1
</PROTEIN> , <PROTEIN> TRAF2 </PROTEIN> , and AIPl .
.... '''[color=blue][color=green][color=darkred]
>>> list(align(text , sgml))[/color][/color][/color]
[('PROTEIN', 0, 3), ('PROTEIN', 31, 35), ('PROTEIN', 37, 43),
('PROTEIN', 50, 55), ('PROTEIN', 128, 159), ('PROTEIN', 173, 178),
('PROTEIN', 180, 184), ('PROTEIN', 186, 191)]

STeVe

[Gerard Flanagan]

Re: aligning SGML to text


Steven Bethard wrote:[color=blue]
> Gerard Flanagan wrote:[color=green]
> > Steven Bethard wrote:[color=darkred]
> >> I have some plain text data and some SGML markup for that text that I
> >> need to align. (The SGML doesn't maintain the original whitespace, so I
> >> have to do some alignment; I can't just calculate the indices directly.)
> >> For example, some of my text looks like:[/color][/color][/color]
[...][color=blue][color=green]
> >
> > Steve
> >
> > This is probably an abuse of itertools...
> >[/color][/color]
[snip hammering][color=blue]
>
> Thanks for taking a look. Yeah, the alignment's a big part of the
> problem. It'd be really nice if the thing that gives me SGML didn't add
> whitespace haphazardly. ;-)
>
> STeVe[/color]

I see, the problem was different than I thought. When all you have is a
hammer... :-)

Gerard