| BioPerl documentation | Contained in the BioPerl distribution. |
Bio::DB::SeqFeature::Store::DBI::mysql -- Mysql implementation of Bio::DB::SeqFeature::Store
use Bio::DB::SeqFeature::Store;
# Open the sequence database
my $db = Bio::DB::SeqFeature::Store->new(-adaptor => 'DBI::mysql',
-dsn => 'dbi:mysql:test');
# get a feature from somewhere
my $feature = Bio::SeqFeature::Generic->new(...);
# store it
$db->store($feature) or die "Couldn't store!";
# primary ID of the feature is changed to indicate its primary ID
# in the database...
my $id = $feature->primary_id;
# get the feature back out
my $f = $db->fetch($id);
# change the feature and update it
$f->start(100);
$f->update($f) or die "Couldn't update!";
# searching...
# ...by id
my @features = $db->fetch_many(@list_of_ids);
# ...by name
@features = $db->get_features_by_name('ZK909');
# ...by alias
@features = $db->get_features_by_alias('sma-3');
# ...by type
@features = $db->get_features_by_name('gene');
# ...by location
@features = $db->get_features_by_location(-seq_id=>'Chr1',-start=>4000,-end=>600000);
# ...by attribute
@features = $db->get_features_by_attribute({description => 'protein kinase'})
# ...by the GFF "Note" field
@result_list = $db->search_notes('kinase');
# ...by arbitrary combinations of selectors
@features = $db->features(-name => $name,
-type => $types,
-seq_id => $seqid,
-start => $start,
-end => $end,
-attributes => $attributes);
# ...using an iterator
my $iterator = $db->get_seq_stream(-name => $name,
-type => $types,
-seq_id => $seqid,
-start => $start,
-end => $end,
-attributes => $attributes);
while (my $feature = $iterator->next_seq) {
# do something with the feature
}
# ...limiting the search to a particular region
my $segment = $db->segment('Chr1',5000=>6000);
my @features = $segment->features(-type=>['mRNA','match']);
# getting & storing sequence information
# Warning: this returns a string, and not a PrimarySeq object
$db->insert_sequence('Chr1','GATCCCCCGGGATTCCAAAA...');
my $sequence = $db->fetch_sequence('Chr1',5000=>6000);
# what feature types are defined in the database?
my @types = $db->types;
# create a new feature in the database
my $feature = $db->new_feature(-primary_tag => 'mRNA',
-seq_id => 'chr3',
-start => 10000,
-end => 11000);
Bio::DB::SeqFeature::Store::mysql is the Mysql adaptor for Bio::DB::SeqFeature::Store. You will not create it directly, but instead use Bio::DB::SeqFeature::Store->new() to do so.
See Bio::DB::SeqFeature::Store for complete usage instructions.
Before you can use the adaptor, you must use the mysqladmin tool to create a database and establish a user account with write permission. In order to use "fast" loading, the user account must have "file" privileges.
To establish a connection to the database, call Bio::DB::SeqFeature::Store->new(-adaptor=>'DBI::mysql',@more_args). The additional arguments are as follows:
Argument name Description
------------- -----------
-dsn The database name. You can abbreviate
"dbi:mysql:foo" as "foo" if you wish.
-user Username for authentication.
-pass Password for authentication.
-namespace A prefix to attach to each table. This allows you
to have several virtual databases in the same
physical database.
-temp Boolean flag. If true, a temporary database
will be created and destroyed as soon as
the Store object goes out of scope. (synonym -temporary)
-autoindex Boolean flag. If true, features in the database will be
reindexed every time they change. This is the default.
-tmpdir Directory in which to place temporary files during "fast" loading.
Defaults to File::Spec->tmpdir(). (synonyms -dump_dir, -dumpdir, -tmp)
-dbi_options A hashref to pass to DBI->connect's 4th argument, the "attributes."
(synonyms -options, -dbi_attr)
-write Pass true to open database for writing or updating.
If successful, a new instance of Bio::DB::SeqFeature::Store::DBI::mysql will be returned.
In addition to the standard methods supported by all well-behaved Bio::DB::SeqFeature::Store databases, several following adaptor-specific methods are provided. These are described in the next sections.
Title : types Usage : @type_list = $db->types Function: Get all the types in the database Returns : array of Bio::DB::GFF::Typename objects Args : none Status : public
Title : toplevel_types Usage : @type_list = $db->toplevel_types Function: Get the toplevel types in the database Returns : array of Bio::DB::GFF::Typename objects Args : none Status : public
This is similar to types() but only returns the types of INDEXED (toplevel) features.
| BioPerl documentation | Contained in the BioPerl distribution. |
package Bio::DB::SeqFeature::Store::DBI::mysql;
use strict; use base 'Bio::DB::SeqFeature::Store'; use Bio::DB::SeqFeature::Store::DBI::Iterator; use DBI; use Memoize; use Cwd 'abs_path'; use Bio::DB::GFF::Util::Rearrange 'rearrange'; use Bio::SeqFeature::Lite; use File::Spec; use Carp 'carp','cluck','croak'; use constant DEBUG=>0; # from the MySQL documentation... # WARNING: if your sequence uses coordinates greater than 2 GB, you are out of luck! use constant MAX_INT => 2_147_483_647; use constant MIN_INT => -2_147_483_648; use constant MAX_BIN => 1_000_000_000; # size of largest feature = 1 Gb use constant MIN_BIN => 1000; # smallest bin we'll make - on a 100 Mb chromosome, there'll be 100,000 of these use constant SUMMARY_BIN_SIZE => 1000; # tier 0 == 1000 bp bins # tier 1 == 10,000 bp bins # etc. memoize('_typeid'); memoize('_locationid'); memoize('_attributeid'); memoize('dump_path'); ### # object initialization # sub init { my $self = shift; my ($dsn, $is_temporary, $autoindex, $namespace, $dump_dir, $user, $pass, $dbi_options, $writeable, $create, ) = rearrange(['DSN', ['TEMP','TEMPORARY'], 'AUTOINDEX', 'NAMESPACE', ['DUMP_DIR','DUMPDIR','TMP','TMPDIR'], 'USER', ['PASS','PASSWD','PASSWORD'], ['OPTIONS','DBI_OPTIONS','DBI_ATTR'], ['WRITE','WRITEABLE'], 'CREATE', ],@_); $dbi_options ||= {}; $writeable = 1 if $is_temporary or $dump_dir; $dsn or $self->throw("Usage: ".__PACKAGE__."->init(-dsn => \$dbh || \$dsn)"); my $dbh; if (ref $dsn) { $dbh = $dsn; } else { $dsn = "dbi:mysql:$dsn" unless $dsn =~ /^dbi:/; $dbh = DBI->connect($dsn,$user,$pass,$dbi_options) or $self->throw($DBI::errstr); $dbh->{mysql_auto_reconnect} = 1; } $self->{dbh} = $dbh; $self->{is_temp} = $is_temporary; $self->{namespace} = $namespace; $self->{writeable} = $writeable; $self->default_settings; $self->autoindex($autoindex) if defined $autoindex; $self->dumpdir($dump_dir) if $dump_dir; if ($self->is_temp) { $self->init_tmp_database(); } elsif ($create) { $self->init_database('erase'); } } sub writeable { shift->{writeable} } sub can_store_parentage { 1 } sub table_definitions { my $self = shift; return { feature => <<END, ( id int(10) auto_increment primary key, typeid int(10) not null, seqid int(10), start int, end int, strand tinyint default 0, tier tinyint, bin int, indexed tinyint default 1, object MEDIUMBLOB not null, index(seqid,tier,bin,typeid), index(typeid) ) END locationlist => <<END, ( id int(10) auto_increment primary key, seqname varchar(255) not null, index(seqname) ) END typelist => <<END, ( id int(10) auto_increment primary key, tag varchar(255) not null, index(tag) ) END name => <<END, ( id int(10) not null, name varchar(255) not null, display_name tinyint default 0, index(id), index(name) ) END attribute => <<END, ( id int(10) not null, attribute_id int(10) not null, attribute_value text, index(id), index(attribute_id,attribute_value(10)) ) END attributelist => <<END, ( id int(10) auto_increment primary key, tag varchar(255) not null, index(tag) ) END parent2child => <<END, ( id int(10) not null, child int(10) not null, unique index(id,child) ) END meta => <<END, ( name varchar(128) primary key, value varchar(128) not null ) END sequence => <<END, ( id int(10) not null, offset int(10) unsigned not null, sequence longblob, primary key(id,offset) ) END interval_stats => <<END, ( typeid integer not null, seqid integer not null, bin integer not null, cum_count integer not null, primary key(typeid,seqid,bin) ) END }; } ### # default settings -- will create and populate meta table if needed # sub default_settings { my $self = shift; $self->maybe_create_meta(); $self->SUPER::default_settings; $self->autoindex(1); $self->dumpdir(File::Spec->tmpdir); } ### # retrieve database handle # sub dbh { my $self = shift; my $d = $self->{dbh}; $self->{dbh} = shift if @_; $d; } sub clone { my $self = shift; $self->{dbh}{InactiveDestroy} = 1; $self->{dbh} = $self->{dbh}->clone unless $self->is_temp; } ### # get/set directory for bulk load tables # sub dumpdir { my $self = shift; my $d = $self->{dumpdir}; $self->{dumpdir} = abs_path(shift) if @_; $d; } ### # table namespace (multiple dbs in one mysql db) # sub namespace { my $self = shift; my $d = $self->{namespace}; $self->{namespace} = shift if @_; $d; } ### # Required for Pg not mysql # sub remove_namespace { return; } ### # find a path that corresponds to a dump table # sub dump_path { my $self = shift; my $table = $self->_qualify(shift); return "$self->{dumpdir}/$table.$$"; } ### # make a filehandle (writeable) that corresponds to a dump table # sub dump_filehandle { my $self = shift; my $table = shift; eval "require IO::File" unless IO::File->can('new'); my $path = $self->dump_path($table); my $fh = $self->{filehandles}{$path} ||= IO::File->new(">$path"); $fh; } ### # find the next ID for a feature (used only during bulk loading) # sub next_id { my $self = shift; $self->{max_id} ||= $self->max_id; return ++$self->{max_id}; } ### # find the maximum ID for a feature (used only during bulk loading) # sub max_id { my $self = shift; my $features = $self->_feature_table; my $sth = $self->_prepare("SELECT max(id) from $features"); $sth->execute or $self->throw($sth->errstr); my ($id) = $sth->fetchrow_array; $id; } ### # wipe database clean and reinstall schema # sub _init_database { my $self = shift; my $erase = shift; my $dbh = $self->dbh; my $tables = $self->table_definitions; for my $t (keys %$tables) { next if $t eq 'meta'; # don't get rid of meta data! my $table = $self->_qualify($t); $dbh->do("DROP table IF EXISTS $table") if $erase; my $query = "CREATE TABLE IF NOT EXISTS $table $tables->{$t}"; $self->_create_table($dbh,$query); } $self->subfeatures_are_indexed(1) if $erase; 1; } sub init_tmp_database { my $self = shift; my $dbh = $self->dbh; my $tables = $self->table_definitions; for my $t (keys %$tables) { next if $t eq 'meta'; # done earlier my $table = $self->_qualify($t); my $query = "CREATE TEMPORARY TABLE $table $tables->{$t}"; $self->_create_table($dbh,$query); } 1; } sub _create_table { my $self = shift; my ($dbh,$query) = @_; for my $q (split ';',$query) { chomp($q); next unless $q =~ /\S/; $dbh->do("$q;\n") or $self->throw($dbh->errstr); } } sub maybe_create_meta { my $self = shift; return unless $self->writeable; my $meta = $self->_meta_table; my $tables = $self->table_definitions; my $temporary = $self->is_temp ? 'TEMPORARY' : ''; $self->dbh->do("CREATE $temporary TABLE IF NOT EXISTS $meta $tables->{meta}"); } ### # use temporary tables # sub is_temp { shift->{is_temp}; } sub attributes { my $self = shift; my $dbh = $self->dbh; my $attributelist_table = $self->_attributelist_table; my $a = $dbh->selectcol_arrayref("SELECT tag FROM $attributelist_table") or $self->throw($dbh->errstr); return @$a; } sub _store { my $self = shift; # special case for bulk updates return $self->_dump_store(@_) if $self->{bulk_update_in_progress}; my $indexed = shift; my $count = 0; my $autoindex = $self->autoindex; my $dbh = $self->dbh; local $dbh->{RaiseError} = 1; $self->begin_work; eval { for my $obj (@_) { $self->replace($obj,$indexed); $self->_update_indexes($obj) if $indexed && $autoindex; $count++; } }; if ($@) { warn "Transaction aborted because $@"; $self->rollback; } else { $self->commit; } # remember whether we are have ever stored a non-indexed feature unless ($indexed or $self->{indexed_flag}++) { $self->subfeatures_are_indexed(0); } $count; } # we memoize this in order to avoid making zillions of calls sub autoindex { my $self = shift; # special case for bulk update -- need to build the indexes # at the same time we build the main feature table return 1 if $self->{bulk_update_in_progress}; my $d = $self->setting('autoindex'); $self->setting(autoindex=>shift) if @_; $d; } sub _start_bulk_update { my $self = shift; my $dbh = $self->dbh; $self->begin_work; $self->{bulk_update_in_progress}++; } sub _finish_bulk_update { my $self = shift; my $dbh = $self->dbh; my $dir = $self->{dumpdir} || '.'; for my $table ($self->_feature_table,$self->index_tables) { my $fh = $self->dump_filehandle($table); my $path = $self->dump_path($table); $fh->close; print STDERR "$path\n"; $dbh->do("LOAD DATA LOCAL INFILE '$path' REPLACE INTO TABLE $table FIELDS OPTIONALLY ENCLOSED BY '\\''") or $self->throw($dbh->errstr); unlink $path; } delete $self->{bulk_update_in_progress}; delete $self->{ filehandles}; $self->commit; } ### # Add a subparts to a feature. Both feature and all subparts must already be in database. # sub _add_SeqFeature { my $self = shift; # special purpose method for case when we are doing a bulk update return $self->_dump_add_SeqFeature(@_) if $self->{bulk_update_in_progress}; my $parent = shift; my @children = @_; my $dbh = $self->dbh; local $dbh->{RaiseError} = 1; my $parent2child = $self->_parent2child_table(); my $count = 0; my $sth = $self->_prepare(<<END); REPLACE INTO $parent2child (id,child) VALUES (?,?) END my $parent_id = (ref $parent ? $parent->primary_id : $parent) or $self->throw("$parent should have a primary_id"); $self->begin_work or $self->throw($dbh->errstr); eval { for my $child (@children) { my $child_id = ref $child ? $child->primary_id : $child; defined $child_id or die "no primary ID known for $child"; $sth->execute($parent_id,$child_id); $count++; } }; if ($@) { warn "Transaction aborted because $@"; $self->rollback; } else { $self->commit; } $sth->finish; $count; } sub _fetch_SeqFeatures { my $self = shift; my $parent = shift; my @types = @_; my $parent_id = $parent->primary_id or $self->throw("$parent should have a primary_id"); my $features = $self->_feature_table; my $parent2child = $self->_parent2child_table(); my @from = ("$features as f","$parent2child as c"); my @where = ('f.id=c.child','c.id=?'); my @args = $parent_id; if (@types) { my ($from,$where,undef,@a) = $self->_types_sql(\@types,'f'); push @from,$from if $from; push @where,$where if $where; push @args,@a; } my $from = join ', ',@from; my $where = join ' AND ',@where; my $query = <<END; SELECT f.id,f.object FROM $from WHERE $where END $self->_print_query($query,@args) if DEBUG || $self->debug; my $sth = $self->_prepare($query) or $self->throw($self->dbh->errstr); $sth->execute(@args) or $self->throw($sth->errstr); return $self->_sth2objs($sth); } ### # get primary sequence between start and end # sub _fetch_sequence { my $self = shift; my ($seqid,$start,$end) = @_; # backward compatibility to the old days when I liked reverse complementing # dna by specifying $start > $end my $reversed; if (defined $start && defined $end && $start > $end) { $reversed++; ($start,$end) = ($end,$start); } $start-- if defined $start; $end-- if defined $end; my $id = $self->_locationid($seqid); my $offset1 = $self->_offset_boundary($id,$start || 'left'); my $offset2 = $self->_offset_boundary($id,$end || 'right'); my $sequence_table = $self->_sequence_table; my $sql = <<END; SELECT sequence,offset FROM $sequence_table as s WHERE s.id=? AND s.offset >= ? AND s.offset <= ? ORDER BY s.offset END my $sth = $self->_prepare($sql); my $seq = ''; $self->_print_query($sql,$id,$offset1,$offset2) if DEBUG || $self->debug; $sth->execute($id,$offset1,$offset2) or $self->throw($sth->errstr); while (my($frag,$offset) = $sth->fetchrow_array) { substr($frag,0,$start-$offset) = '' if defined $start && $start > $offset; $seq .= $frag; } substr($seq,$end-$start+1) = '' if defined $end && $end-$start+1 < length($seq); if ($reversed) { $seq = reverse $seq; $seq =~ tr/gatcGATC/ctagCTAG/; } $sth->finish; $seq; } sub _offset_boundary { my $self = shift; my ($seqid,$position) = @_; my $sequence_table = $self->_sequence_table; my $locationlist_table = $self->_locationlist_table; my $sql; $sql = $position eq 'left' ? "SELECT min(offset) FROM $sequence_table as s WHERE s.id=?" :$position eq 'right' ? "SELECT max(offset) FROM $sequence_table as s WHERE s.id=?" :"SELECT max(offset) FROM $sequence_table as s WHERE s.id=? AND offset<=?"; my $sth = $self->_prepare($sql); my @args = $position =~ /^-?\d+$/ ? ($seqid,$position) : ($seqid); $self->_print_query($sql,@args) if DEBUG || $self->debug; $sth->execute(@args) or $self->throw($sth->errstr); my $boundary = $sth->fetchall_arrayref->[0][0]; $sth->finish; return $boundary; } ### # add namespace to tablename # sub _qualify { my $self = shift; my $table_name = shift; my $namespace = $self->namespace; return $table_name if (!defined $namespace || # is namespace already present in table name? index($table_name, $namespace) == 0); return "${namespace}_${table_name}"; } ### # Fetch a Bio::SeqFeatureI from database using its primary_id # sub _fetch { my $self = shift; @_ or $self->throw("usage: fetch(\$primary_id)"); my $primary_id = shift; my $features = $self->_feature_table; my $sth = $self->_prepare(<<END); SELECT id,object FROM $features WHERE id=? END $sth->execute($primary_id) or $self->throw($sth->errstr); my $obj = $self->_sth2obj($sth); $sth->finish; $obj; } ### # Efficiently fetch a series of IDs from the database # Can pass an array or an array ref # sub _fetch_many { my $self = shift; @_ or $self->throw('usage: fetch_many($id1,$id2,$id3...)'); my $ids = join ',',map {ref($_) ? @$_ : $_} @_ or return; my $features = $self->_feature_table; my $sth = $self->_prepare(<<END); SELECT id,object FROM $features WHERE id IN ($ids) END $sth->execute() or $self->throw($sth->errstr); return $self->_sth2objs($sth); } sub _features { my $self = shift; my ($seq_id,$start,$end,$strand, $name,$class,$allow_aliases, $types, $attributes, $range_type, $fromtable, $iterator, $sources, ) = rearrange([['SEQID','SEQ_ID','REF'],'START',['STOP','END'],'STRAND', 'NAME','CLASS','ALIASES', ['TYPES','TYPE','PRIMARY_TAG'], ['ATTRIBUTES','ATTRIBUTE'], 'RANGE_TYPE', 'FROM_TABLE', 'ITERATOR', ['SOURCE','SOURCES'], ],@_); my (@from,@where,@args,@group); $range_type ||= 'overlaps'; my $features = $self->_feature_table; @from = "$features as f"; if (defined $name) { # hacky backward compatibility workaround undef $class if $class && $class eq 'Sequence'; $name = "$class:$name" if defined $class && length $class > 0; # last argument is the join field my ($from,$where,$group,@a) = $self->_name_sql($name,$allow_aliases,'f.id'); push @from,$from if $from; push @where,$where if $where; push @group,$group if $group; push @args,@a; } if (defined $seq_id) { # last argument is the name of the features table my ($from,$where,$group,@a) = $self->_location_sql($seq_id,$start,$end,$range_type,$strand,'f'); push @from,$from if $from; push @where,$where if $where; push @group,$group if $group; push @args,@a; } if (defined($sources)) { my @sources = ref($sources) eq 'ARRAY' ? @{$sources} : ($sources); if (defined($types)) { my @types = ref($types) eq 'ARRAY' ? @{$types} : ($types); my @final_types; foreach my $type (@types) { # *** not sure what to do if user supplies both -source and -type # where the type includes a source! if ($type =~ /:/) { push(@final_types, $type); } else { foreach my $source (@sources) { push(@final_types, $type.':'.$source); } } } $types = \@final_types; } else { $types = [map { ':'.$_ } @sources]; } } if (defined($types)) { # last argument is the name of the features table my ($from,$where,$group,@a) = $self->_types_sql($types,'f'); push @from,$from if $from; push @where,$where if $where; push @group,$group if $group; push @args,@a; } if (defined $attributes) { # last argument is the join field my ($from,$where,$group,@a) = $self->_attributes_sql($attributes,'f.id'); push @from,$from if $from; push @where,$where if $where; push @group,$group if $group; push @args,@a; } if (defined $fromtable) { # last argument is the join field my ($from,$where,$group,@a) = $self->_from_table_sql($fromtable,'f.id'); push @from,$from if $from; push @where,$where if $where; push @group,$group if $group; push @args,@a; } # if no other criteria are specified, then # only fetch indexed (i.e. top level objects) @where = 'indexed=1' unless @where; my $from = join ', ',@from; my $where = join ' AND ',map {"($_)"} @where; my $group = join ', ',@group; $group = "GROUP BY $group" if @group; my $query = <<END; SELECT f.id,f.object,f.typeid,f.seqid,f.start,f.end,f.strand FROM $from WHERE $where $group END ; $self->_print_query($query,@args) if DEBUG || $self->debug; my $sth = $self->_prepare($query) or $self->throw($self->dbh->errstr); $sth->execute(@args) or $self->throw($sth->errstr); return $iterator ? Bio::DB::SeqFeature::Store::DBI::Iterator->new($sth,$self) : $self->_sth2objs($sth); } sub _aggregate_bins { my $self = shift; my $sth = shift; my (%types,$binsize,$binstart); while (my ($type,$seqname,$bin,$count,$bins,$start,$end) = $sth->fetchrow_array) { $binsize ||= ($end-$start+1)/$bins; $binstart ||= int($start/$binsize); $types{$type}{seqname} ||= $seqname; $types{$type}{min} ||= $start; $types{$type}{max} ||= $end; $types{$type}{bins} ||= [(0) x $bins]; $types{$type}{bins}[$bin-$binstart] = $count; $types{$type}{count} += $count; } my @results; for my $type (keys %types) { my $min = $types{$type}{min}; my $max = $types{$type}{max}; my $seqid= $types{$type}{seqname}; my $f = Bio::SeqFeature::Lite->new(-seq_id => $seqid, -start => $min, -end => $max, -type => "$type:bins", -score => $types{$type}{count}, -attributes => {coverage => join ',',@{$types{$type}{bins}}}); push @results,$f; } return @results; } sub _name_sql { my $self = shift; my ($name,$allow_aliases,$join) = @_; my $name_table = $self->_name_table; my $from = "$name_table as n"; my ($match,$string) = $self->_match_sql($name); my $where = "n.id=$join AND n.name $match"; $where .= " AND n.display_name>0" unless $allow_aliases; return ($from,$where,'',$string); } sub _search_attributes { my $self = shift; my ($search_string,$attribute_names,$limit) = @_; my @words = map {quotemeta($_)} split /\s+/,$search_string; my $name_table = $self->_name_table; my $attribute_table = $self->_attribute_table; my $attributelist_table = $self->_attributelist_table; my $type_table = $self->_type_table; my $typelist_table = $self->_typelist_table; my @tags = @$attribute_names; my $tag_sql = join ' OR ',("al.tag=?") x @tags; my $perl_regexp = join '|',@words; my $sql_regexp = join ' OR ',("a.attribute_value REGEXP ?") x @words; my $sql = <<END; SELECT name,attribute_value,tl.tag,n.id FROM $name_table as n,$attribute_table as a,$attributelist_table as al,$type_table as t,$typelist_table as tl WHERE n.id=a.id AND al.id=a.attribute_id AND n.id=t.id AND t.typeid=tl.id AND n.display_name=1 AND ($tag_sql) AND ($sql_regexp) END $sql .= "LIMIT $limit" if defined $limit; $self->_print_query($sql,@tags,@words) if DEBUG || $self->debug; my $sth = $self->_prepare($sql); $sth->execute(@tags,@words) or $self->throw($sth->errstr); my @results; while (my($name,$value,$type,$id) = $sth->fetchrow_array) { my (@hits) = $value =~ /$perl_regexp/ig; my @words_in_row = split /\b/,$value; my $score = int(@hits * 10); push @results,[$name,$value,$score,$type,$id]; } $sth->finish; @results = sort {$b->[2]<=>$a->[2]} @results; return @results; } sub _match_sql { my $self = shift; my $name = shift; my ($match,$string); if ($name =~ /(?:^|[^\\])[*?]/) { $name =~ s/(^|[^\\])([%_])/$1\\$2/g; $name =~ s/(^|[^\\])\*/$1%/g; $name =~ s/(^|[^\\])\?/$1_/g; $match = "LIKE ?"; $string = $name; } else { $match = "= ?"; $string = $name; } return ($match,$string); } sub _from_table_sql { my $self = shift; my ($from_table,$join) = @_; my $from = "$from_table as ft"; my $where = "ft.id=$join"; return ($from,$where,''); } sub _attributes_sql { my $self = shift; my ($attributes,$join) = @_; my ($wf,@bind_args) = $self->_make_attribute_where('a','al',$attributes); my ($group_by,@group_args)= $self->_make_attribute_group('a',$attributes); my $attribute_table = $self->_attribute_table; my $attributelist_table = $self->_attributelist_table; my $from = "$attribute_table as a use index(attribute_id), $attributelist_table as al"; my $where = <<END; a.id=$join AND a.attribute_id=al.id AND ($wf) END my $group = $group_by; my @args = (@bind_args,@group_args); return ($from,$where,$group,@args); } sub subfeature_types_are_indexed { 1 } sub subfeature_locations_are_indexed { 1 } sub _types_sql { my $self = shift; my ($types,$type_table) = @_; my ($primary_tag,$source_tag); my @types = ref $types eq 'ARRAY' ? @$types : $types; my $typelist = $self->_typelist_table; my $from = "$typelist AS tl"; my (@matches,@args); for my $type (@types) { if (ref $type && $type->isa('Bio::DB::GFF::Typename')) { $primary_tag = $type->method; $source_tag = $type->source; } else { ($primary_tag,$source_tag) = split ':',$type,2; } if (length $source_tag) { if (length($primary_tag)) { push @matches,"tl.tag=?"; push @args,"$primary_tag:$source_tag"; } else { push @matches,"tl.tag LIKE ?"; push @args,"%:$source_tag"; } } else { push @matches,"tl.tag LIKE ?"; push @args,"$primary_tag:%"; } } my $matches = join ' OR ',@matches; my $where = <<END; tl.id=$type_table.typeid AND ($matches) END return ($from,$where,'',@args); } sub _location_sql { my $self = shift; my ($seq_id,$start,$end,$range_type,$strand,$location) = @_; # the additional join on the location_list table badly impacts performance # so we build a copy of the table in memory my $seqid = $self->_locationid_nocreate($seq_id) || 0; # zero is an invalid primary ID, so will return empty $start = MIN_INT unless defined $start; $end = MAX_INT unless defined $end; my ($bin_where,@bin_args) = $self->bin_where($start,$end,$location); my ($range,@range_args); if ($range_type eq 'overlaps') { $range = "$location.end>=? AND $location.start<=? AND ($bin_where)"; @range_args = ($start,$end,@bin_args); } elsif ($range_type eq 'contains') { $range = "$location.start>=? AND $location.end<=? AND ($bin_where)"; @range_args = ($start,$end,@bin_args); } elsif ($range_type eq 'contained_in') { $range = "$location.start<=? AND $location.end>=?"; @range_args = ($start,$end); } else { $self->throw("range_type must be one of 'overlaps', 'contains' or 'contained_in'"); } if (defined $strand) { $range .= " AND strand=?"; push @range_args,$strand; } my $where = <<END; $location.seqid=? AND $range END my $from = ''; my $group = ''; my @args = ($seqid,@range_args); return ($from,$where,$group,@args); } ### # force reindexing # sub reindex { my $self = shift; my $from_update_table = shift; # if present, will take ids from "update_table" my $dbh = $self->dbh; my $count = 0; my $now; # try to bring in highres time() function eval "require Time::HiRes"; my $last_time = $self->time(); # tell _delete_index() not to bother removing the index rows corresponding # to each individual feature local $self->{reindexing} = 1; $self->begin_work; eval { my $update = $from_update_table; for my $table ($self->index_tables) { my $query = $from_update_table ? "DELETE $table FROM $table,$update WHERE $table.id=$update.id" : "DELETE FROM $table"; $dbh->do($query); $self->_disable_keys($dbh,$table); } my $iterator = $self->get_seq_stream(-from_table=>$from_update_table ? $update : undef); while (my $f = $iterator->next_seq) { if (++$count %1000 == 0) { $now = $self->time(); my $elapsed = sprintf(" in %5.2fs",$now - $last_time); $last_time = $now; print STDERR "$count features indexed$elapsed...",' 'x60; print STDERR -t STDOUT && !$ENV{EMACS} ? "\r" : "\n"; } $self->_update_indexes($f); } }; for my $table ($self->index_tables) { $self->_enable_keys($dbh,$table); } if (@_) { warn "Couldn't complete transaction: $@"; $self->rollback; return; } else { $self->commit; return 1; } } sub optimize { my $self = shift; $self->dbh->do("ANALYZE TABLE $_") foreach $self->index_tables; } sub all_tables { my $self = shift; my @index_tables = $self->index_tables; my $features = $self->_feature_table; return ($features,@index_tables); } sub index_tables { my $self = shift; return map {$self->_qualify($_)} qw(name attribute parent2child) } sub _firstid { my $self = shift; my $features = $self->_feature_table; my $query = <<END; SELECT min(id) FROM $features END my $sth=$self->_prepare($query); $sth->execute(); my ($first) = $sth->fetchrow_array; $sth->finish; $first; } sub _nextid { my $self = shift; my $lastkey = shift; my $features = $self->_feature_table; my $query = <<END; SELECT min(id) FROM $features WHERE id>? END my $sth=$self->_prepare($query); $sth->execute($lastkey); my ($next) = $sth->fetchrow_array; $sth->finish; $next; } sub _existsid { my $self = shift; my $key = shift; my $features = $self->_feature_table; my $query = <<END; SELECT count(*) FROM $features WHERE id=? END my $sth=$self->_prepare($query); $sth->execute($key); my ($count) = $sth->fetchrow_array; $sth->finish; $count > 0; } sub _deleteid { my $self = shift; my $key = shift; my $dbh = $self->dbh; my $parent2child = $self->_parent2child_table; my $query = "SELECT child FROM $parent2child WHERE id=?"; my $sth=$self->_prepare($query); $sth->execute($key); my $success = 0; while (my ($cid) = $sth->fetchrow_array) { # Backcheck looking for multiple parents, delete only if one is present. I'm # sure there is a nice way to left join the parent2child table onto itself # to get this in one query above, just haven't worked it out yet... my $sth2 = $self->_prepare("SELECT count(id) FROM $parent2child WHERE child=?"); $sth2->execute($cid); my ($count) = $sth2->fetchrow_array; if ($count == 1) { $self->_deleteid($cid) || warn "An error occurred while removing subfeature id=$cid. Perhaps it was previously deleted?\n"; } } for my $table ($self->all_tables) { $success += $dbh->do("DELETE FROM $table WHERE id=$key") || 0; } return $success; } sub _clearall { my $self = shift; my $dbh = $self->dbh; for my $table ($self->all_tables) { $dbh->do("DELETE FROM $table"); } } sub _featurecount { my $self = shift; my $dbh = $self->dbh; my $features = $self->_feature_table; my $query = <<END; SELECT count(*) FROM $features END my $sth=$self->_prepare($query); $sth->execute(); my ($count) = $sth->fetchrow_array; $sth->finish; $count; } sub _seq_ids { my $self = shift; my $dbh = $self->dbh; my $location = $self->_locationlist_table; my $sth = $self->_prepare("SELECT DISTINCT seqname FROM $location"); $sth->execute() or $self->throw($sth->errstr); my @result; while (my ($id) = $sth->fetchrow_array) { push @result,$id; } return @result; } sub setting { my $self = shift; my ($variable_name,$value) = @_; my $meta = $self->_meta_table; if (defined $value && $self->writeable) { my $query = <<END; REPLACE INTO $meta (name,value) VALUES (?,?) END my $sth = $self->_prepare($query); $sth->execute($variable_name,$value) or $self->throw($sth->errstr); $sth->finish; $self->{settings_cache}{$variable_name} = $value; } else { return $self->{settings_cache}{$variable_name} if exists $self->{settings_cache}{$variable_name}; my $query = <<END; SELECT value FROM $meta as m WHERE m.name=? END my $sth = $self->_prepare($query); $sth->execute($variable_name) or $self->throw($sth->errstr); my ($value) = $sth->fetchrow_array; $sth->finish; return $self->{settings_cache}{$variable_name} = $value; } } ### # Replace Bio::SeqFeatureI into database. # sub replace { my $self = shift; my $object = shift; my $index_flag = shift || undef; # ?? shouldn't need to do this # $self->_load_class($object); my $id = $object->primary_id; my $features = $self->_feature_table; my $sth = $self->_prepare(<<END); REPLACE INTO $features (id,object,indexed,seqid,start,end,strand,tier,bin,typeid) VALUES (?,?,?,?,?,?,?,?,?,?) END my @location = $index_flag ? $self->_get_location_and_bin($object) : (undef)x6; my $primary_tag = $object->primary_tag; my $source_tag = $object->source_tag || ''; $primary_tag .= ":$source_tag"; my $typeid = $self->_typeid($primary_tag,1); my $frozen = $self->no_blobs() ? 0 : $self->freeze($object); $sth->execute($id,$frozen,$index_flag||0,@location,$typeid) or $self->throw($sth->errstr); my $dbh = $self->dbh; $object->primary_id($dbh->{mysql_insertid}) unless defined $id; $self->flag_for_indexing($dbh->{mysql_insertid}) if $self->{bulk_update_in_progress}; } # doesn't work with this schema, since we have to update name and attribute # tables which need object ids, which we can only know by replacing feats in # the feature table one by one sub bulk_replace { my $self = shift; my $index_flag = shift || undef; my @objects = @_; my $features = $self->_feature_table; my @insert_values; foreach my $object (@objects) { my $id = $object->primary_id; my @location = $index_flag ? $self->_get_location_and_bin($object) : (undef)x6; my $primary_tag = $object->primary_tag; my $source_tag = $object->source_tag || ''; $primary_tag .= ":$source_tag"; my $typeid = $self->_typeid($primary_tag,1); push(@insert_values, ($id,0,$index_flag||0,@location,$typeid)); } my @value_blocks; for (1..@objects) { push(@value_blocks, '(?,?,?,?,?,?,?,?,?,?)'); } my $value_blocks = join(',', @value_blocks); my $sql = qq{REPLACE INTO $features (id,object,indexed,seqid,start,end,strand,tier,bin,typeid) VALUES $value_blocks}; my $sth = $self->_prepare($sql); $sth->execute(@insert_values) or $self->throw($sth->errstr); } ### # Insert one Bio::SeqFeatureI into database. primary_id must be undef # sub insert { my $self = shift; my $object = shift; my $index_flag = shift || 0; $self->_load_class($object); defined $object->primary_id and $self->throw("$object already has a primary id"); my $features = $self->_feature_table; my $sth = $self->_prepare(<<END); INSERT INTO $features (id,object,indexed) VALUES (?,?,?) END $sth->execute(undef,$self->freeze($object),$index_flag) or $self->throw($sth->errstr); my $dbh = $self->dbh; $object->primary_id($dbh->{mysql_insertid}); $self->flag_for_indexing($dbh->{mysql_insertid}) if $self->{bulk_update_in_progress}; }
sub types { my $self = shift; eval "require Bio::DB::GFF::Typename" unless Bio::DB::GFF::Typename->can('new'); my $typelist = $self->_typelist_table; my $sql = <<END; SELECT tag from $typelist END ; $self->_print_query($sql) if DEBUG || $self->debug; my $sth = $self->_prepare($sql); $sth->execute() or $self->throw($sth->errstr); my @results; while (my($tag) = $sth->fetchrow_array) { push @results,Bio::DB::GFF::Typename->new($tag); } $sth->finish; return @results; }
sub toplevel_types { my $self = shift; eval "require Bio::DB::GFF::Typename" unless Bio::DB::GFF::Typename->can('new'); my $typelist = $self->_typelist_table; my $features = $self->_feature_table; my $sql = <<END; SELECT distinct(tag) from $typelist as tl,$features as f WHERE tl.id=f.typeid AND f.indexed=1 END ; $self->_print_query($sql) if DEBUG || $self->debug; my $sth = $self->_prepare($sql); $sth->execute() or $self->throw($sth->errstr); my @results; while (my($tag) = $sth->fetchrow_array) { push @results,Bio::DB::GFF::Typename->new($tag); } $sth->finish; return @results; } ### # Insert a bit of DNA or protein into the database # sub _insert_sequence { my $self = shift; my ($seqid,$seq,$offset) = @_; my $id = $self->_locationid($seqid); my $sequence = $self->_sequence_table; my $sth = $self->_prepare(<<END); REPLACE INTO $sequence (id,offset,sequence) VALUES (?,?,?) END $sth->execute($id,$offset,$seq) or $self->throw($sth->errstr); } ### # This subroutine flags the given primary ID for later reindexing # sub flag_for_indexing { my $self = shift; my $id = shift; my $needs_updating = $self->_update_table; my $sth = $self->_prepare("REPLACE INTO $needs_updating VALUES (?)"); $sth->execute($id) or $self->throw($self->dbh->errstr); } ### # Update indexes for given object # sub _update_indexes { my $self = shift; my $obj = shift; defined (my $id = $obj->primary_id) or return; if ($self->{bulk_update_in_progress}) { $self->_dump_update_name_index($obj,$id); $self->_dump_update_attribute_index($obj,$id); } else { $self->_update_name_index($obj,$id); $self->_update_attribute_index($obj,$id); } } sub _update_name_index { my $self = shift; my ($obj,$id) = @_; my $name = $self->_name_table; my $primary_id = $obj->primary_id; $self->_delete_index($name,$id); my ($names,$aliases) = $self->feature_names($obj); my $sth = $self->_prepare("INSERT INTO $name (id,name,display_name) VALUES (?,?,?)"); $sth->execute($id,$_,1) or $self->throw($sth->errstr) foreach @$names; $sth->execute($id,$_,0) or $self->throw($sth->errstr) foreach @$aliases; $sth->finish; } sub _update_attribute_index { my $self = shift; my ($obj,$id) = @_; my $attribute = $self->_attribute_table; $self->_delete_index($attribute,$id); my $sth = $self->_prepare("INSERT INTO $attribute (id,attribute_id,attribute_value) VALUES (?,?,?)"); for my $tag ($obj->get_all_tags) { my $tagid = $self->_attributeid($tag); for my $value ($obj->get_tag_values($tag)) { $sth->execute($id,$tagid,$value) or $self->throw($sth->errstr); } } $sth->finish; } sub _genericid { my $self = shift; my ($table,$namefield,$name,$add_if_missing) = @_; my $qualified_table = $self->_qualify($table); my $sth = $self->_prepare(<<END); SELECT id FROM $qualified_table WHERE $namefield=? END $sth->execute($name) or die $sth->errstr; my ($id) = $sth->fetchrow_array; $sth->finish; return $id if defined $id; return unless $add_if_missing; $sth = $self->_prepare(<<END); INSERT INTO $qualified_table ($namefield) VALUES (?) END $sth->execute($name) or die $sth->errstr; my $dbh = $self->dbh; return $dbh->{mysql_insertid}; } sub _typeid { shift->_genericid('typelist','tag',shift,1); } sub _locationid { shift->_genericid('locationlist','seqname',shift,1); } sub _locationid_nocreate { shift->_genericid('locationlist','seqname',shift,0); } sub _attributeid { shift->_genericid('attributelist','tag',shift,1); } sub _get_location_and_bin { my $self = shift; my $feature = shift; my $seqid = $self->_locationid($feature->seq_id||''); my $start = $feature->start; my $end = $feature->end; my $strand = $feature->strand || 0; my ($tier,$bin) = $self->get_bin($start,$end); return ($seqid,$start,$end,$strand,$tier,$bin); } sub get_bin { my $self = shift; my ($start,$end) = @_; my $binsize = MIN_BIN; my ($bin_start,$bin_end,$tier); $tier = 0; while (1) { $bin_start = int $start/$binsize; $bin_end = int $end/$binsize; last if $bin_start == $bin_end; $binsize *= 10; $tier++; } return ($tier,$bin_start); } sub bin_where { my $self = shift; my ($start,$end,$f) = @_; my (@bins,@args); my $tier = 0; my $binsize = MIN_BIN; while ($binsize <= MAX_BIN) { my $bin_start = int($start/$binsize); my $bin_end = int($end/$binsize); push @bins,"($f.tier=? AND $f.bin between ? AND ?)"; push @args,($tier,$bin_start,$bin_end); $binsize *= 10; $tier++; } my $query = join ("\n\t OR ",@bins); return wantarray ? ($query,@args) : substitute($query,@args); } sub _delete_index { my $self = shift; my ($table_name,$id) = @_; return if $self->{reindexing}; my $sth = $self->_prepare("DELETE FROM $table_name WHERE id=?") or $self->throw($self->dbh->errstr); $sth->execute($id); } # given a statement handler that is expected to return rows of (id,object) # unthaw each object and return a list of 'em sub _sth2objs { my $self = shift; my $sth = shift; my @result; while (my ($id,$o,$typeid,$seqid,$start,$end,$strand) = $sth->fetchrow_array) { my $obj; if ($o eq '0') { # rebuild a new feat object from the data stored in the db $obj = $self->_rebuild_obj($id,$typeid,$seqid,$start,$end,$strand); } else { $obj = $self->thaw($o,$id); } push @result,$obj; } $sth->finish; return @result; } # given a statement handler that is expected to return rows of (id,object) # unthaw each object and return a list of 'em sub _sth2obj { my $self = shift; my $sth = shift; my ($id,$o,$typeid,$seqid,$start,$end,$strand) = $sth->fetchrow_array; return unless defined $o; my $obj; if ($o eq '0') { # I don't understand why an object ever needs to be rebuilt! # rebuild a new feat object from the data stored in the db $obj = $self->_rebuild_obj($id,$typeid,$seqid,$start,$end,$strand); } else { $obj = $self->thaw($o,$id); } $obj; } sub _rebuild_obj { my ($self, $id, $typeid, $db_seqid, $start, $end, $strand) = @_; my ($type, $source, $seqid); # convert typeid to type and source if (exists $self->{_type_cache}->{$typeid}) { ($type, $source) = @{$self->{_type_cache}->{$typeid}}; } else { my $sql = qq{ SELECT `tag` FROM typelist WHERE `id` = ? }; my $sth = $self->_prepare($sql) or $self->throw($self->dbh->errstr); $sth->execute($typeid); my $result; $sth->bind_columns(\$result); while ($sth->fetch()) { # there should be only one row returned, but we ensure to get all rows } ($type, $source) = split(':', $result); $self->{_type_cache}->{$typeid} = [$type, $source]; } # convert the db seqid to the sequence name if (exists $self->{_seqid_cache}->{$db_seqid}) { $seqid = $self->{_seqid_cache}->{$db_seqid}; } else { my $sql = qq{ SELECT `seqname` FROM locationlist WHERE `id` = ? }; my $sth = $self->_prepare($sql) or $self->throw($self->dbh->errstr); $sth->execute($db_seqid); $sth->bind_columns(\$seqid); while ($sth->fetch()) { # there should be only one row returned, but we ensure to get all rows } $self->{_seqid_cache}->{$db_seqid} = $seqid; } # get the names from name table? # get the attributes and store those in obj my $sql = qq{ SELECT attribute_id,attribute_value FROM attribute WHERE `id` = ? }; my $sth = $self->_prepare($sql) or $self->throw($self->dbh->errstr); $sth->execute($id); my ($attribute_id, $attribute_value); $sth->bind_columns(\($attribute_id, $attribute_value)); my %attribs; while ($sth->fetch()) { # convert the attribute_id to its real name my $attribute; if (exists $self->{_attribute_cache}->{$attribute_id}) { $attribute = $self->{_attribute_cache}->{$attribute_id}; } else { my $sql = qq{ SELECT `tag` FROM attributelist WHERE `id` = ? }; my $sth2 = $self->_prepare($sql) or $self->throw($self->dbh->errstr); $sth2->execute($attribute_id); $sth2->bind_columns(\$attribute); while ($sth2->fetch()) { # there should be only one row returned, but we ensure to get all rows } $self->{_attribute_cache}->{$attribute_id} = $attribute; } if ($source && $attribute eq 'source' && $attribute_value eq $source) { next; } $attribs{$attribute} = $attribute_value; } # if we weren't called with all the params, pull those out of the database too if ( not ( grep { defined($_) } ( $typeid, $db_seqid, $start, $end, $strand ))) { my $sql = qq{ SELECT start,end,tag,strand,seqname FROM feature,feature_location,typelist,locationlist WHERE feature.id=feature_location.id AND feature.typeid=typelist.id AND seqid=locationlist.id AND feature.id = ? }; my $sth = $self->_prepare($sql) or $self->throw($self->dbh->errstr); $sth->execute($id); my ($feature_start, $feature_end, $feature_type, $feature_strand,$feature_seqname); $sth->bind_columns(\($feature_start, $feature_end, $feature_type, $feature_strand, $feature_seqname)); while ($sth->fetch()) { # there should be only one row returned, but we call like this to # ensure we get all rows } $start ||= $feature_start; $end ||= $feature_end; $strand ||= $feature_strand; $seqid ||= $feature_seqname; my( $feature_typename , $feature_typesource ) = split /:/ , $feature_type; $type ||= $feature_typename; $source ||= $feature_typesource; } my $obj = Bio::SeqFeature::Lite->new(-primary_id => $id, $type ? (-type => $type) : (), $source ? (-source => $source) : (), $seqid ? (-seq_id => $seqid) : (), defined $start ? (-start => $start) : (), defined $end ? (-end => $end) : (), defined $strand ? (-strand => $strand) : (), keys %attribs ? (-attributes => \%attribs) : ()); return $obj; } sub _prepare { my $self = shift; my $query = shift; my $dbh = $self->dbh; my $sth = $dbh->prepare_cached($query, {}, 3) or $self->throw($dbh->errstr); $sth; } #################################################################################################### # SQL Fragment generators #################################################################################################### sub _attribute_table { shift->_qualify('attribute') } sub _attributelist_table { shift->_qualify('attributelist') } sub _feature_table { shift->_qualify('feature') } sub _interval_stats_table { shift->_qualify('interval_stats') } sub _location_table { shift->_qualify('location') } sub _locationlist_table { shift->_qualify('locationlist') } sub _meta_table { shift->_qualify('meta') } sub _name_table { shift->_qualify('name') } sub _parent2child_table { shift->_qualify('parent2child') } sub _sequence_table { shift->_qualify('sequence') } sub _type_table { shift->_qualify('feature') } sub _typelist_table { shift->_qualify('typelist') } sub _update_table { shift->_qualify('update_table') } sub _make_attribute_where { my $self = shift; my ($attributetable,$attributenametable,$attributes) = @_; my @args; my @sql; my $dbh = $self->dbh; foreach (keys %$attributes) { my @match_values; my @values = ref($attributes->{$_}) && ref($attributes->{$_}) eq 'ARRAY' ? @{$attributes->{$_}} : $attributes->{$_}; foreach (@values) { # convert * into % for wildcard matches s/\*/%/g; } my $match = join ' OR ',map { /%/ ? "$attributetable.attribute_value LIKE ?" : "$attributetable.attribute_value=?" } @values; push @sql,"($attributenametable.tag=? AND ($match))"; push @args,($_,@values); } return (join(' OR ',@sql),@args); } sub _make_attribute_group { my $self = shift; my ($table_name,$attributes) = @_; my $key_count = keys %$attributes or return; return "f.id,f.object,f.typeid,f.seqid,f.start,f.end,f.strand HAVING count(f.id)>?",$key_count-1; } sub _print_query { my $self = shift; my ($query,@args) = @_; while ($query =~ /\?/) { my $arg = $self->dbh->quote(shift @args); $query =~ s/\?/$arg/; } warn $query,"\n"; } ### # special-purpose store for bulk loading - write to a file rather than to the db # sub _dump_store { my $self = shift; my $indexed = shift; my $count = 0; my $store_fh = $self->dump_filehandle('feature'); my $dbh = $self->dbh; my $autoindex = $self->autoindex; for my $obj (@_) { my $id = $self->next_id; my ($seqid,$start,$end,$strand,$tier,$bin) = $indexed ? $self->_get_location_and_bin($obj) : (undef)x6; my $primary_tag = $obj->primary_tag; my $source_tag = $obj->source_tag || ''; $primary_tag .= ":$source_tag"; my $typeid = $self->_typeid($primary_tag,1); print $store_fh join("\t",$id,$typeid,$seqid,$start,$end,$strand,$tier,$bin,$indexed,$dbh->quote($self->freeze($obj))),"\n"; $obj->primary_id($id); $self->_update_indexes($obj) if $indexed && $autoindex; $count++; } # remember whether we are have ever stored a non-indexed feature unless ($indexed or $self->{indexed_flag}++) { $self->subfeatures_are_indexed(0); } $count; } sub _dump_add_SeqFeature { my $self = shift; my $parent = shift; my @children = @_; my $dbh = $self->dbh; my $fh = $self->dump_filehandle('parent2child'); my $parent_id = (ref $parent ? $parent->primary_id : $parent) or $self->throw("$parent should have a primary_id"); my $count = 0; for my $child_id (@children) { print $fh join("\t",$parent_id,$child_id),"\n"; $count++; } $count; } sub _dump_update_name_index { my $self = shift; my ($obj,$id) = @_; my $fh = $self->dump_filehandle('name'); my $dbh = $self->dbh; my ($names,$aliases) = $self->feature_names($obj); print $fh join("\t",$id,$dbh->quote($_),1),"\n" foreach @$names; print $fh join("\t",$id,$dbh->quote($_),0),"\n" foreach @$aliases; } sub _dump_update_attribute_index { my $self = shift; my ($obj,$id) = @_; my $fh = $self->dump_filehandle('attribute'); my $dbh = $self->dbh; for my $tag ($obj->all_tags) { my $tagid = $self->_attributeid($tag); for my $value ($obj->each_tag_value($tag)) { print $fh join("\t",$id,$tagid,$dbh->quote($value)),"\n"; } } } sub coverage_array { my $self = shift; my ($seq_name,$start,$end,$types,$bins) = rearrange([['SEQID','SEQ_ID','REF'],'START',['STOP','END'], ['TYPES','TYPE','PRIMARY_TAG'],'BINS'],@_); $bins ||= 1000; $start ||= 1; unless ($end) { my $segment = $self->segment($seq_name) or $self->throw("unknown seq_id $seq_name"); $end = $segment->end; } my $binsize = ($end-$start+1)/$bins; my $seqid = $self->_locationid_nocreate($seq_name) || 0; return [] unless $seqid; # where each bin starts my @his_bin_array = map {$start + $binsize * $_} (0..$bins-1); my @sum_bin_array = map {int(($_-1)/SUMMARY_BIN_SIZE)} @his_bin_array; my $interval_stats = $self->_interval_stats_table; my ($sth,@a); if ($types) { # pick up the type ids my ($from,$where,$group); ($from,$where,$group,@a) = $self->_types_sql($types,'b'); $where =~ s/.+AND//s; $sth = $self->_prepare(<<END); SELECT id,tag FROM $from WHERE $where END ; } else { $sth = $self->_prepare(<<END); SELECT id,tag FROM typelist END } my (@t,$report_tag); $sth->execute(@a); while (my ($t,$tag) = $sth->fetchrow_array) { $report_tag ||= $tag; push @t,$t; } my %bins; my $sql = <<END; SELECT bin,cum_count FROM $interval_stats WHERE typeid=? AND seqid=? AND bin >= ? LIMIT 1 END ; $sth = $self->_prepare($sql); eval { for my $typeid (@t) { for (my $i=0;$i<@sum_bin_array;$i++) { my @args = ($typeid,$seqid,$sum_bin_array[$i]); $self->_print_query($sql,@args) if $self->debug; $sth->execute(@args) or $self->throw($sth->errstr); my ($bin,$cum_count) = $sth->fetchrow_array; push @{$bins{$typeid}},[$bin,$cum_count]; } } }; return unless %bins; my @merged_bins; my $firstbin = int(($start-1)/$binsize); for my $type (keys %bins) { my $arry = $bins{$type}; my $last_count = $arry->[0][1]; my $last_bin = -1; my $i = 0; my $delta; for my $b (@$arry) { my ($bin,$count) = @$b; $delta = $count - $last_count if $bin > $last_bin; $merged_bins[$i++] += $delta; $last_count = $count; $last_bin = $bin; } } return wantarray ? (\@merged_bins,$report_tag) : \@merged_bins; } sub build_summary_statistics { my $self = shift; my $interval_stats = $self->_interval_stats_table; my $dbh = $self->dbh; $self->begin_work; my $sbs = SUMMARY_BIN_SIZE; my $result = eval { $self->_add_interval_stats_table; $self->_disable_keys($dbh,$interval_stats); $dbh->do("DELETE FROM $interval_stats"); my $insert = $dbh->prepare(<<END) or $self->throw($dbh->errstr); INSERT INTO $interval_stats (typeid,seqid,bin,cum_count) VALUES (?,?,?,?) END my $sql = $self->_fetch_indexed_features_sql; my $select = $dbh->prepare($sql) or $self->throw($dbh->errstr); my $current_bin = -1; my ($current_type,$current_seqid,$count); my $cum_count = 0; my (%residuals,$last_bin); my $le = -t \*STDERR ? "\r" : "\n"; print STDERR "\n"; $select->execute; while (my($typeid,$seqid,$start,$end) = $select->fetchrow_array) { print STDERR $count," features processed$le" if ++$count % 1000 == 0; my $bin = int($start/$sbs); $current_type ||= $typeid; $current_seqid ||= $seqid; # because the input is sorted by start, no more features will contribute to the # current bin so we can dispose of it if ($bin != $current_bin) { if ($seqid != $current_seqid or $typeid != $current_type) { # load all bins left over $self->_load_bins($insert,\%residuals,\$cum_count,$current_type,$current_seqid); %residuals = () ; $cum_count = 0; } else { # load all up to current one $self->_load_bins($insert,\%residuals,\$cum_count,$current_type,$current_seqid,$current_bin); } } $last_bin = $current_bin; ($current_seqid,$current_type,$current_bin) = ($seqid,$typeid,$bin); # summarize across entire spanned region my $last_bin = int(($end-1)/$sbs); for (my $b=$bin;$b<=$last_bin;$b++) { $residuals{$b}++; } } # handle tail case # load all bins left over $self->_load_bins($insert,\%residuals,\$cum_count,$current_type,$current_seqid); $self->_enable_keys($dbh,$interval_stats); 1; }; if ($result) { $self->commit } else { warn "Can't build summary statistics: $@"; $self->rollback }; print STDERR "\n"; } sub _load_bins { my $self = shift; my ($insert,$residuals,$cum_count,$type,$seqid,$stop_after) = @_; for my $b (sort {$a<=>$b} keys %$residuals) { last if defined $stop_after and $b > $stop_after; $$cum_count += $residuals->{$b}; my @args = ($type,$seqid,$b,$$cum_count); $insert->execute(@args); delete $residuals->{$b}; # no longer needed } } sub _add_interval_stats_table { my $self = shift; my $tables = $self->table_definitions; my $interval_stats = $self->_interval_stats_table; $self->dbh->do("CREATE TABLE IF NOT EXISTS $interval_stats $tables->{interval_stats}"); } sub _fetch_indexed_features_sql { my $self = shift; my $features = $self->_feature_table; return <<END; SELECT typeid,seqid,start-1,end FROM $features as f WHERE f.indexed=1 ORDER BY typeid,seqid,start END } sub _disable_keys { my $self = shift; my ($dbh,$table) = @_; $dbh->do("ALTER TABLE $table DISABLE KEYS"); } sub _enable_keys { my $self = shift; my ($dbh,$table) = @_; $dbh->do("ALTER TABLE $table ENABLE KEYS"); } sub time { return Time::HiRes::time() if Time::HiRes->can('time'); return time(); } sub DESTROY { my $self = shift; if ($self->{bulk_update_in_progress}) { # be sure to remove temp files for my $table ($self->_feature_table,$self->index_tables) { my $path = $self->dump_path($table); unlink $path; } } } sub begin_work { my $self = shift; return if $self->{_in_transaction}++; my $dbh = $self->dbh; return unless $dbh->{AutoCommit}; $dbh->begin_work; } sub commit { my $self = shift; return unless $self->{_in_transaction}; delete $self->{_in_transaction}; $self->dbh->commit; } sub rollback { my $self = shift; return unless $self->{_in_transaction}; delete $self->{_in_transaction}; $self->dbh->rollback; } 1;