/*=============================================================================#
 # Copyright (c) 2011-2016 Stephan Wahlbrink (WalWare.de) and others.
 # All rights reserved. This program and the accompanying materials
 # are made available under the terms of either (per the licensee's choosing)
 #   - the Eclipse Public License v1.0
 #     which accompanies this distribution, and is available at
 #     http://www.eclipse.org/legal/epl-v10.html, or
 #   - the GNU Lesser General Public License v2.1 or newer
 #     which accompanies this distribution, and is available at
 #     http://www.gnu.org/licenses/lgpl.html
 # 
 # Contributors:
 #     Stephan Wahlbrink - initial API and implementation
 #=============================================================================*/

package de.walware.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;
	}
	
}