RegistrationTableModel.java

/*-
 * #%L
 * Fiji distribution of ImageJ for the life sciences.
 * %%
 * Copyright (C) 2007 - 2017 Fiji developers.
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 2 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-2.0.html>.
 * #L%
 */
package spim.fiji.spimdata.explorer.registration;

import java.util.ArrayList;
import java.util.List;

import javax.swing.table.AbstractTableModel;

import spim.fiji.spimdata.explorer.ViewSetupExplorerPanel;
import mpicbg.spim.data.generic.sequence.BasicViewDescription;
import mpicbg.spim.data.registration.ViewRegistration;
import mpicbg.spim.data.registration.ViewRegistrations;
import mpicbg.spim.data.registration.ViewTransform;

public class RegistrationTableModel extends AbstractTableModel
{
	private static final long serialVersionUID = -1263388435427674269L;

	ViewRegistrations viewRegistrations;
	final ArrayList< String > columnNames;
	final RegistrationExplorerPanel panel;
	
	BasicViewDescription< ? > currentVD;
	
	public RegistrationTableModel( final ViewRegistrations viewRegistrations, final RegistrationExplorerPanel panel )
	{
		this.columnNames = new ArrayList< String >();
		this.panel = panel;
		this.columnNames.add( "Transformation Name" );
		
		for ( int row = 0; row < 3; ++row )
			for ( int col = 0; col < 4; ++ col )
				this.columnNames.add( "m" + row + "" + col );
		
		this.viewRegistrations = viewRegistrations;
		this.currentVD = null;
	}

	protected void update( final ViewRegistrations viewRegistrations ) { this.viewRegistrations = viewRegistrations; }
	protected ViewRegistrations getViewRegistrations() { return viewRegistrations; }
	protected BasicViewDescription< ? > getCurrentViewDescription() { return currentVD; } 
	
	protected void updateViewDescription( final BasicViewDescription< ? > vd )
	{
		this.currentVD = vd;
		
		// update everything
		fireTableDataChanged();
	}

	@Override
	public int getColumnCount() { return columnNames.size(); }
	
	@Override
	public int getRowCount()
	{ 
		if ( currentVD == null )
			return 1;
		else
			return Math.max( 1, viewRegistrations.getViewRegistration( currentVD ).getTransformList().size() );
	}

	@Override
	public boolean isCellEditable( final int row, final int column )
	{
		// we can change name and the affine transform itself
		return true;
	}

	@Override
	public void setValueAt( final Object value, final int row, final int column )
	{
		final List< ViewTransform > vtList = viewRegistrations.getViewRegistration( currentVD ).getTransformList(); 
		final ViewTransform vt = vtList.get( row );
		
		if ( column == 0 )
		{
			final String newName = value.toString();
			vtList.set( row, RegistrationExplorerPanel.newName( vt, newName ) );
		}
		else
		{
			try
			{
				final double v = Double.parseDouble( value.toString() );
				vtList.set( row, RegistrationExplorerPanel.newMatrixEntry( vt, v, column - 1 ) );
			}
			catch ( Exception e )
			{
				System.out.println( "Cannot parse: " + value );
			}
		}
		
		viewRegistrations.getViewRegistration( currentVD ).updateModel();
		
		// do something ...
		ViewSetupExplorerPanel.bdvPopup().updateBDV();
		fireTableCellUpdated( row, column );
	}
	
	@Override
	public Object getValueAt( final int row, final int column )
	{
		if ( currentVD == null )
			return column == 0 ? "No View Description selected" : "";

		final ViewRegistration vr = viewRegistrations.getViewRegistration( currentVD );
		
		if ( vr.getTransformList().isEmpty() )
		{
			return column == 0 ? "No transformations present" : "";
		}
		else
		{
			if ( column == 0 )
				return vr.getTransformList().get( row ).getName();
			else
				return vr.getTransformList().get( row ).asAffine3D().getRowPackedCopy()[ column - 1 ];
		}
	}

	@Override
	public String getColumnName( final int column )
	{
		return columnNames.get( column );
	}
}