Implement shifting and resizing

vladtcvs · Dec 15, 2024 · b28730f · b28730f
1 parent 7f9fa9f
commit b28730f
Show file tree

Hide file tree

Showing 6 changed files with 224 additions and 75 deletions.
diff --git a/src/c_modules/movements/lib/sphere.c b/src/c_modules/movements/lib/sphere.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023 Vladislav Tsendrovskii
+ * Copyright (c) 2023-2024 Vladislav Tsendrovskii
  *
  * 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
@@ -90,6 +90,34 @@ void sphere_movement_apply_forward(struct SphereMovement *mov,
     }
 }
 
+void sphere_movement_apply_forward_lonlat(struct SphereMovement *mov,
+                                          const double *posi, double *poso, size_t num)
+{
+    unsigned i;
+    for (i = 0; i < num; i++)
+    {
+        double x, y, z;
+        double lon = posi[i*2];
+        double lat = posi[i*2+1];
+        x = cos(lon)*cos(lat);
+        y = sin(lon)*cos(lat);
+        z = sin(lat);
+        struct quat v = quat_from_vec(x, y, z);
+        struct quat vf = quat_mul(mov->forward, quat_mul(v, mov->reverse));
+        x = vf.x;
+        y = vf.y;
+        z = vf.z;
+        if (z > 1)
+            z = 1;
+        if (z < -1)
+            z = -1;
+        lon = atan2(y, x);
+        lat = asin(z);
+        poso[2*i] = lon;
+        poso[2*i+1] = lat;
+    }
+}
+
 void sphere_movement_apply_reverse(struct SphereMovement *mov,
                                    const double *posi, double *poso, size_t num,
                                    const struct ProjectionDef *in_proj,
@@ -123,3 +151,31 @@ void sphere_movement_apply_reverse(struct SphereMovement *mov,
         poso[2*i+1] = yo;
     }
 }
+
+void sphere_movement_apply_reverse_lonlat(struct SphereMovement *mov,
+                                          const double *posi, double *poso, size_t num)
+{
+    unsigned i;
+    for (i = 0; i < num; i++)
+    {
+        double x, y, z;
+        double lon = posi[i*2];
+        double lat = posi[i*2+1];
+        x = cos(lon)*cos(lat);
+        y = sin(lon)*cos(lat);
+        z = sin(lat);
+        struct quat v = quat_from_vec(x, y, z);
+        struct quat vf = quat_mul(mov->reverse, quat_mul(v, mov->forward));
+        x = vf.x;
+        y = vf.y;
+        z = vf.z;
+        if (z > 1)
+            z = 1;
+        if (z < -1)
+            z = -1;
+        lon = atan2(y, x);
+        lat = asin(z);
+        poso[2*i] = lon;
+        poso[2*i+1] = lat;
+    }
+}
diff --git a/src/c_modules/movements/lib/sphere.h b/src/c_modules/movements/lib/sphere.h
@@ -41,7 +41,13 @@ void sphere_movement_apply_forward(struct SphereMovement *mov,
                                    const struct ProjectionDef *in_proj,
                                    const struct ProjectionDef *out_proj);
 
+void sphere_movement_apply_forward_lonlat(struct SphereMovement *mov,
+                                          const double *posi, double *poso, size_t num);
+
 void sphere_movement_apply_reverse(struct SphereMovement *mov,
                                    const double *posi, double *poso, size_t num,
                                    const struct ProjectionDef *in_proj,
                                    const struct ProjectionDef *out_proj);
+
+void sphere_movement_apply_reverse_lonlat(struct SphereMovement *mov,
+                                          const double *posi, double *poso, size_t num);
diff --git a/src/c_modules/movements/module.c b/src/c_modules/movements/module.c
@@ -111,6 +111,10 @@ typedef void (*action_f)(struct SphereMovement *mov,
                          const struct ProjectionDef *in_proj,
                          const struct ProjectionDef *out_proj);
 
+typedef void (*action_lonlat_f)(struct SphereMovement *mov,
+                                const double *posi, double *poso, size_t num);
+
+
 static PyObject *apply_action(PyObject *_self,
                               PyObject *args,
                               PyObject *kwds,
@@ -187,25 +191,100 @@ static PyObject *apply_action(PyObject *_self,
     return (PyObject *)output_points;
 }
 
+static PyObject *apply_action_lonlat(PyObject *_self,
+                                     PyObject *args,
+                                     PyObject *kwds,
+                                     action_lonlat_f fun)
+{
+    PyArrayObject *points;
+    struct SphereMovementObject *self = (struct SphereMovementObject *)_self;
+
+    static char *kwlist[] = {"points", NULL};
+    if (!PyArg_ParseTupleAndKeywords(args, kwds, "O", kwlist, &points))
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid function arguments");
+        Py_INCREF(Py_None);
+        return Py_None;
+    }
+
+    if (PyArray_TYPE(points) != NPY_DOUBLE)
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid function arguments - should be dtype == double");
+        Py_INCREF(Py_None);
+        return Py_None;
+    }
+
+    if (PyArray_NDIM(points) != 2)
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid function arguments - should be len(shape) == 2");
+        Py_INCREF(Py_None);
+        return Py_None;
+    }
+
+    npy_intp *dims = PyArray_SHAPE(points);
+
+    if (dims[1] != 2)
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid function arguments - should be shape[1] == 2");
+        Py_INCREF(Py_None);
+        return Py_None;
+    }
+
+    size_t num = dims[0];
+    //printf("Processing %i points\n", (int)num);
+
+    const double *posi = PyArray_DATA(points);
+
+    PyArrayObject *output_points = (PyArrayObject *)PyArray_ZEROS(2, dims, NPY_DOUBLE, 0);
+    if (output_points == NULL)
+    {
+        PyErr_SetString(PyExc_ValueError, "can not allocate memory");
+        Py_INCREF(Py_None);
+        return Py_None;
+    }
+    double *poso = PyArray_DATA(output_points);
+
+    fun(&self->mov, posi, poso, num);
+    return (PyObject *)output_points;
+}
+
 static PyObject *Sphere_forward(PyObject *_self,
                                 PyObject *args,
                                 PyObject *kwds)
 {
     return apply_action(_self, args, kwds, sphere_movement_apply_forward);
 }
 
+static PyObject *Sphere_forward_lonlat(PyObject *_self,
+                                       PyObject *args,
+                                       PyObject *kwds)
+{
+    return apply_action_lonlat(_self, args, kwds, sphere_movement_apply_forward_lonlat);
+}
+
 static PyObject *Sphere_reverse(PyObject *_self,
                                 PyObject *args,
                                 PyObject *kwds)
 {
    return apply_action(_self, args, kwds, sphere_movement_apply_reverse);
 }
 
+static PyObject *Sphere_reverse_lonlat(PyObject *_self,
+                                       PyObject *args,
+                                       PyObject *kwds)
+{
+   return apply_action_lonlat(_self, args, kwds, sphere_movement_apply_reverse_lonlat);
+}
+
 static PyMethodDef _SphereMovements_methods[] = {
     {"apply_forward", (PyCFunction)Sphere_forward, METH_VARARGS | METH_KEYWORDS,
      "Apply forward rotation"},
+    {"apply_forward_lonlat", (PyCFunction)Sphere_forward_lonlat, METH_VARARGS | METH_KEYWORDS,
+     "Apply forward rotation in lon, lat coordinates"},
     {"apply_reverse", (PyCFunction)Sphere_reverse, METH_VARARGS | METH_KEYWORDS,
      "Apply reverse rotation"},
+    {"apply_reverse_lonlat", (PyCFunction)Sphere_reverse_lonlat, METH_VARARGS | METH_KEYWORDS,
+     "Apply reverse rotation in lon, lat coordinates"},
     {NULL} /* Sentinel */
 };
 

diff --git a/src/c_modules/projections/module.c b/src/c_modules/projections/module.c
@@ -136,7 +136,7 @@ static PyObject *Perspective_reverse(PyObject *_self,
 
 static PyMethodDef _PerspectiveProjection_methods[] = {
     {"project", (PyCFunction)Perspective_forward, METH_VARARGS | METH_KEYWORDS,
-     "Project x,y to lat,lon"},
+     "Project x,y to lon,lat"},
     {"reverse", (PyCFunction)Perspective_reverse, METH_VARARGS | METH_KEYWORDS,
      "Project lon,lat to x,y"},
     {NULL} /* Sentinel */

diff --git a/src/vstarstack/tool/config.py b/src/vstarstack/tool/config.py
@@ -21,7 +21,8 @@
 
 _module_configuration = {
     "output" : {
-        "projection" : (str, "COPY"),
+        "projection_type" : (str, "COPY"),
+        "projection_desc" : (str, "COPY"),
     },
     "paths": {
         "light" : {