Class TObjectsGroupingSubset

catboost/libs/data/objects_grouping.h:125–195

    using TObjectsGroupingPtr = TIntrusivePtr<TObjectsGrouping>;

    class TObjectsGroupingSubset {
    public:
        // needed because of default init in Cython
        TObjectsGroupingSubset() = default;

        TObjectsGroupingSubset(
            TObjectsGroupingPtr subsetGrouping,
            TArraySubsetIndexing<ui32>&& groupsSubset,
            EObjectsOrder groupSubsetOrder,
            TMaybe<TArraySubsetIndexing<ui32>>&& objectsSubsetForNonTrivialGrouping
                = TMaybe<TArraySubsetIndexing<ui32>>(),

            // used only if objectsSubsetForNonTrivialGrouping is specified
            EObjectsOrder objectSubsetOrder = EObjectsOrder::Undefined
        )
            : SubsetGrouping(std::move(subsetGrouping))
            , GroupsSubset(std::move(groupsSubset))
            , GroupSubsetOrder(groupSubsetOrder)
            , ObjectsSubsetForNonTrivialGrouping(std::move(objectsSubsetForNonTrivialGrouping))
            , ObjectSubsetOrder(objectSubsetOrder)
        {
            CB_ENSURE(SubsetGrouping, "subsetGrouping must be initialized");
        }

        bool operator==(const TObjectsGroupingSubset& rhs) const {
            if (!(*SubsetGrouping == *rhs.SubsetGrouping) ||
                (GroupsSubset != rhs.GroupsSubset))
            {
                return false;
            }
            if (ObjectsSubsetForNonTrivialGrouping) {
                if (rhs.ObjectsSubsetForNonTrivialGrouping) {
                    return *ObjectsSubsetForNonTrivialGrouping == *rhs.ObjectsSubsetForNonTrivialGrouping;
                }
                return false;
            }
            return !rhs.ObjectsSubsetForNonTrivialGrouping;
        }

        TObjectsGroupingPtr GetSubsetGrouping() const {
            return SubsetGrouping;
        }

        const TArraySubsetIndexing<ui32>& GetGroupsIndexing() const {
            return GroupsSubset;
        }

        const TArraySubsetIndexing<ui32>& GetObjectsIndexing() const {
            return ObjectsSubsetForNonTrivialGrouping ?
                *ObjectsSubsetForNonTrivialGrouping : GroupsSubset;
        }

        EObjectsOrder GetGroupSubsetOrder() const {
            return GroupSubsetOrder;
        }

        EObjectsOrder GetObjectSubsetOrder() const {
            return ObjectsSubsetForNonTrivialGrouping ? GroupSubsetOrder : ObjectSubsetOrder;