X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FText%2FTradition%2FCollation.pm;h=50903a28916f44eee15dd705c5a4f039dd2304b2;hb=a753cc846a37032aca109851f8de6ce760283383;hp=abdb4415a11f8d9ec0864095478b811b77041510;hpb=3a2ebbf4607bc7ab83788e5a57c44a960829cd1c;p=scpubgit%2Fstemmatology.git

diff --git a/lib/Text/Tradition/Collation.pm b/lib/Text/Tradition/Collation.pm
index abdb441..50903a2 100644
--- a/lib/Text/Tradition/Collation.pm
+++ b/lib/Text/Tradition/Collation.pm
@@ -312,15 +312,20 @@ sub relationship_valid {
 
 sub related_readings {
 	my( $self, $reading, $colocated ) = @_;
-	$reading = $reading->id 
-		if ref( $reading ) eq 'Text::Tradition::Collation::Reading';
+	my $return_object;
+	if( ref( $reading ) eq 'Text::Tradition::Collation::Reading' ) {
+		$reading = $reading->id;
+		$return_object = 1;
+		print STDERR "Returning related objects\n";
+	} else {
+		print STDERR "Returning related object names\n";
+	}
 	my @related = $self->relations->all_reachable( $reading );
 	if( $colocated ) {
 		my @colo = grep { $self->relations->has_edge_attribute( $reading, $_, 'colocated' ) } @related;
-		return @colo;
-	} else {
-		return @related;
-	}
+		@related = @colo;
+	} 
+	return $return_object ? map { $self->reading( $_ ) } @related : @related;
 }
 
 =head2 Output method(s)