store tradition objects in a KiokuDB instance

[scpubgit/stemmatology.git] / script / save_to_db.pl

diff --git a/script/save_to_db.pl b/script/save_to_db.pl
new file mode 100644 (file)
index 0000000..45ebb17
--- /dev/null
+++ b/script/save_to_db.pl
@@ -0,0 +1,85 @@
+#!/usr/bin/env perl
+
+use lib 'lib';
+use strict;
+use warnings;
+use File::Basename;
+use KiokuDB;
+use KiokuDB::TypeMap::Entry::Naive;
+use Text::Tradition;
+use Text::Tradition::Stemma;
+
+# Make a KiokuDB store from the traditions data we have.
+
+my $kdb = KiokuDB->connect( "dbi:SQLite:dbname=db/traditions.db", 
+       create => 1,
+       typemap => KiokuDB::TypeMap->new(
+            isa_entries => {
+                "Graph::Easy::Base" => KiokuDB::TypeMap::Entry::Naive->new,
+                "Graph" => KiokuDB::TypeMap::Entry::Naive->new,
+                "Graph::AdjacencyMap" => KiokuDB::TypeMap::Entry::Naive->new,
+            },
+        ),
+    );
+    
+my %stemma_map = (
+       'florilegium.xml' => 'stemma_a.dot',
+       'besoin.xml' => 'stemma_b.dot',
+       'heinrichi.xml' => 'stemma_h.dot',
+       'parzival.xml' => 'stemma_p.dot',
+       's158.xml' => 'stemma_s.dot',
+       );
+
+my $dir = $ARGV[0];
+if( $dir ) {
+       $dir =~ s/\/$//;
+       opendir( DIR, $dir ) or die "Could not open directory $dir";
+       while( readdir DIR ) {
+               next unless /\.xml$/;
+               my $stemmafile = "$dir/" . $stemma_map{$_};     
+               my $tradition = Text::Tradition->new( 
+                       'input' => 'Self',
+                       'file' => "$dir/$_",
+                       'linear' => 1,
+                       );
+               open my $stemma_fh, '<', $stemmafile or die "Could not read stemma file $stemmafile";           
+               my $stemma = Text::Tradition::Stemma->new(
+                       'collation' => $tradition->collation,
+                       'dot' => $stemma_fh,
+                       );
+                       
+               my $scope = $kdb->new_scope;
+               my $tid = $kdb->store( $tradition );
+               my $sid = $kdb->store( $stemma );
+               
+               print STDERR "Stored tradition and stemma for " . $tradition->name 
+                       . ", got $tid / $sid as the ref\n";
+       }
+}
+
+# Now try reading the objects from the DB.
+
+my $scope = $kdb->new_scope;
+
+my $stream = $kdb->root_set;
+until( $stream->is_done ) {
+       foreach my $t ( $stream->items ) {
+               print STDERR "*** Object " . $kdb->object_to_id( $t ) . " ***\n";
+               if( ref( $t ) eq 'Text::Tradition' ) {
+                       print STDERR "Got tradition " . $t->name . " out of the database\n";
+                       my @wits = map { $_->sigil } $t->witnesses;
+                       print STDERR "...with witnesses @wits\n";
+                       my $c = $t->collation;
+                       print STDERR "Collation has " . scalar( $c->readings ) . " readings\n";
+                       print STDERR "Collation has " . scalar( $c->paths ) . " paths\n";
+                       print STDERR "Collation has " . scalar( $c->relationships ) . " relationship links\n";
+               } elsif( ref( $t ) eq 'Text::Tradition::Stemma' ) {
+                       print STDERR "Got stemma for tradition " . $t->collation->tradition->name 
+                               . " out of the database\n";
+                       print STDERR "Stemma graph is " . $t->graph . "\n";
+               } else {
+                       print STDERR "Got unexpected object of type " . ref( $t ) 
+                               . " out of the database\n";
+               }
+       }
+}
\ No newline at end of file