/*
 * Grid transfer operators
 * Copyright 2018 Anton Khirnov <anton@khirnov.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MG2D_TRANSFER_H
#define MG2D_TRANSFER_H

#include <stddef.h>
#include <threadpool.h>

#include "ndarray.h"

enum GridTransferOperator {
    GRID_TRANSFER_LAGRANGE_1,
    GRID_TRANSFER_LAGRANGE_3,
    GRID_TRANSFER_LAGRANGE_5,
    GRID_TRANSFER_LAGRANGE_7,
    GRID_TRANSFER_FW_1,
    GRID_TRANSFER_FW_3,
    GRID_TRANSFER_FW_5,
};

typedef struct RegularGrid2D {
    size_t size[2];
    double step[2];
} RegularGrid2D;

typedef struct GridTransferContext {
    void *priv;

    TPContext *tp;
    int cpuflags;

    enum GridTransferOperator op;

    RegularGrid2D src;
    RegularGrid2D dst;
} GridTransferContext;

GridTransferContext *mg2di_gt_alloc(enum GridTransferOperator op);
int mg2di_gt_init(GridTransferContext *ctx);

void mg2di_gt_free(GridTransferContext **ctx);

int mg2di_gt_transfer(GridTransferContext *ctx,
                      NDArray *dst, const NDArray *src);

#endif // MG2D_TRANSFER_H