nom.bdezonia.zorbage.algorithm.CreateMask Maven / Gradle / Ivy

Go to download
/*
 * Zorbage: an algebraic data hierarchy for use in numeric processing.
 *
 * Copyright (c) 2016-2021 Barry DeZonia All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 * 
 * Redistributions of source code must retain the above copyright notice, this list
 * of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright notice, this
 * list of conditions and the following disclaimer in the documentation and/or other
 * materials provided with the distribution.
 * 
 * Neither the name of the  nor the names of its contributors may
 * be used to endorse or promote products derived from this software without specific
 * prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL  BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
 * DAMAGE.
 */
package nom.bdezonia.zorbage.algorithm;

import nom.bdezonia.zorbage.storage.Storage;
import nom.bdezonia.zorbage.type.integer.int1.UnsignedInt1Member;
import nom.bdezonia.zorbage.algebra.Algebra;
import nom.bdezonia.zorbage.algebra.G;
import nom.bdezonia.zorbage.data.DimensionedDataSource;
import nom.bdezonia.zorbage.data.NdData;
import nom.bdezonia.zorbage.datasource.IndexedDataSource;
import nom.bdezonia.zorbage.function.Function1;

/**
 * 
 * @author Barry DeZonia
 *
 */
public class CreateMask {

	// do not instantiate
	
	private CreateMask() { }
	
	/**
	 * Make and return a one dimensional mask by applying a boolean mapping function
	 * across a list of values.
	 * 
	 * @param 
	 * @param 
	 * @param alg
	 * @param condition A function that maps values to booleans
	 * @param a The input list
	 * @return
	 */
	public static , U>
		IndexedDataSource compute(T alg, Function1 condition, IndexedDataSource a)
	{
		U value = alg.construct();
		UnsignedInt1Member zero = G.UINT1.construct();
		UnsignedInt1Member one = G.UINT1.construct();
		G.UINT1.unity().call(one);
		IndexedDataSource result = Storage.allocate(zero, a.size());
		for (long i = 0; i < a.size(); i++) {
			a.get(i, value);
			if (condition.call(value))
				result.set(i, one);
			else
				result.set(i, zero);
		}
		return result;
	}

	/**
	 * 
	 * @param 
	 * @param 
	 * @param alg
	 * @param condition
	 * @param a
	 * @return
	 */
	public static , U>
		DimensionedDataSource compute(T alg, Function1 condition, DimensionedDataSource a)
	{
		long[] dims = new long[a.numDimensions()];
		for (int i = 0; i < dims.length; i++) {
			dims[i] = a.dimension(i);
		}
		IndexedDataSource data = compute(alg, condition, a.rawData());
		return new NdData<>(dims, data);
	}
}