Hexlaser paper fix fig S11 axis label crop

Former-commit-id: d8cb511a77fd47a9a4dcfc7bbe36381adc79c7e3

qpms/latticegens.c

@@ -1,4 +1,7 @@
 #include "lattices.h"
+#include <limits.h>
+#include <math.h>
+
 // here, various "classes" of the PGenSph point generators are implemented.
 
 
@@ -107,3 +110,104 @@ const PGenSphClassInfo PGenSph_FromPoint2DArray = {
   PGenSph_FromPoint2DArray_destructor,
 };
 
+
+
+//==== PGenSph_zAxis ====
+//equidistant points along the z-axis;
+
+extern const PGenSphClassInfo PGenSph_zAxis; // forward declaration needed by constructor (may be placed in header file instead)
+
+enum PGenSph_zAxis_incrementDirection{
+    //PGENSPH_ZAXIS_POSITIVE_INC, // not implemented
+    //PGENSPH_ZAXIS_NEGATIVE_INC, // not implemented
+    PGENSPH_ZAXIS_INC_FROM_ORIGIN,
+    PGENSPH_ZAXIS_INC_TOWARDS_ORIGIN
+};    
+
+// Internal state structure
+typedef struct PGenSph_zAxis_StateData {
+  long ptindex;
+  long stopindex;
+  double minR, maxR;
+  bool inc_minR, inc_maxR;
+  double a; // lattice period
+  enum PGenSph_zAxis_incrementDirection incdir;
+  bool skip_origin;
+} PGenSph_zAxis_StateData;
+
+// Constructor, specified by maximum and maximum absolute value
+PGenSph PGenSph_zAxis_new_minMaxR(double period, double offset, double minR, bool inc_minR, double maxR, bool inc_maxR, 
+    PGenSph_zAxis_incrementDirection incdir) {
+  PGenSph_zAxis_StateData *s = g->stateData = malloc(sizeof(PGenSph_zAxis_StateData));
+  s->minR = minR;
+  s->maxR = maxR;
+  s->inc_minR = inc_minR;
+  s->inc_maxR = inc_maxR;
+  s->incdir = incdir;
+  period = fabs(period);
+  double offset_normalised = offset - period * floor(offset / period); // shift to interval [0, period]
+  if (offset_normalised > period / 2) offset_normalised -= period; // and to interval [-period/2, period/2]
+  s->offset = offset_normalised;
+  if (offset_normalised > 0) // reverse the direction so that the conditions in _next() are hit in correct order
+    period *= -1;
+  // !!!!!!!! ZDE JSEM SKONČIL !!!!!!!!!!!!!!
+  
+  PGenSph g = {&PGenSph_zAxis, (void *) stateData};
+  return g;
+}
+
+// Dectructor
+void PGenSph_zAxis_dectructor(PGenSph *g) {
+  free(g->stateData);
+  g->stateData = NULL;
+}
+
+// Extractor
+PGenSphReturnData PGenSph_zAxis_next(PGenSph *g) {
+  if (g->stateData == NULL) // already destroyed
+    return PGenSphDoneVal;
+  PGenSph_zAxis_StateData *s = (PGenSph_zAxis_StateData *) g->stateData;
+  const double zval = s->ptindex * s->a + s->offset;
+  const double r = fabs(zval);
+  bool theEnd = false;
+  switch (s->incdir) {
+    case PGENSPH_ZAXIS_INC_FROM_ORIGIN:
+      if (r < s->maxR || (inc_maxR && r == s->maxR)) {
+        if (s->ptindex > 0)
+          s->ptindex *= -1;
+        else
+          s->ptindex = -s->ptindex + 1;
+      } else theEnd = true;
+      break; 
+    case PGENSPH_ZAXIS_INC_TOWARDS_ORIGIN:
+      if (r > s->minR || (inc_minR && r == s->minR)) {
+        if (s->ptindex == 0) // handle "underflow"
+          s->minR = INFINITY;
+        else if (s->ptindex > 0)
+          s->ptindex = -s->ptindex - 1;
+        else // s->ptindex < 0
+          s->ptindex *= -1;
+      } else theEnd = true;
+      break;
+    default:
+        abort(); // invalid value
+  }
+  if (!theEnd) {
+      const PGenSphReturnData retval = {PGEN_NOTDONE | PGEN_NEWR | PGEN_AT_Z,
+        {r, zval >= 0 ? 0 : M_PI, 0}};
+      return retval;
+  } else {
+      PGenSph_destroy(g);
+      return PGenSphDoneVal;
+  }
+}
+
+// Class metadata structure; TODO maybe this can rather be done by macro.
+const PGenSphClassInfo PGenSph_zAxis = {
+  "PGenSph_zAxis",
+  PGenSph_zAxis_next,
+  PGenSph_zAxis_destructor
+};
+
+
+