core/org.eclipse.statet.rj.data/src/org/eclipse/statet/rj/data/RDataJConverter.java

/*=============================================================================#
 # Copyright (c) 2011, 2021 Stephan Wahlbrink and others.
 # 
 # This program and the accompanying materials are made available under the
 # terms of the Eclipse Public License 2.0 which is available at
 # https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 # which is available at https://www.apache.org/licenses/LICENSE-2.0.
 # 
 # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 # 
 # Contributors:
 #     Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
 #=============================================================================*/

package org.eclipse.statet.rj.data;

import java.util.Map;
import java.util.Map.Entry;



/**
 * Converts R data objects to classic Java objects and the other way around.
 */
public class RDataJConverter {
	
	
	private boolean keepArray1;
	
	private RObjectFactory rObjectFactory;
	
	
	public void setKeepArray1(final boolean enable) {
		this.keepArray1= enable;
	}
	
	public void setRObjectFactory(final RObjectFactory factory) {
		this.rObjectFactory= factory;
	}
	
	public RObjectFactory getRObjectFactory() {
		return this.rObjectFactory;
	}
	
	
	public Object toJava(final RObject rObject) {
		switch (rObject.getRObjectType()) {
		case RObject.TYPE_VECTOR:
			return toJava(rObject.getData());
		}
		return null;
	}
	
	public Object toJava(final RStore<?> rData) {
		final Object[] array= rData.toArray();
		if (!this.keepArray1 && array != null && array.length == 1) {
			return array[0];
		}
		return array;
	}
	
	
	
	public RObject toRJ(final Object javaObj) {
		if (javaObj instanceof Boolean) {
			return this.rObjectFactory.createVector(
					this.rObjectFactory.createLogiData(new boolean[] {
							((Boolean) javaObj).booleanValue() }));
		}
		if (javaObj instanceof String) {
			return this.rObjectFactory.createVector(
					this.rObjectFactory.createCharData(new String[] {
							(String) javaObj }));
		}
		if (javaObj instanceof String[]) {
			return this.rObjectFactory.createVector(
					this.rObjectFactory.createCharData((String[]) javaObj) );
		}
		if (javaObj instanceof Map) {
			final Map<?, ?> map= (Map<?, ?>) javaObj;
			final String[] names= new String[map.size()];
			final RObject[] components= new RObject[map.size()];
			int i= 0;
			for (final Entry<?, ?> entry : map.entrySet()) {
				names[i]= (String) entry.getKey();
				components[i]= toRJ(entry.getValue());
				i++;
			}
			return this.rObjectFactory.createList(components, names);
		}
		return null;
	}
	
}