CMreturn Configure(bool shrink, bool flat, char *expStr) {
size_t i;
DBInt recID, layerID;
DBFloat floatVal;
DBObject *obj;
DBObjTableField *fieldPTR;
Operand = new DBMathOperand(expStr);
if (Operand->Expand(Variables) == DBFault) return (CMfailed);
CellSize.X = CellSize.Y = DBHugeVal;
for (recID = 0; recID < Variables->ItemNum(); ++recID) {
obj = Variables->Item(recID);
if ((fieldPTR = Table->Field(obj->Name())) != (DBObjTableField *) NULL) continue;
if ((GrdVar = (CMDgrdVariable **) realloc(GrdVar, sizeof(CMDgrdVariable *) * (VarNum + 1))) ==
(CMDgrdVariable **) NULL) {
CMmsgPrint(CMmsgSysError, "Memory Allocation Error in: %s %d", __FILE__, __LINE__);
return (CMfailed);
}
GrdVar[VarNum] = new CMDgrdVariable(obj->Name());
if (GrdVar[VarNum]->Configure(Table, flat) == DBFault) {
for (i = 0; i <= VarNum; ++i) delete GrdVar[i];
free(GrdVar);
return (CMfailed);
}
Extent.Expand(GrdVar[VarNum]->Extent());
floatVal = GrdVar[VarNum]->CellWidth();
if (CellSize.X > floatVal) CellSize.X = floatVal;
floatVal = GrdVar[VarNum]->CellHeight();
if (CellSize.Y > floatVal) CellSize.Y = floatVal;
layerID = (GrdVar[VarNum])->LayerNum();
if (LayerNum < layerID) {
LayerNum = layerID;
MasterVar = VarNum;
}
VarNum++;
}
for (i = 0; i < ExpNum; ++i) if (Expressions[i]->Configure(Table) == DBFault) return (CMfailed);
Operand->Configure(Table->Fields());
if (shrink) for (i = 0; i < VarNum; ++i) Extent.Shrink(GrdVar[i]->Extent());
return (CMsucceeded);
}
void ComputeTask(DBObjRecord *record, size_t taskId) {
size_t i;
DBPosition pos;
DBCoordinate coord;
pos.Row = taskId / GridIF->ColNum();
pos.Col = taskId % GridIF->ColNum();
GridIF->Pos2Coord(pos, coord);
for (i = 0; i < VarNum; ++i) GrdVar[i]->GetVariable(record, coord);
for (i = 0; i < ExpNum; ++i) Expressions[i]->Evaluate(record);
GridIF->Value(LayerRec, pos, Operand->Float(record));
}
int main(int argc, char *argv[]) {
int argPos, argNum = argc, ret, verbose = false;
DBInt fromSelection = false, selectMode = true, recID;
char *tableName = (char *) NULL;
char *expr = (char *) NULL;
DBMathOperand *operand;
DBObjectLIST<DBObject> *variables = new DBObjectLIST<DBObject>("Variables");
DBObjData *data;
DBObjTable *table;
DBObjRecord *record;
for (argPos = 1; argPos < argNum;) {
if (CMargTest (argv[argPos], "-a", "--table")) {
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) {
CMmsgPrint(CMmsgUsrError, "Missing table name!");
return (CMfailed);
}
tableName = argv[argPos];
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv[argPos], "-c", "--condition")) {
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) {
CMmsgPrint(CMmsgUsrError, "Missing expression!");
return (CMfailed);
}
if (expr != (char *) NULL) {
CMmsgPrint(CMmsgUsrError, "Expression is already set");
return (CMfailed);
}
expr = argv[argPos];
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv[argPos], "-f", "--from")) {
int modeCodes[] = {false, true};
const char *modes[] = {"all", "selection", (char *) NULL};
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) {
CMmsgPrint(CMmsgUsrError, "Missing selection mode!");
return (CMfailed);
}
if ((fromSelection = CMoptLookup(modes, argv[argPos], true)) == DBFault) {
CMmsgPrint(CMmsgUsrError, "Invalid selection mode!");
return (CMfailed);
}
fromSelection = modeCodes[fromSelection];
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv[argPos], "-m", "--mode")) {
int modeCodes[] = {true, false};
const char *modes[] = {"select", "unselect", (char *) NULL};
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) {
CMmsgPrint(CMmsgUsrError, "Missing selection mode!");
return (CMfailed);
}
if ((selectMode = CMoptLookup(modes, argv[argPos], true)) == DBFault) {
CMmsgPrint(CMmsgUsrError, "Invalid selection mode!");
return (CMfailed);
}
selectMode = modeCodes[selectMode];
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv[argPos], "-V", "--verbose")) {
verbose = true;
if ((argNum = CMargShiftLeft(argPos, argv, argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv[argPos], "-h", "--help")) {
CMmsgPrint(CMmsgInfo, "%s [options] <input file> <output file>", CMfileName(argv[0]));
CMmsgPrint(CMmsgInfo, " -a,--table [ table name ]");
CMmsgPrint(CMmsgInfo, " -c,--condition [ fieldname expression ]");
CMmsgPrint(CMmsgInfo, " -f,--from [ [all] | selection ]");
CMmsgPrint(CMmsgInfo, " -m,--mode [ [select] | unselect ]");
CMmsgPrint(CMmsgInfo, " -V,--verbose");
CMmsgPrint(CMmsgInfo, " -h,--help");
return (DBSuccess);
}
if ((argv[argPos][0] == '-') && ((int) strlen(argv[argPos]) > 1)) {
CMmsgPrint(CMmsgUsrError, "Unknown option: %s!", argv[argPos]);
return (CMfailed);
}
argPos++;
}
if (argNum > 3) {
CMmsgPrint(CMmsgUsrError, "Extra arguments!");
return (CMfailed);
}
if (verbose) RGlibPauseOpen(argv[0]);
if (expr != (char *) NULL) {
operand = new DBMathOperand(expr);
if (operand->Expand(variables) == DBFault) return (CMfailed);
}
data = new DBObjData();
if (((argNum > 1) && (strcmp(argv[1], "-") != 0) ? data->Read(argv[1]) : data->Read(stdin)) == DBFault) {
delete data;
delete operand;
return (CMfailed);
}
if (tableName == (char *) NULL) tableName = DBrNItems;
if ((table = data->Table(tableName)) == (DBObjTable *) NULL) {
CMmsgPrint(CMmsgUsrError, "Invalid table!");
delete data;
delete operand;
return (CMfailed);
}
if (expr != (char *) NULL) {
if (operand->Configure(table->Fields()) == DBFault) {
delete data;
delete operand;
return (CMfailed);
}
for (recID = 0; recID < table->ItemNum(); ++recID) {
record = table->Item(recID);
if (fromSelection && ((record->Flags() & DBObjectFlagSelected) != DBObjectFlagSelected))
continue;
if ((ret = operand->Int(record)) == true)
record->Flags(DBObjectFlagSelected, selectMode ? DBSet : DBClear);
else if (selectMode) record->Flags(DBObjectFlagSelected, DBClear);
}
}
else {
for (recID = 0; recID < table->ItemNum(); ++recID) {
record = table->Item(recID);
if (fromSelection && ((record->Flags() & DBObjectFlagSelected) != DBObjectFlagSelected))
continue;
record->Flags(DBObjectFlagSelected, selectMode ? DBSet : DBClear);
}
}
ret = (argNum > 2) && (strcmp(argv[2], "-") != 0) ? data->Write(argv[2]) : data->Write(stdout);
if (expr != (char *) NULL) delete operand;
delete data;
delete variables;
if (verbose) RGlibPauseClose();
return (ret);
}
int main (int argc,char *argv [])
{
int argPos, argNum = argc, ret, verbose = false;
DBInt fromSelection = false, recID;
char *tableName = (char *) NULL;
char *expr = (char *) NULL;
DBMathOperand *operand;
DBObjectLIST<DBObject> *variables = new DBObjectLIST<DBObject> ("Variables");
DBObjData *data;
DBObjTable *table, *saveTable, *groups = (DBObjTable *) NULL, *saveGroups;
DBObjRecord *record;
for (argPos = 1;argPos < argNum; )
{
if (CMargTest (argv [argPos],"-a","--table"))
{
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos)
{ CMmsgPrint (CMmsgUsrError,"Missing table name!"); return (CMfailed); }
tableName = argv [argPos];
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv [argPos],"-c","--condition"))
{
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos)
{ CMmsgPrint (CMmsgUsrError,"Missing expression!"); return (CMfailed); }
if (expr != (char *) NULL) { CMmsgPrint (CMmsgUsrError,"Expression is already set"); return (CMfailed); }
expr = argv [argPos];
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv [argPos],"-s","--selection"))
{
fromSelection = true;
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv [argPos],"-V","--verbose"))
{
verbose = true;
if ((argNum = CMargShiftLeft (argPos,argv,argNum)) <= argPos) break;
continue;
}
if (CMargTest (argv [argPos],"-h","--help"))
{
CMmsgPrint (CMmsgInfo,"%s [options] <input file> <output file>",CMfileName(argv[0]));
CMmsgPrint (CMmsgInfo," -a,--table [table name]");
CMmsgPrint (CMmsgInfo," -c,--condition [expression]");
CMmsgPrint (CMmsgInfo," -s,--selection");
CMmsgPrint (CMmsgInfo," -V,--verbose");
CMmsgPrint (CMmsgInfo," -h,--help");
return (DBSuccess);
}
if ((argv [argPos][0] == '-') && ((int) strlen (argv [argPos]) > 1))
{ CMmsgPrint (CMmsgUsrError,"Unknown option: %s!",argv [argPos]); return (CMfailed); }
argPos++;
}
if (argNum > 3) { CMmsgPrint (CMmsgUsrError,"Extra arguments!"); return (CMfailed); }
if (verbose) RGlibPauseOpen (argv[0]);
if (expr != (char *) NULL)
{
operand = new DBMathOperand (expr);
if (operand->Expand (variables) == DBFault) return (CMfailed);
}
data = new DBObjData ();
if (((argNum > 1) && (strcmp (argv [1],"-") != 0) ? data->Read (argv [1]) : data->Read (stdin)) == DBFault)
{ delete data; delete operand; return (CMfailed); }
if (tableName == (char *) NULL) tableName = DBrNItems;
if ((table = data->Table (tableName)) == (DBObjTable *) NULL)
{ CMmsgPrint (CMmsgUsrError,"Invalid table!"); delete data; delete operand; return (CMfailed); }
saveTable = new DBObjTable (*table);
table->DeleteAll ();
if ((strcmp (tableName,DBrNItems) == 0) && ((groups = data->Table (DBrNGroups)) != (DBObjTable *) NULL))
{
saveGroups = new DBObjTable (*groups);
groups->DeleteAll ();
}
if (expr != (char *) NULL)
{
if (operand->Configure (saveTable->Fields()) == DBFault)
{ delete data; delete operand; return (CMfailed); }
for (recID = 0;recID < saveTable->ItemNum ();++recID)
{
record = saveTable->Item (recID);
if ((ret = operand->Int (record)) == true)
{
if (!fromSelection) continue;
if ((record->Flags () & DBObjectFlagSelected) == DBObjectFlagSelected) continue;
}
table->Add (new DBObjRecord (*record));
if (groups != (DBObjTable *) NULL) groups->Add (new DBObjRecord (*(saveGroups->Item (recID))));
}
}
else
{
if (fromSelection)
{
for (recID = 0;recID < saveTable->ItemNum ();++recID)
{
record = saveTable->Item (recID);
if ((record->Flags () & DBObjectFlagSelected) == DBObjectFlagSelected)
{
table->Add (new DBObjRecord (*record));
if (groups != (DBObjTable *) NULL)
groups->Add (new DBObjRecord (*(saveGroups->Item (recID))));
}
}
}
}
if (expr != (char *) NULL) delete operand;
delete variables;
delete saveTable;
if (groups != (DBObjTable *) NULL) delete saveGroups;
ret = (argNum > 2) && (strcmp (argv [2],"-") != 0) ? data->Write (argv [2]) : data->Write (stdout);
delete data;
if (verbose) RGlibPauseClose ();
return (ret);
}
DBObjData *Compute(char *title, CMthreadUserExecFunc userFunc) {
DBInt i, layerID, dataLayerID;
size_t threadId;
DBPosition pos;
DBCoordinate coord;
char *layerName;
DBObjData *data;
DBObjRecord *record;
size_t threadsNum = CMthreadProcessorNum();
CMthreadTeam_t team;
CMthreadJob_p job = (CMthreadJob_p) NULL;
if (CMthreadTeamInitialize(&team, threadsNum) == (CMthreadTeam_p) NULL) {
CMmsgPrint (CMmsgUsrError,"Team initialization error %s, %d",__FILE__,__LINE__);
return ((DBObjData *) NULL);
}
if ((data = DBGridCreate(title, Extent, CellSize)) == (DBObjData *) NULL) return ((DBObjData *) NULL);
data->Projection(GrdVar[0]->Projection()); // Taking projection from first grid variable
GridIF = new DBGridIF(data);
if (team.ThreadNum > 0) {
if ((job = CMthreadJobCreate(GridIF->RowNum() * GridIF->ColNum(), userFunc, (void *) this)) ==
(CMthreadJob_p) NULL) {
CMmsgPrint(CMmsgAppError, "Job creation error in %s:%d", __FILE__, __LINE__);
CMthreadTeamDestroy(&team);
return ((DBObjData *) NULL);
}
for (threadId = 0; threadId < team.ThreadNum; ++threadId)
if (Table->Add("TEMPRecord") == (DBObjRecord *) NULL) {
CMthreadTeamDestroy(&team);
return ((DBObjData *) NULL);
}
}
else {
if ((record = Table->Add("TEMPRecord")) == (DBObjRecord *) NULL) return ((DBObjData *) NULL);
}
for (layerID = 0; layerID < LayerNum; ++layerID) {
layerName = GrdVar[MasterVar]->CurrentLayer(layerID);
for (i = 0; i < (DBInt) VarNum; ++i) {
if ((dataLayerID = GrdVar[i]->FindLayer(layerName)) != DBFault)
GrdVar[i]->CurrentLayer(dataLayerID);
else {
if (GrdVar[i]->LayerIsDated(0)) continue;
GrdVar[i]->CurrentLayer(dataLayerID);
}
}
if (layerID > 0) GridIF->AddLayer((char *) "New Layer");
LayerRec = GridIF->Layer(layerID);
GridIF->RenameLayer(LayerRec, layerName);
if (job != (CMthreadJob_p) NULL) CMthreadJobExecute(&team, job);
else
for (pos.Row = 0; pos.Row < GridIF->RowNum(); ++pos.Row)
for (pos.Col = 0; pos.Col < GridIF->ColNum(); ++pos.Col) {
GridIF->Pos2Coord(pos, coord);
for (i = 0; i < (DBInt) VarNum; ++i) GrdVar[i]->GetVariable(record, coord);
for (i = 0; i < (DBInt) ExpNum; ++i) Expressions[i]->Evaluate(record);
GridIF->Value(LayerRec, pos, Operand->Float(record));
}
GridIF->RecalcStats(LayerRec);
}
delete GridIF;
CMthreadTeamDestroy(&team);
return (data);
}
