allpy
view allpy/base.py @ 644:0386dc31676e
Added a test for pickling
author | Daniil Alexeyevsky <dendik@kodomo.fbb.msu.ru> |
---|---|
date | Fri, 03 Jun 2011 16:59:45 +0400 |
parents | 61de9b5744c8 |
children | 88c246f20918 612c618fb7b0 |
line source
8 # import this very module as means of having all related classes in one place
12 """Set of characters to recoginze as gaps when parsing alignment."""
15 """Monomer object."""
18 """Either of 'dna', 'rna', 'protein'."""
21 """Mapping of related types. SHOULD be redefined in subclasses."""
24 """A mapping from 1-letter code to Monomer subclass."""
27 """A mapping from 3-letter code to Monomer subclass."""
30 """A mapping from full monomer name to Monomer subclass."""
32 @classmethod
34 """Create new subclass of Monomer for given monomer type."""
36 pass
46 # Save the class in data.monomers so that it can be pickled
47 # Some names are not unique, we append underscores to them
48 # in order to fix it.
49 # XXX: this WILL fail with dna 0AV != rna A2M, which both have
50 # name "2'-O-METHYLADENOSINE 5'-(DIHYDROGEN PHOSPHATE)"
58 # We duplicate distinguished long names into Monomer itself, so that we
59 # can use Monomer.from_code3 to create the relevant type of monomer.
63 @classmethod
65 """Create all relevant subclasses of Monomer."""
69 @classmethod
71 """Create new monomer from 1-letter code."""
74 @classmethod
76 """Create new monomer from 3-letter code."""
79 @classmethod
81 """Create new monomer from full name."""
88 """Returns one-letter code"""
92 """Monomers within same monomer type are compared by code1."""
102 """Sequence of Monomers.
104 This behaves like list of monomer objects. In addition to standard list
105 behaviour, Sequence has the following attributes:
107 * name -- str with the name of the sequence
108 * description -- str with description of the sequence
109 * source -- str denoting source of the sequence
111 Any of them may be empty (i.e. hold empty string)
112 """
115 """Mapping of related types. SHOULD be redefined in subclasses."""
121 @classmethod
123 """Create sequence from a list of monomer objecst."""
133 @classmethod
135 """Create sequences from string of one-letter codes."""
144 """Returns sequence of one-letter codes."""
148 """Hash sequence by identity."""
152 """Alignment. It is a list of Columns."""
155 """Mapping of related types. SHOULD be redefined in subclasses."""
158 """Ordered list of sequences in alignment. Read, but DO NOT FIDDLE!"""
161 """Initialize empty alignment."""
165 # Alignment grow & IO methods
166 # ==============================
169 """Add sequence to alignment. Return self.
171 If sequence is too short, pad it with gaps on the right.
172 """
181 """Add row from a string of one-letter codes and gaps. Return self."""
189 ]
196 """Add row from row_as_list representation and sequence. Return self."""
205 """Pad alignment with empty columns on the right to width n."""
210 """Append sequences from file to alignment. Return self.
212 If sequences in file have gaps (detected as characters belonging to
213 `gaps` set), treat them accordingly.
214 """
222 """Write alignment in FASTA file as sequences with gaps."""
227 # Data access methods for alignment
228 # =================================
231 """Return list of rows (temporary objects) in alignment.
233 Each row is a dictionary of { column : monomer }.
235 For gap positions there is no key for the column in row.
237 Each row has attribute `sequence` pointing to the sequence the row is
238 describing.
240 Modifications of row have no effect on the alignment.
241 """
242 # For now, the function returns a list rather than iterator.
243 # It is yet to see, whether memory performance here becomes critical,
244 # or is random access useful.
256 """Return list of rows (temporary objects) in alignment.
258 Each row here is a list of either monomer or None (for gaps).
260 Each row has attribute `sequence` pointing to the sequence of row.
262 Modifications of row have no effect on the alignment.
263 """
274 """Return list of string representation of rows in alignment.
276 Each row has attribute `sequence` pointing to the sequence of row.
278 `gap` is the symbol to use for gap.
279 """
294 """Return list of columns (temorary objects) in alignment.
296 Each column here is a list of either monomer or None (for gaps).
298 Items of column are sorted in the same way as alignment.sequences.
300 Modifications of column have no effect on the alignment.
301 """
310 # Alignment / Block editing methods
311 # =================================
314 """Remove all gaps from alignment and flush results to one side.
316 `whence` must be one of 'left', 'right' or 'center'
317 """
329 """Remove all empty columns."""
335 """Turn all row positions into gaps (but keep sequences intact)."""
341 """Replace contents of `dst` with those of `new`.
343 Replace contents of elements using function `merge(dst_el, new_le)`.
344 """
351 """Replace contents of sequences with those of `new` alignment."""
352 # XXX: we manually copy sequence contents here
353 # XXX: we only copy, overlapping parts and link to the rest
363 """Replace column contents with those of `new` alignment.
365 In other words: copy gap patterns from `new` to `self`.
367 `self.sequences` and `new.sequences` should have the same contents.
368 """
377 ]
382 """Replace alignment contents with those of other alignment."""
388 """Apply function to the alignment (or block); inject results back.
390 - `function(block)` must return block with same line order.
391 - if `copy_descriptions` is False, ignore new sequence names.
392 - if `copy_contents` is False, don't copy sequence contents too.
394 `function` (object) may have attributes `copy_descriptions` and
395 `copy_contents`, which override the same named arguments.
396 """
405 """Realign self.
407 I.e.: apply function to self to produce a new alignment, then update
408 self to have the same gap patterns as the new alignment.
410 This is the same as process(function, False, False)
411 """
416 """Column of alignment.
418 Column is a dict of { sequence : monomer }.
420 For sequences that have gaps in current row, given key is not present in
421 the column.
422 """
425 """Mapping of related types. SHOULD be redefined in subclasses."""
428 """Return hash by identity."""
432 """Block of alignment.
434 Block is an intersection of several rows & columns. (The collections of
435 rows and columns are represented as ordered lists, to retain display order
436 of Alignment or add ability to tweak it). Most of blocks look like
437 rectangular part of alignment if you shuffle alignment rows the right way.
438 """
441 """Alignment the block belongs to."""
444 """List of sequences in block."""
447 """List of columns in block."""
449 @classmethod
451 """Build new block from alignment.
453 If sequences are not given, the block uses all sequences in alignment.
455 If columns are not given, the block uses all columns in alignment.
457 In both cases we use exactly the list used in alignment, thus, if new
458 sequences or columns are added to alignment, the block tracks this too.
459 """
470 # vim: set ts=4 sts=4 sw=4 et: