com.github.ojil.algorithm.RgbAffineWarp Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ojil-core Show documentation
Open Java Imaging Library.
There is a newer version: 0.0.11
/*
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 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 Lesser General Public License for more details.
 *  
 *  You should have received a copy of the Lesser GNU General Public License
 *  along with this program.  If not, see .
 */

package com.github.ojil.algorithm;

import com.github.ojil.core.Error;
import com.github.ojil.core.Gray8OffsetImage;
import com.github.ojil.core.Image;
import com.github.ojil.core.PipelineStage;
import com.github.ojil.core.RgbImage;
import com.github.ojil.core.RgbOffsetImage;
import com.github.ojil.core.Vec2;

/**
 * Copyright 2008 by Jon A. Webb
 * @author webb
 */
public class RgbAffineWarp extends PipelineStage {
    private Gray8AffineWarp grayWarp;
    private RgbSelectGray selectRed, selectGreen, selectBlue;
    
    public RgbAffineWarp(Integer[][] warp) throws Error {
        this.grayWarp = new Gray8AffineWarp(warp);
        this.selectRed = new RgbSelectGray(RgbSelectGray.RED);
        this.selectGreen = new RgbSelectGray(RgbSelectGray.GREEN);
        this.selectBlue = new RgbSelectGray(RgbSelectGray.BLUE);
    }
    
    public void push(Image imageInput) throws Error {
        if (!(imageInput instanceof RgbImage)) {
            throw new Error(
    				Error.PACKAGE.ALGORITHM,
    				ErrorCodes.IMAGE_NOT_RGBIMAGE,
    				imageInput.toString(),
    				null,
    				null);
        }
        this.selectRed.push(imageInput);
        this.grayWarp.push(this.selectRed.getFront());
        Gray8OffsetImage warpedRed = (Gray8OffsetImage) this.grayWarp.getFront();
        this.selectGreen.push(imageInput);
        this.grayWarp.push(this.selectGreen.getFront());
        Gray8OffsetImage warpedGreen = (Gray8OffsetImage) this.grayWarp.getFront();
        this.selectBlue.push(imageInput);
        this.grayWarp.push(this.selectBlue.getFront());
        Gray8OffsetImage warpedBlue = (Gray8OffsetImage) this.grayWarp.getFront();
        RgbImage rgb = Gray3Bands2Rgb.push(warpedRed, warpedGreen, warpedBlue);
        super.setOutput(new RgbOffsetImage(
                rgb, 
                warpedRed.getXOffset(), 
                warpedRed.getYOffset()));
    }

    public void setWarp(Integer[][] warp) throws Error {
        this.grayWarp.setWarp(warp);
    }
    
    public Vec2 warpVec(Vec2 v) {
        return this.grayWarp.warpVec(v);
    }
}