Visualize with Lambdas

lambdaVisualize.cpp

#include <iostream>
#include <functional>

class ConstitutiveLaw {};

class ElementBase
{
public:
    ElementBase(ConstitutiveLaw* law) : mLaw(law) {}

    ConstitutiveLaw* GetConstitutiveLawPtr()
    {
        return mLaw;
    }
private:
    ConstitutiveLaw* mLaw;
};

class Visualize
{
public:
    static void VisualizeElementFunction(ElementBase& element, std::function<int(ElementBase&)> function,
            std::string paraviewName)
    {
        std::cout << paraviewName << ": " << function(element) << std::endl;
    }
};

int main()
{
    ConstitutiveLaw* heatConduction = new ConstitutiveLaw;
    ConstitutiveLaw* linearElastic = new ConstitutiveLaw;

    std::function<int(ElementBase&)> integerLaws = [=](ElementBase& element)
    {
        return heatConduction == element.GetConstitutiveLawPtr() ? 1 : 0;
    };

    ElementBase elementOne(heatConduction);
    ElementBase elementTwo(linearElastic);

    Visualize::VisualizeElementFunction(elementOne, integerLaws, "ConstitutiveLaw");
    Visualize::VisualizeElementFunction(elementTwo, integerLaws, "ConstitutiveLaw");
}