hqlcset.cpp 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924
  1. /*##############################################################################
  2. Copyright (C) 2011 HPCC Systems.
  3. All rights reserved. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU Affero General Public License as
  5. published by the Free Software Foundation, either version 3 of the
  6. License, or (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU Affero General Public License for more details.
  11. You should have received a copy of the GNU Affero General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ############################################################################## */
  14. #include "jliball.hpp"
  15. #include "hql.hpp"
  16. #include "platform.h"
  17. #include "jlib.hpp"
  18. #include "jexcept.hpp"
  19. #include "jmisc.hpp"
  20. #include "javahash.hpp"
  21. #include "eclhelper.hpp"
  22. #include "hqlfunc.hpp"
  23. #include "hqlattr.hpp"
  24. #include "hqlhtcpp.ipp"
  25. #include "hqlwcpp.hpp"
  26. #include "hqlcpputil.hpp"
  27. #include "hqlcerrors.hpp"
  28. #include "hqlcatom.hpp"
  29. #include "hqlpmap.hpp"
  30. #include "hqlthql.hpp"
  31. #include "hqlcset.ipp"
  32. #include "hqlfold.hpp"
  33. #include "hqltcppc.ipp"
  34. #include "hqlutil.hpp"
  35. #include "hqliter.ipp"
  36. #ifdef CREATE_DEAULT_ROW_IF_NULL
  37. #define CREATE_DEAULT_ROW_IF_NULL_VALUE 1
  38. #else
  39. #define CREATE_DEAULT_ROW_IF_NULL_VALUE 0
  40. #endif
  41. //===========================================================================
  42. IHqlExpression * getOutOfRangeValue(IHqlExpression * indexExpr)
  43. {
  44. IHqlExpression * dft = indexExpr->queryProperty(defaultAtom);
  45. if (dft)
  46. return LINK(dft->queryChild(0));
  47. else
  48. return createNullExpr(indexExpr);
  49. }
  50. //===========================================================================
  51. BaseDatasetCursor::BaseDatasetCursor(HqlCppTranslator & _translator, IHqlExpression * _ds, CHqlBoundExpr * _boundDs) : translator(_translator)
  52. {
  53. ds.set(_ds);
  54. record.set(ds->queryRecord());
  55. if (_boundDs)
  56. boundDs.set(*_boundDs);
  57. }
  58. BoundRow * BaseDatasetCursor::buildIterateLoop(BuildCtx & ctx, bool needToBreak)
  59. {
  60. StringBuffer iterName;
  61. buildIterateClass(ctx, iterName, NULL);
  62. StringBuffer s, rowName;
  63. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  64. //row = iter.first()
  65. s.clear().append(rowName).append(" = ").append(iterName).append(".first();");
  66. ctx.addQuoted(s);
  67. //while (row)
  68. ctx.addLoop(row, NULL, false);
  69. BoundRow * cursor = translator.bindTableCursor(ctx, ds, row);
  70. // row = iter.next();
  71. ctx.setNextDestructor();
  72. s.clear().append(rowName).append(" = (byte *)").append(iterName).append(".next();");
  73. ctx.addQuoted(s);
  74. return cursor;
  75. }
  76. void BaseDatasetCursor::buildIterateClass(BuildCtx & ctx, SharedHqlExpr & iter, SharedHqlExpr & row)
  77. {
  78. StringBuffer cursorName, rowName;
  79. buildIterateClass(ctx, cursorName, NULL);
  80. iter.setown(createVariable(cursorName.str(), makeBoolType()));
  81. row.setown(createRow(ctx, "row", rowName));
  82. }
  83. void BaseDatasetCursor::buildIterateMembers(BuildCtx & declarectx, BuildCtx & initctx)
  84. {
  85. StringBuffer iterName;
  86. buildIterateClass(declarectx, iterName, &initctx);
  87. StringBuffer s, rowName;
  88. OwnedHqlExpr row = createRow(declarectx, "row", rowName);
  89. //row = iter.first()
  90. BuildCtx firstctx(declarectx);
  91. firstctx.addQuotedCompound("virtual bool first()");
  92. s.clear().append(rowName).append(" = (byte *)").append(iterName).append(".first();");
  93. firstctx.addQuoted(s);
  94. s.clear().append("return ").append(rowName).append(" != NULL;");
  95. firstctx.addQuoted(s);
  96. //row = iter.first()
  97. BuildCtx nextctx(declarectx);
  98. nextctx.addQuotedCompound("virtual bool next()");
  99. s.clear().append(rowName).append(" = (byte *)").append(iterName).append(".next();");
  100. nextctx.addQuoted(s);
  101. s.clear().append("return ").append(rowName).append(" != NULL;");
  102. nextctx.addQuoted(s);
  103. //iterate
  104. translator.bindTableCursor(declarectx, ds, row);
  105. }
  106. BoundRow * BaseDatasetCursor::buildSelect(BuildCtx & ctx, IHqlExpression * indexExpr)
  107. {
  108. //MORE: Check if the cursor already exists....
  109. StringBuffer cursorName;
  110. buildIterateClass(ctx, cursorName, NULL);
  111. //create a unique dataset and associate it with a call to select
  112. //set value to be the field selection from the dataset
  113. StringBuffer s, rowName;
  114. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  115. CHqlBoundExpr boundIndex;
  116. OwnedHqlExpr index = adjustIndexBaseToZero(indexExpr->queryChild(1));
  117. translator.buildExpr(ctx, index, boundIndex);
  118. //MORE: CREATE_DEAULT_ROW_IF_NULL - pass the default row to the select() function.
  119. //row = iter.select(n)
  120. s.clear().append(rowName).append(" = (byte *)").append(cursorName).append(".select(");
  121. translator.generateExprCpp(s, boundIndex.expr);
  122. s.append(");");
  123. ctx.addQuoted(s);
  124. bool conditional = !indexExpr->hasProperty(noBoundCheckAtom);
  125. #ifdef CREATE_DEAULT_ROW_IF_NULL
  126. if (conditional)
  127. {
  128. CHqlBoundExpr boundCleared;
  129. translator.buildDefaultRow(ctx, ds, boundCleared);
  130. OwnedHqlExpr defaultRowPtr = getPointer(boundCleared.expr);
  131. BuildCtx subctx(ctx);
  132. OwnedHqlExpr test = createValue(no_not, makeBoolType(), LINK(row));
  133. subctx.addFilter(test);
  134. subctx.addAssign(row, defaultRowPtr);
  135. conditional = false;
  136. }
  137. #endif
  138. BoundRow * cursor = translator.bindRow(ctx, indexExpr, row);
  139. cursor->setConditional(conditional);
  140. return cursor;
  141. }
  142. IHqlExpression * BaseDatasetCursor::createRow(BuildCtx & ctx, const char * prefix, StringBuffer & rowName)
  143. {
  144. translator.getUniqueId(rowName.append(prefix));
  145. OwnedITypeInfo type;
  146. if (boundDs.expr && boundDs.expr->queryRecord())
  147. type.setown(makeConstantModifier(makeRowReferenceType(boundDs)));
  148. else
  149. type.setown(makeConstantModifier(makeRowReferenceType(ds)));
  150. OwnedHqlExpr row = createVariable(rowName, type.getClear());
  151. ctx.addDeclare(row);
  152. return row.getClear();
  153. }
  154. //---------------------------------------------------------------------------
  155. BlockDatasetCursor::BlockDatasetCursor(HqlCppTranslator & _translator, IHqlExpression * _ds, CHqlBoundExpr & _boundDs) : BaseDatasetCursor(_translator, _ds, &_boundDs)
  156. {
  157. boundDs.set(_boundDs);
  158. assertex(boundDs.expr->isDatarow() || !isArrayRowset(boundDs.expr->queryType())); // I don't think this can ever be called at the moment
  159. }
  160. void BlockDatasetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  161. {
  162. tgt.expr.setown(translator.getBoundCount(boundDs));
  163. }
  164. void BlockDatasetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  165. {
  166. if (boundDs.count)
  167. tgt.expr.setown(createValue(no_ne, makeBoolType(), LINK(boundDs.count), getZero()));
  168. else
  169. tgt.expr.setown(createValue(no_ne, makeBoolType(), LINK(boundDs.length), getZero()));
  170. }
  171. void BlockDatasetCursor::buildIterateClass(BuildCtx & ctx, StringBuffer & cursorName, BuildCtx * initctx)
  172. {
  173. translator.getUniqueId(cursorName.append("iter"));
  174. StringBuffer extraParams;
  175. StringBuffer decl,args;
  176. if (translator.isFixedRecordSize(record))
  177. {
  178. //RtlFixedDatasetCursor cursor(len, data, size)
  179. decl.append("RtlFixedDatasetCursor");
  180. extraParams.append(", ").append(translator.getFixedRecordSize(record));
  181. }
  182. else
  183. {
  184. //RtlVariableDatasetCursor cursor(len, data, recordSize)
  185. decl.append("RtlVariableDatasetCursor");
  186. translator.buildMetaForRecord(extraParams.append(", "), record);
  187. }
  188. OwnedHqlExpr size = translator.getBoundSize(boundDs);
  189. decl.append(" ").append(cursorName);
  190. translator.generateExprCpp(args, size);
  191. args.append(", ");
  192. translator.generateExprCpp(args, boundDs.expr);
  193. args.append(extraParams);
  194. if (initctx)
  195. {
  196. StringBuffer s;
  197. s.append(cursorName).append(".init(").append(args).append(");");
  198. initctx->addQuoted(s);
  199. }
  200. else
  201. {
  202. decl.append("(").append(args).append(")");
  203. }
  204. decl.append(";");
  205. ctx.addQuoted(decl);
  206. }
  207. //---------------------------------------------------------------------------
  208. bool isEmptyDataset(const CHqlBoundExpr & bound)
  209. {
  210. IValue * value = NULL;
  211. if (bound.length)
  212. value = bound.length->queryValue();
  213. else if (bound.count)
  214. value = bound.count->queryValue();
  215. return (value && value->getIntValue() == 0);
  216. }
  217. InlineBlockDatasetCursor::InlineBlockDatasetCursor(HqlCppTranslator & _translator, IHqlExpression * _ds, CHqlBoundExpr & _boundDs) : BlockDatasetCursor(_translator, _ds, _boundDs)
  218. {
  219. }
  220. BoundRow * InlineBlockDatasetCursor::buildIterateLoop(BuildCtx & ctx, bool needToBreak)
  221. {
  222. StringBuffer rowName;
  223. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  224. if (isEmptyDataset(boundDs))
  225. {
  226. ctx.addFilter(queryBoolExpr(false));
  227. return translator.bindTableCursor(ctx, ds, row);
  228. }
  229. StringBuffer s;
  230. //row = ds;
  231. OwnedHqlExpr address = getPointer(boundDs.expr);
  232. OwnedHqlExpr cast = createValue(no_implicitcast, row->getType(), LINK(address));
  233. ctx.addAssign(row, cast);
  234. OwnedHqlExpr test;
  235. if (boundDs.length)
  236. {
  237. OwnedHqlExpr length = translator.getBoundLength(boundDs);
  238. StringBuffer endName;
  239. OwnedHqlExpr end = createRow(ctx, "end", endName);
  240. //end = row+length;
  241. s.clear().append(endName).append(" = ").append(rowName).append("+");
  242. translator.generateExprCpp(s, length).append(";");
  243. ctx.addQuoted(s);
  244. //while (row < end)
  245. test.setown(createValue(no_lt, makeBoolType(), LINK(row), LINK(end)));
  246. }
  247. else if (matchesConstantValue(boundDs.count, 1) && !needToBreak)
  248. {
  249. //Optimize count=1, needToBreak = false;
  250. ctx.addGroup();
  251. return translator.bindTableCursor(ctx, ds, row);
  252. }
  253. else
  254. {
  255. OwnedHqlExpr count = translator.getBoundCount(boundDs);
  256. //count = <n>
  257. OwnedHqlExpr counter = ctx.getTempDeclare(unsignedType, count);
  258. //while (count--)
  259. test.setown(createValue(no_postdec, LINK(counter)));
  260. }
  261. ctx.addLoop(test, NULL, false);
  262. BoundRow * cursor = translator.bindTableCursor(ctx, ds, row);
  263. //row = row + recordSize
  264. OwnedHqlExpr size = translator.getRecordSize(cursor->querySelector());
  265. CHqlBoundExpr boundSize;
  266. translator.buildExpr(ctx, size, boundSize);
  267. ctx.setNextDestructor();
  268. if (translator.queryOptions().optimizeIncrement)
  269. {
  270. ctx.addAssignIncrement(row, boundSize.expr);
  271. }
  272. else
  273. {
  274. OwnedHqlExpr inc = createValue(no_add, row->getType(), LINK(row), LINK(boundSize.expr));
  275. ctx.addAssign(row, inc);
  276. }
  277. return cursor;
  278. }
  279. BoundRow * InlineBlockDatasetCursor::buildSelectFirst(BuildCtx & ctx, IHqlExpression * indexExpr, bool createDefaultRowIfNull)
  280. {
  281. StringBuffer s, rowName;
  282. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  283. BuildCtx subctx(ctx);
  284. bool conditional = !indexExpr->hasProperty(noBoundCheckAtom);
  285. if (conditional)
  286. {
  287. HqlExprAttr test;
  288. if (boundDs.count)
  289. {
  290. IValue * countValue = boundDs.count->queryValue();
  291. if (countValue)
  292. {
  293. if (countValue->getIntValue() == 0)
  294. return NULL;
  295. }
  296. else
  297. {
  298. OwnedHqlExpr max = createTranslated(boundDs.count);
  299. test.setown(createCompare(no_ne, max, queryZero()));
  300. }
  301. }
  302. else
  303. {
  304. OwnedHqlExpr max = createTranslated(boundDs.length);
  305. test.setown(createCompare(no_gt, max, queryZero()));
  306. }
  307. if (test)
  308. {
  309. CHqlBoundExpr boundCleared;
  310. if (createDefaultRowIfNull)
  311. {
  312. translator.buildDefaultRow(ctx, ds, boundCleared);
  313. conditional = false;
  314. }
  315. else
  316. translator.buildNullRow(ctx, ds, boundCleared);
  317. OwnedHqlExpr defaultRowPtr = getPointer(boundCleared.expr);
  318. ctx.addAssign(row, defaultRowPtr);
  319. translator.buildFilter(subctx, test);
  320. }
  321. else
  322. conditional = false;
  323. }
  324. if (isArrayRowset(boundDs.expr->queryType()))
  325. {
  326. s.clear().append(rowName).append(" = ");
  327. translator.generateExprCpp(s, boundDs.expr).append("[0];");
  328. subctx.addQuoted(s);
  329. }
  330. else
  331. {
  332. OwnedHqlExpr address = getPointer(boundDs.expr);
  333. s.clear().append(rowName).append(" = (byte *)(void *)"); // more: should really be const...
  334. translator.generateExprCpp(s, address);
  335. s.append(";");
  336. subctx.addQuoted(s);
  337. }
  338. BoundRow * cursor = translator.bindRow(ctx, indexExpr, row);
  339. cursor->setConditional(conditional);
  340. return cursor;
  341. }
  342. BoundRow * InlineBlockDatasetCursor::buildSelect(BuildCtx & ctx, IHqlExpression * indexExpr)
  343. {
  344. assertex(!isArrayRowset(boundDs.expr->queryType())); // I don't think this can ever be called at the moment
  345. OwnedHqlExpr index = foldHqlExpression(indexExpr->queryChild(1));
  346. if (!translator.isFixedRecordSize(record))
  347. {
  348. if (matchesConstantValue(index, 1))
  349. return buildSelectFirst(ctx, indexExpr, CREATE_DEAULT_ROW_IF_NULL_VALUE);
  350. return BlockDatasetCursor::buildSelect(ctx, indexExpr);
  351. }
  352. if (matchesConstantValue(index, 1))
  353. return buildSelectFirst(ctx, indexExpr, CREATE_DEAULT_ROW_IF_NULL_VALUE);
  354. //row = NULL
  355. StringBuffer s, rowName;
  356. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  357. //if (index > 0 && (index <= count) or (index * fixedSize <= size)
  358. //MORE: Need to be very careful about the types...
  359. OwnedHqlExpr base0Index;
  360. unsigned fixedSize = translator.getFixedRecordSize(record);
  361. BuildCtx subctx(ctx);
  362. bool conditional = !indexExpr->hasProperty(noBoundCheckAtom);
  363. if (conditional)
  364. {
  365. OwnedHqlExpr simpleIndex = translator.buildSimplifyExpr(ctx, index);
  366. base0Index.setown(adjustIndexBaseToZero(simpleIndex));
  367. IValue * indexValue = index->queryValue();
  368. OwnedHqlExpr test;
  369. if (indexValue)
  370. {
  371. if (indexValue->getIntValue() <= 0)
  372. return NULL;
  373. }
  374. else
  375. test.setown(createCompare(no_gt, simpleIndex, queryZero()));
  376. IHqlExpression * test2 = NULL;
  377. if (boundDs.count)
  378. {
  379. IValue * countValue = boundDs.count->queryValue();
  380. if (countValue && indexValue)
  381. {
  382. if (indexValue->getIntValue() > countValue->getIntValue())
  383. return NULL;
  384. }
  385. else
  386. {
  387. OwnedHqlExpr max = createTranslated(boundDs.count);
  388. test2 = createCompare(no_le, simpleIndex, max);
  389. }
  390. }
  391. else
  392. {
  393. OwnedHqlExpr max = createTranslated(boundDs.length);
  394. OwnedHqlExpr offset = multiplyValue(simpleIndex, fixedSize);
  395. test2 = createCompare(no_le, offset, max);
  396. }
  397. extendConditionOwn(test, no_and, test2);
  398. if (test)
  399. {
  400. CHqlBoundExpr boundCleared;
  401. #ifdef CREATE_DEAULT_ROW_IF_NULL
  402. translator.buildDefaultRow(ctx, ds, boundCleared);
  403. conditional = false;
  404. #else
  405. translator.buildNullRow(ctx, ds, boundCleared);
  406. #endif
  407. OwnedHqlExpr defaultRowPtr = getPointer(boundCleared.expr);
  408. ctx.addAssign(row, defaultRowPtr);
  409. translator.buildFilter(subctx, test);
  410. }
  411. else
  412. conditional = false;
  413. }
  414. else
  415. {
  416. CHqlBoundExpr boundIndex;
  417. OwnedHqlExpr base0 = adjustIndexBaseToZero(index);
  418. translator.buildExpr(ctx, base0, boundIndex);
  419. base0Index.setown(boundIndex.getTranslatedExpr());
  420. }
  421. //row = base + index * fixedSize;
  422. OwnedHqlExpr address = LINK(boundDs.expr);//getPointer(boundDs.expr);
  423. s.clear().append(rowName).append(" = (byte *)(void *)"); // more: should really be const...
  424. translator.generateExprCpp(s, address);
  425. CHqlBoundExpr boundOffset;
  426. OwnedHqlExpr offset = multiplyValue(base0Index, fixedSize);
  427. translator.buildExpr(subctx, offset, boundOffset);
  428. s.append(" + (");
  429. translator.generateExprCpp(s, boundOffset.expr).append(")");
  430. s.append(";");
  431. subctx.addQuoted(s);
  432. BoundRow * cursor = translator.bindRow(ctx, indexExpr, row);
  433. cursor->setConditional(conditional);
  434. return cursor;
  435. }
  436. //---------------------------------------------------------------------------
  437. InlineLinkedDatasetCursor::InlineLinkedDatasetCursor(HqlCppTranslator & _translator, IHqlExpression * _ds, CHqlBoundExpr & _boundDs) : BaseDatasetCursor(_translator, _ds, &_boundDs)
  438. {
  439. assertex(boundDs.count != NULL);
  440. assertex(isArrayRowset(boundDs.expr->queryType()));
  441. }
  442. void InlineLinkedDatasetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  443. {
  444. tgt.expr.set(boundDs.count);
  445. }
  446. void InlineLinkedDatasetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  447. {
  448. tgt.expr.setown(createValue(no_ne, makeBoolType(), LINK(boundDs.count), getZero()));
  449. }
  450. void InlineLinkedDatasetCursor::buildIterateClass(BuildCtx & ctx, StringBuffer & cursorName, BuildCtx * initctx)
  451. {
  452. translator.getUniqueId(cursorName.append("iter"));
  453. //RtlFixedDatasetCursor cursor(len, data, size)
  454. StringBuffer decl;
  455. decl.append("RtlLinkedDatasetCursor ").append(cursorName);
  456. StringBuffer args;
  457. translator.generateExprCpp(args, boundDs.count);
  458. args.append(", ");
  459. translator.generateExprCpp(args, boundDs.expr);
  460. if (initctx)
  461. {
  462. StringBuffer s;
  463. s.append(cursorName).append(".init(").append(args).append(");");
  464. initctx->addQuoted(s);
  465. }
  466. else
  467. {
  468. decl.append("(").append(args).append(")");
  469. }
  470. decl.append(";");
  471. ctx.addQuoted(decl);
  472. }
  473. BoundRow * InlineLinkedDatasetCursor::buildIterateLoop(BuildCtx & ctx, bool needToBreak)
  474. {
  475. StringBuffer rowName;
  476. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  477. if (isEmptyDataset(boundDs))
  478. {
  479. ctx.addFilter(queryBoolExpr(false));
  480. return translator.bindTableCursor(ctx, ds, row);
  481. }
  482. if (matchesConstantValue(boundDs.count, 1) && !needToBreak)
  483. {
  484. CHqlBoundExpr boundRow;
  485. boundRow.set(boundDs);
  486. translator.convertBoundDatasetToFirstRow(ds, boundRow);
  487. //Optimize count=1, needToBreak = false;
  488. ctx.addGroup();
  489. return translator.bindTableCursor(ctx, ds, boundRow.expr);
  490. }
  491. StringBuffer cursorName, s;
  492. translator.getUniqueId(cursorName.append("cur"));
  493. //row = ds;
  494. OwnedHqlExpr address = getPointer(boundDs.expr); // ensure no longer a wrapped item
  495. s.clear().append("byte * * ").append(cursorName).append(" = ");
  496. translator.generateExprCpp(s, address).append(";");
  497. ctx.addQuoted(s);
  498. OwnedHqlExpr test;
  499. OwnedHqlExpr count = translator.getBoundCount(boundDs);
  500. //count = <n>
  501. OwnedHqlExpr counter = ctx.getTempDeclare(unsignedType, count);
  502. //while (count--)
  503. test.setown(createValue(no_postdec, LINK(counter)));
  504. ctx.addLoop(test, NULL, false);
  505. ctx.addQuoted(s.clear().append(rowName).append(" = *").append(cursorName).append("++;"));
  506. BoundRow * cursor = translator.bindTableCursor(ctx, ds, row);
  507. return cursor;
  508. }
  509. BoundRow * InlineLinkedDatasetCursor::buildSelect(BuildCtx & ctx, IHqlExpression * indexExpr)
  510. {
  511. OwnedHqlExpr index = foldHqlExpression(indexExpr->queryChild(1));
  512. //row = NULL
  513. StringBuffer s, rowName;
  514. OwnedHqlExpr row = createRow(ctx, "row", rowName);
  515. //if (index > 0 && (index <= count)
  516. //MORE: Need to be very careful about the types...
  517. CHqlBoundExpr boundBase0Index;
  518. BuildCtx subctx(ctx);
  519. bool conditional = !indexExpr->hasProperty(noBoundCheckAtom);
  520. if (conditional)
  521. {
  522. IValue * indexValue = index->queryValue();
  523. if (indexValue)
  524. {
  525. if (indexValue->getIntValue() <= 0)
  526. return NULL;
  527. if (indexValue->getIntValue() > (size32_t)-1)
  528. return NULL;
  529. if (indexValue->queryType()->getSize() > sizeof(size32_t))
  530. index.setown(ensureExprType(index, sizetType));
  531. }
  532. OwnedHqlExpr simpleIndex = translator.buildSimplifyExpr(ctx, index);
  533. OwnedHqlExpr base0Index = adjustIndexBaseToZero(simpleIndex);
  534. translator.buildExpr(ctx, base0Index, boundBase0Index);
  535. OwnedHqlExpr test;
  536. if (!indexValue)
  537. test.setown(createCompare(no_gt, simpleIndex, queryZero()));
  538. IHqlExpression * test2 = NULL;
  539. IValue * countValue = boundDs.count->queryValue();
  540. if (countValue && indexValue)
  541. {
  542. if (indexValue->getIntValue() > countValue->getIntValue())
  543. return NULL;
  544. }
  545. else
  546. {
  547. OwnedHqlExpr max = createTranslated(boundDs.count);
  548. test2 = createCompare(no_le, simpleIndex, max);
  549. }
  550. extendConditionOwn(test, no_and, test2);
  551. if (test)
  552. {
  553. CHqlBoundExpr boundCleared;
  554. #ifdef CREATE_DEAULT_ROW_IF_NULL
  555. translator.buildDefaultRow(ctx, ds, boundCleared);
  556. conditional = false;
  557. #else
  558. translator.buildNullRow(ctx, ds, boundCleared);
  559. #endif
  560. OwnedHqlExpr defaultRowPtr = getPointer(boundCleared.expr);
  561. ctx.addAssign(row, defaultRowPtr);
  562. translator.buildFilter(subctx, test);
  563. }
  564. else
  565. conditional = false;
  566. }
  567. else
  568. {
  569. OwnedHqlExpr base0 = adjustIndexBaseToZero(index);
  570. translator.buildExpr(ctx, base0, boundBase0Index);
  571. }
  572. //row = base[index]
  573. OwnedHqlExpr address = getPointer(boundDs.expr);
  574. OwnedHqlExpr indexedValue = createValue(no_index, row->getType(), LINK(address), LINK(boundBase0Index.expr));
  575. subctx.addAssign(row, indexedValue);
  576. //MORE: Should mark as linked if it is.
  577. BoundRow * cursor = translator.bindRow(ctx, indexExpr, row);
  578. cursor->setConditional(conditional);
  579. return cursor;
  580. }
  581. //---------------------------------------------------------------------------
  582. MultiLevelDatasetCursor::MultiLevelDatasetCursor(HqlCppTranslator & _translator, IHqlExpression * _ds)
  583. : BaseDatasetCursor(_translator, _ds, NULL)
  584. {
  585. }
  586. void MultiLevelDatasetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  587. {
  588. throwUnexpected();
  589. }
  590. void MultiLevelDatasetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  591. {
  592. throwUnexpected();
  593. }
  594. BoundRow * MultiLevelDatasetCursor::buildIterateLoop(BuildCtx & ctx, bool needToBreak)
  595. {
  596. OwnedHqlExpr breakVar;
  597. if (needToBreak)
  598. {
  599. CHqlBoundTarget bound;
  600. translator.createTempFor(ctx, boolType, bound, typemod_none, FormatNatural);
  601. breakVar.set(bound.expr);
  602. ctx.addAssign(breakVar, queryBoolExpr(false));
  603. }
  604. return doBuildIterateLoop(ctx, ds, breakVar, true);
  605. }
  606. BoundRow * MultiLevelDatasetCursor::buildSelect(BuildCtx & ctx, IHqlExpression * indexExpr)
  607. {
  608. //Declare row for final level, iterate the appropriate number of times, and then assign and break.
  609. BuildCtx initctx(ctx);
  610. IHqlExpression * selector = ds->queryNormalizedSelector();
  611. StringBuffer cursorName;
  612. translator.getUniqueId(cursorName.append("row"));
  613. OwnedHqlExpr rowExpr = createVariable(cursorName, makeRowReferenceType(selector));
  614. initctx.addDeclare(rowExpr);
  615. CHqlBoundExpr boundCleared;
  616. translator.buildDefaultRow(initctx, selector, boundCleared);
  617. OwnedHqlExpr defaultRowPtr = getPointer(boundCleared.expr);
  618. initctx.addAssign(rowExpr, defaultRowPtr);
  619. HqlExprAssociation * savedMarker = ctx.associateExpr(queryConditionalRowMarker(), rowExpr);
  620. CHqlBoundTarget boundCount;
  621. IHqlExpression * index = indexExpr->queryChild(1);
  622. bool selectFirst = matchesConstValue(index, 1);
  623. if (!selectFirst)
  624. {
  625. translator.createTempFor(initctx, index, boundCount);
  626. translator.buildExprAssign(initctx, boundCount, index);
  627. }
  628. BuildCtx subctx(ctx);
  629. buildIterateLoop(subctx, true);
  630. if (!selectFirst)
  631. {
  632. OwnedHqlExpr test = createValue(no_eq, makeBoolType(), createValue(no_predec, LINK(boundCount.expr)), getZero());
  633. subctx.addFilter(test);
  634. }
  635. //Now we have the correct element, assign it to the pointer.
  636. //Need to be careful that the row we are pointing at is preserved, and doesn't go out of scope. (Don't need to worry about t can't be reused).
  637. BoundRow * curIter = translator.resolveSelectorDataset(subctx, selector);
  638. OwnedHqlExpr source = getPointer(curIter->queryBound());
  639. subctx.addAssign(rowExpr, source);
  640. subctx.addBreak();
  641. //Bind the expression as a row - so that the same select expression will get commoned up (e.g. sqagg)
  642. ctx.removeAssociation(savedMarker);
  643. return translator.bindRow(ctx, indexExpr, rowExpr);
  644. }
  645. BoundRow * MultiLevelDatasetCursor::doBuildIterateLoop(BuildCtx & ctx, IHqlExpression * expr, IHqlExpression * breakVar, bool topLevel)
  646. {
  647. IHqlExpression * root = queryRoot(expr);
  648. if (root)
  649. {
  650. if (isMultiLevelDatasetSelector(root, false))
  651. doBuildIterateLoop(ctx, root->queryChild(0), breakVar, false);
  652. }
  653. BuildCtx oldctx(ctx);
  654. BoundRow * row;
  655. if (root)
  656. {
  657. OwnedHqlExpr thisLevel = replaceExpression(expr, root, root->queryNormalizedSelector());
  658. row = translator.buildDatasetIterate(ctx, thisLevel, breakVar != NULL);
  659. }
  660. else
  661. {
  662. //Unusual... Something like (no_select(no_select(somethingComplex))) Assert on topLevel to prevent recursive stack fault
  663. //(see dlingle4.xhql for an example)
  664. assertex(!topLevel);
  665. root = expr->queryChild(0);
  666. row = translator.buildDatasetIterate(ctx, expr, breakVar != NULL);
  667. }
  668. if (breakVar)
  669. {
  670. if (topLevel)
  671. {
  672. ctx.addAssign(breakVar, queryBoolExpr(true));
  673. ctx.setNextDestructor();
  674. ctx.addAssign(breakVar, queryBoolExpr(false));
  675. }
  676. if (isMultiLevelDatasetSelector(root, false))
  677. {
  678. oldctx.addFilter(breakVar);
  679. oldctx.addBreak();
  680. }
  681. }
  682. return row;
  683. }
  684. //---------------------------------------------------------------------------
  685. BaseSetCursor::BaseSetCursor(HqlCppTranslator & _translator, IHqlExpression * _expr) : translator(_translator)
  686. {
  687. expr.set(_expr);
  688. }
  689. ListSetCursor::ListSetCursor(HqlCppTranslator & _translator, IHqlExpression * _expr) : BaseSetCursor(_translator, _expr)
  690. {
  691. }
  692. void ListSetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  693. {
  694. tgt.expr.setown(getCountExpr());
  695. }
  696. void ListSetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  697. {
  698. tgt.expr.set(queryBoolExpr(expr->numChildren() != 0));
  699. }
  700. void ListSetCursor::buildIsAll(BuildCtx & ctx, CHqlBoundExpr & tgt)
  701. {
  702. tgt.expr.set(queryBoolExpr(false));
  703. }
  704. void ListSetCursor::buildIterateLoop(BuildCtx & ctx, CHqlBoundExpr & curBound, bool needToBreak)
  705. {
  706. if (expr->numChildren() == 0)
  707. {
  708. ctx.addFilter(queryBoolExpr(false));
  709. curBound.expr.setown(createNullExpr(expr->queryType()->queryChildType()));
  710. return;
  711. }
  712. if (!needToBreak && (expr->numChildren() == 1))
  713. {
  714. translator.buildExpr(ctx, expr->queryChild(0), curBound);
  715. return;
  716. }
  717. CHqlBoundExpr boundList;
  718. translator.buildSimpleExpr(ctx, expr, boundList);
  719. OwnedHqlExpr loopVar = ctx.getTempDeclare(unsignedType, NULL);
  720. OwnedHqlExpr loopTest = createValue(no_lt, makeBoolType(), LINK(loopVar), getCountExpr());
  721. OwnedHqlExpr inc = createValue(no_postinc, loopVar->getType(), LINK(loopVar));
  722. translator.buildAssignToTemp(ctx, loopVar, queryZero());
  723. ctx.addLoop(loopTest, inc, false);
  724. curBound.expr.setown(createValue(no_index, LINK(expr->queryType()->queryChildType()), LINK(boundList.expr), LINK(loopVar)));
  725. }
  726. void ListSetCursor::buildIterateClass(BuildCtx & ctx, CHqlBoundExpr & tgt)
  727. {
  728. CHqlBoundExpr boundList;
  729. translator.buildSimpleExpr(ctx, expr, boundList);
  730. UNIMPLEMENTED;
  731. ctx.addQuoted("create fixed iterate (bound.length, bound.getAddress()");
  732. }
  733. void ListSetCursor::gatherSelect(BuildCtx & ctx, IHqlExpression * indexExpr, CHqlBoundExpr & value, HqlExprAttr & cond)
  734. {
  735. if (expr->numChildren() == 0)
  736. {
  737. OwnedHqlExpr null = getOutOfRangeValue(indexExpr);
  738. translator.buildExpr(ctx, null, value);
  739. return;
  740. }
  741. IHqlExpression * index = indexExpr->queryChild(1);
  742. if (index->isConstant())
  743. {
  744. OwnedHqlExpr folded = foldHqlExpression(index);
  745. unsigned which = (unsigned)folded->queryValue()->getIntValue()-1;
  746. if (which < expr->numChildren())
  747. translator.buildExpr(ctx, expr->queryChild(which), value);
  748. else
  749. {
  750. OwnedHqlExpr null = getOutOfRangeValue(indexExpr);
  751. translator.buildExpr(ctx, null, value);
  752. }
  753. }
  754. else
  755. {
  756. CHqlBoundExpr boundList;
  757. translator.buildSimpleExpr(ctx, expr, boundList);
  758. CHqlBoundExpr boundIndex;
  759. ITypeInfo * elementType = expr->queryType()->queryChildType(); // not indexExpr->getType() because may now be more specific
  760. OwnedHqlExpr base0Index = adjustIndexBaseToZero(index);
  761. if (indexExpr->hasProperty(noBoundCheckAtom))
  762. translator.buildExpr(ctx, base0Index, boundIndex);
  763. else
  764. translator.buildSimpleExpr(ctx, base0Index, boundIndex);
  765. value.expr.setown(createValue(no_index, LINK(elementType), LINK(boundList.expr), LINK(boundIndex.expr)));
  766. if (!indexExpr->hasProperty(noBoundCheckAtom))
  767. {
  768. ITypeInfo * indexType = boundIndex.expr->queryType();
  769. //ok to subtract early and remove a check for > 0 on unsigned values because they will wrap and fail upper limit test
  770. if (indexType->isSigned())
  771. cond.setown(createBoolExpr(no_ge, LINK(boundIndex.expr), getZero()));
  772. if (indexType->getCardinality() > expr->numChildren())
  773. extendConditionOwn(cond, no_and, createBoolExpr(no_lt, LINK(boundIndex.expr), getCountExpr()));
  774. }
  775. }
  776. }
  777. void ListSetCursor::buildExprSelect(BuildCtx & ctx, IHqlExpression * indexExpr, CHqlBoundExpr & tgt)
  778. {
  779. CHqlBoundExpr value;
  780. HqlExprAttr cond;
  781. gatherSelect(ctx, indexExpr, value, cond);
  782. if (cond)
  783. {
  784. translator.buildTempExpr(ctx, indexExpr, tgt);
  785. return;
  786. CHqlBoundTarget tempTarget;
  787. translator.createTempFor(ctx, indexExpr, tempTarget);
  788. buildAssignSelect(ctx, tempTarget, indexExpr);
  789. tgt.setFromTarget(tempTarget);
  790. }
  791. else
  792. tgt.set(value);
  793. }
  794. void ListSetCursor::buildAssignSelect(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * indexExpr)
  795. {
  796. CHqlBoundExpr value;
  797. HqlExprAttr cond;
  798. gatherSelect(ctx, indexExpr, value, cond);
  799. if (cond)
  800. {
  801. BuildCtx subctx(ctx);
  802. IHqlStmt * e = subctx.addFilter(cond);
  803. translator.assign(subctx, target, value);
  804. subctx.selectElse(e);
  805. OwnedHqlExpr null = getOutOfRangeValue(indexExpr);
  806. translator.buildExprAssign(subctx, target, null);
  807. }
  808. else
  809. translator.assign(ctx, target, value);
  810. }
  811. IHqlExpression * ListSetCursor::getCountExpr()
  812. {
  813. return getSizetConstant(expr->numChildren());
  814. }
  815. //---------------------------------------------------------------------------
  816. AllSetCursor::AllSetCursor(HqlCppTranslator & _translator) : BaseSetCursor(_translator, NULL)
  817. {
  818. }
  819. void AllSetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  820. {
  821. translator.throwError(HQLERR_CountAllSet);
  822. }
  823. void AllSetCursor::buildIsAll(BuildCtx & ctx, CHqlBoundExpr & tgt)
  824. {
  825. tgt.expr.set(queryBoolExpr(true));
  826. }
  827. void AllSetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  828. {
  829. tgt.expr.set(queryBoolExpr(true));
  830. }
  831. void AllSetCursor::buildIterateLoop(BuildCtx & ctx, CHqlBoundExpr & curBound, bool needToBreak)
  832. {
  833. translator.throwError(HQLERR_IndexAllSet);
  834. }
  835. void AllSetCursor::buildIterateClass(BuildCtx & ctx, CHqlBoundExpr & tgt)
  836. {
  837. translator.throwError(HQLERR_IndexAllSet);
  838. }
  839. void AllSetCursor::buildExprSelect(BuildCtx & ctx, IHqlExpression * indexExpr, CHqlBoundExpr & tgt)
  840. {
  841. translator.throwError(HQLERR_IndexAllSet);
  842. }
  843. void AllSetCursor::buildAssignSelect(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * indexExpr)
  844. {
  845. translator.throwError(HQLERR_IndexAllSet);
  846. }
  847. //---------------------------------------------------------------------------
  848. GeneralSetCursor::GeneralSetCursor(HqlCppTranslator & _translator, IHqlExpression * _expr, CHqlBoundExpr & boundSet) : BaseSetCursor(_translator, _expr)
  849. {
  850. isAll.setown(boundSet.getIsAll());
  851. ITypeInfo * elementType = LINK(expr->queryType()->queryChildType());
  852. if (!elementType)
  853. elementType = makeStringType(UNKNOWN_LENGTH, NULL, NULL);
  854. element.setown(createField(valueAtom, elementType, NULL));
  855. HqlExprArray fields;
  856. fields.append(*LINK(element));
  857. ds.setown(createDataset(no_anon, createRecord(fields), LINK(expr)));
  858. dsCursor.setown(new InlineBlockDatasetCursor(translator, ds, boundSet));
  859. }
  860. void GeneralSetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  861. {
  862. checkNotAll(ctx);
  863. dsCursor->buildCount(ctx, tgt);
  864. }
  865. void GeneralSetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  866. {
  867. if (isAll->queryValue())
  868. {
  869. if (isAll->queryValue()->getBoolValue())
  870. {
  871. tgt.expr.set(queryBoolExpr(true));
  872. return;
  873. }
  874. dsCursor->buildExists(ctx, tgt);
  875. }
  876. else
  877. {
  878. dsCursor->buildExists(ctx, tgt);
  879. tgt.expr.setown(createBoolExpr(no_or, LINK(isAll), LINK(tgt.expr)));
  880. }
  881. }
  882. void GeneralSetCursor::buildIsAll(BuildCtx & ctx, CHqlBoundExpr & tgt)
  883. {
  884. tgt.expr.set(isAll);
  885. }
  886. void GeneralSetCursor::buildIterateLoop(BuildCtx & ctx, CHqlBoundExpr & curBound, bool needToBreak)
  887. {
  888. BoundRow * cursor = dsCursor->buildIterateLoop(ctx, needToBreak);
  889. OwnedHqlExpr select = createSelectExpr(LINK(cursor->querySelector()), LINK(element));
  890. translator.buildExpr(ctx, select, curBound);
  891. }
  892. void GeneralSetCursor::buildIterateClass(BuildCtx & ctx, CHqlBoundExpr & tgt)
  893. {
  894. UNIMPLEMENTED;
  895. HqlExprAttr row;
  896. dsCursor->buildIterateClass(ctx, tgt.expr, row);
  897. }
  898. IHqlExpression * GeneralSetCursor::createDatasetSelect(IHqlExpression * indexExpr)
  899. {
  900. HqlExprArray args;
  901. args.append(*LINK(ds));
  902. unwindChildren(args, indexExpr, 1);
  903. return createRow(no_selectnth, args);
  904. }
  905. void GeneralSetCursor::buildExprSelect(BuildCtx & ctx, IHqlExpression * indexExpr, CHqlBoundExpr & tgt)
  906. {
  907. if (indexExpr->hasProperty(noBoundCheckAtom))
  908. {
  909. if (indexExpr->hasProperty(forceAllCheckAtom))
  910. checkNotAll(ctx);
  911. OwnedHqlExpr dsIndexExpr = createDatasetSelect(indexExpr);
  912. BoundRow * cursor = dsCursor->buildSelect(ctx, dsIndexExpr);
  913. OwnedHqlExpr select = createSelectExpr(LINK(dsIndexExpr), LINK(element));
  914. translator.buildExpr(ctx, select, tgt);
  915. }
  916. else
  917. {
  918. translator.buildTempExpr(ctx, indexExpr, tgt);
  919. }
  920. }
  921. void GeneralSetCursor::buildAssignSelect(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * indexExpr)
  922. {
  923. if (!indexExpr->hasProperty(noBoundCheckAtom) || indexExpr->hasProperty(forceAllCheckAtom))
  924. checkNotAll(ctx);
  925. OwnedHqlExpr dsIndexExpr = createDatasetSelect(indexExpr);
  926. BoundRow * cursor = dsCursor->buildSelect(ctx, dsIndexExpr);
  927. if (cursor)
  928. {
  929. OwnedHqlExpr select = createSelectExpr(LINK(dsIndexExpr), LINK(element));
  930. if (!cursor->isConditional())
  931. translator.buildExprAssign(ctx, target, select);
  932. else
  933. {
  934. // if (row) tgt = x else tgt = dft;
  935. BuildCtx subctx(ctx);
  936. IHqlStmt * e = subctx.addFilter(cursor->queryBound());
  937. cursor->setConditional(false); // yuk!
  938. translator.buildExprAssign(subctx, target, select);
  939. cursor->setConditional(true);
  940. subctx.selectElse(e);
  941. OwnedHqlExpr null = getOutOfRangeValue(indexExpr);
  942. translator.buildExprAssign(subctx, target, null);
  943. }
  944. }
  945. else
  946. {
  947. OwnedHqlExpr null = getOutOfRangeValue(indexExpr);
  948. translator.buildExprAssign(ctx, target, null);
  949. }
  950. }
  951. void GeneralSetCursor::checkNotAll(BuildCtx & ctx)
  952. {
  953. if (isAll->queryValue())
  954. {
  955. if (isAll->queryValue()->getBoolValue())
  956. translator.throwError(HQLERR_IndexAllSet);
  957. }
  958. else
  959. {
  960. //MORE: Should only really do this once...
  961. BuildCtx subctx(ctx);
  962. subctx.addFilter(isAll);
  963. IHqlExpression * msg = translator.createFailMessage("Cannot index ALL", NULL, NULL, translator.queryCurrentActivityId(ctx));
  964. OwnedHqlExpr fail = createValue(no_fail, makeVoidType(), getZero(), msg, getDefaultAttr());
  965. translator.buildStmt(subctx, fail);
  966. }
  967. }
  968. bool GeneralSetCursor::isSingleValued()
  969. {
  970. if (!matchesBoolean(isAll, false))
  971. return false;
  972. // return dsCursor->hasSingleRow();
  973. return false;
  974. }
  975. //---------------------------------------------------------------------------
  976. CreateSetCursor::CreateSetCursor(HqlCppTranslator & _translator, IHqlExpression * _expr, IHqlCppDatasetCursor * _dsCursor) : BaseSetCursor(_translator, _expr)
  977. {
  978. ds.set(expr->queryChild(0));
  979. value.set(expr->queryChild(1));
  980. dsCursor.set(_dsCursor);
  981. }
  982. void CreateSetCursor::buildCount(BuildCtx & ctx, CHqlBoundExpr & tgt)
  983. {
  984. dsCursor->buildCount(ctx, tgt);
  985. }
  986. void CreateSetCursor::buildExists(BuildCtx & ctx, CHqlBoundExpr & tgt)
  987. {
  988. dsCursor->buildExists(ctx, tgt);
  989. }
  990. void CreateSetCursor::buildIsAll(BuildCtx & ctx, CHqlBoundExpr & tgt)
  991. {
  992. tgt.expr.set(queryBoolExpr(false));
  993. }
  994. void CreateSetCursor::buildIterateLoop(BuildCtx & ctx, CHqlBoundExpr & curBound, bool needToBreak)
  995. {
  996. BoundRow * cursor = dsCursor->buildIterateLoop(ctx, needToBreak);
  997. translator.buildExpr(ctx, value, curBound);
  998. }
  999. void CreateSetCursor::buildIterateClass(BuildCtx & ctx, CHqlBoundExpr & tgt)
  1000. {
  1001. UNIMPLEMENTED;
  1002. }
  1003. IHqlExpression * CreateSetCursor::createDatasetSelect(IHqlExpression * indexExpr)
  1004. {
  1005. if (value->getOperator() == no_select &&
  1006. (value->queryChild(0)->queryNormalizedSelector() == ds->queryNormalizedSelector()))
  1007. {
  1008. HqlExprArray args;
  1009. args.append(*LINK(ds));
  1010. unwindChildren(args, indexExpr, 1);
  1011. IHqlExpression * select = createRow(no_selectnth, args);
  1012. return createNewSelectExpr(select, LINK(value->queryChild(1)));
  1013. }
  1014. else
  1015. {
  1016. OwnedHqlExpr field = createField(createIdentifierAtom("__f1__"), value->getType(), NULL);
  1017. IHqlExpression * aggregateRecord = createRecord(field);
  1018. IHqlExpression * assign = createAssign(createSelectExpr(getSelf(aggregateRecord), LINK(field)), LINK(value));
  1019. IHqlExpression * transform = createValue(no_newtransform, makeTransformType(aggregateRecord->getType()), assign);
  1020. HqlExprArray args;
  1021. args.append(*createDataset(no_newusertable, LINK(ds), createComma(aggregateRecord, transform)));
  1022. unwindChildren(args, indexExpr, 1);
  1023. IHqlExpression * select = createRow(no_selectnth, args);
  1024. return createNewSelectExpr(select, LINK(field));
  1025. }
  1026. }
  1027. void CreateSetCursor::buildExprSelect(BuildCtx & ctx, IHqlExpression * indexExpr, CHqlBoundExpr & tgt)
  1028. {
  1029. OwnedHqlExpr newExpr = createDatasetSelect(indexExpr);
  1030. translator.buildExpr(ctx, newExpr, tgt);
  1031. }
  1032. void CreateSetCursor::buildAssignSelect(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * indexExpr)
  1033. {
  1034. OwnedHqlExpr newExpr = createDatasetSelect(indexExpr);
  1035. translator.buildExprAssign(ctx, target, newExpr);
  1036. }
  1037. bool CreateSetCursor::isSingleValued()
  1038. {
  1039. return hasSingleRow(ds);
  1040. }
  1041. //---------------------------------------------------------------------------
  1042. IHqlCppSetCursor * HqlCppTranslator::createSetSelector(BuildCtx & ctx, IHqlExpression * expr)
  1043. {
  1044. OwnedHqlExpr normalized = normalizeListCasts(expr);
  1045. switch (normalized->getOperator())
  1046. {
  1047. case no_alias_scope:
  1048. {
  1049. unsigned max = normalized->numChildren();
  1050. for (unsigned idx = 1; idx < max; idx++)
  1051. expandAliases(ctx, normalized->queryChild(idx));
  1052. return createSetSelector(ctx, normalized->queryChild(0));
  1053. }
  1054. case no_null:
  1055. return new ListSetCursor(*this, normalized);
  1056. case no_all:
  1057. return new AllSetCursor(*this);
  1058. case no_list:
  1059. if ((normalized->numChildren() == 0) || (normalized->queryType()->queryChildType()->getSize() != UNKNOWN_LENGTH))
  1060. return new ListSetCursor(*this, normalized);
  1061. break; // default
  1062. case no_createset:
  1063. {
  1064. Owned<IHqlCppDatasetCursor> dsCursor = createDatasetSelector(ctx, expr->queryChild(0));
  1065. return new CreateSetCursor(*this, expr, dsCursor);
  1066. }
  1067. }
  1068. CHqlBoundExpr bound;
  1069. buildSimpleExpr(ctx, normalized, bound);
  1070. return new GeneralSetCursor(*this, normalized, bound);
  1071. }
  1072. //---------------------------------------------------------------------------
  1073. IHqlCppDatasetCursor * HqlCppTranslator::createDatasetSelector(BuildCtx & ctx, IHqlExpression * expr)
  1074. {
  1075. // OwnedHqlExpr normalized = normalizeDatasetCasts(expr);
  1076. switch (expr->getOperator())
  1077. {
  1078. case no_null:
  1079. break;
  1080. case no_select:
  1081. if (isMultiLevelDatasetSelector(expr, false))
  1082. return new MultiLevelDatasetCursor(*this, expr);
  1083. break;
  1084. }
  1085. CHqlBoundExpr bound;
  1086. buildDataset(ctx, expr, bound, FormatNatural);
  1087. if (bound.expr->isDatarow() || !isArrayRowset(bound.expr->queryType()))
  1088. return new InlineBlockDatasetCursor(*this, expr, bound);
  1089. return new InlineLinkedDatasetCursor(*this, expr, bound);
  1090. }
  1091. //---------------------------------------------------------------------------
  1092. CHqlCppDatasetBuilder::CHqlCppDatasetBuilder(HqlCppTranslator & _translator, IHqlExpression * _record)
  1093. : translator(_translator), record(_record)
  1094. {
  1095. }
  1096. DatasetBuilderBase::DatasetBuilderBase(HqlCppTranslator & _translator, IHqlExpression * _record, bool _buildLinkedRows) : CHqlCppDatasetBuilder(_translator, _record)
  1097. {
  1098. StringBuffer rowName;
  1099. unique_id_t id = translator.getUniqueId();
  1100. appendUniqueId(instanceName.append("cr"), id);
  1101. builderName.append(instanceName).append(".rowBuilder()");
  1102. rowName.append(instanceName).append(".rowBuilder().row()"); // more!
  1103. IHqlExpression * linkAttr = _buildLinkedRows ? getLinkCountedAttr() : NULL;
  1104. ITypeInfo * rowType = makeRowReferenceType(record);
  1105. if (_buildLinkedRows)
  1106. rowType = makeAttributeModifier(rowType, getLinkCountedAttr());
  1107. OwnedHqlExpr cursorVar = createVariable(rowName.str(), rowType);
  1108. dataset.setown(createDataset(no_anon, LINK(record), createComma(getSelfAttr(), linkAttr)));
  1109. }
  1110. BoundRow * DatasetBuilderBase::buildCreateRow(BuildCtx & ctx)
  1111. {
  1112. StringBuffer s;
  1113. OwnedHqlExpr cond = createQuoted(s.append(instanceName).append(".createRow()"), makeBoolType());
  1114. if (isRestricted())
  1115. ctx.addFilter(cond);
  1116. else
  1117. ctx.addExpr(cond);
  1118. return translator.bindSelf(ctx, dataset, builderName);
  1119. }
  1120. BoundRow * DatasetBuilderBase::buildDeserializeRow(BuildCtx & ctx, IHqlExpression * serializedInput)
  1121. {
  1122. StringBuffer serializerInstanceName;
  1123. translator.ensureRowSerializer(serializerInstanceName, ctx, record, deserializerAtom);
  1124. StringBuffer s;
  1125. s.append(instanceName).append(".deserializeRow(*");
  1126. s.append(serializerInstanceName).append(", ");
  1127. translator.generateExprCpp(s, serializedInput).append(");");
  1128. ctx.addQuoted(s);
  1129. return translator.bindSelf(ctx, dataset, builderName);
  1130. }
  1131. void DatasetBuilderBase::finishRow(BuildCtx & ctx, BoundRow * selfCursor)
  1132. {
  1133. OwnedHqlExpr size = createSizeof(selfCursor->querySelector());
  1134. CHqlBoundExpr boundSize;
  1135. translator.buildExpr(ctx, size, boundSize);
  1136. StringBuffer s;
  1137. s.append(instanceName).append(".finalizeRow(");
  1138. translator.generateExprCpp(s, boundSize.expr).append(");");
  1139. ctx.addQuoted(s);
  1140. ctx.removeAssociation(selfCursor);
  1141. }
  1142. //---------------------------------------------------------------------------
  1143. BlockedDatasetBuilder::BlockedDatasetBuilder(HqlCppTranslator & _translator, IHqlExpression * _record) : DatasetBuilderBase(_translator, _record, false)
  1144. {
  1145. forceLength = false;
  1146. }
  1147. void BlockedDatasetBuilder::buildDeclare(BuildCtx & ctx)
  1148. {
  1149. StringBuffer decl, extra;
  1150. if (count)
  1151. {
  1152. CHqlBoundExpr boundCount;
  1153. translator.buildExpr(ctx, count, boundCount);
  1154. if (translator.isFixedRecordSize(record))
  1155. {
  1156. //RtlFixedDatasetCreator cursor(len, data, size)
  1157. decl.append("RtlLimitedFixedDatasetBuilder");
  1158. extra.append(translator.getFixedRecordSize(record));
  1159. }
  1160. else
  1161. {
  1162. //RtlVariableDatasetCursor cursor(len, data, recordSize)
  1163. decl.append("RtlLimitedVariableDatasetBuilder");
  1164. translator.buildMetaForRecord(extra, record);
  1165. }
  1166. translator.ensureContextAvailable(ctx);
  1167. decl.append(" ").append(instanceName).append("(").append(extra).append(",");
  1168. translator.generateExprCpp(decl, boundCount.expr).append(",");
  1169. if (forceLength)
  1170. {
  1171. OwnedHqlExpr clearFunc = translator.getClearRecordFunction(record);
  1172. translator.generateExprCpp(decl, clearFunc).append(", ctx);");
  1173. }
  1174. else
  1175. decl.append("NULL,NULL);");
  1176. }
  1177. else
  1178. {
  1179. if (translator.isFixedRecordSize(record))
  1180. {
  1181. //RtlFixedDatasetCreator cursor(len, data, size)
  1182. decl.append("RtlFixedDatasetBuilder");
  1183. extra.append(translator.getFixedRecordSize(record)).append(", 0");
  1184. }
  1185. else
  1186. {
  1187. //RtlVariableDatasetCursor cursor(len, data, recordSize)
  1188. decl.append("RtlVariableDatasetBuilder");
  1189. translator.buildMetaForRecord(extra, record);
  1190. }
  1191. decl.append(" ").append(instanceName).append("(").append(extra).append(");");
  1192. }
  1193. ctx.addQuoted(decl);
  1194. }
  1195. void BlockedDatasetBuilder::buildFinish(BuildCtx & ctx, const CHqlBoundTarget & target)
  1196. {
  1197. //more: should I do this by really calling a function?
  1198. StringBuffer s;
  1199. s.append(instanceName).append(".getData(");
  1200. translator.generateExprCpp(s, target.length);
  1201. s.append(",");
  1202. OwnedHqlExpr ref = createValue(no_reference, target.getType(), LINK(target.expr));
  1203. translator.generateExprCpp(s, ref);
  1204. s.append(");");
  1205. ctx.addQuoted(s);
  1206. }
  1207. void BlockedDatasetBuilder::buildFinish(BuildCtx & ctx, CHqlBoundExpr & bound)
  1208. {
  1209. StringBuffer s;
  1210. s.clear().append(instanceName).append(".getSize()");
  1211. bound.length.setown(createQuoted(s.str(), LINK(unsignedType)));
  1212. s.clear().append(instanceName).append(".queryData()");
  1213. bound.expr.setown(createQuoted(s.str(), makeReferenceModifier(dataset->getType())));
  1214. }
  1215. //---------------------------------------------------------------------------
  1216. SingleRowTempDatasetBuilder::SingleRowTempDatasetBuilder(HqlCppTranslator & _translator, IHqlExpression * _record, BoundRow * _row) : CHqlCppDatasetBuilder(_translator, _record)
  1217. {
  1218. row.set(_row);
  1219. cursor.set(row);
  1220. }
  1221. void SingleRowTempDatasetBuilder::buildDeclare(BuildCtx & ctx)
  1222. {
  1223. }
  1224. BoundRow * SingleRowTempDatasetBuilder::buildCreateRow(BuildCtx & ctx)
  1225. {
  1226. cursor.set(row);
  1227. return row;
  1228. }
  1229. void SingleRowTempDatasetBuilder::buildFinish(BuildCtx & ctx, const CHqlBoundTarget & target)
  1230. {
  1231. assertex(cursor != NULL);
  1232. }
  1233. void SingleRowTempDatasetBuilder::buildFinish(BuildCtx & ctx, CHqlBoundExpr & target)
  1234. {
  1235. assertex(cursor != NULL);
  1236. }
  1237. void SingleRowTempDatasetBuilder::finishRow(BuildCtx & ctx, BoundRow * selfCursor)
  1238. {
  1239. }
  1240. //---------------------------------------------------------------------------
  1241. InlineDatasetBuilder::InlineDatasetBuilder(HqlCppTranslator & _translator, IHqlExpression * _record, IHqlExpression * _size, IHqlExpression * _address) : CHqlCppDatasetBuilder(_translator, _record)
  1242. {
  1243. StringBuffer cursorName;
  1244. getUniqueId(cursorName.append("p"));
  1245. ITypeInfo * rowType = makeRowReferenceType(record);
  1246. cursorVar.setown(createVariable(cursorName.str(), rowType));
  1247. dataset.setown(createDataset(no_anon, LINK(record), getSelfAttr()));
  1248. size.set(_size);
  1249. address.set(_address);
  1250. }
  1251. void InlineDatasetBuilder::buildDeclare(BuildCtx & ctx)
  1252. {
  1253. //NB: This is only ever used where the target has already been checked to ensure there is enough room
  1254. //If we wanted to be clever we would need to use a RtlNestedRowBuilder(parent, <start-of-this-row>, ...);
  1255. ctx.addDeclare(cursorVar, address);
  1256. }
  1257. BoundRow * InlineDatasetBuilder::buildCreateRow(BuildCtx & ctx)
  1258. {
  1259. Owned<BoundRow> cursor = translator.createTableCursor(dataset, cursorVar, no_self, NULL);
  1260. ctx.associate(*cursor);
  1261. return cursor;
  1262. }
  1263. void InlineDatasetBuilder::buildFinish(BuildCtx & ctx, const CHqlBoundTarget & target)
  1264. {
  1265. ctx.addAssign(target.length, size);
  1266. }
  1267. void InlineDatasetBuilder::buildFinish(BuildCtx & ctx, CHqlBoundExpr & bound)
  1268. {
  1269. bound.length.set(size);
  1270. bound.expr.set(address);
  1271. }
  1272. void InlineDatasetBuilder::finishRow(BuildCtx & ctx, BoundRow * selfCursor)
  1273. {
  1274. CHqlBoundExpr bound;
  1275. translator.getRecordSize(ctx, selfCursor->querySelector(), bound);
  1276. if (translator.queryOptions().optimizeIncrement)
  1277. {
  1278. ctx.addAssignIncrement(selfCursor->queryBound(), bound.expr);
  1279. }
  1280. else
  1281. {
  1282. OwnedHqlExpr inc = createValue(no_add, LINK(selfCursor->queryBound()), LINK(bound.expr));
  1283. ctx.addAssign(selfCursor->queryBound(), inc);
  1284. }
  1285. }
  1286. //---------------------------------------------------------------------------
  1287. LinkedDatasetBuilder::LinkedDatasetBuilder(HqlCppTranslator & _translator, IHqlExpression * _record, IHqlExpression * _choosenLimit) : DatasetBuilderBase(_translator, _record, true)
  1288. {
  1289. choosenLimit.set(_choosenLimit);
  1290. }
  1291. void LinkedDatasetBuilder::buildDeclare(BuildCtx & ctx)
  1292. {
  1293. StringBuffer decl, allocatorName;
  1294. OwnedHqlExpr curActivityId = translator.getCurrentActivityId(ctx);
  1295. translator.ensureRowAllocator(allocatorName, ctx, record, curActivityId);
  1296. decl.append("RtlLinkedDatasetBuilder ").append(instanceName).append("(");
  1297. decl.append(allocatorName);
  1298. if (choosenLimit)
  1299. {
  1300. CHqlBoundExpr boundLimit;
  1301. translator.buildExpr(ctx, choosenLimit, boundLimit);
  1302. translator.generateExprCpp(decl.append(", "), boundLimit.expr);
  1303. }
  1304. decl.append(");");
  1305. ctx.addQuoted(decl);
  1306. }
  1307. void LinkedDatasetBuilder::finishRow(BuildCtx & ctx, BoundRow * selfCursor)
  1308. {
  1309. OwnedHqlExpr size = translator.getRecordSize(selfCursor->querySelector());
  1310. CHqlBoundExpr boundSize;
  1311. translator.buildExpr(ctx, size, boundSize);
  1312. StringBuffer s;
  1313. s.append(instanceName).append(".finalizeRow(");
  1314. translator.generateExprCpp(s, boundSize.expr).append(");");
  1315. ctx.addQuoted(s);
  1316. }
  1317. void LinkedDatasetBuilder::buildFinish(BuildCtx & ctx, const CHqlBoundTarget & target)
  1318. {
  1319. //more: should I do this by really calling a function?
  1320. StringBuffer s;
  1321. s.append(instanceName).append(".getcount()");
  1322. if (hasWrapperModifier(target.queryType()))
  1323. {
  1324. translator.generateExprCpp(s.clear(), target.expr);
  1325. s.append(".setown(").append(instanceName).append(".getcount()");
  1326. s.append(",").append(instanceName).append(".linkrows());");
  1327. ctx.addQuoted(s);
  1328. }
  1329. else
  1330. {
  1331. OwnedHqlExpr countExpr = createQuoted(s.str(), LINK(unsignedType));
  1332. ctx.addAssign(target.count, countExpr);
  1333. s.clear().append(instanceName).append(".linkrows()");
  1334. OwnedHqlExpr rowsExpr = createQuoted(s.str(), dataset->getType());
  1335. ctx.addAssign(target.expr, rowsExpr);
  1336. }
  1337. }
  1338. void LinkedDatasetBuilder::buildFinish(BuildCtx & ctx, CHqlBoundExpr & bound)
  1339. {
  1340. StringBuffer s;
  1341. s.clear().append(instanceName).append(".getcount()");
  1342. bound.count.setown(createQuoted(s.str(), LINK(unsignedType)));
  1343. s.clear().append(instanceName).append(".queryrows()");
  1344. bound.expr.setown(createQuoted(s.str(), makeReferenceModifier(dataset->getType())));
  1345. }
  1346. bool LinkedDatasetBuilder::buildLinkRow(BuildCtx & ctx, BoundRow * sourceRow)
  1347. {
  1348. IHqlExpression * sourceRecord = sourceRow->queryRecord();
  1349. if (recordTypesMatch(sourceRecord, record) && sourceRow->isBinary())
  1350. {
  1351. OwnedHqlExpr source = getPointer(sourceRow->queryBound());
  1352. BuildCtx subctx(ctx);
  1353. if (sourceRow->isConditional())
  1354. subctx.addFilter(source);
  1355. if (sourceRow->isLinkCounted())
  1356. {
  1357. StringBuffer s;
  1358. s.append(instanceName).append(".append(");
  1359. translator.generateExprCpp(s, source);
  1360. s.append(");");
  1361. subctx.addQuoted(s);
  1362. return true;
  1363. }
  1364. IHqlExpression * sourceExpr = sourceRow->querySelector();
  1365. OwnedHqlExpr rowExpr = sourceExpr->isDataset() ? ensureActiveRow(sourceExpr) : LINK(sourceExpr);
  1366. OwnedHqlExpr size = createSizeof(rowExpr);
  1367. CHqlBoundExpr boundSize;
  1368. translator.buildExpr(ctx, size, boundSize);
  1369. StringBuffer s;
  1370. s.append(instanceName).append(".cloneRow(");
  1371. translator.generateExprCpp(s, boundSize.expr).append(",");
  1372. translator.generateExprCpp(s, source);
  1373. s.append(");");
  1374. subctx.addQuoted(s);
  1375. return true;
  1376. }
  1377. return false;
  1378. }
  1379. bool LinkedDatasetBuilder::buildAppendRows(BuildCtx & ctx, IHqlExpression * expr)
  1380. {
  1381. IHqlExpression * sourceRecord = expr->queryRecord();
  1382. if (recordTypesMatch(sourceRecord, record))
  1383. {
  1384. CHqlBoundExpr bound;
  1385. if (!ctx.getMatchExpr(expr, bound))
  1386. {
  1387. bool tryToOptimize = false;
  1388. switch (expr->getOperator())
  1389. {
  1390. case no_select:
  1391. if (isMultiLevelDatasetSelector(expr, false))
  1392. break;
  1393. if (!hasLinkedRow(expr->queryType()))
  1394. break;
  1395. tryToOptimize = true;
  1396. break;
  1397. default:
  1398. //Don't speculatively evaluate if the expression isn't pure
  1399. tryToOptimize = alwaysEvaluatesToBound(expr) && expr->isPure();
  1400. break;
  1401. }
  1402. if (tryToOptimize)
  1403. translator.buildDataset(ctx, expr, bound, FormatNatural);
  1404. }
  1405. if (bound.expr)
  1406. {
  1407. if (hasLinkedRow(bound.queryType()))
  1408. {
  1409. OwnedHqlExpr source = getPointer(bound.expr);
  1410. StringBuffer s;
  1411. s.append(instanceName).append(".appendRows(");
  1412. translator.generateExprCpp(s, bound.count);
  1413. s.append(",");
  1414. translator.generateExprCpp(s, source);
  1415. s.append(");");
  1416. ctx.addQuoted(s);
  1417. return true;
  1418. }
  1419. }
  1420. }
  1421. return false;
  1422. }
  1423. //---------------------------------------------------------------------------
  1424. SetBuilder::SetBuilder(HqlCppTranslator & _translator, ITypeInfo * fieldType, IHqlExpression * _allVar) : translator(_translator)
  1425. {
  1426. HqlExprArray fields;
  1427. fields.append(*createField(valueAtom, LINK(fieldType), NULL));
  1428. record.setown(createRecord(fields));
  1429. allVar.set(_allVar);
  1430. activeRow = NULL;
  1431. }
  1432. void SetBuilder::buildDeclare(BuildCtx & ctx)
  1433. {
  1434. datasetBuilder->buildDeclare(ctx);
  1435. }
  1436. IReferenceSelector * SetBuilder::buildCreateElement(BuildCtx & ctx)
  1437. {
  1438. activeRow = datasetBuilder->buildCreateRow(ctx);
  1439. OwnedHqlExpr select = createSelectExpr(LINK(activeRow->querySelector()), LINK(record->queryChild(0)));
  1440. return translator.buildReference(ctx, select);
  1441. }
  1442. void SetBuilder::buildFinish(BuildCtx & ctx, const CHqlBoundTarget & target)
  1443. {
  1444. if (target.isAll && (allVar != target.isAll))
  1445. {
  1446. assertex(allVar != NULL);
  1447. ctx.addAssign(target.isAll, allVar);
  1448. }
  1449. datasetBuilder->buildFinish(ctx, target);
  1450. }
  1451. void SetBuilder::finishElement(BuildCtx & ctx)
  1452. {
  1453. datasetBuilder->finishRow(ctx, activeRow);
  1454. activeRow = NULL;
  1455. }
  1456. void SetBuilder::setAll(BuildCtx & ctx, IHqlExpression * isAll)
  1457. {
  1458. if (allVar)
  1459. {
  1460. CHqlBoundExpr bound;
  1461. translator.buildExpr(ctx, isAll, bound);
  1462. ctx.addAssign(allVar, bound.expr);
  1463. }
  1464. else
  1465. {
  1466. if (!matchesBoolean(isAll, false))
  1467. throwUnexpected();
  1468. }
  1469. }
  1470. TempSetBuilder::TempSetBuilder(HqlCppTranslator & _translator, ITypeInfo * fieldType, IHqlExpression * _allVar) : SetBuilder(_translator, fieldType, _allVar)
  1471. {
  1472. datasetBuilder.setown(new BlockedDatasetBuilder(translator, record));
  1473. }
  1474. InlineSetBuilder::InlineSetBuilder(HqlCppTranslator & _translator, ITypeInfo * fieldType, IHqlExpression * _allVar, IHqlExpression * _size, IHqlExpression * _address) : SetBuilder(_translator, fieldType, _allVar)
  1475. {
  1476. datasetBuilder.setown(new InlineDatasetBuilder(translator, record, _size, _address));
  1477. }
  1478. IHqlCppSetBuilder * HqlCppTranslator::createTempSetBuilder(ITypeInfo * type, IHqlExpression * allVar)
  1479. {
  1480. return new TempSetBuilder(*this, type, allVar);
  1481. }
  1482. IHqlCppSetBuilder * HqlCppTranslator::createInlineSetBuilder(ITypeInfo * type, IHqlExpression * allVar, IHqlExpression * size, IHqlExpression * address)
  1483. {
  1484. assertex(allVar);
  1485. return new InlineSetBuilder(*this, type, allVar, size, address);
  1486. }
  1487. IHqlCppDatasetBuilder * HqlCppTranslator::createBlockedDatasetBuilder(IHqlExpression * record)
  1488. {
  1489. return new BlockedDatasetBuilder(*this, record);
  1490. }
  1491. IHqlCppDatasetBuilder * HqlCppTranslator::createLinkedDatasetBuilder(IHqlExpression * record, IHqlExpression * choosenLimit)
  1492. {
  1493. return new LinkedDatasetBuilder(*this, record, choosenLimit);
  1494. }
  1495. IHqlCppDatasetBuilder * HqlCppTranslator::createSingleRowTempDatasetBuilder(IHqlExpression * record, BoundRow * row)
  1496. {
  1497. // if (translator.isFixedRecordSize(record))
  1498. return new SingleRowTempDatasetBuilder(*this, record, row);
  1499. return createBlockedDatasetBuilder(record);
  1500. }
  1501. IHqlCppDatasetBuilder * HqlCppTranslator::createInlineDatasetBuilder(IHqlExpression * record, IHqlExpression * size, IHqlExpression * address)
  1502. {
  1503. assertex(isFixedRecordSize(record));
  1504. return new InlineDatasetBuilder(*this, record, size, address);
  1505. }
  1506. IHqlCppDatasetBuilder * HqlCppTranslator::createChoosenDatasetBuilder(IHqlExpression * record, IHqlExpression * maxCount)
  1507. {
  1508. BlockedDatasetBuilder * builder = new BlockedDatasetBuilder(*this, record);
  1509. builder->setLimit(maxCount, false);
  1510. return builder;
  1511. }
  1512. IHqlCppDatasetBuilder * HqlCppTranslator::createLimitedDatasetBuilder(IHqlExpression * record, IHqlExpression * maxCount)
  1513. {
  1514. BlockedDatasetBuilder * builder = new BlockedDatasetBuilder(*this, record);
  1515. builder->setLimit(maxCount, true);
  1516. return builder;
  1517. }
  1518. //---------------------------------------------------------------------------
  1519. void HqlCppTranslator::doBuildSetAssignAndCast(BuildCtx & ctx, IHqlCppSetBuilder * builder, IHqlExpression * value)
  1520. {
  1521. Owned<IHqlCppSetCursor> cursor = createSetSelector(ctx, value);
  1522. CHqlBoundExpr srcIsAll;
  1523. cursor->buildIsAll(ctx, srcIsAll);
  1524. OwnedHqlExpr translated = srcIsAll.getTranslatedExpr();
  1525. builder->setAll(ctx, translated);
  1526. BuildCtx loopctx(ctx);
  1527. CHqlBoundExpr boundCurElement;
  1528. cursor->buildIterateLoop(loopctx, boundCurElement, false);
  1529. Owned<IReferenceSelector> selector = builder->buildCreateElement(loopctx);
  1530. OwnedHqlExpr translatedCurElement = boundCurElement.getTranslatedExpr();
  1531. selector->set(loopctx, translatedCurElement);
  1532. builder->finishElement(loopctx);
  1533. }
  1534. void HqlCppTranslator::buildSetAssign(BuildCtx & ctx, IHqlCppSetBuilder * builder, IHqlExpression * expr)
  1535. {
  1536. switch (expr->getOperator())
  1537. {
  1538. case no_cast:
  1539. doBuildSetAssignAndCast(ctx, builder, expr->queryChild(0));
  1540. break;
  1541. case no_addsets:
  1542. doBuildSetAssignAndCast(ctx, builder, expr);
  1543. break;
  1544. //MORE: This is wrong because needs to cope with all.
  1545. buildSetAssign(ctx, builder, expr->queryChild(0));
  1546. buildSetAssign(ctx, builder, expr->queryChild(1));
  1547. break;
  1548. case no_all:
  1549. builder->setAll(ctx, queryBoolExpr(true));
  1550. break;
  1551. case no_list:
  1552. {
  1553. unsigned max = expr->numChildren();
  1554. if ((max < 3) || isComplexSet(expr) || !isConstantSet(expr))
  1555. {
  1556. for (unsigned i=0; i < max; i++)
  1557. {
  1558. //Need a subcontext otherwise sizeof(target-row) gets cached.
  1559. BuildCtx subctx(ctx);
  1560. subctx.addGroup();
  1561. Owned<IReferenceSelector> selector = builder->buildCreateElement(subctx);
  1562. selector->set(subctx, expr->queryChild(i));
  1563. builder->finishElement(subctx);
  1564. }
  1565. builder->setAll(ctx, queryBoolExpr(false));
  1566. }
  1567. else
  1568. doBuildSetAssignAndCast(ctx, builder, expr);
  1569. }
  1570. break;
  1571. case no_createset:
  1572. {
  1573. IHqlExpression * ds = expr->queryChild(0);
  1574. IHqlExpression * value = expr->queryChild(1);
  1575. builder->setAll(ctx, queryBoolExpr(false));
  1576. BuildCtx subctx(ctx);
  1577. BoundRow * cursor = buildDatasetIterate(subctx, ds, false);
  1578. Owned<IReferenceSelector> selector = builder->buildCreateElement(subctx);
  1579. selector->set(subctx, value);
  1580. builder->finishElement(subctx);
  1581. break;
  1582. }
  1583. default:
  1584. doBuildSetAssignAndCast(ctx, builder, expr);
  1585. break;
  1586. }
  1587. }
  1588. void HqlCppTranslator::buildSetAssignViaBuilder(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * value)
  1589. {
  1590. ITypeInfo * to = target.queryType();
  1591. Owned<IHqlCppSetBuilder> builder = createTempSetBuilder(to->queryChildType(), target.isAll);
  1592. builder->buildDeclare(ctx);
  1593. buildSetAssign(ctx, builder, value);
  1594. builder->buildFinish(ctx, target);
  1595. }
  1596. void HqlCppTranslator::doBuildAssignAddSets(BuildCtx & ctx, const CHqlBoundTarget & target, IHqlExpression * value)
  1597. {
  1598. IHqlExpression * left = value->queryChild(0);
  1599. IHqlExpression * right = value->queryChild(1);
  1600. assertex(left->queryType() == right->queryType());
  1601. //a poor implementation, but at least it works.
  1602. HqlExprArray args;
  1603. args.append(*LINK(left));
  1604. args.append(*LINK(right));
  1605. OwnedHqlExpr call = bindFunctionCall(appendSetXAtom, args, left->queryType());
  1606. buildExprAssign(ctx, target, call);
  1607. }