BenchmarkWaveletTransform.java example

Explorer
BoofCV-master
/*
 * Copyright (c) 2011-2016, Peter Abeles. All Rights Reserved.
 *
 * This file is part of BoofCV (http://boofcv.org).
 *
 * 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.
 */

package boofcv.alg.transform.wavelet;

import boofcv.alg.misc.ImageMiscOps;
import boofcv.alg.transform.wavelet.impl.ImplWaveletTransformNaive;
import boofcv.core.image.border.BorderType;
import boofcv.factory.transform.wavelet.FactoryWaveletDaub;
import boofcv.misc.PerformerBase;
import boofcv.misc.ProfileOperation;
import boofcv.struct.image.GrayF32;
import boofcv.struct.image.GrayS32;
import boofcv.struct.image.ImageDimension;
import boofcv.struct.wavelet.WaveletDescription;
import boofcv.struct.wavelet.WlCoef_F32;
import boofcv.struct.wavelet.WlCoef_I32;

import java.util.Random;


/**
 * @author Peter Abeles
 */
public class BenchmarkWaveletTransform {
	static int imgWidth = 640;
	static int imgHeight = 480;
	static long TEST_TIME = 1000;

	static WaveletDescription<WlCoef_F32> desc_F32 = FactoryWaveletDaub.biorthogonal_F32(5, BorderType.REFLECT);
	static WaveletDescription<WlCoef_I32> desc_I32 = FactoryWaveletDaub.biorthogonal_I32(5,BorderType.REFLECT);

	static GrayF32 orig_F32 = new GrayF32(imgWidth,imgHeight);
	static GrayF32 temp1_F32 = new GrayF32(imgWidth,imgHeight);
	static GrayF32 temp2_F32 = new GrayF32(imgWidth,imgHeight);
	static GrayS32 orig_I32 = new GrayS32(imgWidth,imgHeight);
	static GrayS32 temp1_I32 = new GrayS32(imgWidth,imgHeight);
	static GrayS32 temp2_I32 = new GrayS32(imgWidth,imgHeight);

	public static class Naive_F32 extends PerformerBase {

		@Override
		public void process() {
			ImplWaveletTransformNaive.horizontal(desc_F32.getBorder(),desc_F32.getForward(),orig_F32,temp1_F32);
			ImplWaveletTransformNaive.vertical(desc_F32.getBorder(),desc_F32.getForward(),temp1_F32,temp2_F32);
		}
	}

	public static class Standard_F32 extends PerformerBase {

		@Override
		public void process() {
			WaveletTransformOps.transform1(desc_F32,orig_F32,temp1_F32,temp1_F32);
		}
	}

	public static class Naive_I32 extends PerformerBase {

		@Override
		public void process() {
			ImplWaveletTransformNaive.horizontal(desc_I32.getBorder(),desc_I32.getForward(),orig_I32,temp1_I32);
			ImplWaveletTransformNaive.vertical(desc_I32.getBorder(),desc_I32.getForward(),temp1_I32,temp2_I32);
		}
	}

	public static class Standard_I32 extends PerformerBase {

		@Override
		public void process() {
			WaveletTransformOps.transform1(desc_I32,orig_I32,temp1_I32,temp1_I32);
		}
	}

	public static class FullLevel3_F32 extends PerformerBase {

		static GrayF32 copy = new GrayF32(imgWidth,imgHeight);
		GrayF32 tran;
		GrayF32 storage;

		public FullLevel3_F32() {
			ImageDimension dim = UtilWavelet.transformDimension(copy,3);
			tran = new GrayF32(dim.width,dim.height);
			storage = new GrayF32(dim.width,dim.height);
		}

		@Override
		public void process() {
			// don't modify the input image
			copy.setTo(orig_F32);
			WaveletTransformOps.transformN(desc_F32,copy,tran,storage,3);
		}
	}


	public static void main(String args[]) {

		Random rand = new Random(234);
		ImageMiscOps.fillUniform(orig_F32, rand, 0, 100);
		ImageMiscOps.fillUniform(orig_I32, rand, 0, 100);

		System.out.println("=========  Profile Image Size " + imgWidth + " x " + imgHeight + " ==========");
		System.out.println();

		ProfileOperation.printOpsPerSec(new FullLevel3_F32(), TEST_TIME);
		ProfileOperation.printOpsPerSec(new Naive_F32(), TEST_TIME);
		ProfileOperation.printOpsPerSec(new Standard_F32(), TEST_TIME);
		ProfileOperation.printOpsPerSec(new Naive_I32(), TEST_TIME);
		ProfileOperation.printOpsPerSec(new Standard_I32(), TEST_TIME);
	}
}