net.raumzeitfalle.registration.ReferencesMatrix Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of image-registration Show documentation

A collection of functions for image registration based on control points using rigid and affine transforms.

There is a newer version: 0.0.7

/*-
 * #%L
 * Image-Registration
 * %%
 * Copyright (C) 2019, 2021 Oliver Loeffler, Raumzeitfalle.net
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */
package net.raumzeitfalle.registration;

import net.raumzeitfalle.registration.alignment.RigidModelEquation;
import net.raumzeitfalle.registration.distortions.AffineModelEquation;
import net.raumzeitfalle.registration.solver.References;

public final class ReferencesMatrix implements References {
	
	private final double[][] matrix;
	
	public ReferencesMatrix(int rows, int cols) {
		this.matrix = new double[rows][cols];
	}

	public final double[][] getArray() {
		return this.matrix;
	}

	public final void set(int row, int column, double value) {
		this.matrix[row][column] = value;
	}
	
	public final void set(int row, RigidModelEquation eq, Orientation direction) {
		if (Orientations.X.equals(direction)) {
			set(row, 0, eq.getXf());
			set(row, 1, eq.getDesignValue());
		}
		
		if (Orientations.Y.equals(direction)) {
			set(row, 0, eq.getYf());
			set(row, 1, eq.getDesignValue());
		}
		
		if (Orientations.XY.equals(direction)) {
			set(row, 0, eq.getXf());
			set(row, 1, eq.getYf());
			set(row, 2, eq.getDesignValue());
		}
	}
	
	public final void set2D(int row, AffineModelEquation eq, Orientation direction) {
		if (Orientations.X.equals(direction)) {
			set(row, 0, eq.getSx());
			set(row, 1, eq.getOy());
			set(row, 2, eq.getTx());
		}
		
		if (Orientations.Y.equals(direction)) {
			set(row, 0, eq.getSy());
			set(row, 1, eq.getOx());
			set(row, 2, eq.getTy());
		}
		
		if (Orientations.XY.equals(direction)) {
			set(row, 0, eq.getSx());
			set(row, 1, eq.getSy());
			set(row, 2, eq.getOx());
			set(row, 3, eq.getOy());
			set(row, 4, eq.getTx());
			set(row, 5, eq.getTy());
		}
	}
	
	public final void set1D(int row, AffineModelEquation eq, Orientation direction) {
		if (eq.matches(Orientations.X)) {
			set(row, 0, eq.getSx());
			set(row, 1, eq.getOy());
			set(row, 2, eq.getTx());
			set(row, 3, eq.getTy());
		}
		
		if (eq.matches(Orientations.Y)) {
			set(row, 0, eq.getSy());
			set(row, 1, eq.getOx());
			set(row, 2, eq.getTx());
			set(row, 3, eq.getTy());
		}
	}

}