org.solovyev.common.math.AbstractMathRegistry Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-other Show documentation
The newest version!
/*
 * Copyright (c) 2009-2011. Created by serso aka se.solovyev.
 * For more information, please, contact [email protected]
 * or visit http://se.solovyev.org
 */

package org.solovyev.common.math;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.solovyev.common.JPredicate;
import org.solovyev.common.collections.CollectionsUtils;
import org.solovyev.common.collections.SortedList;
import org.solovyev.common.JBuilder;

import java.util.*;

/**
 * User: serso
 * Date: 9/29/11
 * Time: 4:57 PM
 */
public abstract class AbstractMathRegistry implements MathRegistry {

	private static final MathEntityComparator MATH_ENTITY_COMPARATOR = new MathEntityComparator();

	static class MathEntityComparator implements Comparator {

		MathEntityComparator() {
		}

		@Override
		public int compare(T l, T r) {
			int result = r.getName().length() - l.getName().length();
			if (result == 0) {
				result = l.getName().compareTo(r.getName());
			}
			return result;
		}
	}

	@NotNull
	private static Integer counter = 0;

	@NotNull
	protected final List entities = new SortedList(new ArrayList(30), MATH_ENTITY_COMPARATOR);

	@NotNull
	protected final List systemEntities = new SortedList(new ArrayList(30), MATH_ENTITY_COMPARATOR);

	protected AbstractMathRegistry() {
	}

	@NotNull
	@Override
	public List getEntities() {
		return Collections.unmodifiableList(new ArrayList(entities));
	}

	@NotNull
	@Override
	public List getSystemEntities() {
		return Collections.unmodifiableList(new ArrayList(systemEntities));
	}

	protected void add(@NotNull T entity) {
		if (entity.isSystem()) {
			if (contains(entity.getName(), this.systemEntities)) {
				throw new IllegalArgumentException("Trying to add two system entities with same name: " + entity.getName());
			}

			this.systemEntities.add(entity);
		}

		if (!contains(entity.getName(), this.entities)) {
			addEntity(entity, this.entities);
		}
	}

	private void addEntity(@NotNull T entity, @NotNull List list) {
		entity.setId(count());
		list.add(entity);
	}

	@Override
	public T add(@NotNull JBuilder JBuilder) {
		final T entity = JBuilder.create();

		T varFromRegister;

		if (entity.isIdDefined()) {
			varFromRegister = getById(entity.getId());
		} else {
			varFromRegister = get(entity.getName());
		}

		if (varFromRegister == null) {
			varFromRegister = entity;

			addEntity(entity, this.entities);
			if (entity.isSystem()) {
				this.systemEntities.add(entity);
			}

		} else {
			varFromRegister.copy(entity);
			Collections.sort(this.entities, MATH_ENTITY_COMPARATOR);
			Collections.sort(this.systemEntities, MATH_ENTITY_COMPARATOR);
		}

		return varFromRegister;
	}

	@Override
	public void remove(@NotNull T entity) {
		if (!entity.isSystem()) {
			CollectionsUtils.removeFirst(this.entities, new MathEntity.Finder(entity.getName()));
		}
	}

	@Override
	@NotNull
	public List getNames() {
		final List result = new ArrayList();

		for (T entity : entities) {
			result.add(entity.getName());
		}

		return result;
	}

	@Override
	@Nullable
	public T get(@NotNull final String name) {
		return CollectionsUtils.find(entities, new MathEntity.Finder(name));
	}

	@Override
	public T getById(@NotNull final Integer id) {
		return CollectionsUtils.find(entities, new JPredicate() {
			@Override
			public boolean apply(@Nullable T t) {
				return t != null && t.getId().equals(id);
			}
		});
	}

	@Override
	public boolean contains(@NotNull final String name) {
		return contains(name, this.entities);
	}

	private boolean contains(final String name, @NotNull Collection entities) {
		return CollectionsUtils.find(entities, new MathEntity.Finder(name)) != null;
	}

	@NotNull
	private static synchronized Integer count() {
		final Integer result = counter;
		counter++;
		return result;
	}
}