org.atemsource.atem.utility.snapshot.SnapshotBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of atem-utility Show documentation
Atem utilities for cloning, comparing and transforming
The newest version!
/*******************************************************************************
 * Stefan Meyer, 2012 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 org.atemsource.atem.utility.snapshot;

import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.atomic.AtomicLong;

import javax.inject.Inject;

import org.atemsource.atem.api.BeanLocator;
import org.atemsource.atem.api.EntityTypeRepository;
import org.atemsource.atem.api.attribute.Attribute;
import org.atemsource.atem.api.type.EntityType;
import org.atemsource.atem.api.view.View;
import org.atemsource.atem.api.view.ViewVisitor;
import org.atemsource.atem.api.view.Visitor;
import org.atemsource.atem.spi.DynamicEntityTypeSubrepository;
import org.atemsource.atem.utility.path.AttributePath;
import org.atemsource.atem.utility.path.AttributePathBuilderFactory;
import org.atemsource.atem.utility.transform.api.Transformation;
import org.atemsource.atem.utility.transform.api.TransformationBuilderFactory;
import org.atemsource.atem.utility.transform.api.TypeTransformationBuilder;
import org.atemsource.atem.utility.transform.impl.builder.OneToOneAttributeTransformationBuilder;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;


@Component
@Scope("prototype")
public class SnapshotBuilder implements ViewVisitor
{

	private static final AtomicLong typeId = new AtomicLong(1);

	protected AttributePathBuilderFactory attributePathBuilderFactory;

	@Inject
	private BeanLocator beanLocator;

	private DynamicEntityTypeSubrepository dynamicEntityTypeRepository;

	private EntityType entityType;

	@Inject
	private EntityTypeRepository entityTypeRepository;

	private SnapshotBuilderFactory factory;

	private final Set subBuilders = new HashSet();

	private TypeTransformationBuilder transformationBuilder;

	private TransformationBuilderFactory transformationBuilderFactory;

	public Transformation create()
	{
		for (SnapshotAttributeBuilder subBuilder : subBuilders)
		{
			subBuilder.preCreate();
		}
		return transformationBuilder.buildTypeTransformation();
	}

	public AttributePathBuilderFactory getAttributePathBuilderFactory()
	{
		return attributePathBuilderFactory;
	}

	public DynamicEntityTypeSubrepository getDynamicEntityTypeRepository()
	{
		return dynamicEntityTypeRepository;
	}

	private EntityType getEntityType()
	{
		return entityType;
	}

	public SnapshotBuilderFactory getFactory()
	{
		return factory;
	}

	public TransformationBuilderFactory getTransformationBuilderFactory()
	{
		return transformationBuilderFactory;
	}

	public SnapshotAttributeBuilder include(Attribute attribute)
	{
		OneToOneAttributeTransformationBuilder attributeTransformationBuilder =
			transformationBuilder.transform(attribute.getClass()).from(attribute.getCode()).to(attribute.getCode());
		SnapshotAttributeBuilder attributeBuilder = beanLocator.getInstance(SnapshotAttributeBuilder.class);
		attributeBuilder.setSnapshotBuilderFactory(this.factory);
		attributeBuilder.initialize(attributeTransformationBuilder, attribute.getTargetType());
		subBuilders.add(attributeBuilder);
		return attributeBuilder;
	}

	public SnapshotAttributeBuilder include(String path)
	{
		AttributePath attributePath = attributePathBuilderFactory.createAttributePath(path, entityType);
		String targetAttributePath = path.replace('.', '_');
		OneToOneAttributeTransformationBuilder attributeTransformationBuilder =
			transformationBuilder.transform(attributePath.getAttribute().getClass()).from(path).to(targetAttributePath);
		SnapshotAttributeBuilder attributeBuilder = beanLocator.getInstance(SnapshotAttributeBuilder.class);
		attributeBuilder.initialize(attributeTransformationBuilder, attributePath.getAttribute().getTargetType());
		subBuilders.add(attributeBuilder);
		return attributeBuilder;
	}

	public void include(View view)
	{
		view.visit(this, this);
	}

	public void initialize()
	{
		transformationBuilder = (TypeTransformationBuilder) transformationBuilderFactory.create();
		transformationBuilder.setSourceType(entityType);
		String typeCode = "snapshot::" + entityType.getCode();
		if (entityTypeRepository.getEntityType(typeCode) != null)
		{
			typeCode += "-" + typeId.getAndIncrement();
		}
		transformationBuilder.setTargetTypeBuilder(dynamicEntityTypeRepository.createBuilder(typeCode));
	}

	public void setAttributePathBuilderFactory(AttributePathBuilderFactory attributePathBuilderFactory)
	{
		this.attributePathBuilderFactory = attributePathBuilderFactory;
	}

	public void setDynamicEntityTypeRepository(DynamicEntityTypeSubrepository dynamicEntityTypeRepository)
	{
		this.dynamicEntityTypeRepository = dynamicEntityTypeRepository;
	}

	public void setEntityType(EntityType entityType)
	{
		this.entityType = entityType;
	}

	public void setFactory(SnapshotBuilderFactory factory)
	{
		this.factory = factory;
	}

	public void setTransformationBuilderFactory(TransformationBuilderFactory transformationBuilderFactory)
	{
		this.transformationBuilderFactory = transformationBuilderFactory;
	}

	@Override
	public void visit(SnapshotBuilder context, Attribute attribute)
	{
		context.include(context.getEntityType().getAttribute(attribute.getCode()));
	}

	@Override
	public void visit(SnapshotBuilder context, Attribute attribute, Visitor attributeVisitor)
	{

		SnapshotBuilder snapshotBuilder =
			context.include(context.getEntityType().getAttribute(attribute.getCode())).cascade();
		attributeVisitor.visit(snapshotBuilder);

	}

	@Override
	public boolean visitSubView(SnapshotBuilder context, View view)
	{
		return true;
	}

	@Override
	public boolean visitSuperView(SnapshotBuilder context, View view)
	{
		return true;
	}
}