Fix graphml output / input format

[scpubgit/stemmatology.git] / lib / Text / Tradition / Collation / Reading.pm

package Text::Tradition::Collation::Reading;

use Moose;
use MooseX::NonMoose;
use Text::Tradition::Collation::Position;

extends 'Graph::Easy::Node';

has 'position' => (
    is => 'rw',
    isa => 'Text::Tradition::Collation::Position',
    predicate => 'has_position',
    );
    
has 'rank' => (
    is => 'rw',
    isa => 'Int',
    predicate => 'has_rank',
    );
    
has 'is_lacuna' => (
    is => 'rw',
    isa => 'Bool',
    );

# This contains an array of reading objects; the array is a pool,
# shared by the reading objects inside the pool.  When a reading is
# added to the pool, all the same_as attributes should be updated.
has 'same_as' => (
    is => 'rw',
    isa => 'ArrayRef[Text::Tradition::Collation::Reading]',
    );

# Deal with the non-arg option for Graph::Easy's constructor.
around BUILDARGS => sub {
    my $orig = shift;
    my $class = shift;

    my %args;
    if( @_ == 1 && ref( $_[0] ) ne 'HASH' ) {
	return $class->$orig( 'name' => $_[0] );
    } else {
	return $class->$orig( @_ );
    }
};

# Take constructor args as well as a Position argument.
around position => sub {
    my $orig = shift;
    my $self = shift;
    return $self->$orig() unless @_;

    my @args = @_;
    unless( @_ == 1 && ref( $_[0] ) eq 'Text::Tradition::Collation::Position' ) {
	# We have constructor arguments; pass them to Position.
	@args = ( Text::Tradition::Collation::Position->new( @_ ) );
    }
    $self->$orig( @args );
};

# A lacuna node is also a meta node.
before is_lacuna => sub {
	my( $self, $arg ) = @_;
	if( $arg ) {
		$self->is_meta( 1 );
	}
};

# Initialize the identity pool. 
sub BUILD {
    my( $self, $args ) = @_;
    $self->same_as( [ $self ] );
}

sub text {
    # Wrapper function around 'label' attribute.
    my $self = shift;
    if( @_ ) {
        if( defined $_[0] ) {
        	$self->set_attribute( 'label', $_[0] );
        } else {
            $self->del_attribute( 'label' );
        }
    }
    return $self->label;
}

sub witnessed_by {
    my( $self, $sigil, $backup ) = @_;
    my @wits = $self->witnesses;
    return 1 if grep { $_ eq $sigil } @wits;
    if( $backup ) {
        return 1 if grep { $_ eq $backup } @wits;
    }
    return 0;
}
    
sub witnesses {
    my( $self ) = @_;
    my @paths = grep { $_->get_attribute( 'class' ) eq 'path' } $self->outgoing;
    push( @paths, grep { $_->get_attribute( 'class' ) eq 'path' } $self->incoming );
    my %wits;
    foreach my $p ( @paths ) {
        if( $p->has_hidden_witnesses ) {
            foreach ( @{$p->hidden_witnesses} ) {
                $wits{$_} = 1;
            }
        } else {
            $wits{$p->label} = 1;
        }
    }
    return keys %wits;
}

sub merge_from {
    my( $self, $merged_node ) = @_;
    # Adopt the identity pool of the other node.
    my @now_identical = grep { $_ ne $merged_node } @{$merged_node->same_as};
    my $new_pool = _merge_array_pool( \@now_identical, $self->same_as )
	if @now_identical;

    # TODO Adopt the relationship attributes and segment memberships of the other node.
}

## Dealing with transposed readings.  These methods are only really
## applicable if we have a linear collation graph.

sub set_identical {
    my( $self, $other_node ) = @_; 
    my $enlarged_pool = _merge_array_pool( $self->same_as, 
					   $other_node->same_as );

    # ...and set this node to point to the enlarged pool.
    $self->same_as( $enlarged_pool );
}   

sub identical_readings {
    my $self = shift;
    my @same = grep { $_ ne $self } @{$self->same_as};
    return @same;
}

sub _merge_array_pool {
    my( $pool, $main_pool ) = @_;
    my %poolhash;
    foreach ( @$main_pool ) {
	# Note which nodes are already in the main pool so that we
	# don't re-add them.
	$poolhash{$_->name} = 1;
    }

    foreach( @$pool ) {
	# Add the remaining nodes to the main pool...
	push( @$main_pool, $_ ) unless $poolhash{$_->name};
    }
    return $main_pool;
}

sub has_primary {
    my $self = shift;
    my $pool = $self->same_as;
    return $pool->[0]->name ne $self->name;
}

sub primary {
    my $self = shift;
    return $self->same_as->[0];
}

sub is_at_position {
    my $self = shift;
    return undef unless $self->has_position;
    return $self->position->is_colocated( @_ );
}

# Looks from the outside like an accessor for a Boolean, but really 
# sets the node's class.  Should apply to start, end, and lacunae.

sub is_meta {
	my $self = shift;
	my $arg = shift;
	if( defined $arg && $arg ) {
		$self->set_attribute( 'class', 'meta' );
	} elsif ( defined $arg ) {
		$self->del_attribute( 'class' );
	}
	return $self->sub_class eq 'meta';	
}

# Returns all readings that adjoin this one on any path.
sub neighbor_readings {
    my( $self, $direction ) = @_;
    $direction = 'both' unless $direction;
    my @paths = grep { $_->isa( 'Text::Tradition::Collation::Path' ) } $self->edges;
    my %connected;
    foreach my $p ( @paths ) {
	if( $p->to eq $self ) {
	    next if $direction eq 'forward';
	    $connected{$p->from->name} = $p->from;
	} else { # $p->from eq $self
	    next if $direction =~ /^back/;
	    $connected{$p->to->name} = $p->to;
	}
    }
    return values( %connected );
}

# Returns all readings related to the one we've got.
sub related_readings {
    my( $self, $colocated ) = @_;
    my @related;
    foreach my $e ( $self->edges ) {
        next unless $e->isa( 'Text::Tradition::Collation::Relationship' );
        next if $colocated && $e->type eq 'repetition';
        push( @related, $e->from eq $self ? $e->to : $e->from );
    }
    return @related;
}

## Keep track of which readings are unchanged across witnesses.
sub is_common {
    my( $self ) = shift;
    return $self->get_attribute( 'class' ) eq 'common';
}

## TODO Rationalize make_common, is_meta, etc.
sub make_common {
    my( $self ) = shift;
    $self->set_attribute( 'class', 'common' );
}

sub make_variant {
    my( $self ) = shift;
    $self->set_attribute( 'class', 'variant' );
}

no Moose;
__PACKAGE__->meta->make_immutable;

1;

######################################################
## copied from Graph::Easy::Parser docs
######################################################
# when overriding nodes, we also need ::Anon

package Text::Tradition::Collation::Reading::Anon;
use Moose;
use MooseX::NonMoose;
extends 'Text::Tradition::Collation::Reading';
extends 'Graph::Easy::Node::Anon';
no Moose;
__PACKAGE__->meta->make_immutable;

1;
# use base qw/Text::Tradition::Collation::Reading/;
# use base qw/Graph::Easy::Node::Anon/;

######################################################
# and :::Empty

package Text::Tradition::Collation::Reading::Empty;
use Moose;
use MooseX::NonMoose;
extends 'Graph::Easy::Node::Empty';
no Moose;
__PACKAGE__->meta->make_immutable;

1;
# use base qw/Text::Tradition::Collation::Reading/;

######################################################