McRogueFace/src/PyHeightMap.h

#pragma once
#include "Common.h"
#include "Python.h"
#include <libtcod.h>

// Forward declaration
class PyHeightMap;

// Python object structure
typedef struct {
    PyObject_HEAD
    TCOD_heightmap_t* heightmap;  // libtcod heightmap pointer
} PyHeightMapObject;

class PyHeightMap
{
public:
    // Python type interface
    static PyObject* pynew(PyTypeObject* type, PyObject* args, PyObject* kwds);
    static int init(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static void dealloc(PyHeightMapObject* self);
    static PyObject* repr(PyObject* obj);

    // Properties
    static PyObject* get_size(PyHeightMapObject* self, void* closure);

    // Scalar operations (all return self for chaining, support region parameters)
    static PyObject* fill(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* clear(PyHeightMapObject* self, PyObject* Py_UNUSED(args));
    static PyObject* add_constant(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* scale(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* clamp(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* normalize(PyHeightMapObject* self, PyObject* args, PyObject* kwds);

    // Query methods (#196)
    static PyObject* get(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* get_interpolated(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* get_slope(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* get_normal(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* min_max(PyHeightMapObject* self, PyObject* Py_UNUSED(args));
    static PyObject* count_in_range(PyHeightMapObject* self, PyObject* args);

    // Threshold operations (#197) - return NEW HeightMaps
    static PyObject* threshold(PyHeightMapObject* self, PyObject* args);
    static PyObject* threshold_binary(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* inverse(PyHeightMapObject* self, PyObject* Py_UNUSED(args));

    // Terrain generation methods (#195) - mutate self, return self for chaining
    static PyObject* add_hill(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* dig_hill(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* add_voronoi(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* mid_point_displacement(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* rain_erosion(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* dig_bezier(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* smooth(PyHeightMapObject* self, PyObject* args, PyObject* kwds);

    // Correct convolution methods (using new libtcod functions)
    static PyObject* sparse_kernel(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* sparse_kernel_from(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    // NOTE: gradients waiting for jmccardle:feature/heightmap-gradients to be merged into libtcod:main
    // static PyObject* gradients(PyHeightMapObject* self, PyObject* args, PyObject* kwds);

    // Subscript support for hmap[x, y] syntax
    static PyObject* subscript(PyHeightMapObject* self, PyObject* key);
    static int subscript_assign(PyHeightMapObject* self, PyObject* key, PyObject* value);

    // Combination operations (#194) - mutate self, return self for chaining, support region parameters
    static PyObject* add(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* subtract(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* multiply(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* lerp(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* copy_from(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* hmap_max(PyHeightMapObject* self, PyObject* args, PyObject* kwds);  // 'max' conflicts with macro
    static PyObject* hmap_min(PyHeightMapObject* self, PyObject* args, PyObject* kwds);  // 'min' conflicts with macro

    // Direct source sampling (#209) - sample from NoiseSource/BSP directly
    static PyObject* add_noise(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* multiply_noise(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* add_bsp(PyHeightMapObject* self, PyObject* args, PyObject* kwds);
    static PyObject* multiply_bsp(PyHeightMapObject* self, PyObject* args, PyObject* kwds);

    // Mapping methods for subscript support
    static PyMappingMethods mapping_methods;

    // Method and property definitions
    static PyMethodDef methods[];
    static PyGetSetDef getsetters[];
};

namespace mcrfpydef {
    inline PyTypeObject PyHeightMapType = {
        .ob_base = {.ob_base = {.ob_refcnt = 1, .ob_type = NULL}, .ob_size = 0},
        .tp_name = "mcrfpy.HeightMap",
        .tp_basicsize = sizeof(PyHeightMapObject),
        .tp_itemsize = 0,
        .tp_dealloc = (destructor)PyHeightMap::dealloc,
        .tp_repr = PyHeightMap::repr,
        .tp_as_mapping = &PyHeightMap::mapping_methods,  // hmap[x, y] subscript
        .tp_flags = Py_TPFLAGS_DEFAULT,
        .tp_doc = PyDoc_STR(
            "HeightMap(size: tuple[int, int], fill: float = 0.0)\n\n"
            "A 2D grid of float values for procedural generation.\n\n"
            "HeightMap is the universal canvas for procedural generation. It stores "
            "float values that can be manipulated, combined, and applied to Grid and "
            "Layer objects.\n\n"
            "Args:\n"
            "    size: (width, height) dimensions of the heightmap. Immutable after creation.\n"
            "    fill: Initial value for all cells. Default 0.0.\n\n"
            "Example:\n"
            "    hmap = mcrfpy.HeightMap((100, 100))\n"
            "    hmap.fill(0.5).scale(2.0).clamp(0.0, 1.0)\n"
            "    value = hmap[5, 5]  # Subscript shorthand for get()\n"
        ),
        .tp_methods = nullptr,  // Set in McRFPy_API.cpp before PyType_Ready
        .tp_getset = nullptr,   // Set in McRFPy_API.cpp before PyType_Ready
        .tp_init = (initproc)PyHeightMap::init,
        .tp_new = PyHeightMap::pynew,
    };
}