com.github.TKnudsen.ComplexDataObject.model.processors.complexDataObject.CollectionToBooleanAttributesConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of complex-data-object Show documentation

A library that models real-world objects in Java, referred to as ComplexDataObjects. Other features: IO and preprocessing of ComplexDataObjects.

There is a newer version: 0.2.13

Show newest version

package com.github.TKnudsen.ComplexDataObject.model.processors.complexDataObject;

import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import com.github.TKnudsen.ComplexDataObject.data.complexDataObject.ComplexDataContainer;
import com.github.TKnudsen.ComplexDataObject.data.complexDataObject.ComplexDataObject;
import com.github.TKnudsen.ComplexDataObject.data.enums.FuzzyBooleanCategory;

/**
 * 
 * Title: CollectionToBooleanAttributesConverter
 * 
 * 
 * 
 * Description:
 * 
 * 
 * 
 * Copyright: Copyright (c) 2016
 * 
 * 
 * @author Juergen Bernard
 * @version 1.01
 */
public class CollectionToBooleanAttributesConverter implements IComplexDataObjectProcessor {

	private String attributeString;

	public CollectionToBooleanAttributesConverter(String attributeString) {
		this.attributeString = attributeString;
	}

	@Override
	public void process(ComplexDataContainer container) {

		// 1) gather all observations
		Set observations = new HashSet<>();
		Iterator iterator = container.iterator();
		while (iterator.hasNext()) {
			ComplexDataObject next = iterator.next();
			Object object = next.getAttribute(attributeString);
			if (object instanceof Collection) {
				Collection collection = (Collection) object;
				for (Object element : collection)
					if (element instanceof String)
						observations.add((String) element);
			}
		}

		// 2) create new FuzzyBoolenCategory attributes
		for (String string : observations)
			container.addAttribute(attributeString + "_" + string, FuzzyBooleanCategory.class, FuzzyBooleanCategory.NO_INFORMATION);

		// 3) add entries for the objects
		iterator = container.iterator();
		while (iterator.hasNext()) {
			ComplexDataObject next = iterator.next();
			Object object = next.getAttribute(attributeString);
			Collection collection = null;
			if (object != null && object instanceof Collection)
				collection = (Collection) object;

			for (String string : observations) {
				if (collection == null)
					next.add(attributeString + "_" + string, FuzzyBooleanCategory.NO_INFORMATION);
				else if (collection != null && collection.contains(string))
					next.add(attributeString + "_" + string, FuzzyBooleanCategory.YES);
				else
					next.add(attributeString + "_" + string, FuzzyBooleanCategory.NO);
			}
		}
	}

	@Override
	public void process(List data) {
		ComplexDataContainer container = new ComplexDataContainer(data);
		process(container);
	}

	public String getAttributeString() {
		return attributeString;
	}

	public void setAttributeString(String attributeString) {
		this.attributeString = attributeString;
	}

	@Override
	public DataProcessingCategory getPreprocessingCategory() {
		return DataProcessingCategory.SECONDARY_DATA_PROVIDER;
	}
}