com.scudata.expression.mfn.file.FileGroupReset 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.file;

import java.io.File;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.FileGroup;
import com.scudata.dm.FileObject;
import com.scudata.dm.cursor.ICursor;
import com.scudata.expression.FileGroupFunction;
import com.scudata.expression.IParam;
import com.scudata.resources.EngineMessage;

/**
 * ????????????ݻ??߸?????????ݵ???һ???????
 * f.reset(f??:b;cs) f.reset(f??:b,x;cs)
 * @author RunQian
 *
 */
public class FileGroupReset extends FileGroupFunction {
	public Object calculate(Context ctx) {
		if (param == null) {
			return fg.resetGroupTable(option, null, null, ctx);
		}

		Object obj = null; // ???ļ????????ļ???????????ļ???
		String distribute = null; // ?ֲ?????ʽ
		Integer blockSize = null;
		
		ICursor cs = null;
		IParam param = this.param;
		if (param != null && param.getType() == IParam.Semicolon && param.getSubSize() == 2) {
			IParam csParam = param.getSub(1);
			if (csParam != null) {
				Object csObj = csParam.getLeafExpression().calculate(ctx);
				if (csObj instanceof ICursor) {
					cs = (ICursor) csObj;
				} else {
					MessageManager mm = EngineMessage.get();
					throw new RQException("reset" + mm.getMessage("function.paramTypeError"));
				}
			}
			param = param.getSub(0);
		}
		
		if (param == null) {
			
		} else if (param.isLeaf()) {
			obj = param.getLeafExpression().calculate(ctx);
		} else if (param.getType() == IParam.Colon) {
			//f:b
			IParam sub0 = param.getSub(0);
			if (sub0 != null) {
				obj = sub0.getLeafExpression().calculate(ctx);
			}
			IParam blockSizeParam = param.getSub(1);
			if (blockSizeParam != null) {
				String b = blockSizeParam.getLeafExpression().calculate(ctx).toString();
				try {
					blockSize = Integer.parseInt(b);
				} catch (NumberFormatException e) {
				}
			}
		} else if (param.getType() == IParam.Comma) {
			//x
			IParam expParam = param.getSub(1);
			if (expParam != null) {
				distribute = expParam.getLeafExpression().toString();
			}
			
			param = param.getSub(0);
			if (param.isLeaf()) {
				obj = param.getLeafExpression().calculate(ctx);
			} else if (param.getType() == IParam.Colon) {
				//f:b
				IParam sub0 = param.getSub(0);
				if (sub0 != null) {
					obj = sub0.getLeafExpression().calculate(ctx);
				}
				IParam blockSizeParam = param.getSub(1);
				if (blockSizeParam != null) {
					String b = blockSizeParam.getLeafExpression().calculate(ctx).toString();
					try {
						blockSize = Integer.parseInt(b);
					} catch (NumberFormatException e) {
					}
				}
			}
		}
		
		if (obj == null) {
			//return fg.resetGroupTable(option, blockSize, cs, ctx);
			MessageManager mm = EngineMessage.get();
			throw new RQException("reset" + mm.getMessage("function.paramTypeError"));
		} else if (obj instanceof FileObject) {
			File newFile = ((FileObject)obj).getLocalFile().file();
			return fg.resetGroupTable(newFile, option, blockSize, cs, ctx);
		} else if (obj instanceof String) {
			FileObject fileObject = new FileObject((String)obj);
			File newFile = fileObject.getLocalFile().file();
			return fg.resetGroupTable(newFile, option, blockSize, cs, ctx);
		} else if (obj instanceof FileGroup) {
			return fg.resetGroupTable((FileGroup)obj, option, distribute, blockSize, cs, ctx);
		} else {
			MessageManager mm = EngineMessage.get();
			throw new RQException("reset" + mm.getMessage("function.paramTypeError"));
		}
	}
}