@itwin/tree-widget-react 3.0.0-dev.0 → 3.0.0-dev.1

package/lib/esm/components/trees/imodel-content-tree/IModelContentTreeDefinition.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelContentTreeDefinition.js","sourceRoot":"","sources":["../../../../../src/components/trees/imodel-content-tree/IModelContentTreeDefinition.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,mCAAmC,EAAE,6BAA6B,EAAE,aAAa,EAAE,2BAA2B,GAC/G,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,yCAAyC,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAqB5E,MAAM,OAAO,2BAA2B;IAMtC,YAAmB,KAAuC;QACxD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,KAAK,GAAG,mCAAmC,CAAC;YAC/C,uBAAuB,EAAE,KAAK,CAAC,YAAY;YAC3C,SAAS,EAAE;gBACT,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,YAAY,CAAC;gBACxF,UAAU,EAAE;oBACV;wBACE,mBAAmB,EAAE,iBAAiB;wBACtC,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC;qBAC/H;oBACD;wBACE,mBAAmB,EAAE,4BAA4B;wBACjD,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,qCAAqC,CAAC,YAAY,CAAC;qBAC1I;oBACD;wBACE,mBAAmB,EAAE,0BAA0B;wBAC/C,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAC9E,IAAI,CAAC,iCAAiC,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;qBAC9E;oBACD;wBACE,mBAAmB,EAAE,0BAA0B;wBAC/C,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAC9E,IAAI,CAAC,iCAAiC,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;qBAC9E;oBACD;wBACE,mBAAmB,EAAE,yBAAyB;wBAC9C,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAC9E,IAAI,CAAC,2BAA2B,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;qBACxE;oBACD;wBACE,mBAAmB,EAAE,yBAAyB;wBAC9C,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAC9E,IAAI,CAAC,2BAA2B,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;qBACxE;oBACD;wBACE,mBAAmB,EAAE,eAAe;wBACpC,gBAAgB,EAAE,IAAI;wBACtB,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,YAAY,CAAC;qBAC7H;oBACD;wBACE,mBAAmB,EAAE,iCAAiC;wBACtD,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,+CAA+C,CAAC,YAAY,CAAC;qBACpJ;oBACD;wBACE,mBAAmB,EAAE,cAAc;wBACnC,WAAW,EAAE,KAAK,EAAE,YAAsD,EAAE,EAAE,CAAC,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC;qBAClI;oBACD;wBACE,mBAAmB,EAAE,aAAa;wBAClC,WAAW,EAAE,KAAK,EAAE,YAAsD,EAAE,EAAE,CAAC,IAAI,CAAC,gDAAgD,CAAC,YAAY,CAAC;qBACnJ;oBACD;wBACE,mBAAmB,EAAE,iBAAiB;wBACtC,gBAAgB,EAAE,IAAI;wBACtB,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,YAAY,CAAC;qBAC/H;iBACF;aACF;SACF,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,GAAG,6BAA6B,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QAC/F,IAAI,CAAC,6BAA6B,GAAG,yCAAyC,CAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;IAClI,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,IAA4B;QACvD,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YAC5F,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,CAAC;SAC7F;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,KAAgC;QAChE,OAAO,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,kCAAkC,CAAC,KAAoC;QACnF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC/E,MAAM,EAAE,KAAK,CAAC,cAAc;YAC5B,YAAY,EAAE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE;SAC7D,CAAC,CAAC;QACH,OAAO;YACL;gBACE,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE;wBAClF,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,iBAAiB;6BAC7B,CAAC;yBACH;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE,sBAAsB;yBAChC;wBACD,UAAU,EAAE,IAAI;wBAChB,iBAAiB,EAAE,IAAI;qBACxB,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;;gBAGzB,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,EACvC,qBAAqB,EAAE,UAAU,EACjC,cAAc,GAC6B;QAC3C,MAAM,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnE,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC3C,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE;aAC7D,CAAC;YACF,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC3C,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE;aAC3D,CAAC;SACH,CAAC,CAAC;QACH,MAAM,CAAC,eAAe,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACzD,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,UAAU,CAAC;YAC7C,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,UAAU,CAAC;SACnD,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,KAAK,EAA4B,CAAC;QACnD,eAAe,CAAC,MAAM;YACpB,IAAI,CAAC,IAAI,CAAC;gBACR,aAAa,EAAE,iBAAiB;gBAChC,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,iBAAiB;6BAC7B,CAAC;yBACH;wBACD,WAAW,EAAE,EAAE,QAAQ,EAAE,oCAAoC,EAAE;wBAC/D,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE;wBAC9D,YAAY,EAAE;4BACZ,OAAO,EAAE,aAAa;yBACvB;wBACD,iBAAiB,EAAE,IAAI;qBACxB,CAAC;mBACG,oBAAoB,CAAC,IAAI;cAC9B,oBAAoB,CAAC,KAAK;;sCAEF,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC9D,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,oBAAoB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC1E;oBACD,QAAQ,EAAE;wBACR,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE,EAAE;wBACpE,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;qBAC1E;iBACF;aACF,CAAC,CAAC;QACL,aAAa,CAAC,MAAM;YAClB,IAAI,CAAC,IAAI,CAAC;gBACR,aAAa,EAAE,eAAe;gBAC9B,KAAK,EAAE;oBACL,KAAK,EAAE;;;;kBAIC,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE;wBACtC,YAAY,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBAC5C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,WAAW;gCACvB,SAAS,EAAE,qCAAqC;6BACjD,CAAC;yBACH;wBACD,mBAAmB,EAAE;4BACnB,QAAQ,EAAE;;;;;;;;qBAQT;yBACF;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE,YAAY;yBACtB;wBACD,iBAAiB,EAAE,IAAI;qBACxB,CAAC;;;;qCAImB,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;;mBAExD,kBAAkB,CAAC,IAAI;cAC5B,kBAAkB,CAAC,KAAK;cACxB,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,gBAAgB,2BAA2B,CAAC,mBAAmB,OAAO,kBAAkB,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE;WACpI;oBACD,QAAQ,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC/E;aACF,CAAC,CAAC;QACL,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,qCAAqC,CAAC,EAClD,qBAAqB,EAAE,UAAU,GACU;QAC3C,+FAA+F;QAC/F,qEAAqE;QACrE,OAAO;YACL;gBACE,aAAa,EAAE,eAAe;gBAC9B,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE,EAAE;wBACb,mBAAmB,EAAE,IAAI;qBAC1B,CAAC;;;2CAG2B,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;;WAEnE;oBACD,QAAQ,EAAE,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;iBACjF;aACF;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,iCAAiC,CAAC,EAC9C,qBAAqB,EAAE,QAAQ,EAC/B,cAAc,EACd,QAAQ,GAC+D;QACvE,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAE3E,MAAM,EAAE,aAAa,EAAE,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAC3D,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC/E,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;SACzD,CAAC,CAAC;QACH,MAAM,sCAAsC,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAChG,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,mCAAmC,EAAE,KAAK,EAAE,MAAM,EAAE;SAC/E,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,KAAK,EAA4B,CAAC;QACnD,gBAAgB,CAAC,MAAM;YACrB,IAAI,CAAC,IAAI,CAAC;gBACR,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE;wBAC/D,YAAY,EAAE;4BACZ,OAAO,EAAE,aAAa;4BACtB,QAAQ,EAAE,EAAE,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,CAAC,EAAE;yBACpD;wBACD,WAAW,EAAE,IAAI;wBACjB,iBAAiB,EAAE,IAAI;qBACxB,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;sCAEH,gBAAgB,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC/D,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;oBACD,QAAQ,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAClF;aACF,CAAC,CAAC;QACL,IAAI,CAAC,IAAI,CAAC;YACR,aAAa,EAAE,mCAAmC;YAClD,KAAK,EAAE;gBACL,KAAK,EAAE;;gBAEC,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;oBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;oBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;oBAC/C,SAAS,EAAE;wBACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;4BACpE,UAAU,EAAE,MAAM;4BAClB,SAAS,EAAE,mCAAmC;yBAC/C,CAAC;qBACH;oBACD,WAAW,EAAE,IAAI;oBACjB,QAAQ,EAAE;wBACR,OAAO,EAAE,IAAI;qBACd;oBACD,iBAAiB,EAAE,IAAI;iBACxB,CAAC;mBACG,sCAAsC,CAAC,IAAI;cAChD,sCAAsC,CAAC,KAAK;;kCAExB,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACnD,sCAAsC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,sCAAsC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC9G;gBACH,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;aAC5D;SACF,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,KAA6E;QACrH,MAAM,EAAE,qBAAqB,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;QAC3F,MAAM,QAAQ,GAAG,sBAAsB,CAAC,UAAU,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAE3E,kGAAkG;QAClG,kEAAkE;QAClE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YACzB,OAAO,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC;SAC/C;QAED,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACtE,OAAO,OAAO,CAAC,GAAG,CAChB,sBAAsB,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;YAC9F,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC/E,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE;aACxD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,YAAY;gBAC3B,KAAK,EAAE;oBACL,KAAK,EAAE;;kBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,YAAY;6BACxB,CAAC;yBACH;wBACD,QAAQ,EAAE;4BACR,OAAO,EAAE,IAAI;yBACd;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE,WAAW;yBACrB;wBACD,WAAW,EAAE,WAAW,CAAC,WAAW;wBACpC,iBAAiB,EAAE,WAAW,CAAC,iBAAiB;qBACjD,CAAC;qBACG,qBAAqB,CAAC,IAAI;gBAC/B,qBAAqB,CAAC,KAAK;;uCAEJ,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;wCACnC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;;kBAEvD,WAAW,CAAC,CAAC,CAAC,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;kBACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;aAC5E;oBACD,QAAQ,EAAE,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAmB;iBAC5I;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,EACrC,qBAAqB,EAAE,QAAQ,EAC/B,cAAc,GAC6B;QAC3C,OAAO,OAAO,CAAC,GAAG,CAChB,sBAAsB,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;YACjF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC/E,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;aACzD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAEH,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,QAAQ,EAAE;4BACR,OAAO,EAAE,IAAI;yBACd;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE,WAAW;yBACrB;wBACD,WAAW,EAAE,WAAW,EAAE,WAAW;wBACrC,iBAAiB,EAAE,WAAW,EAAE,iBAAiB;qBAClD,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;;sCAGH,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACvD,WAAW,CAAC,CAAC,CAAC,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;gBACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;oBACC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC1E;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,+CAA+C,CAAC,EAC5D,qBAAqB,EAAE,QAAQ,GACY;QAC3C,OAAO;YACL;gBACE,IAAI,EAAE;oBACJ,GAAG,EAAE,cAAc;oBACnB,KAAK,EAAE,UAAU;oBACjB,YAAY,EAAE;wBACZ,QAAQ;wBACR,OAAO,EAAE,qBAAqB;qBAC/B;oBACD,iBAAiB,EAAE,IAAI;oBACvB,gBAAgB,EAAE;wBAChB,gBAAgB,EAAE,IAAI;qBACvB;iBACF;aACF;YACD;gBACE,IAAI,EAAE;oBACJ,GAAG,EAAE,aAAa;oBAClB,KAAK,EAAE,SAAS;oBAChB,YAAY,EAAE;wBACZ,QAAQ;wBACR,OAAO,EAAE,YAAY;qBACtB;oBACD,iBAAiB,EAAE,IAAI;oBACvB,gBAAgB,EAAE;wBAChB,gBAAgB,EAAE,IAAI;qBACvB;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,+BAA+B,CAAC,EAAE,UAAU,EAAE,cAAc,EAA4C;QACpH,MAAM,QAAQ,GAAa,UAAU,CAAC,YAAY,EAAE,QAAQ,CAAC;QAC7D,OAAO,OAAO,CAAC,GAAG,CAChB,sBAAsB,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;YACjF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC/E,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;aACzD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;kBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,YAAY,EAAE;4BACZ,OAAO,EAAE,WAAW;yBACrB;wBACD,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;wBAC3B,WAAW,EAAE,WAAW,CAAC,WAAW;wBACpC,iBAAiB,EAAE,WAAW,CAAC,iBAAiB;qBACjD,CAAC;qBACG,qBAAqB,CAAC,IAAI;gBAC/B,qBAAqB,CAAC,KAAK;;qCAEN,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;kBACpD,WAAW,CAAC,CAAC,CAAC,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;kBACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;aAC5E;oBACD,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC5D;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gDAAgD,CAAC,EAC7D,UAAU,EACV,cAAc,GAC2B;QACzC,MAAM,QAAQ,GAAa,UAAU,CAAC,YAAY,EAAE,QAAQ,CAAC;QAC7D,OAAO,OAAO,CAAC,GAAG,CAChB,sBAAsB,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;YACjF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC/E,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;aACzD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;kBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;wBAC3B,YAAY,EAAE;4BACZ,OAAO,EAAE,WAAW;yBACrB;wBACD,WAAW,EAAE,WAAW,CAAC,WAAW;wBACpC,iBAAiB,EAAE,WAAW,CAAC,iBAAiB;qBACjD,CAAC;qBACG,qBAAqB,CAAC,IAAI;;gBAE/B,qBAAqB,CAAC,KAAK;;6CAEE,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;kBAC5D,WAAW,CAAC,CAAC,CAAC,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;kBACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;aAC5E;oBACD,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC5D;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,EACvC,qBAAqB,EAAE,UAAU,EACjC,cAAc,GAC6B;QAC3C,OAAO,OAAO,CAAC,GAAG,CAChB,sBAAsB,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;YACjF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBAC/E,MAAM,EAAE,cAAc;gBACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;aACzD,CAAC,CAAC;YACH,OAAO;gBACL,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;kBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;wBAC3B,YAAY,EAAE;4BACZ,OAAO,EAAE,WAAW;yBACrB;wBACD,WAAW,EAAE,WAAW,CAAC,WAAW;wBACpC,iBAAiB,EAAE,WAAW,CAAC,iBAAiB;qBACjD,CAAC;qBACG,qBAAqB,CAAC,IAAI;;gBAE/B,qBAAqB,CAAC,KAAK;;qCAEN,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;kBACtD,WAAW,CAAC,CAAC,CAAC,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE;kBACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;aAC5E;oBACD,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC9D;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AAED,SAAS,sBAAsB,CAAC,IAAiB;IAC/C,IAAI,IAAI,KAAK,IAAI,EAAE;QACjB,OAAO,EAAE,aAAa,EAAE,yBAAyB,EAAE,YAAY,EAAE,4BAA4B,EAAE,UAAU,EAAE,0BAA0B,EAAW,CAAC;KAClJ;IACD,OAAO,EAAE,aAAa,EAAE,yBAAyB,EAAE,YAAY,EAAE,4BAA4B,EAAE,UAAU,EAAE,0BAA0B,EAAW,CAAC;AACnJ,CAAC;AAED,SAAS,sBAAsB,CAAC,KAA2G;IACzI,MAAM,kBAAkB,GAAG,KAAK,EAAE,UAAU,IAAI,eAAe,CAAC;IAChE,MAAM,oBAAoB,GAAG,KAAK,EAAE,YAAY,IAAI,iBAAiB,CAAC;IACtE,MAAM,MAAM,GAAG;QACb;YACE,aAAa,EAAE,oBAAoB;YACnC,WAAW,EAAE,yDAAyD;YACtE,WAAW,EAAE;gBACX,WAAW,EAAE;oBACX,QAAQ,EAAE;;;;+CAI2B,oBAAoB;;0DAET,kBAAkB,mCAAmC,oBAAoB;;;;;SAK1H;iBACA;gBACD,iBAAiB,EAAE,IAAI;aACxB;SACF;KACF,CAAC;IAEF,IAAI,CAAC,KAAK,EAAE,YAAY,EAAE;QACxB,MAAM,CAAC,IAAI,CAAC;YACV,aAAa,EAAE,iCAAiC;YAChD,WAAW,EAAE,EAAE;YACf,WAAW,EAAE;gBACX,WAAW,EAAE;oBACX,QAAQ,EAAE;;;;;;;;;;;WAWT;iBACF;gBACD,iBAAiB,EAAE,KAAK;aACzB;SACF,CAAC,CAAC;KACJ;IAED,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport {\n  createClassBasedHierarchyDefinition, createNodesQueryClauseFactory, HierarchyNode, NodeSelectClauseColumnNames,\n} from \"@itwin/presentation-hierarchies\";\nimport { createBisInstanceLabelSelectClauseFactory, ECSql } from \"@itwin/presentation-shared\";\nimport { createIdsSelector, parseIdsSelectorResult } from \"../common/Utils\";\n\nimport type { ECClassHierarchyInspector, ECSchemaProvider, ECSqlBinding, IInstanceLabelSelectClauseFactory } from \"@itwin/presentation-shared\";\nimport type {\n  DefineCustomNodeChildHierarchyLevelProps,\n  DefineHierarchyLevelProps,\n  DefineInstanceNodeChildHierarchyLevelProps,\n  DefineRootHierarchyLevelProps,\n  HierarchyDefinition,\n  HierarchyLevelDefinition,\n  HierarchyNodesDefinition,\n  NodesQueryClauseFactory,\n  ProcessedHierarchyNode,\n} from \"@itwin/presentation-hierarchies\";\nimport type { IModelContentTreeIdsCache } from \"./internal/IModelContentTreeIdsCache\";\n\ninterface IModelContentTreeDefinitionProps {\n  imodelAccess: ECSchemaProvider & ECClassHierarchyInspector;\n  idsCache: IModelContentTreeIdsCache;\n}\n\nexport class IModelContentTreeDefinition implements HierarchyDefinition {\n  private _impl: HierarchyDefinition;\n  private _idsCache: IModelContentTreeIdsCache;\n  private _selectQueryFactory: NodesQueryClauseFactory;\n  private _nodeLabelSelectClauseFactory: IInstanceLabelSelectClauseFactory;\n\n  public constructor(props: IModelContentTreeDefinitionProps) {\n    this._idsCache = props.idsCache;\n    this._impl = createClassBasedHierarchyDefinition({\n      classHierarchyInspector: props.imodelAccess,\n      hierarchy: {\n        rootNodes: async (requestProps) => this.createRootHierarchyLevelDefinition(requestProps),\n        childNodes: [\n          {\n            parentNodeClassName: \"BisCore.Subject\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createSubjectChildrenQuery(requestProps),\n          },\n          {\n            parentNodeClassName: \"BisCore.ISubModeledElement\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createISubModeledElementChildrenQuery(requestProps),\n          },\n          {\n            parentNodeClassName: \"BisCore.GeometricModel3d\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) =>\n              this.createGeometricModelChildrenQuery({ ...requestProps, viewType: \"3d\" }),\n          },\n          {\n            parentNodeClassName: \"BisCore.GeometricModel2d\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) =>\n              this.createGeometricModelChildrenQuery({ ...requestProps, viewType: \"2d\" }),\n          },\n          {\n            parentNodeClassName: \"BisCore.SpatialCategory\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) =>\n              this.createCategoryChildrenQuery({ ...requestProps, viewType: \"3d\" }),\n          },\n          {\n            parentNodeClassName: \"BisCore.DrawingCategory\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) =>\n              this.createCategoryChildrenQuery({ ...requestProps, viewType: \"2d\" }),\n          },\n          {\n            parentNodeClassName: \"BisCore.Model\",\n            onlyIfNotHandled: true,\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createModelChildrenQuery(requestProps),\n          },\n          {\n            parentNodeClassName: \"BisCore.GroupInformationElement\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createGroupInformationElementChildElementsQuery(requestProps),\n          },\n          {\n            customParentNodeKey: \"ChildrenNode\",\n            definitions: async (requestProps: DefineCustomNodeChildHierarchyLevelProps) => this.createChildrenNodeChildrenQuery(requestProps),\n          },\n          {\n            customParentNodeKey: \"MembersNode\",\n            definitions: async (requestProps: DefineCustomNodeChildHierarchyLevelProps) => this.createGroupInformationElementMemberElementsQuery(requestProps),\n          },\n          {\n            parentNodeClassName: \"BisCore.Element\",\n            onlyIfNotHandled: true,\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createElementChildrenQuery(requestProps),\n          },\n        ],\n      },\n    });\n    this._selectQueryFactory = createNodesQueryClauseFactory({ imodelAccess: props.imodelAccess });\n    this._nodeLabelSelectClauseFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });\n  }\n\n  public async postProcessNode(node: ProcessedHierarchyNode): Promise<ProcessedHierarchyNode> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      const label = node.children.length ? `${node.label} (${node.children.length})` : node.label;\n      return { ...node, label, extendedData: { ...node.extendedData, imageId: \"icon-ec-class\" } };\n    }\n    return node;\n  }\n\n  public async defineHierarchyLevel(props: DefineHierarchyLevelProps) {\n    return this._impl.defineHierarchyLevel(props);\n  }\n\n  private async createRootHierarchyLevelDefinition(props: DefineRootHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: props.instanceFilter,\n      contentClass: { fullName: \"BisCore.Subject\", alias: \"this\" },\n    });\n    return [\n      {\n        fullClassName: \"BisCore.Subject\",\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: ECSql.createRawPropertyValueSelector(\"this\", \"ECClassId\") },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: \"BisCore.Subject\",\n                  }),\n                },\n                extendedData: {\n                  imageId: \"icon-imodel-hollow-2\",\n                },\n                autoExpand: true,\n                supportsFiltering: true,\n              })}\n            FROM ${instanceFilterClauses.from} this\n            ${instanceFilterClauses.joins}\n            WHERE\n              this.Parent IS NULL\n              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n          `,\n        },\n      },\n    ];\n  }\n\n  private async createSubjectChildrenQuery({\n    parentNodeInstanceIds: subjectIds,\n    instanceFilter,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const [subjectFilterClauses, modelFilterClauses] = await Promise.all([\n      this._selectQueryFactory.createFilterClauses({\n        filter: instanceFilter,\n        contentClass: { fullName: \"BisCore.Subject\", alias: \"this\" },\n      }),\n      this._selectQueryFactory.createFilterClauses({\n        filter: instanceFilter,\n        contentClass: { fullName: \"BisCore.Model\", alias: \"this\" },\n      }),\n    ]);\n    const [childSubjectIds, childModelIds] = await Promise.all([\n      this._idsCache.getChildSubjectIds(subjectIds),\n      this._idsCache.getChildSubjectModelIds(subjectIds),\n    ]);\n    const defs = new Array<HierarchyNodesDefinition>();\n    childSubjectIds.length &&\n      defs.push({\n        fullClassName: \"BisCore.Subject\",\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: \"BisCore.Subject\",\n                  }),\n                },\n                hasChildren: { selector: `InVirtualSet(?, this.ECInstanceId)` },\n                grouping: { byLabel: { action: \"merge\", groupId: \"subject\" } },\n                extendedData: {\n                  imageId: \"icon-folder\",\n                },\n                supportsFiltering: true,\n              })}\n            FROM ${subjectFilterClauses.from} this\n            ${subjectFilterClauses.joins}\n            WHERE\n              this.ECInstanceId IN (${childSubjectIds.map(() => \"?\").join(\",\")})\n              ${subjectFilterClauses.where ? `AND ${subjectFilterClauses.where}` : \"\"}\n          `,\n          bindings: [\n            { type: \"idset\", value: await this._idsCache.getParentSubjectIds() },\n            ...childSubjectIds.map((id): ECSqlBinding => ({ type: \"id\", value: id })),\n          ],\n        },\n      });\n    childModelIds.length &&\n      defs.push({\n        fullClassName: \"BisCore.Model\",\n        query: {\n          ecsql: `\n            SELECT model.ECInstanceId AS ECInstanceId, model.*\n            FROM (\n              SELECT\n                ${await this._selectQueryFactory.createSelectClause({\n                  ecClassId: { selector: \"m.ECClassId\" },\n                  ecInstanceId: { selector: \"m.ECInstanceId\" },\n                  nodeLabel: {\n                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                      classAlias: \"partition\",\n                      className: \"BisCore.InformationPartitionElement\",\n                    }),\n                  },\n                  hideNodeInHierarchy: {\n                    selector: `\n                      CASE\n                        WHEN (\n                          json_extract([partition].JsonProperties, '$.PhysicalPartition.Model.Content') IS NOT NULL\n                          OR json_extract([partition].JsonProperties, '$.GraphicalPartition3d.Model.Content') IS NOT NULL\n                        ) THEN 1\n                        ELSE 0\n                      END\n                    `,\n                  },\n                  extendedData: {\n                    imageId: \"icon-model\",\n                  },\n                  supportsFiltering: true,\n                })}\n              FROM BisCore.Model m\n              JOIN BisCore.InformationPartitionElement [partition] ON [partition].ECInstanceId = m.ModeledElement.Id\n              WHERE\n                m.ECInstanceId IN (${childModelIds.map(() => \"?\").join(\",\")})\n            ) model\n            JOIN ${modelFilterClauses.from} this ON this.ECInstanceId = model.ECInstanceId\n            ${modelFilterClauses.joins}\n            ${modelFilterClauses.where ? `WHERE (model.${NodeSelectClauseColumnNames.HideNodeInHierarchy} OR ${modelFilterClauses.where})` : \"\"}\n          `,\n          bindings: childModelIds.map((id): ECSqlBinding => ({ type: \"id\", value: id })),\n        },\n      });\n    return defs;\n  }\n\n  private async createISubModeledElementChildrenQuery({\n    parentNodeInstanceIds: elementIds,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    // note: we do not apply hierarchy level filtering on this hierarchy level, because it's always\n    // hidden - the filter will get applied on the child hierarchy levels\n    return [\n      {\n        fullClassName: \"BisCore.Model\",\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: \"\", // doesn't matter - the node is always hidden\n                hideNodeInHierarchy: true,\n              })}\n            FROM BisCore.Model this\n            WHERE\n              this.ModeledElement.Id IN (${elementIds.map(() => \"?\").join(\",\")})\n              AND NOT this.IsPrivate\n          `,\n          bindings: [...elementIds.map((id): ECSqlBinding => ({ type: \"id\", value: id }))],\n        },\n      },\n    ];\n  }\n\n  private async createGeometricModelChildrenQuery({\n    parentNodeInstanceIds: modelIds,\n    instanceFilter,\n    viewType,\n  }: DefineInstanceNodeChildHierarchyLevelProps & { viewType: \"2d\" | \"3d\" }): Promise<HierarchyLevelDefinition> {\n    const childCategoryIds = await this._idsCache.getModelCategories(modelIds);\n\n    const { categoryClass } = getClassNameByViewType(viewType);\n    const categoryFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: instanceFilter,\n      contentClass: { fullName: categoryClass, alias: \"this\" },\n    });\n    const informationContentElementFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: instanceFilter,\n      contentClass: { fullName: \"BisCore.InformationContentElement\", alias: \"this\" },\n    });\n\n    const defs = new Array<HierarchyNodesDefinition>();\n    childCategoryIds.length &&\n      defs.push({\n        fullClassName: categoryClass,\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: categoryClass,\n                  }),\n                },\n                grouping: { byLabel: { action: \"merge\", groupId: \"category\" } },\n                extendedData: {\n                  imageId: \"icon-layers\",\n                  modelIds: { selector: createIdsSelector(modelIds) },\n                },\n                hasChildren: true,\n                supportsFiltering: true,\n              })}\n            FROM ${categoryFilterClauses.from} this\n            ${categoryFilterClauses.joins}\n            WHERE\n              this.ECInstanceId IN (${childCategoryIds.map(() => \"?\").join(\",\")})\n              ${categoryFilterClauses.where ? `AND ${categoryFilterClauses.where}` : \"\"}\n          `,\n          bindings: childCategoryIds.map((id): ECSqlBinding => ({ type: \"id\", value: id })),\n        },\n      });\n    defs.push({\n      fullClassName: \"BisCore.InformationContentElement\",\n      query: {\n        ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: \"BisCore.InformationContentElement\",\n                  }),\n                },\n                hasChildren: true,\n                grouping: {\n                  byClass: true,\n                },\n                supportsFiltering: true,\n              })}\n            FROM ${informationContentElementFilterClauses.from} this\n            ${informationContentElementFilterClauses.joins}\n            WHERE\n              this.Model.Id IN (${modelIds.map(() => \"?\").join(\",\")})\n              ${informationContentElementFilterClauses.where ? `AND ${informationContentElementFilterClauses.where}` : \"\"}\n          `,\n        bindings: modelIds.map((id) => ({ type: \"id\", value: id })),\n      },\n    });\n    return defs;\n  }\n\n  private async createCategoryChildrenQuery(props: DefineInstanceNodeChildHierarchyLevelProps & { viewType: \"2d\" | \"3d\" }): Promise<HierarchyLevelDefinition> {\n    const { parentNodeInstanceIds: categoryIds, parentNode, instanceFilter, viewType } = props;\n    const modelIds = parseIdsSelectorResult(parentNode.extendedData?.modelIds);\n\n    // We only want to handle a category added as a child of `GeometricModel2d` or `GeometricModel3d`.\n    // ModelIds is not empty only if parent node is a geometric model.\n    if (modelIds.length === 0) {\n      return this.createElementChildrenQuery(props);\n    }\n\n    const { elementClass, modelClass } = getClassNameByViewType(viewType);\n    return Promise.all(\n      getElementsSelectProps({ modelClass, elementClass }).map(async ({ selectProps, whereClause }) => {\n        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: elementClass, alias: \"this\" },\n        });\n        return {\n          fullClassName: elementClass,\n          query: {\n            ecsql: `\n              SELECT\n                ${await this._selectQueryFactory.createSelectClause({\n                  ecClassId: { selector: \"this.ECClassId\" },\n                  ecInstanceId: { selector: \"this.ECInstanceId\" },\n                  nodeLabel: {\n                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                      classAlias: \"this\",\n                      className: elementClass,\n                    }),\n                  },\n                  grouping: {\n                    byClass: true,\n                  },\n                  extendedData: {\n                    imageId: \"icon-item\",\n                  },\n                  hasChildren: selectProps.hasChildren,\n                  supportsFiltering: selectProps.supportsFiltering,\n                })}\n              FROM ${instanceFilterClauses.from} this\n              ${instanceFilterClauses.joins}\n              WHERE\n                this.Category.Id IN (${categoryIds.map(() => \"?\").join(\",\")})\n                AND this.Model.Id IN (${modelIds.map(() => \"?\").join(\",\")})\n                AND this.Parent IS NULL\n                ${whereClause ? `AND ${whereClause}` : \"\"}\n                ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n            `,\n            bindings: [...categoryIds.map((id) => ({ type: \"id\", value: id })), ...modelIds.map((id) => ({ type: \"id\", value: id }))] as ECSqlBinding[],\n          },\n        };\n      }),\n    );\n  }\n\n  private async createModelChildrenQuery({\n    parentNodeInstanceIds: modelIds,\n    instanceFilter,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    return Promise.all(\n      getElementsSelectProps().map(async ({ classFullName, whereClause, selectProps }) => {\n        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: classFullName, alias: \"this\" },\n        });\n        return {\n          fullClassName: classFullName,\n          query: {\n            ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: classFullName,\n                  }),\n                },\n                grouping: {\n                  byClass: true,\n                },\n                extendedData: {\n                  imageId: \"icon-item\",\n                },\n                hasChildren: selectProps?.hasChildren,\n                supportsFiltering: selectProps?.supportsFiltering,\n              })}\n            FROM ${instanceFilterClauses.from} this\n            ${instanceFilterClauses.joins}\n            WHERE\n              this.Parent IS NULL\n              AND this.Model.Id IN (${modelIds.map(() => \"?\").join(\",\")})\n              ${whereClause ? `AND ${whereClause}` : \"\"}\n              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n          `,\n            bindings: modelIds.map((id): ECSqlBinding => ({ type: \"id\", value: id })),\n          },\n        };\n      }),\n    );\n  }\n\n  private async createGroupInformationElementChildElementsQuery({\n    parentNodeInstanceIds: groupIds,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    return [\n      {\n        node: {\n          key: \"ChildrenNode\",\n          label: \"Children\",\n          extendedData: {\n            groupIds,\n            imageId: \"icon-hierarchy-tree\",\n          },\n          supportsFiltering: true,\n          processingParams: {\n            hideIfNoChildren: true,\n          },\n        },\n      },\n      {\n        node: {\n          key: \"MembersNode\",\n          label: \"Members\",\n          extendedData: {\n            groupIds,\n            imageId: \"icon-group\",\n          },\n          supportsFiltering: true,\n          processingParams: {\n            hideIfNoChildren: true,\n          },\n        },\n      },\n    ];\n  }\n\n  private async createChildrenNodeChildrenQuery({ parentNode, instanceFilter }: DefineCustomNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const groupIds: string[] = parentNode.extendedData?.groupIds;\n    return Promise.all(\n      getElementsSelectProps().map(async ({ classFullName, whereClause, selectProps }) => {\n        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: classFullName, alias: \"this\" },\n        });\n        return {\n          fullClassName: classFullName,\n          query: {\n            ecsql: `\n              SELECT\n                ${await this._selectQueryFactory.createSelectClause({\n                  ecClassId: { selector: \"this.ECClassId\" },\n                  ecInstanceId: { selector: \"this.ECInstanceId\" },\n                  nodeLabel: {\n                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                      classAlias: \"this\",\n                      className: classFullName,\n                    }),\n                  },\n                  extendedData: {\n                    imageId: \"icon-item\",\n                  },\n                  grouping: { byClass: true },\n                  hasChildren: selectProps.hasChildren,\n                  supportsFiltering: selectProps.supportsFiltering,\n                })}\n              FROM ${instanceFilterClauses.from} this\n              ${instanceFilterClauses.joins}\n              WHERE\n                this.Parent.Id IN (${groupIds.map(() => \"?\").join(\",\")})\n                ${whereClause ? `AND ${whereClause}` : \"\"}\n                ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n            `,\n            bindings: groupIds.map((id) => ({ type: \"id\", value: id })),\n          },\n        };\n      }),\n    );\n  }\n\n  private async createGroupInformationElementMemberElementsQuery({\n    parentNode,\n    instanceFilter,\n  }: DefineCustomNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const groupIds: string[] = parentNode.extendedData?.groupIds;\n    return Promise.all(\n      getElementsSelectProps().map(async ({ classFullName, whereClause, selectProps }) => {\n        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: classFullName, alias: \"this\" },\n        });\n        return {\n          fullClassName: classFullName,\n          query: {\n            ecsql: `\n              SELECT\n                ${await this._selectQueryFactory.createSelectClause({\n                  ecClassId: { selector: \"this.ECClassId\" },\n                  ecInstanceId: { selector: \"this.ECInstanceId\" },\n                  nodeLabel: {\n                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                      classAlias: \"this\",\n                      className: classFullName,\n                    }),\n                  },\n                  grouping: { byClass: true },\n                  extendedData: {\n                    imageId: \"icon-item\",\n                  },\n                  hasChildren: selectProps.hasChildren,\n                  supportsFiltering: selectProps.supportsFiltering,\n                })}\n              FROM ${instanceFilterClauses.from} this\n              JOIN BisCore.ElementGroupsMembers egm ON egm.TargetECInstanceId = this.ECInstanceId\n              ${instanceFilterClauses.joins}\n              WHERE\n                egm.SourceECInstanceId IN (${groupIds.map(() => \"?\").join(\",\")})\n                ${whereClause ? `AND ${whereClause}` : \"\"}\n                ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n            `,\n            bindings: groupIds.map((id) => ({ type: \"id\", value: id })),\n          },\n        };\n      }),\n    );\n  }\n\n  private async createElementChildrenQuery({\n    parentNodeInstanceIds: elementIds,\n    instanceFilter,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    return Promise.all(\n      getElementsSelectProps().map(async ({ classFullName, whereClause, selectProps }) => {\n        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: classFullName, alias: \"this\" },\n        });\n        return {\n          fullClassName: classFullName,\n          query: {\n            ecsql: `\n              SELECT\n                ${await this._selectQueryFactory.createSelectClause({\n                  ecClassId: { selector: \"this.ECClassId\" },\n                  ecInstanceId: { selector: \"this.ECInstanceId\" },\n                  nodeLabel: {\n                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                      classAlias: \"this\",\n                      className: classFullName,\n                    }),\n                  },\n                  grouping: { byClass: true },\n                  extendedData: {\n                    imageId: \"icon-item\",\n                  },\n                  hasChildren: selectProps.hasChildren,\n                  supportsFiltering: selectProps.supportsFiltering,\n                })}\n              FROM ${instanceFilterClauses.from} this\n              JOIN BisCore.Element p ON p.ECInstanceId = this.Parent.Id\n              ${instanceFilterClauses.joins}\n              WHERE\n                p.ECInstanceId IN (${elementIds.map(() => \"?\").join(\",\")})\n                ${whereClause ? `AND ${whereClause}` : \"\"}\n                ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n            `,\n            bindings: elementIds.map((id) => ({ type: \"id\", value: id })),\n          },\n        };\n      }),\n    );\n  }\n}\n\nfunction getClassNameByViewType(view: \"2d\" | \"3d\") {\n  if (view === \"2d\") {\n    return { categoryClass: \"BisCore.DrawingCategory\", elementClass: \"BisCore.GeometricElement2d\", modelClass: \"BisCore.GeometricModel2d\" } as const;\n  }\n  return { categoryClass: \"BisCore.SpatialCategory\", elementClass: \"BisCore.GeometricElement3d\", modelClass: \"BisCore.GeometricModel3d\" } as const;\n}\n\nfunction getElementsSelectProps(props?: { modelClass?: string; elementClass?: \"BisCore.GeometricElement3d\" | \"BisCore.GeometricElement2d\" }) {\n  const modelClassFullName = props?.modelClass ?? \"BisCore.Model\";\n  const elementClassFullName = props?.elementClass ?? \"BisCore.Element\";\n  const result = [\n    {\n      classFullName: elementClassFullName,\n      whereClause: \"this.ECClassId IS NOT (BisCore.GroupInformationElement)\",\n      selectProps: {\n        hasChildren: {\n          selector: `\n          IFNULL((\n            SELECT 1\n            FROM (\n              SELECT Parent.Id ParentId FROM ${elementClassFullName}\n              UNION ALL\n              SELECT sm.ModeledElement.Id ParentId FROM ${modelClassFullName} sm WHERE EXISTS (SELECT 1 FROM ${elementClassFullName} WHERE Model.Id = sm.ECInstanceId)\n            )\n            WHERE ParentId = this.ECInstanceId\n            LIMIT 1\n          ), 0)\n        `,\n        },\n        supportsFiltering: true,\n      },\n    },\n  ];\n\n  if (!props?.elementClass) {\n    result.push({\n      classFullName: \"BisCore.GroupInformationElement\",\n      whereClause: \"\",\n      selectProps: {\n        hasChildren: {\n          selector: `\n            IFNULL((\n              SELECT 1\n              FROM (\n                SELECT Parent.Id ParentId FROM BisCore.Element\n                UNION ALL\n                SELECT SourceECInstanceId ParentId FROM BisCore.ElementGroupsMembers\n              )\n              WHERE ParentId = this.ECInstanceId\n              LIMIT 1\n            ), 0)\n          `,\n        },\n        supportsFiltering: false,\n      },\n    });\n  }\n\n  return result;\n}\n"]}

package/lib/esm/components/trees/imodel-content-tree/internal/IModelContentTreeIdsCache.d.ts

@@ -0,0 +1,26 @@
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+/** @internal */
+export declare class IModelContentTreeIdsCache {
+    private _queryExecutor;
+    private _subjectInfos;
+    private _parentSubjectIds;
+    private _modelInfos;
+    constructor(_queryExecutor: LimitingECSqlQueryExecutor);
+    private querySubjects;
+    private queryModels;
+    private getSubjectInfos;
+    /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */
+    getParentSubjectIds(): Promise<Id64String[]>;
+    /**
+     * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into
+     * account `hideInHierarchy` flag.
+     */
+    getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]>;
+    /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */
+    getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]>;
+    private queryModelCategories;
+    private getModelInfos;
+    getModelCategories(modelIds: Id64String[]): Promise<Id64Array>;
+}
+//# sourceMappingURL=IModelContentTreeIdsCache.d.ts.map

package/lib/esm/components/trees/imodel-content-tree/internal/IModelContentTreeIdsCache.js

@@ -0,0 +1,199 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { assert } from "@itwin/core-bentley";
+import { pushToMap } from "../../common/Utils";
+/** @internal */
+export class IModelContentTreeIdsCache {
+    constructor(_queryExecutor) {
+        this._queryExecutor = _queryExecutor;
+    }
+    async *querySubjects() {
+        const subjectsQuery = `
+      SELECT
+        s.ECInstanceId id,
+        s.Parent.Id parentId,
+        (
+          SELECT m.ECInstanceId
+          FROM bis.Model m
+          WHERE
+            m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))
+            AND NOT m.IsPrivate
+        ) targetPartitionId,
+        CASE
+          WHEN (
+            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL
+            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'
+          ) THEN 1
+          ELSE 0
+        END hideInHierarchy
+      FROM bis.Subject s
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };
+        }
+    }
+    async *queryModels() {
+        const modelsQuery = `
+      SELECT p.ECInstanceId id, p.Parent.Id parentId
+      FROM bis.InformationPartitionElement p
+      INNER JOIN bis.Model m ON m.ModeledElement.Id = p.ECInstanceId
+      WHERE NOT m.IsPrivate
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, parentId: row.parentId };
+        }
+    }
+    async getSubjectInfos() {
+        this._subjectInfos ??= (async () => {
+            const [subjectInfos, targetPartitionSubjects] = await Promise.all([
+                (async () => {
+                    const result = new Map();
+                    for await (const subject of this.querySubjects()) {
+                        const subjectInfo = {
+                            parentSubject: subject.parentId,
+                            hideInHierarchy: subject.hideInHierarchy,
+                            childSubjects: new Set(),
+                            childModels: new Set(),
+                        };
+                        if (subject.targetPartitionId) {
+                            subjectInfo.childModels.add(subject.targetPartitionId);
+                        }
+                        result.set(subject.id, subjectInfo);
+                    }
+                    return result;
+                })(),
+                (async () => {
+                    const result = new Map();
+                    for await (const model of this.queryModels()) {
+                        pushToMap(result, model.id, model.parentId);
+                    }
+                    return result;
+                })(),
+            ]);
+            for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {
+                if (parentSubjectId) {
+                    const parentSubjectInfo = subjectInfos.get(parentSubjectId);
+                    assert(!!parentSubjectInfo);
+                    parentSubjectInfo.childSubjects.add(subjectId);
+                }
+            }
+            for (const [partitionId, subjectIds] of targetPartitionSubjects) {
+                subjectIds.forEach((subjectId) => {
+                    const subjectInfo = subjectInfos.get(subjectId);
+                    assert(!!subjectInfo);
+                    subjectInfo.childModels.add(partitionId);
+                });
+            }
+            return subjectInfos;
+        })();
+        return this._subjectInfos;
+    }
+    /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */
+    async getParentSubjectIds() {
+        this._parentSubjectIds ??= (async () => {
+            const subjectInfos = await this.getSubjectInfos();
+            const parentSubjectIds = new Set();
+            subjectInfos.forEach((subjectInfo, subjectId) => {
+                if (subjectInfo.childModels.size === 0 && subjectInfo.childSubjects.size === 0) {
+                    return;
+                }
+                parentSubjectIds.add(subjectId);
+                let currParentId = subjectInfo.parentSubject;
+                while (currParentId) {
+                    parentSubjectIds.add(currParentId);
+                    currParentId = subjectInfos.get(currParentId)?.parentSubject;
+                }
+            });
+            return [...parentSubjectIds];
+        })();
+        return this._parentSubjectIds;
+    }
+    /**
+     * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into
+     * account `hideInHierarchy` flag.
+     */
+    async getChildSubjectIds(parentSubjectIds) {
+        const childSubjectIds = new Array();
+        const subjectInfos = await this.getSubjectInfos();
+        parentSubjectIds.forEach((subjectId) => {
+            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
+                if (!childSubjectInfo.hideInHierarchy) {
+                    childSubjectIds.push(childSubjectId);
+                    return "break";
+                }
+                return "continue";
+            });
+        });
+        return childSubjectIds;
+    }
+    /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */
+    async getChildSubjectModelIds(parentSubjectIds) {
+        const subjectInfos = await this.getSubjectInfos();
+        const hiddenSubjectIds = new Array();
+        parentSubjectIds.forEach((subjectId) => {
+            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
+                if (childSubjectInfo.hideInHierarchy) {
+                    hiddenSubjectIds.push(childSubjectId);
+                    return "continue";
+                }
+                return "break";
+            });
+        });
+        const modelIds = new Array();
+        [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {
+            const subjectInfo = subjectInfos.get(subjectId);
+            subjectInfo && modelIds.push(...subjectInfo.childModels);
+        });
+        return modelIds;
+    }
+    async *queryModelCategories() {
+        const query = /* sql */ `
+      SELECT Model.Id modelId, Category.Id categoryId
+      FROM BisCore.GeometricElement3d
+      WHERE Parent IS NULL
+      GROUP BY Model.Id, Category.Id
+      UNION ALL
+      SELECT Model.Id modelId, Category.Id categoryId
+      FROM BisCore.GeometricElement2d
+      WHERE Parent IS NULL
+      GROUP BY Model.Id, Category.Id
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { modelId: row.modelId, categoryId: row.categoryId };
+        }
+    }
+    async getModelInfos() {
+        this._modelInfos ??= (async () => {
+            const modelInfos = new Map();
+            for await (const { modelId, categoryId } of this.queryModelCategories()) {
+                const entry = modelInfos.get(modelId);
+                if (entry) {
+                    entry.categories.add(categoryId);
+                }
+                else {
+                    modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });
+                }
+            }
+            return modelInfos;
+        })();
+        return this._modelInfos;
+    }
+    async getModelCategories(modelIds) {
+        const modelInfos = await this.getModelInfos();
+        return modelIds.map((modelId) => modelInfos.get(modelId)?.categories).flatMap((categories) => (categories ? [...categories] : []));
+    }
+}
+function forEachChildSubject(subjectInfos, parentSubject, cb) {
+    const parentSubjectInfo = typeof parentSubject === "string" ? subjectInfos.get(parentSubject) : parentSubject;
+    parentSubjectInfo &&
+        parentSubjectInfo.childSubjects.forEach((childSubjectId) => {
+            const childSubjectInfo = subjectInfos.get(childSubjectId);
+            if (cb(childSubjectId, childSubjectInfo) === "break") {
+                return;
+            }
+            forEachChildSubject(subjectInfos, childSubjectInfo, cb);
+        });
+}
+//# sourceMappingURL=IModelContentTreeIdsCache.js.map

package/lib/esm/components/trees/imodel-content-tree/internal/IModelContentTreeIdsCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelContentTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../src/components/trees/imodel-content-tree/internal/IModelContentTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAgB/C,gBAAgB;AAChB,MAAM,OAAO,yBAAyB;IAKpC,YAAoB,cAA0C;QAA1C,mBAAc,GAAd,cAAc,CAA4B;IAAG,CAAC;IAE1D,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;KAmBrB,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAChJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;SAChI;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;KAKnB,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAC9I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;SAC9C;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;wBAChD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE;4BAC7B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;yBACxD;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;qBACrC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE;wBAC5C,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;qBAC7C;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE;gBACpF,IAAI,eAAe,EAAE;oBACnB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;iBAChD;aACF;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;aACJ;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,IAAI,WAAW,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE;oBAC9E,OAAO;iBACR;gBACD,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;gBAC7C,OAAO,YAAY,EAAE;oBACnB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;iBAC9D;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBACrC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;iBAChB;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE;oBACpC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;iBACnB;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC;;;;;;;;;;KAUvB,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAC;YACxF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;gBACvE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,KAAK,EAAE;oBACT,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;iBAClC;qBAAM;oBACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;iBACjF;aACF;YACD,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,QAAsB;QACpD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACrI,CAAC;CACF;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE;gBACpD,OAAO;aACR;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { assert } from \"@itwin/core-bentley\";\nimport { pushToMap } from \"../../common/Utils\";\n\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  categories: Id64Set;\n}\n\n/** @internal */\nexport class IModelContentTreeIdsCache {\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n\n  constructor(private _queryExecutor: LimitingECSqlQueryExecutor) {}\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.Model m\n          WHERE\n            m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.Model m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE NOT m.IsPrivate\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size === 0 && subjectInfo.childSubjects.size === 0) {\n          return;\n        }\n        parentSubjectIds.add(subjectId);\n        let currParentId = subjectInfo.parentSubject;\n        while (currParentId) {\n          parentSubjectIds.add(currParentId);\n          currParentId = subjectInfos.get(currParentId)?.parentSubject;\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  private async *queryModelCategories() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM BisCore.GeometricElement3d\n      WHERE Parent IS NULL\n      GROUP BY Model.Id, Category.Id\n      UNION ALL\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM BisCore.GeometricElement2d\n      WHERE Parent IS NULL\n      GROUP BY Model.Id, Category.Id\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Id64Set; elementCount: number }>();\n      for await (const { modelId, categoryId } of this.queryModelCategories()) {\n        const entry = modelInfos.get(modelId);\n        if (entry) {\n          entry.categories.add(categoryId);\n        } else {\n          modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });\n        }\n      }\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelIds: Id64String[]): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    return modelIds.map((modelId) => modelInfos.get(modelId)?.categories).flatMap((categories) => (categories ? [...categories] : []));\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n"]}

package/lib/esm/components/trees/index.d.ts

@@ -1,20 +1,5 @@
-export * from "./common/Types";
-export * from "./common/ContextMenu";
-export * from "./common/TreeRenderer";
-export * from "./common/TreeNodeRenderer";
-export * from "./VisibilityTreeEventHandler";
-export * from "./VisibilityTreeRenderer";
-export * from "./imodel-content-tree/IModelContentTree";
-export * from "./imodel-content-tree/IModelContentTreeComponent";
-export { CategoryInfo, CategoriesTreeHeaderButtonProps } from "./category-tree/CategoriesTreeButtons";
-export * from "./category-tree/CategoriesTree";
-export * from "./category-tree/CategoriesTreeComponent";
-export * from "./category-tree/CategoryVisibilityHandler";
-export * from "./external-sources-tree/ExternalSourcesTree";
-export * from "./external-sources-tree/ExternalSourcesTreeComponent";
-export { ModelInfo, ModelsTreeHeaderButtonProps } from "./models-tree/ModelsTreeButtons";
-export * from "./models-tree/ModelsTree";
-export * from "./models-tree/ModelsTreeComponent";
-export * from "./models-tree/ModelsVisibilityHandler";
-export * from "./stateless";
+export { ModelsTreeComponent } from "./models-tree/ModelsTreeComponent";
+export { CategoriesTreeComponent } from "./categories-tree/CategoriesTreeComponent";
+export { IModelContentTreeComponent } from "./imodel-content-tree/IModelContentTreeComponent";
+export { ExternalSourcesTreeComponent } from "./external-sources-tree/ExternalSourcesTreeComponent";
 //# sourceMappingURL=index.d.ts.map

package/lib/esm/components/trees/index.js

@@ -2,21 +2,8 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-export * from "./common/Types";
-export * from "./common/ContextMenu";
-export * from "./common/TreeRenderer";
-export * from "./common/TreeNodeRenderer";
-export * from "./VisibilityTreeEventHandler";
-export * from "./VisibilityTreeRenderer";
-export * from "./imodel-content-tree/IModelContentTree";
-export * from "./imodel-content-tree/IModelContentTreeComponent";
-export * from "./category-tree/CategoriesTree";
-export * from "./category-tree/CategoriesTreeComponent";
-export * from "./category-tree/CategoryVisibilityHandler";
-export * from "./external-sources-tree/ExternalSourcesTree";
-export * from "./external-sources-tree/ExternalSourcesTreeComponent";
-export * from "./models-tree/ModelsTree";
-export * from "./models-tree/ModelsTreeComponent";
-export * from "./models-tree/ModelsVisibilityHandler";
-export * from "./stateless";
+export { ModelsTreeComponent } from "./models-tree/ModelsTreeComponent";
+export { CategoriesTreeComponent } from "./categories-tree/CategoriesTreeComponent";
+export { IModelContentTreeComponent } from "./imodel-content-tree/IModelContentTreeComponent";
+export { ExternalSourcesTreeComponent } from "./external-sources-tree/ExternalSourcesTreeComponent";
 //# sourceMappingURL=index.js.map

package/lib/esm/components/trees/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/components/trees/index.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,cAAc,gBAAgB,CAAC;AAC/B,cAAc,sBAAsB,CAAC;AACrC,cAAc,uBAAuB,CAAC;AACtC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,0BAA0B,CAAC;AAEzC,cAAc,yCAAyC,CAAC;AACxD,cAAc,kDAAkD,CAAC;AAGjE,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yCAAyC,CAAC;AACxD,cAAc,2CAA2C,CAAC;AAE1D,cAAc,6CAA6C,CAAC;AAC5D,cAAc,sDAAsD,CAAC;AAGrE,cAAc,0BAA0B,CAAC;AACzC,cAAc,mCAAmC,CAAC;AAClD,cAAc,uCAAuC,CAAC;AAEtD,cAAc,aAAa,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nexport * from \"./common/Types\";\nexport * from \"./common/ContextMenu\";\nexport * from \"./common/TreeRenderer\";\nexport * from \"./common/TreeNodeRenderer\";\nexport * from \"./VisibilityTreeEventHandler\";\nexport * from \"./VisibilityTreeRenderer\";\n\nexport * from \"./imodel-content-tree/IModelContentTree\";\nexport * from \"./imodel-content-tree/IModelContentTreeComponent\";\n\nexport { CategoryInfo, CategoriesTreeHeaderButtonProps } from \"./category-tree/CategoriesTreeButtons\";\nexport * from \"./category-tree/CategoriesTree\";\nexport * from \"./category-tree/CategoriesTreeComponent\";\nexport * from \"./category-tree/CategoryVisibilityHandler\";\n\nexport * from \"./external-sources-tree/ExternalSourcesTree\";\nexport * from \"./external-sources-tree/ExternalSourcesTreeComponent\";\n\nexport { ModelInfo, ModelsTreeHeaderButtonProps } from \"./models-tree/ModelsTreeButtons\";\nexport * from \"./models-tree/ModelsTree\";\nexport * from \"./models-tree/ModelsTreeComponent\";\nexport * from \"./models-tree/ModelsVisibilityHandler\";\n\nexport * from \"./stateless\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/components/trees/index.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AACxE,OAAO,EAAE,uBAAuB,EAAE,MAAM,2CAA2C,CAAC;AACpF,OAAO,EAAE,0BAA0B,EAAE,MAAM,kDAAkD,CAAC;AAC9F,OAAO,EAAE,4BAA4B,EAAE,MAAM,sDAAsD,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nexport { ModelsTreeComponent } from \"./models-tree/ModelsTreeComponent\";\nexport { CategoriesTreeComponent } from \"./categories-tree/CategoriesTreeComponent\";\nexport { IModelContentTreeComponent } from \"./imodel-content-tree/IModelContentTreeComponent\";\nexport { ExternalSourcesTreeComponent } from \"./external-sources-tree/ExternalSourcesTreeComponent\";\n"]}

package/lib/esm/components/trees/models-tree/ModelsTree.d.ts

@@ -1,77 +1,22 @@
-/// <reference types="react" />
-import "../VisibilityTreeBase.scss";
-import { ClassGroupingOption } from "../common/Types";
-import { ModelsVisibilityHandler } from "./ModelsVisibilityHandler";
-import type { SingleSchemaClassSpecification } from "@itwin/presentation-common";
+import { VisibilityTree } from "../common/components/VisibilityTree";
+import { ModelsTreeDefinition } from "./ModelsTreeDefinition";
+import type { ComponentPropsWithoutRef } from "react";
 import type { Viewport } from "@itwin/core-frontend";
-import type { BaseFilterableTreeProps, HierarchyLevelConfig } from "../common/Types";
-import type { ModelsTreeSelectionPredicate, ModelsVisibilityHandlerProps } from "./ModelsVisibilityHandler";
-/**
- * Props for configuring the hierarchy in [[ModelsTree]].
- * @public
- */
-export interface ModelsTreeHierarchyConfiguration {
-    /** Should the tree group displayed element nodes by class. Defaults to `ClassGroupingOption.No`. */
-    enableElementsClassGrouping?: ClassGroupingOption;
-    /**
-     * Defines the `bis.GeometricElement3d` sub-class that should be used to load element nodes.
-     * Defaults to `bis.GeometricElement3d`. It's expected for the given class to derive from it.
-     */
-    elementClassSpecification?: SingleSchemaClassSpecification;
-    /** Should the tree show models without elements. */
-    showEmptyModels?: boolean;
-}
-/**
- * Props for [[ModelsTree]] component.
- * @public
- */
-export interface ModelsTreeProps extends BaseFilterableTreeProps {
-    /**
-     * Predicate which indicates whether node can be selected or no
-     */
-    selectionPredicate?: ModelsTreeSelectionPredicate;
-    /**
-     * Active view used to determine and control visibility
-     */
+interface ModelsTreeOwnProps {
     activeView: Viewport;
-    /**
-     * Configuration options for the hierarchy loaded in the component.
-     */
-    hierarchyConfig?: ModelsTreeHierarchyConfiguration;
-    /**
-     * Auto-update the hierarchy when data in the iModel changes.
-     * @alpha
-     * @deprecated in 2.0.1. It does not have any effect, auto update is always on.
-     */
-    enableHierarchyAutoUpdate?: boolean;
-    /**
-     * Custom visibility handler.
-     */
-    modelsVisibilityHandler?: ModelsVisibilityHandler | ((props: ModelsVisibilityHandlerProps) => ModelsVisibilityHandler);
-    /**
-     * Props for configuring hierarchy level.
-     * @beta
-     */
-    hierarchyLevelConfig?: HierarchyLevelConfig;
-    /**
-     * Reports performance of a feature.
-     * @param featureId ID of the feature.
-     * @param elapsedTime Elapsed time of the feature.
-     * @beta
-     */
-    onPerformanceMeasured?: (featureId: string, elapsedTime: number) => void;
-    /**
-     * Callback that is invoked when a tracked feature is used.
-     * @param featureId ID of the feature.
-     * @beta
-     */
+    hierarchyLevelConfig?: {
+        sizeLimit?: number;
+    };
+    filter?: string;
+    onPerformanceMeasured?: (featureId: string, duration: number) => void;
     onFeatureUsed?: (feature: string) => void;
 }
-/**
- * A tree component that shows a subject - model - category - element
- * hierarchy along with checkboxes that represent and allow changing
- * the display of those instances.
- * @public
- */
-export declare function ModelsTree(props: ModelsTreeProps): JSX.Element | null;
+type VisibilityTreeProps = ComponentPropsWithoutRef<typeof VisibilityTree>;
+type ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0]["hierarchyConfig"];
+type ModelsTreeProps = ModelsTreeOwnProps & Pick<VisibilityTreeProps, "imodel" | "getSchemaContext" | "height" | "width" | "density" | "selectionMode"> & {
+    hierarchyConfig?: Partial<ModelsTreeHierarchyConfiguration>;
+};
+/** @internal */
+export declare function ModelsTree({ imodel, getSchemaContext, height, width, activeView, filter, density, hierarchyLevelConfig, hierarchyConfig, selectionMode, onPerformanceMeasured, onFeatureUsed, }: ModelsTreeProps): JSX.Element;
+export {};
 //# sourceMappingURL=ModelsTree.d.ts.map