Function setup_surface

src/api/c/surface.cpp:47–144

template<typename T>
fg_chart setup_surface(fg_window window, const af_array xVals,
                       const af_array yVals, const af_array zVals,
                       const af_cell* const props) {
    ForgeModule& _ = forgePlugin();
    Array<T> xIn   = getArray<T>(xVals);
    Array<T> yIn   = getArray<T>(yVals);
    Array<T> zIn   = getArray<T>(zVals);

    const ArrayInfo& Xinfo = getInfo(xVals);
    const ArrayInfo& Yinfo = getInfo(yVals);
    const ArrayInfo& Zinfo = getInfo(zVals);

    dim4 X_dims = Xinfo.dims();
    dim4 Y_dims = Yinfo.dims();
    dim4 Z_dims = Zinfo.dims();

    if (Xinfo.isVector()) {
        // Convert xIn is a column vector
        xIn = modDims(xIn, xIn.elements());
        // Now tile along second dimension
        dim4 x_tdims(1, Y_dims[0], 1, 1);
        xIn = arrayfire::common::tile(xIn, x_tdims);

        // Convert yIn to a row vector
        yIn = modDims(yIn, dim4(1, yIn.elements()));
        // Now tile along first dimension
        dim4 y_tdims(X_dims[0], 1, 1, 1);
        yIn = arrayfire::common::tile(yIn, y_tdims);
    }

    // Flatten xIn, yIn and zIn into row vectors
    dim4 rowDims = dim4(1, zIn.elements());
    xIn          = modDims(xIn, rowDims);
    yIn          = modDims(yIn, rowDims);
    zIn          = modDims(zIn, rowDims);

    // Now join along first dimension, skip reorder
    std::vector<Array<T>> inputs{xIn, yIn, zIn};

    dim4 odims(3, rowDims[1]);
    Array<T> out = createEmptyArray<T>(odims);
    join(out, 0, inputs);
    Array<T> Z = out;

    ForgeManager& fgMngr = forgeManager();

    // Get the chart for the current grid position (if any)
    fg_chart chart = NULL;
    if (props->col > -1 && props->row > -1) {
        chart = fgMngr.getChart(window, props->row, props->col, FG_CHART_3D);
    } else {
        chart = fgMngr.getChart(window, 0, 0, FG_CHART_3D);
    }

    fg_surface surface =
        fgMngr.getSurface(chart, Z_dims[0], Z_dims[1], getGLType<T>());

    FG_CHECK(_.fg_set_surface_color(surface, 0.0, 1.0, 0.0, 1.0));