Method classCounts

test/unit/filters/GroundFilterTest.cpp:49–92

{
    protected:
        std::unordered_map<uint8_t, size_t> classCounts(
            const std::string filterTypeName,
            const uint8_t groundClass,
            const uint8_t otherClass,
            const uint8_t startingClass,
            const bool onlyGround
        )
        {
            StageFactory f;

            Stage* reader(f.createStage("readers.las"));
            Options rOpts;
            rOpts.add("filename", Support::datapath("las/warsaw_small.las"));
            reader->setOptions(rOpts);

            Stage* assign(f.createStage("filters.assign"));
            Options aOpts;
            aOpts.add("assignment", "Classification[:]=" + std::to_string(startingClass));
            assign->setInput(*reader);
            assign->setOptions(aOpts);

            const std::string filterType = "filters." + filterTypeName;
            Stage* filter(f.createStage(filterType));
            Options fOpts;
            fOpts.add("ground_class", groundClass);
            fOpts.add("other_class", otherClass);
            fOpts.add("only_ground", onlyGround);
            filter->setInput(*assign);
            filter->setOptions(fOpts);

            PointTable t;
            filter->prepare(t);
            PointViewSet s = filter->execute(t);

            PointViewPtr v = *s.begin();

            std::unordered_map<uint8_t, size_t> classCounts;
            for (PointId id = 0; id < v->size(); ++id)
            {
                uint8_t cl = v->getFieldAs<uint8_t>(Dimension::Id::Classification, id);
                classCounts[cl]++;
            }
            return classCounts;
        }
};

TEST_P(CustomGroundClassTest, CustomGroundClass)