genvecwhere.cpp

A C++ class library for scientific computing

#include "bzfstream.h"
#include "optuple.h"

int main()
{
    std::cout << "Generating <vecwhere.cc>" << std::endl;

    bzofstream ofs("../vecwhere.cc", "where(X,Y,Z) function for vectors",
        __FILE__, "BZ_VECWHERE_CC");

    ofs.beginNamespace();

    OperandTuple ops(3);

    do {
        if (ops[0].isScalar())
            continue;

        if (ops[1].isScalar() && ops[2].isScalar())
        {
            if (ops.operandIndex(1) != ops.operandIndex(2))
                continue;
        }

        ofs << "// where(";
        ops.printTypes(ofs);
        ofs << ")" << std::endl;

        int complexFlag = 0;
        if (ops.anyComplex())
        {
            ofs << "#ifdef BZ_HAVE_COMPLEX" << std::endl;
            complexFlag = 1;
        }

        ops.printTemplates(ofs);
        ofs << std::endl << "inline" << std::endl;
        ofs << "_bz_VecExpr<_bz_VecWhere<";
        ops.printIterators(ofs, 1);
        ofs << " > >" << std::endl;

        ofs << "where(";
        ops.printArgumentList(ofs, 1);
        ofs << ")" << std::endl
            << "{ " << std::endl;

        ofs << "    typedef _bz_VecWhere<";
        ops.printIterators(ofs, 1);
        ofs << " > T_expr;" << std::endl << std::endl;

        ofs << "    return _bz_VecExpr<T_expr>(T_expr(";
        ops.printInitializationList(ofs, 1);
        ofs << "));" << std::endl
            << "}" << std::endl;
      
        if (complexFlag)
            ofs << "#endif // BZ_HAVE_COMPLEX" << std::endl;

        ofs << std::endl;
    } while (++ops);

    std::cout << ops.numSpecializations() << " specializations written." << std::endl;
}