libcmo21/CodeGen/BMapBindings/PythonWriter.java

import java.io.OutputStreamWriter;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Vector;
import java.util.stream.Collectors;

public class PythonWriter {

	private static final Map<String, String> g_CppTypeMap = cppTypeMapCreator();

	private static Map<String, String> cppTypeMapCreator() {
		Map<String, String> cache = new HashMap<String, String>();
		cache.put("CKSTRING", "CKSTRING");
		cache.put("CKDWORD", "CKDWORD");
		cache.put("CKWORD", "CKWORD");
		cache.put("CKINT", "CKINT");
		cache.put("bool", "bool");
		cache.put("CKFLOAT", "CKFLOAT");
		cache.put("CKBYTE", "CKBYTE");
		cache.put("CK_ID", "CKID");
		cache.put("NakedOutputCallback", "callback");
		cache.put("BMFile", "void");
		cache.put("BMMeshTransition", "void");
		cache.put("VxVector3", "VxVector3");
		cache.put("VxVector2", "VxVector2");
		cache.put("VxColor", "VxColor");
		cache.put("VxMatrix", "VxMatrix");
		cache.put("CK_TEXTURE_SAVEOPTIONS", "enum");
		cache.put("VX_PIXELFORMAT", "enum");
		cache.put("VXTEXTURE_BLENDMODE", "enum");
		cache.put("VXTEXTURE_FILTERMODE", "enum");
		cache.put("VXTEXTURE_ADDRESSMODE", "enum");
		cache.put("VXBLEND_MODE", "enum");
		cache.put("VXFILL_MODE", "enum");
		cache.put("VXSHADE_MODE", "enum");
		cache.put("VXCMPFUNC", "enum");
		cache.put("VXMESH_LITMODE", "enum");
		return Collections.unmodifiableMap(cache);
	}

	private static String pythonTypeGetter(ExpFctParamDecl paramdecl) throws IllegalArgumentException {
		VariableType vt = paramdecl.mVarType;
		if (!paramdecl.mIsInput) {
			vt = vt.getPointerOfThis();
		}

		// create string builder for build final type string
		StringBuilder sb = new StringBuilder();
		// add type prefix
		sb.append("bm_");
		// try getting cpp type from base type
		String cpp_type = g_CppTypeMap.get(vt.getBaseType());
		if (cpp_type == null) {
			throw new IllegalArgumentException("Unexpected type: " + vt.toCType());
		}
		// assign cpp type
		sb.append(cpp_type);
		// add pointer suffix
		if (vt.isPointer()) {
			sb.append("_");
			sb.append(String.join("", Collections.nCopies(vt.getPointerLevel(), "p")));
		}
		// return built type string.
		return sb.toString();
	}

	public static void writePythonCode(Vector<ExpFctDecl> data) throws Exception {
		OutputStreamWriter writer = CommonHelper.openWriter("dest/BMExports.py");
		IndentHelper helper = new IndentHelper(writer);

		// write function decls
		for (ExpFctDecl fctdecl : data) {
			// write annotation
			// function name
			helper.printf("## %s", fctdecl.mFctName);
			// param
			for (ExpFctParamDecl paramdecl : fctdecl.mFctParams) {
				helper.printf("#  @param %s[%s] Type: %s. %s%s", paramdecl.mVarName,
						CommonHelper.getDoxygenInOutStr(paramdecl.mIsInput), paramdecl.mVarType.toCType(),
						(paramdecl.mIsInput ? "" : "Use ctypes.byref(data) pass it. "), paramdecl.mVarDesc);
			}
			// return val
			helper.puts("#  @return True if no error, otherwise False.");

			// write real declaration
			// first, we need join all param
			helper.printf("%s = _create_bmap_func('%s', [%s])", fctdecl.mFctName, fctdecl.mFctName, fctdecl.mFctParams
					.stream().map(value -> pythonTypeGetter(value)).collect(Collectors.joining(", ")));
		}

		writer.close();
	}

}