com.scudata.expression.mfn.sequence.Export Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
There is a newer version: 20241126
package com.scudata.expression.mfn.sequence;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.BaseRecord;
import com.scudata.dm.ComputeStack;
import com.scudata.dm.Context;
import com.scudata.dm.Current;
import com.scudata.dm.DataStruct;
import com.scudata.dm.FileObject;
import com.scudata.dm.Sequence;
import com.scudata.expression.Expression;
import com.scudata.expression.IParam;
import com.scudata.expression.ParamInfo2;
import com.scudata.expression.SequenceFunction;
import com.scudata.resources.EngineMessage;
import com.scudata.util.Variant;

/**
 * ?????и?ʽ?????ַ???
 * A.export(x:F,??;s)
 * @author RunQian
 *
 */
public class Export extends SequenceFunction {
	public Object calculate(Context ctx) {
		Expression []exps = null;
		String []names = null;
		String s = null;

		IParam param = this.param;
		if (param != null && param.getType() == IParam.Semicolon) { // ;
			if (param.getSubSize() != 2) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("export" + mm.getMessage("function.invalidParam"));
			}

			IParam param1 = param.getSub(1);
			if (param1 == null) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("export" + mm.getMessage("function.invalidParam"));
			}

			Object obj = param1.getLeafExpression().calculate(ctx);
			if (!(obj instanceof String)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("export" + mm.getMessage("function.paramTypeError"));
			}

			s = (String)obj;
			param = param.getSub(0);
		}

		if (param != null) {
			ParamInfo2 pi2 = ParamInfo2.parse(param, "export", true, false);
			exps = pi2.getExpressions1();
			names = pi2.getExpressionStrs2();
		}

		return export(srcSequence, exps, names, s, option, ctx);
	}

	/**
	 * ?????б???ַ???
	 * @param sequence ????
	 * @param exps ?ֶα???ʽ???ɿ?
	 * @param names ?????ֶ??????ɿ?
	 * @param s ?зָ???
	 * @param option ѡ??
	 * @param ctx
	 * @return
	 */
	public static String export(Sequence sequence, Expression []exps, String []names, 
				String s, String option, Context ctx) {
		boolean isTitle = false, isCsv = false, isQuote = false;
		String LINE_SEPARATOR = new String(FileObject.LINE_SEPARATOR);
		char escapeChar = '\\';
		if (option != null) {
			if (option.indexOf('t') != -1) isTitle = true;
			if (option.indexOf('c') != -1) isCsv = true;
			if (option.indexOf('w') != -1) {
				LINE_SEPARATOR = new String(FileObject.DM_LINE_SEPARATOR);
			}
			
			if (option.indexOf('q') != -1) isQuote = true;
			if (option.indexOf('o') != -1) escapeChar = '"';
		}
				
		StringBuffer sb = new StringBuffer(1024 * 1024);
		if (s == null || s.length() == 0) {
			if (isCsv) {
				s = ",";
			} else {
				s = "\t";
			}
		}

		if (exps == null) {
			int fcount = 1;
			DataStruct ds = sequence.dataStruct();
			if (ds == null) {
				fcount = FileObject.getMaxMemberCount(sequence);
				if (isTitle && fcount < 1) {
					if (isQuote) {
						writeLine(sb, new String[]{FileObject.S_FIELDNAME}, s, LINE_SEPARATOR, escapeChar);
					} else {
						writeLine(sb, new String[]{FileObject.S_FIELDNAME}, s, LINE_SEPARATOR);
					}
				}
			} else {
				fcount = ds.getFieldCount();
				if (isTitle) {
					if (isQuote) {
						writeLine(sb, ds.getFieldNames(), s, LINE_SEPARATOR, escapeChar);
					} else {
						writeLine(sb, ds.getFieldNames(), s, LINE_SEPARATOR);
					}
				}
			}

			if (ds == null) {
				if (fcount < 1) {
					Object []lineObjs = new Object[1];
					for (int i = 1, len = sequence.length(); i <= len; ++i) {
						lineObjs[0] = sequence.getMem(i);
						if (isQuote) {
							writeLine(sb, lineObjs, s, LINE_SEPARATOR, escapeChar);
						} else {
							writeLine(sb, lineObjs, s, LINE_SEPARATOR);
						}
					}
				} else {
					// A?????е?????ʱ???????ޱ???Ķ????ı?
					Object []lineObjs = new Object[fcount];
					for (int i = 1, len = sequence.length(); i <= len; ++i) {
						Sequence seq = (Sequence)sequence.getMem(i);
						if (seq == null) {
							for (int f = 0; f < fcount; ++f) {
								lineObjs[f] = null;
							}
						} else {
							seq.toArray(lineObjs);
							for (int f = seq.length(); f < fcount; ++f) {
								lineObjs[f] = null;
							}
						}
						
						if (isQuote) {
							writeLine(sb, lineObjs, s, LINE_SEPARATOR, escapeChar);
						} else {
							writeLine(sb, lineObjs, s, LINE_SEPARATOR);
						}
					}
				}
			} else {
				for (int i = 1, len = sequence.length(); i <= len; ++i) {
					BaseRecord r = (BaseRecord)sequence.getMem(i);
					Object []vals = r.getFieldValues();

					if (isQuote) {
						writeLine(sb, vals, s, LINE_SEPARATOR, escapeChar);
					} else {
						writeLine(sb, vals, s, LINE_SEPARATOR);
					}
				}
			}
		} else {
			ComputeStack stack = ctx.getComputeStack();
			Current current = new Current(sequence);
			stack.push(current);

			try {
				int fcount = exps.length;
				if (isTitle) {
					if (names == null) names = new String[fcount];
					sequence.getNewFieldNames(exps, names, "export");
					if (isQuote) {
						writeLine(sb, names, s, LINE_SEPARATOR, escapeChar);
					} else {
						writeLine(sb, names, s, LINE_SEPARATOR);
					}
				}

				Object []lineObjs = new Object[fcount];
				for (int i = 1, len = sequence.length(); i <= len; ++i) {
					current.setCurrent(i);
					for (int f = 0; f < fcount; ++f) {
						lineObjs[f] = exps[f].calculate(ctx);
					}

					if (isQuote) {
						writeLine(sb, lineObjs, s, LINE_SEPARATOR, escapeChar);
					} else {
						writeLine(sb, lineObjs, s, LINE_SEPARATOR);
					}
				}
			} finally {
				stack.pop();
			}
		}

		return sb.toString();
	}

	private static void writeLine(StringBuffer sb, Object []items, String s, String ls) {
		String str = Variant.toExportString(items[0]);
		if (str != null) {
			sb.append(str);
		}

		for (int i = 1, len = items.length; i < len; ++i) {
			sb.append(s);
			str = Variant.toExportString(items[i]);
			if (str != null) {
				sb.append(str);
			}
		}

		sb.append(ls);
	}

	private static void writeLine(StringBuffer sb, Object []items, String s, String ls, char escapeChar) {
		String str = Variant.toExportString(items[0], escapeChar);
		if (str != null) {
			sb.append(str);
		}

		for (int i = 1, len = items.length; i < len; ++i) {
			sb.append(s);
			str = Variant.toExportString(items[i], escapeChar);
			if (str != null) {
				sb.append(str);
			}
		}

		sb.append(ls);
	}
}