prefig 0.3.8.dev20250725054648__py3-none-any.whl → 0.3.9.dev20250727054417__py3-none-any.whl

prefig/core/math_utilities.py

@@ -15,6 +15,9 @@ def set_diagram(d):
 # introduce some useful mathematical operations
 #   that are meant to be available to authors
 
+def ln(x):
+    return math.log(x)
+
 def dot(u, v):
     return np.dot(np.array(u), np.array(v))
 
@@ -80,6 +83,17 @@ def rotate(v, theta):
 def deriv(f, a):
     return calculus.derivative(f, a)
 
+def grad(f, a):
+    a = list(a)
+    grad = []
+    for j in range(len(a)):
+        def f_trace(x):
+            b = a[:]
+            b[j] = x
+            return f(*b)
+        grad.append(calculus.derivative(f_trace, a[j]))
+    return np.array(grad)
+
 def zip_lists(a, b):
     return list(zip(a, b))
 

prefig/core/polygon.py

@@ -188,6 +188,9 @@ def spline(element, diagram, parent, outline_status):
 
     cs = CubicSpline(t_vals, points, bc_type=bc)
 
+    if element.get('name', None) is not None:
+        un.enter_function(element.get('name'), cs)
+
     N = un.valid_eval(element.get('N', '100'))
     domain = element.get('domain', None)
     if domain is not None:

prefig/core/slope_field.py

@@ -11,6 +11,7 @@ from . import math_utilities as math_util
 from . import calculus
 
 log = logging.getLogger('prefigure')
+np.seterr(divide="ignore", invalid="ignore")
 
 # Add a graphical element for slope fields
 def slope_field(element, diagram, parent, outline_status):
@@ -140,52 +141,95 @@ def vector_field(element, diagram, parent, outline_status):
     else:
         line_template.set('stroke', element.get('stroke', 'blue'))
     line_template.set('thickness', element.get('thickness', '2'))
-    if element.get('arrows', 'yes') == 'yes':
-        line_template.set('arrows', '1')
+    line_template.set('arrows', '1')
 
     if element.get('arrow-width', None) is not None:
         line_template.set('arrow-width', element.get('arrow-width'))
     if element.get('arrow-angles', None) is not None:
         line_template.set('arrow-angles', element.get('arrow-angles'))
 
-    spacings = element.get('spacings', None)
-    if spacings is not None:
+    field_data = []
+    if element.get('curve', None) is not None:
+        curve = un.valid_eval(element.get('curve'))
         try:
-            spacings = un.valid_eval(spacings)
-            rx, ry = spacings
+            domain = un.valid_eval(element.get('domain'))
         except:
-            log.error(f"Error parsing slope-field attribute @spacings={element.get('spacings')}")
+            log.error('A @domain is needed if adding a vector field to a curve')
+            return
+        try:
+            N = un.valid_eval(element.get('N'))
+        except:
+            log.error('A @N is needed if adding a vector field to a curve')
             return
-    else:
-        rx = grid_axes.find_gridspacing((bbox[0], bbox[2]))
-        ry = grid_axes.find_gridspacing((bbox[1], bbox[3]))
 
-    # we will go through and generate the vectors first
-    # since we'll need to scale them
-    field_data = []
-    max_scale = 0
-    x = rx[0]
-    while x <= rx[2]:
-        y = ry[0]
-        while y <= ry[2]:
-            f_value = f(x, y)
+        t = domain[0]
+        # if "f" a function of t or (x,y)?
+        one_variable = True
+        try:
+            f(t)
+        except TypeError:
+            one_variable = False
+
+        dt = (domain[1]-domain[0])/(N-1)
+        for _ in range(N):
+            position = curve(t)
+            if one_variable:
+                field_data.append([position, f(t)])
+            else:
+                field_data.append([position, f(*position)])
+            t += dt
+        scale_factor = un.valid_eval(element.get('scale', '1'))
+
+    else:
+        spacings = element.get('spacings', None)
+        if spacings is not None:
             try:
-                if len(f_value) != 2:
-                    log.error("Only two-dimensional vector fields are supported")
-                    return;
+                spacings = un.valid_eval(spacings)
+                rx, ry = spacings
             except:
-                pass
-            max_scale = max(max_scale,
-                            abs((f_value[0])/rx[1]),
-                            abs((f_value[1])/ry[1]))
-            field_data.append([np.array([x,y]), f_value])
-            y += ry[1]
-        x += rx[1]
+                log.error(f"Error parsing slope-field attribute @spacings={element.get('spacings')}")
+                return
+        else:
+            rx = grid_axes.find_gridspacing((bbox[0], bbox[2]))
+            ry = grid_axes.find_gridspacing((bbox[1], bbox[3]))
+
+        # we will go through and generate the vectors first
+        # since we'll need to scale them
+        max_scale = 0
+        exponent = un.valid_eval(element.get('exponent', '1'))
+        x = rx[0]
+        while x <= rx[2]:
+            y = ry[0]
+            while y <= ry[2]:
+                f_value = f(x, y)
+                if any(np.isnan(f_value)):
+                    y += ry[1]
+                    continue
 
-    scale_factor = min(1, 0.75 / max_scale)
-    if element.get('scale') is not None:
-        scale = un.valid_eval(element.get('scale'))
-        scale_factor = scale
+                try:
+                    if len(f_value) != 2:
+                        log.error("Only two-dimensional vector fields are supported")
+                        return;
+                except:
+                    pass
+                norm = math_util.length(f_value)
+                if norm < 1e-10:
+                    f_value = np.array((0,0))
+                else:
+                    # we will scale the length by length**exponent
+                    # to promote the length of shorter vectors
+                    f_value = norm**exponent * (1/norm * f_value)
+                max_scale = max(max_scale,
+                                abs((f_value[0])/rx[1]),
+                                abs((f_value[1])/ry[1]))
+                field_data.append([np.array([x,y]), f_value])
+                y += ry[1]
+            x += rx[1]
+
+        scale_factor = min(1, 0.75 / max_scale)
+        if element.get('scale') is not None:
+            scale = un.valid_eval(element.get('scale'))
+            scale_factor = scale
 
     for datum in field_data:
         p, v = datum
@@ -205,18 +249,3 @@ def vector_field(element, diagram, parent, outline_status):
 
     group.group(element, diagram, parent, outline_status)
 
-def gradient(element, diagram, parent, outline_status):
-    try:
-        f = un.valid_eval(element.get("function"))
-    except:
-        log.error("Error retrieving function in gradient element")
-        return
-
-    def grad(a, b):
-        f_x_trace = lambda x: f(x, b)
-        f_y_trace = lambda y: f(a, y)
-        return np.array([calculus.derivative(f_x_trace, a),
-                         calculus.derivative(f_y_trace, b)])
-    un.enter_namespace("__grad", grad)
-    element.set("function", "__grad")
-    vector_field(element, diagram, parent, outline_status)

prefig/core/tags.py

@@ -41,11 +41,11 @@ tag_dict = {
     'caption': label.caption,
     'circle': circle.circle,
     'clip': clip.clip,
+    'contour': implicit.implicit_curve,
     'coordinates': coordinates.coordinates,
     'definition': definition.definition,
     'derivative': definition.derivative,
     'ellipse': circle.ellipse,
-    'gradient': slope_field.gradient,
     'graph': graph.graph,
     'grid': grid_axes.grid,
     'grid-axes': grid_axes.grid_axes,

prefig/core/user_namespace.py

@@ -66,6 +66,8 @@ def validate_node(node, args=None):
         return True
     if isinstance(node, ast.Expression):
         return validate_node(node.body, args)
+    if isinstance(node, ast.Starred):
+        return validate_node(node.value)
     if isinstance(node, ast.Name):
         if node.id in variables:
             return True
@@ -170,6 +172,11 @@ def derivative(f, name):
     functions.add(name)
     variables.add(name)
 
+def enter_function(name, f):
+    globals()[name] = f
+    functions.add(name)
+    variables.add(name)
+
 def enter_namespace(name, value):
     globals()[name] = value
     variables.add(name)

prefig/resources/schema/pf_schema.rnc

@@ -313,6 +313,7 @@ GraphicalElements =
         Area-Under-Curve* &
         Axes* &
         Circle* &
+        Contour* &
         Ellipse* &
         Graph* &
         Grid* &
@@ -337,7 +338,8 @@ GraphicalElements =
         Tick-Mark* &
         Tangent-line* &
         Triangle* &
-        Vector*
+        Vector* & 
+        Vector-Field*
     )
 
 Angle-Marker = element angle-marker {
@@ -433,6 +435,7 @@ Grid = element grid {
     attribute coordinates {text}?,
     attribute spacing-degrees {"yes"|"no"}?,
     attribute scales {"linear"|"semilogx"|"semilogy"|"loglog"}?,
+    Stroke-Attributes,
     CommonAttributes
 }
 
@@ -457,6 +460,7 @@ Grid-Axes = element grid-axes {
     attribute h-zero-label {"yes"|"no"}?,
     attribute v-zero-label {"yes"|"no"}?,
     attribute label-commas {"yes"|"no"}?,
+    StrokeAttributes,
     CommonAttributes,
     XLabel?,
     YLabel?    
@@ -469,10 +473,21 @@ Tick-Mark = element tick-mark {
     attribute stroke {text}?,
     attribute size {text}?,
     attribute thickness {text}?,
+    StrokeAttributes,
     LabelAttributes,
     LabelText?
 }
 
+Contour = element contour {
+    attribute at {text}?,
+    attribute function {text},
+    attribute k {text}?,
+    attribute depth {text}?,
+    attribute initial-depth {text}?,
+    StrokeAttributes,
+    CommonAttributes
+}
+
 Circle = element circle {
     attribute at {text}?,
     attribute center {text},
@@ -744,6 +759,7 @@ Spline = element spline {
     attribute closed {"yes"|"no"}?,
     attribute arrow-width {text}?,
     attribute arrow-angles {text}?,
+    attribute name {text}?,
     FillAttributes,
     CommonAttributes
 }
@@ -781,4 +797,18 @@ Vector = element vector {
     CommonAttributes
 }
 
+Vector-Field = element vector-field {
+    attribute at {text}?,
+    attribute function {text}?,
+    attribute spacings {text}?,
+    attribute scale {text}?,
+    attribute curve {text}?,
+    attribute domain {text}?,
+    attribute N {text}?,
+    attribute arrow-width {text}?,
+    attribute arrow-angles {text?}?,
+    StrokeAttributes,
+    CommonAttributes
+}
+
 }

prefig/resources/schema/pf_schema.rng

@@ -762,6 +762,9 @@
       <zeroOrMore>
         <ref name="Circle"/>
       </zeroOrMore>
+      <zeroOrMore>
+        <ref name="Contour"/>
+      </zeroOrMore>
       <zeroOrMore>
         <ref name="Ellipse"/>
       </zeroOrMore>
@@ -837,6 +840,9 @@
       <zeroOrMore>
         <ref name="Vector"/>
       </zeroOrMore>
+      <zeroOrMore>
+        <ref name="Vector-Field"/>
+      </zeroOrMore>
     </interleave>
   </define>
   <define name="Angle-Marker">
@@ -1176,6 +1182,7 @@
           </choice>
         </attribute>
       </optional>
+      <ref name="Stroke-Attributes"/>
       <ref name="CommonAttributes"/>
     </element>
   </define>
@@ -1291,6 +1298,7 @@
           </choice>
         </attribute>
       </optional>
+      <ref name="StrokeAttributes"/>
       <ref name="CommonAttributes"/>
       <optional>
         <ref name="XLabel"/>
@@ -1318,12 +1326,32 @@
       <optional>
         <attribute name="thickness"/>
       </optional>
+      <ref name="StrokeAttributes"/>
       <ref name="LabelAttributes"/>
       <optional>
         <ref name="LabelText"/>
       </optional>
     </element>
   </define>
+  <define name="Contour">
+    <element name="contour">
+      <optional>
+        <attribute name="at"/>
+      </optional>
+      <attribute name="function"/>
+      <optional>
+        <attribute name="k"/>
+      </optional>
+      <optional>
+        <attribute name="depth"/>
+      </optional>
+      <optional>
+        <attribute name="initial-depth"/>
+      </optional>
+      <ref name="StrokeAttributes"/>
+      <ref name="CommonAttributes"/>
+    </element>
+  </define>
   <define name="Circle">
     <element name="circle">
       <optional>
@@ -2029,6 +2057,9 @@
       <optional>
         <attribute name="arrow-angles"/>
       </optional>
+      <optional>
+        <attribute name="name"/>
+      </optional>
       <ref name="FillAttributes"/>
       <ref name="CommonAttributes"/>
     </element>
@@ -2112,4 +2143,41 @@
       <ref name="CommonAttributes"/>
     </element>
   </define>
+  <define name="Vector-Field">
+    <element name="vector-field">
+      <optional>
+        <attribute name="at"/>
+      </optional>
+      <optional>
+        <attribute name="function"/>
+      </optional>
+      <optional>
+        <attribute name="spacings"/>
+      </optional>
+      <optional>
+        <attribute name="scale"/>
+      </optional>
+      <optional>
+        <attribute name="curve"/>
+      </optional>
+      <optional>
+        <attribute name="domain"/>
+      </optional>
+      <optional>
+        <attribute name="N"/>
+      </optional>
+      <optional>
+        <attribute name="arrow-width"/>
+      </optional>
+      <optional>
+        <attribute name="arrow-angles">
+          <optional>
+            <text/>
+          </optional>
+        </attribute>
+      </optional>
+      <ref name="StrokeAttributes"/>
+      <ref name="CommonAttributes"/>
+    </element>
+  </define>
 </grammar>

{prefig-0.3.8.dev20250725054648.dist-info → prefig-0.3.9.dev20250727054417.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefig
-Version: 0.3.8.dev20250725054648
+Version: 0.3.9.dev20250727054417
 Summary: An authoring system for mathematical diagrams
 Home-page: https://prefigure.org
 License: GPL-3.0-or-later

{prefig-0.3.8.dev20250725054648.dist-info → prefig-0.3.9.dev20250727054417.dist-info}/RECORD

@@ -22,23 +22,23 @@ prefig/core/label.py,sha256=e_gOd2kcdOtcJay1VyRstP66SiDNxK2UK2-obkxQnOc,24746
 prefig/core/label_tools.py,sha256=jOawiyrKC5iQs0vq7133xNXqrkJbMqDZVHflT-hOruo,9167
 prefig/core/legend.py,sha256=0mVfyU3LvqgDCYUOlCeBD04maV-rj2h0fs1aOEBBt6c,13340
 prefig/core/line.py,sha256=HBB9cvJawKAtdubISn5maRrzFFzV0NQuBgwoQoxeXcw,6055
-prefig/core/math_utilities.py,sha256=HyPUmCGhPYPaL52t4uOwLkY8bfTjwDIcDU8J6z7aW38,5644
+prefig/core/math_utilities.py,sha256=sMm3U9rcYKY8oZZWeJjcBokCDO4CxUkvmaX55auR4Eo,5927
 prefig/core/network.py,sha256=1izAX2AKvnj2knElrdLeojE_0Q0zO0aQM-EsEwor0Os,27217
 prefig/core/parametric_curve.py,sha256=ftxdg3gs0M0ME5iPJo3iEX7jT8PHQEMYpx8psr8Cm-w,3046
 prefig/core/parse.py,sha256=plLmR0KKrUjYx0PmHHo5HZonaaT3tW4qjiDOtzuhSkQ,4542
 prefig/core/path.py,sha256=ZWKUNgVU4qeU98Vp4WKpQoy7DehH1UmnyXA4DKRhpGo,19365
 prefig/core/point.py,sha256=a97N1soBcmYn3StZ60HBmaBi3pB1abA7sL0LNJ2nxR0,8341
-prefig/core/polygon.py,sha256=fh9l3XN4qxXGq9ZRskZv8yIXzGdHltIT8cI5ne_uf4o,11203
+prefig/core/polygon.py,sha256=8NVdWSUVXuNE4GiPQ4MySSmLy8aCNwKUqCcfHukizME,11301
 prefig/core/read.py,sha256=Q1Nsluwysg3M5wtUmwIirfNo-Rw9-dFJPEeM4pDCpyo,1988
 prefig/core/rectangle.py,sha256=T0EMtRdLtmG1uY7Bu8kXmxQ81IGG_102g5fLCg9RuzE,3387
 prefig/core/repeat.py,sha256=DAO5w4BxUMGU-8ymsv8EwsifI0zPa_cgIRUcWWS8BWU,2755
 prefig/core/riemann_sum.py,sha256=T2dQgJIY17QuuPDOB8XWIP8HYmg5jOQ-LroFH7tiGAU,3012
 prefig/core/shape.py,sha256=Qb_-upbSHwEoYrrAnbCoFFFuVIb3JgJLdyP9ckD5evc,8675
-prefig/core/slope_field.py,sha256=ctkMFQ34ZFk-Hr9a8HkmKN-CUjWuZ0yutM4pN515wXk,7560
+prefig/core/slope_field.py,sha256=e396uznWNaCUmOIIf70lsxvy9g_SGRFZlpxsIpy4ajE,8640
 prefig/core/statistics.py,sha256=5qN_-ABN0EFVG12ZP95uSPk82INbWRqI3w2fUdR3wsw,3458
-prefig/core/tags.py,sha256=vsaoRMBuxWcTF6o2xlHFCu55EhGTcuJX3jEi6-N64T8,4086
+prefig/core/tags.py,sha256=fj86TjSkSpo0cAYmp0OQ83ULX9B8Nz4PRsNyZsR2JMk,4088
 prefig/core/tangent_line.py,sha256=I9wf9VAqhnmmsh8Lzynwi7_Z1DonHXqHf8ym7eO40tk,3400
-prefig/core/user_namespace.py,sha256=t5D_diLDPQpfek8w7al_DWI6OlAMedgHoMg45DhPJAI,6878
+prefig/core/user_namespace.py,sha256=6xlkYd2g9dSiDPH0vmYj93oPzm1BFZdgyMGQ9ipKOgU,7059
 prefig/core/utilities.py,sha256=R1fBrzj-abXHG4Bh-sS0pZNC8vc9fIt6yJBcuztCVho,2905
 prefig/core/vector.py,sha256=lTw-y9mt6jqmbFxdPwQB4PYBQ9y4pGrQRcpRp9lBIbY,2917
 prefig/engine.py,sha256=Zz7j2sFxeMwPN1f3GFTpaBihqguzMNJD5iBONOVL7eQ,11972
@@ -55,13 +55,13 @@ prefig/resources/examples/tangent.xml,sha256=FLZaNz0iFoTLX6gB4pfLZFtu_TWeCug4Dda
 prefig/resources/fonts/Braille29.ttf,sha256=T4hosDvaUGMVK0Wh0BmBgWeT7sQQmLXyA3gRUR-a0oE,55880
 prefig/resources/js/mj-sre-page.js,sha256=LHVeI4dCH13bTv56nz3SKNo3JGRXa6joPM3D3wpKV40,9251
 prefig/resources/js/package.json,sha256=hwKDg76_GzkhX72T7DRiF3FoMLokCv9PSLQvnW5Ovok,116
-prefig/resources/schema/pf_schema.rnc,sha256=K9CY2wKcsKEQo9S480Mnt7f-qckhUWt9AFRA7-PYNi8,19104
-prefig/resources/schema/pf_schema.rng,sha256=24kDLnXSWgBE1_I4TM5RR1I30XJo6Cq59rVC5ygSG_8,51821
+prefig/resources/schema/pf_schema.rnc,sha256=KAmcFfhLpVRVG9nFzdHcPOMdEj62aUOMs6BXqiXy7jc,19824
+prefig/resources/schema/pf_schema.rng,sha256=OfOY7Pgd_G6V07v6kqRVJlvyAbyclWBSX9Q8Br97CKo,53495
 prefig/resources/template/pf_publication.xml,sha256=eEv8HACv610Apw5DSVNy0reLfELYqHlNy9Oq0GH7C_c,200
 prefig/scripts/__init__.py,sha256=qJcPi1WRh9UAwu4zIFJbmxWalAMilMYqgi6LxRkF7bI,25
 prefig/scripts/install_mj.py,sha256=5A6-oc1xbIXka5mkFE70vlp9-Rh_QoxOxGMrQi9Hkio,1219
-prefig-0.3.8.dev20250725054648.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-prefig-0.3.8.dev20250725054648.dist-info/METADATA,sha256=T3jXMcdrLim_prFihqU2ZP468zTdn_OLa1xOMG5PdTc,8728
-prefig-0.3.8.dev20250725054648.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prefig-0.3.8.dev20250725054648.dist-info/entry_points.txt,sha256=OP4ZQT71q2b0Zfbie-oM2Z1HlxpkuX7qaxJnzwsBOVQ,42
-prefig-0.3.8.dev20250725054648.dist-info/RECORD,,
+prefig-0.3.9.dev20250727054417.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+prefig-0.3.9.dev20250727054417.dist-info/METADATA,sha256=6USAGTlT0nCcH_I6UUkca9UtA9kQrSHhCzyf64mpVPM,8728
+prefig-0.3.9.dev20250727054417.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prefig-0.3.9.dev20250727054417.dist-info/entry_points.txt,sha256=OP4ZQT71q2b0Zfbie-oM2Z1HlxpkuX7qaxJnzwsBOVQ,42
+prefig-0.3.9.dev20250727054417.dist-info/RECORD,,