mapFolding 0.4.3__py3-none-any.whl → 0.5.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- mapFolding/__init__.py +93 -58
- mapFolding/basecamp.py +9 -11
- mapFolding/beDRY.py +24 -23
- mapFolding/oeis.py +47 -45
- mapFolding/theDao.py +48 -48
- mapFolding/theSSOT.py +22 -20
- mapFolding/theSSOTdatatypes.py +20 -32
- {mapFolding-0.4.3.dist-info → mapFolding-0.5.1.dist-info}/METADATA +3 -1
- mapFolding-0.5.1.dist-info/RECORD +14 -0
- {mapFolding-0.4.3.dist-info → mapFolding-0.5.1.dist-info}/top_level.txt +0 -1
- mapFolding/reference/flattened.py +0 -377
- mapFolding/reference/hunterNumba.py +0 -132
- mapFolding/reference/irvineJavaPort.py +0 -120
- mapFolding/reference/jax.py +0 -208
- mapFolding/reference/lunnan.py +0 -153
- mapFolding/reference/lunnanNumpy.py +0 -123
- mapFolding/reference/lunnanWhile.py +0 -121
- mapFolding/reference/rotatedEntryPoint.py +0 -240
- mapFolding/reference/total_countPlus1vsPlusN.py +0 -211
- mapFolding/someAssemblyRequired/__init__.py +0 -5
- mapFolding/someAssemblyRequired/getLLVMforNoReason.py +0 -19
- mapFolding/someAssemblyRequired/makeJob.py +0 -55
- mapFolding/someAssemblyRequired/synthesizeModuleJAX.py +0 -29
- mapFolding/someAssemblyRequired/synthesizeNumba.py +0 -340
- mapFolding/someAssemblyRequired/synthesizeNumbaGeneralized.py +0 -396
- mapFolding/someAssemblyRequired/synthesizeNumbaJob.py +0 -162
- mapFolding/someAssemblyRequired/synthesizeNumbaModules.py +0 -129
- mapFolding/syntheticModules/numbaCount.py +0 -158
- mapFolding/syntheticModules/numba_doTheNeedful.py +0 -13
- mapFolding-0.4.3.dist-info/RECORD +0 -40
- tests/__init__.py +0 -1
- tests/conftest.py +0 -306
- tests/test_computations.py +0 -43
- tests/test_oeis.py +0 -129
- tests/test_other.py +0 -171
- tests/test_tasks.py +0 -40
- tests/test_types.py +0 -5
- /mapFolding/{syntheticModules/__init__.py → py.typed} +0 -0
- {mapFolding-0.4.3.dist-info → mapFolding-0.5.1.dist-info}/LICENSE +0 -0
- {mapFolding-0.4.3.dist-info → mapFolding-0.5.1.dist-info}/WHEEL +0 -0
- {mapFolding-0.4.3.dist-info → mapFolding-0.5.1.dist-info}/entry_points.txt +0 -0
mapFolding/reference/jax.py
DELETED
|
@@ -1,208 +0,0 @@
|
|
|
1
|
-
"""I was able to implement the algorithm with JAX, but I didn't see an advantage and it's a pain in the ass.
|
|
2
|
-
I don't maintain this module."""
|
|
3
|
-
from mapFolding import validateListDimensions, getLeavesTotal, makeConnectionGraph
|
|
4
|
-
from typing import List, Tuple
|
|
5
|
-
import jax
|
|
6
|
-
import jaxtyping
|
|
7
|
-
|
|
8
|
-
dtypeMedium = jax.numpy.uint32
|
|
9
|
-
dtypeMaximum = jax.numpy.uint32
|
|
10
|
-
|
|
11
|
-
def countFolds(listDimensions: List[int]) -> int:
|
|
12
|
-
listDimensionsPositive: List[int] = validateListDimensions(listDimensions)
|
|
13
|
-
|
|
14
|
-
n: int = getLeavesTotal(listDimensionsPositive)
|
|
15
|
-
d: int = len(listDimensions)
|
|
16
|
-
import numpy
|
|
17
|
-
D: numpy.ndarray = makeConnectionGraph(listDimensionsPositive)
|
|
18
|
-
connectionGraph = jax.numpy.asarray(D, dtype=dtypeMedium)
|
|
19
|
-
del listDimensionsPositive
|
|
20
|
-
|
|
21
|
-
return foldingsJAX(n, d, connectionGraph)
|
|
22
|
-
|
|
23
|
-
def foldingsJAX(leavesTotal: jaxtyping.UInt32, dimensionsTotal: jaxtyping.UInt32, connectionGraph: jaxtyping.Array) -> jaxtyping.UInt32:
|
|
24
|
-
|
|
25
|
-
def doNothing(argument):
|
|
26
|
-
return argument
|
|
27
|
-
|
|
28
|
-
def while_activeLeaf1ndex_greaterThan_0(comparisonValues: Tuple):
|
|
29
|
-
comparand = comparisonValues[6]
|
|
30
|
-
return comparand > 0
|
|
31
|
-
|
|
32
|
-
def countFoldings(allValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
33
|
-
_0, leafBelow, _2, _3, _4, _5, activeLeaf1ndex, _7 = allValues
|
|
34
|
-
|
|
35
|
-
sentinel = leafBelow.at[0].get().astype(jax.numpy.uint32)
|
|
36
|
-
|
|
37
|
-
allValues = jax.lax.cond(findGapsCondition(sentinel, activeLeaf1ndex),
|
|
38
|
-
lambda argumentX: dao(findGapsDo(argumentX)),
|
|
39
|
-
lambda argumentY: jax.lax.cond(incrementCondition(sentinel, activeLeaf1ndex), lambda argumentZ: dao(incrementDo(argumentZ)), dao, argumentY),
|
|
40
|
-
allValues)
|
|
41
|
-
|
|
42
|
-
return allValues
|
|
43
|
-
|
|
44
|
-
def findGapsCondition(leafBelowSentinel, activeLeafNumber):
|
|
45
|
-
return jax.numpy.logical_or(jax.numpy.logical_and(leafBelowSentinel == 1, activeLeafNumber <= leavesTotal), activeLeafNumber <= 1)
|
|
46
|
-
|
|
47
|
-
def findGapsDo(allValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
48
|
-
def for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1(comparisonValues: Tuple):
|
|
49
|
-
return comparisonValues[-1] <= dimensionsTotal
|
|
50
|
-
|
|
51
|
-
def for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1_do(for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
52
|
-
def ifLeafIsUnconstrainedCondition(comparand):
|
|
53
|
-
return jax.numpy.equal(connectionGraph[comparand, activeLeaf1ndex, activeLeaf1ndex], activeLeaf1ndex)
|
|
54
|
-
|
|
55
|
-
def ifLeafIsUnconstrainedDo(unconstrainedValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
56
|
-
unconstrained_unconstrainedLeaf = unconstrainedValues[3]
|
|
57
|
-
unconstrained_unconstrainedLeaf = 1 + unconstrained_unconstrainedLeaf
|
|
58
|
-
return (unconstrainedValues[0], unconstrainedValues[1], unconstrainedValues[2], unconstrained_unconstrainedLeaf)
|
|
59
|
-
|
|
60
|
-
def ifLeafIsUnconstrainedElse(unconstrainedValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
61
|
-
def while_leaf1ndexConnectee_notEquals_activeLeaf1ndex(comparisonValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
62
|
-
return comparisonValues[-1] != activeLeaf1ndex
|
|
63
|
-
|
|
64
|
-
def countGaps(countGapsDoValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
65
|
-
countGapsCountDimensionsGapped, countGapsPotentialGaps, countGapsGap1ndexLowerBound, countGapsLeaf1ndexConnectee = countGapsDoValues
|
|
66
|
-
|
|
67
|
-
countGapsPotentialGaps = countGapsPotentialGaps.at[countGapsGap1ndexLowerBound].set(countGapsLeaf1ndexConnectee)
|
|
68
|
-
countGapsGap1ndexLowerBound = jax.numpy.where(jax.numpy.equal(countGapsCountDimensionsGapped[countGapsLeaf1ndexConnectee], 0), countGapsGap1ndexLowerBound + 1, countGapsGap1ndexLowerBound)
|
|
69
|
-
countGapsCountDimensionsGapped = countGapsCountDimensionsGapped.at[countGapsLeaf1ndexConnectee].add(1)
|
|
70
|
-
countGapsLeaf1ndexConnectee = connectionGraph.at[dimensionNumber, activeLeaf1ndex, leafBelow.at[countGapsLeaf1ndexConnectee].get()].get().astype(jax.numpy.uint32)
|
|
71
|
-
|
|
72
|
-
return (countGapsCountDimensionsGapped, countGapsPotentialGaps, countGapsGap1ndexLowerBound, countGapsLeaf1ndexConnectee)
|
|
73
|
-
|
|
74
|
-
unconstrained_countDimensionsGapped, unconstrained_gapsWhere, unconstrained_gap1ndexCeiling, unconstrained_unconstrainedLeaf = unconstrainedValues
|
|
75
|
-
|
|
76
|
-
leaf1ndexConnectee = connectionGraph.at[dimensionNumber, activeLeaf1ndex, activeLeaf1ndex].get().astype(jax.numpy.uint32)
|
|
77
|
-
|
|
78
|
-
countGapsValues = (unconstrained_countDimensionsGapped, unconstrained_gapsWhere, unconstrained_gap1ndexCeiling, leaf1ndexConnectee)
|
|
79
|
-
countGapsValues = jax.lax.while_loop(while_leaf1ndexConnectee_notEquals_activeLeaf1ndex, countGaps, countGapsValues)
|
|
80
|
-
unconstrained_countDimensionsGapped, unconstrained_gapsWhere, unconstrained_gap1ndexCeiling, leaf1ndexConnectee = countGapsValues
|
|
81
|
-
|
|
82
|
-
return (unconstrained_countDimensionsGapped, unconstrained_gapsWhere, unconstrained_gap1ndexCeiling, unconstrained_unconstrainedLeaf)
|
|
83
|
-
|
|
84
|
-
dimensions_countDimensionsGapped, dimensions_gapsWhere, dimensions_gap1ndexCeiling, dimensions_unconstrainedLeaf, dimensionNumber = for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values
|
|
85
|
-
|
|
86
|
-
ifLeafIsUnconstrainedValues = (dimensions_countDimensionsGapped, dimensions_gapsWhere, dimensions_gap1ndexCeiling, dimensions_unconstrainedLeaf)
|
|
87
|
-
ifLeafIsUnconstrainedValues = jax.lax.cond(ifLeafIsUnconstrainedCondition(dimensionNumber), ifLeafIsUnconstrainedDo, ifLeafIsUnconstrainedElse, ifLeafIsUnconstrainedValues)
|
|
88
|
-
dimensions_countDimensionsGapped, dimensions_gapsWhere, dimensions_gap1ndexCeiling, dimensions_unconstrainedLeaf = ifLeafIsUnconstrainedValues
|
|
89
|
-
|
|
90
|
-
dimensionNumber = 1 + dimensionNumber
|
|
91
|
-
return (dimensions_countDimensionsGapped, dimensions_gapsWhere, dimensions_gap1ndexCeiling, dimensions_unconstrainedLeaf, dimensionNumber)
|
|
92
|
-
|
|
93
|
-
def almostUselessCondition(comparand):
|
|
94
|
-
return comparand == dimensionsTotal
|
|
95
|
-
|
|
96
|
-
def almostUselessConditionDo(for_leaf1ndex_in_range_activeLeaf1ndexValues: Tuple[jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
97
|
-
def for_leaf1ndex_in_range_activeLeaf1ndex(comparisonValues):
|
|
98
|
-
return comparisonValues[-1] < activeLeaf1ndex
|
|
99
|
-
|
|
100
|
-
def for_leaf1ndex_in_range_activeLeaf1ndex_do(for_leaf1ndex_in_range_activeLeaf1ndexValues: Tuple[jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
101
|
-
leafInRangePotentialGaps, gapNumberLowerBound, leafNumber = for_leaf1ndex_in_range_activeLeaf1ndexValues
|
|
102
|
-
leafInRangePotentialGaps = leafInRangePotentialGaps.at[gapNumberLowerBound].set(leafNumber)
|
|
103
|
-
gapNumberLowerBound = 1 + gapNumberLowerBound
|
|
104
|
-
leafNumber = 1 + leafNumber
|
|
105
|
-
return (leafInRangePotentialGaps, gapNumberLowerBound, leafNumber)
|
|
106
|
-
return jax.lax.while_loop(for_leaf1ndex_in_range_activeLeaf1ndex, for_leaf1ndex_in_range_activeLeaf1ndex_do, for_leaf1ndex_in_range_activeLeaf1ndexValues)
|
|
107
|
-
|
|
108
|
-
def for_range_from_activeGap1ndex_to_gap1ndexCeiling(comparisonValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
109
|
-
return comparisonValues[-1] < gap1ndexCeiling
|
|
110
|
-
|
|
111
|
-
def miniGapDo(gapToGapValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
112
|
-
gapToGapCountDimensionsGapped, gapToGapPotentialGaps, activeGapNumber, index = gapToGapValues
|
|
113
|
-
gapToGapPotentialGaps = gapToGapPotentialGaps.at[activeGapNumber].set(gapToGapPotentialGaps.at[index].get())
|
|
114
|
-
activeGapNumber = jax.numpy.where(jax.numpy.equal(gapToGapCountDimensionsGapped.at[gapToGapPotentialGaps.at[index].get()].get(), dimensionsTotal - unconstrainedLeaf), activeGapNumber + 1, activeGapNumber).astype(jax.numpy.uint32)
|
|
115
|
-
gapToGapCountDimensionsGapped = gapToGapCountDimensionsGapped.at[gapToGapPotentialGaps.at[index].get()].set(0)
|
|
116
|
-
index = 1 + index
|
|
117
|
-
return (gapToGapCountDimensionsGapped, gapToGapPotentialGaps, activeGapNumber, index)
|
|
118
|
-
|
|
119
|
-
_0, leafBelow, countDimensionsGapped, gapRangeStart, gapsWhere, _5, activeLeaf1ndex, activeGap1ndex = allValues
|
|
120
|
-
|
|
121
|
-
unconstrainedLeaf = jax.numpy.uint32(0)
|
|
122
|
-
dimension1ndex = jax.numpy.uint32(1)
|
|
123
|
-
gap1ndexCeiling = gapRangeStart.at[activeLeaf1ndex - 1].get().astype(jax.numpy.uint32)
|
|
124
|
-
activeGap1ndex = gap1ndexCeiling
|
|
125
|
-
for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values = (countDimensionsGapped, gapsWhere, gap1ndexCeiling, unconstrainedLeaf, dimension1ndex)
|
|
126
|
-
for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values = jax.lax.while_loop(for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1, for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1_do, for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values)
|
|
127
|
-
countDimensionsGapped, gapsWhere, gap1ndexCeiling, unconstrainedLeaf, dimension1ndex = for_dimension1ndex_in_range_1_to_dimensionsTotalPlus1Values
|
|
128
|
-
del dimension1ndex
|
|
129
|
-
|
|
130
|
-
leaf1ndex = jax.numpy.uint32(0)
|
|
131
|
-
for_leaf1ndex_in_range_activeLeaf1ndexValues = (gapsWhere, gap1ndexCeiling, leaf1ndex)
|
|
132
|
-
for_leaf1ndex_in_range_activeLeaf1ndexValues = jax.lax.cond(almostUselessCondition(unconstrainedLeaf), almostUselessConditionDo, doNothing, for_leaf1ndex_in_range_activeLeaf1ndexValues)
|
|
133
|
-
gapsWhere, gap1ndexCeiling, leaf1ndex = for_leaf1ndex_in_range_activeLeaf1ndexValues
|
|
134
|
-
del leaf1ndex
|
|
135
|
-
|
|
136
|
-
indexMiniGap = activeGap1ndex
|
|
137
|
-
miniGapValues = (countDimensionsGapped, gapsWhere, activeGap1ndex, indexMiniGap)
|
|
138
|
-
miniGapValues = jax.lax.while_loop(for_range_from_activeGap1ndex_to_gap1ndexCeiling, miniGapDo, miniGapValues)
|
|
139
|
-
countDimensionsGapped, gapsWhere, activeGap1ndex, indexMiniGap = miniGapValues
|
|
140
|
-
del indexMiniGap
|
|
141
|
-
|
|
142
|
-
return (allValues[0], leafBelow, countDimensionsGapped, gapRangeStart, gapsWhere, allValues[5], activeLeaf1ndex, activeGap1ndex)
|
|
143
|
-
|
|
144
|
-
def incrementCondition(leafBelowSentinel, activeLeafNumber):
|
|
145
|
-
return jax.numpy.logical_and(activeLeafNumber > leavesTotal, leafBelowSentinel == 1)
|
|
146
|
-
|
|
147
|
-
def incrementDo(allValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
148
|
-
foldingsSubTotal = allValues[5]
|
|
149
|
-
foldingsSubTotal = leavesTotal + foldingsSubTotal
|
|
150
|
-
return (allValues[0], allValues[1], allValues[2], allValues[3], allValues[4], foldingsSubTotal, allValues[6], allValues[7])
|
|
151
|
-
|
|
152
|
-
def dao(allValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
153
|
-
def whileBacktrackingCondition(backtrackingValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32]):
|
|
154
|
-
comparand = backtrackingValues[2]
|
|
155
|
-
return jax.numpy.logical_and(comparand > 0, jax.numpy.equal(activeGap1ndex, gapRangeStart.at[comparand - 1].get()))
|
|
156
|
-
|
|
157
|
-
def whileBacktrackingDo(backtrackingValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32]):
|
|
158
|
-
backtrackAbove, backtrackBelow, activeLeafNumber = backtrackingValues
|
|
159
|
-
|
|
160
|
-
activeLeafNumber = activeLeafNumber - 1
|
|
161
|
-
backtrackBelow = backtrackBelow.at[backtrackAbove.at[activeLeafNumber].get()].set(backtrackBelow.at[activeLeafNumber].get())
|
|
162
|
-
backtrackAbove = backtrackAbove.at[backtrackBelow.at[activeLeafNumber].get()].set(backtrackAbove.at[activeLeafNumber].get())
|
|
163
|
-
|
|
164
|
-
return (backtrackAbove, backtrackBelow, activeLeafNumber)
|
|
165
|
-
|
|
166
|
-
def if_activeLeaf1ndex_greaterThan_0(activeLeafNumber):
|
|
167
|
-
return activeLeafNumber > 0
|
|
168
|
-
|
|
169
|
-
def if_activeLeaf1ndex_greaterThan_0_do(leafPlacementValues: Tuple[jaxtyping.Array, jaxtyping.Array, jaxtyping.Array, jaxtyping.UInt32, jaxtyping.UInt32]):
|
|
170
|
-
placeLeafAbove, placeLeafBelow, placeGapRangeStart, activeLeafNumber, activeGapNumber = leafPlacementValues
|
|
171
|
-
activeGapNumber = activeGapNumber - 1
|
|
172
|
-
placeLeafAbove = placeLeafAbove.at[activeLeafNumber].set(gapsWhere.at[activeGapNumber].get())
|
|
173
|
-
placeLeafBelow = placeLeafBelow.at[activeLeafNumber].set(placeLeafBelow.at[placeLeafAbove.at[activeLeafNumber].get()].get())
|
|
174
|
-
placeLeafBelow = placeLeafBelow.at[placeLeafAbove.at[activeLeafNumber].get()].set(activeLeafNumber)
|
|
175
|
-
placeLeafAbove = placeLeafAbove.at[placeLeafBelow.at[activeLeafNumber].get()].set(activeLeafNumber)
|
|
176
|
-
placeGapRangeStart = placeGapRangeStart.at[activeLeafNumber].set(activeGapNumber)
|
|
177
|
-
|
|
178
|
-
activeLeafNumber = 1 + activeLeafNumber
|
|
179
|
-
return (placeLeafAbove, placeLeafBelow, placeGapRangeStart, activeLeafNumber, activeGapNumber)
|
|
180
|
-
|
|
181
|
-
leafAbove, leafBelow, _2, gapRangeStart, gapsWhere, _5, activeLeaf1ndex, activeGap1ndex = allValues
|
|
182
|
-
|
|
183
|
-
whileBacktrackingValues = (leafAbove, leafBelow, activeLeaf1ndex)
|
|
184
|
-
whileBacktrackingValues = jax.lax.while_loop(whileBacktrackingCondition, whileBacktrackingDo, whileBacktrackingValues)
|
|
185
|
-
leafAbove, leafBelow, activeLeaf1ndex = whileBacktrackingValues
|
|
186
|
-
|
|
187
|
-
if_activeLeaf1ndex_greaterThan_0_values = (leafAbove, leafBelow, gapRangeStart, activeLeaf1ndex, activeGap1ndex)
|
|
188
|
-
if_activeLeaf1ndex_greaterThan_0_values = jax.lax.cond(if_activeLeaf1ndex_greaterThan_0(activeLeaf1ndex), if_activeLeaf1ndex_greaterThan_0_do, doNothing, if_activeLeaf1ndex_greaterThan_0_values)
|
|
189
|
-
leafAbove, leafBelow, gapRangeStart, activeLeaf1ndex, activeGap1ndex = if_activeLeaf1ndex_greaterThan_0_values
|
|
190
|
-
|
|
191
|
-
return (leafAbove, leafBelow, allValues[2], gapRangeStart, gapsWhere, allValues[5], activeLeaf1ndex, activeGap1ndex)
|
|
192
|
-
|
|
193
|
-
# Dynamic values
|
|
194
|
-
A = jax.numpy.zeros(leavesTotal + 1, dtype=dtypeMedium)
|
|
195
|
-
B = jax.numpy.zeros(leavesTotal + 1, dtype=dtypeMedium)
|
|
196
|
-
count = jax.numpy.zeros(leavesTotal + 1, dtype=dtypeMedium)
|
|
197
|
-
gapter = jax.numpy.zeros(leavesTotal + 1, dtype=dtypeMedium)
|
|
198
|
-
gap = jax.numpy.zeros(leavesTotal * leavesTotal + 1, dtype=dtypeMaximum)
|
|
199
|
-
|
|
200
|
-
foldingsTotal = jax.numpy.uint32(0)
|
|
201
|
-
l = jax.numpy.uint32(1)
|
|
202
|
-
g = jax.numpy.uint32(0)
|
|
203
|
-
|
|
204
|
-
foldingsValues = (A, B, count, gapter, gap, foldingsTotal, l, g)
|
|
205
|
-
foldingsValues = jax.lax.while_loop(while_activeLeaf1ndex_greaterThan_0, countFoldings, foldingsValues)
|
|
206
|
-
return foldingsValues[5]
|
|
207
|
-
|
|
208
|
-
foldingsJAX = jax.jit(foldingsJAX, static_argnums=(0, 1))
|
mapFolding/reference/lunnan.py
DELETED
|
@@ -1,153 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
An unnecessarily literal translation of the original Atlas Autocode code by W. F. Lunnon to Python.
|
|
3
|
-
W. F. Lunnon, Multi-dimensional map-folding, The Computer Journal, Volume 14, Issue 1, 1971, Pages 75-80, https://doi.org/10.1093/comjnl/14.1.75
|
|
4
|
-
"""# NOTE not functional yet
|
|
5
|
-
def foldings(p, job=None):
|
|
6
|
-
"""An unnecessarily literal translation of the original Atlas Autocode code."""
|
|
7
|
-
p = list(p)
|
|
8
|
-
p.append(None) # NOTE mimics Atlas `array` type
|
|
9
|
-
p.insert(0, None) # NOTE mimics Atlas `array` type
|
|
10
|
-
|
|
11
|
-
if job is None:
|
|
12
|
-
global G
|
|
13
|
-
G = 0
|
|
14
|
-
def job(A, B):
|
|
15
|
-
global G
|
|
16
|
-
G = G + 1
|
|
17
|
-
return foldings(p, job)
|
|
18
|
-
# perform job (A, B) on each folding of a p[1] x ... x p[d] map,
|
|
19
|
-
# where A and B are the above and below vectors. p[d + 1] < 0 terminates p;
|
|
20
|
-
|
|
21
|
-
d: int
|
|
22
|
-
n: int
|
|
23
|
-
j: int
|
|
24
|
-
i: int
|
|
25
|
-
m: int
|
|
26
|
-
l: int
|
|
27
|
-
g: int
|
|
28
|
-
gg: int
|
|
29
|
-
dd: int
|
|
30
|
-
|
|
31
|
-
n = 1
|
|
32
|
-
i, d = 0, 0
|
|
33
|
-
|
|
34
|
-
while (i := i + 1) and p[i] is not None:
|
|
35
|
-
d = i
|
|
36
|
-
n = n * p[i]
|
|
37
|
-
|
|
38
|
-
# d dimensions and n leaves;
|
|
39
|
-
|
|
40
|
-
# A: list[int] = [None] * (n + 1) # type: ignore
|
|
41
|
-
# B: list[int] = [None] * (n + 1) # type: ignore
|
|
42
|
-
# count: list[int] = [None] * (n + 1) # type: ignore
|
|
43
|
-
# gapter: list[int] = [None] * (n + 1) # type: ignore
|
|
44
|
-
# gap: list[int] = [None] * (n * n + 1) # type: ignore
|
|
45
|
-
A: list[int] = [0] * (n + 1) # type: ignore
|
|
46
|
-
B: list[int] = [0] * (n + 1) # type: ignore
|
|
47
|
-
count: list[int] = [0] * (n + 1) # type: ignore
|
|
48
|
-
gapter: list[int] = [0] * (n + 1) # type: ignore
|
|
49
|
-
gap: list[int] = [0] * (n * n + 1) # type: ignore
|
|
50
|
-
|
|
51
|
-
# B[m] is the leaf below leaf m in the current folding,
|
|
52
|
-
# A[m] the leaf above. count[m] is the no. of sections in which
|
|
53
|
-
# there is a gap for the new leaf l below leaf m,
|
|
54
|
-
# gap[gapter[l - 1] + j] is the j-th (possible or actual) gap for leaf l,
|
|
55
|
-
# and later gap[gapter[l]] is the gap where leaf l is currently inserted;
|
|
56
|
-
|
|
57
|
-
P: list[int] = [0] * (d + 1) # type: ignore
|
|
58
|
-
C: list[list[int]] = [[0] * (n + 1) for dimension1 in range(d + 1)] # type: ignore
|
|
59
|
-
# D: list[list[list[int]]] = [[[None] * (n + 1) for dimension2 in range(n + 1)] for dimension1 in range(d + 1)] # type: ignore
|
|
60
|
-
D: list[list[list[int]]] = [[[0] * (n + 1) for dimension2 in range(n + 1)] for dimension1 in range(d + 1)]
|
|
61
|
-
|
|
62
|
-
P[0] = 1
|
|
63
|
-
for i in range(1, d + 1):
|
|
64
|
-
P[i] = P[i - 1] * p[i]
|
|
65
|
-
|
|
66
|
-
for i in range(1, d + 1):
|
|
67
|
-
for m in range(1, n + 1):
|
|
68
|
-
C[i][m] = ((m - 1) // P[i - 1]) - ((m - 1) // P[i]) * p[i] + 1
|
|
69
|
-
|
|
70
|
-
for i in range(1, d + 1):
|
|
71
|
-
for l in range(1, n + 1):
|
|
72
|
-
for m in range(1, l + 1):
|
|
73
|
-
D[i][l][m] = (0 if m == 0
|
|
74
|
-
else
|
|
75
|
-
((m if C[i][m] == 1
|
|
76
|
-
else m - P[i - 1])
|
|
77
|
-
if C[i][l] - C[i][m] == (C[i][l] - C[i][m]) // 2 * 2
|
|
78
|
-
else
|
|
79
|
-
(m if C[i][m] == p[i] or m + P[i - 1] > l
|
|
80
|
-
else m + P[i - 1])))
|
|
81
|
-
# P[i] = p[1] x ... x p[i], C[i][m] = i-th co-ordinate of leaf m,
|
|
82
|
-
# D[i][l][m] = leaf connected to m in section i when inserting l;
|
|
83
|
-
|
|
84
|
-
for m in range(n + 1):
|
|
85
|
-
count[m] = 0
|
|
86
|
-
|
|
87
|
-
A[0], B[0], g, l = 0, 0, 0, 0
|
|
88
|
-
|
|
89
|
-
state = 'entry'
|
|
90
|
-
while True:
|
|
91
|
-
if state == 'entry':
|
|
92
|
-
gapter[l] = g
|
|
93
|
-
l = l + 1
|
|
94
|
-
if l <= n:
|
|
95
|
-
state = 'down'
|
|
96
|
-
continue
|
|
97
|
-
else:
|
|
98
|
-
job(A, B)
|
|
99
|
-
state = 'up'
|
|
100
|
-
continue
|
|
101
|
-
|
|
102
|
-
elif state == 'down':
|
|
103
|
-
dd = 0
|
|
104
|
-
gg = gapter[l - 1]
|
|
105
|
-
g = gg
|
|
106
|
-
for i in range(1, d + 1):
|
|
107
|
-
if D[i][l][l] == l:
|
|
108
|
-
dd = dd + 1
|
|
109
|
-
else:
|
|
110
|
-
m = D[i][l][l]
|
|
111
|
-
while m != l:
|
|
112
|
-
gap[gg] = m
|
|
113
|
-
if count[m] == 0:
|
|
114
|
-
gg = gg + 1
|
|
115
|
-
count[m] = count[m] + 1
|
|
116
|
-
m = D[i][l][B[m]]
|
|
117
|
-
|
|
118
|
-
if dd == d:
|
|
119
|
-
for m in range(l):
|
|
120
|
-
gap[gg] = m
|
|
121
|
-
gg = gg + 1
|
|
122
|
-
|
|
123
|
-
for j in range(g, gg):
|
|
124
|
-
gap[g] = gap[j]
|
|
125
|
-
if count[gap[j]] == d - dd:
|
|
126
|
-
g = g + 1
|
|
127
|
-
count[gap[j]] = 0
|
|
128
|
-
state = 'along'
|
|
129
|
-
continue
|
|
130
|
-
|
|
131
|
-
elif state == 'along':
|
|
132
|
-
if g == gapter[l - 1]:
|
|
133
|
-
state = 'up'
|
|
134
|
-
continue
|
|
135
|
-
g = g - 1
|
|
136
|
-
A[l] = gap[g]
|
|
137
|
-
B[l] = B[A[l]]
|
|
138
|
-
B[A[l]] = l
|
|
139
|
-
A[B[l]] = l
|
|
140
|
-
state = 'entry'
|
|
141
|
-
continue
|
|
142
|
-
|
|
143
|
-
elif state == 'up':
|
|
144
|
-
l = l - 1
|
|
145
|
-
B[A[l]] = B[l]
|
|
146
|
-
A[B[l]] = A[l]
|
|
147
|
-
if l > 0:
|
|
148
|
-
state = 'along'
|
|
149
|
-
continue
|
|
150
|
-
else:
|
|
151
|
-
break
|
|
152
|
-
|
|
153
|
-
return G #if job.__closure__ else None
|
|
@@ -1,123 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
A generally faithful translation of the original Atlas Autocode code by W. F. Lunnon to Python using NumPy.
|
|
3
|
-
W. F. Lunnon, Multi-dimensional map-folding, The Computer Journal, Volume 14, Issue 1, 1971, Pages 75-80, https://doi.org/10.1093/comjnl/14.1.75
|
|
4
|
-
"""
|
|
5
|
-
from typing import List
|
|
6
|
-
import numpy
|
|
7
|
-
|
|
8
|
-
def foldings(p: List[int]) -> int:
|
|
9
|
-
"""
|
|
10
|
-
Run loop with (A, B) on each folding of a p[1] x ... x p[d] map, where A and B are the above and below vectors.
|
|
11
|
-
|
|
12
|
-
Parameters:
|
|
13
|
-
p: A list of integers representing the dimensions of the map.
|
|
14
|
-
|
|
15
|
-
Returns:
|
|
16
|
-
G: The number of distinct foldings for the given map dimensions.
|
|
17
|
-
|
|
18
|
-
NOTE If there are fewer than two dimensions, any dimensions are not positive, or any dimensions are not integers, the output will be unreliable.
|
|
19
|
-
"""
|
|
20
|
-
|
|
21
|
-
g: int = 0
|
|
22
|
-
d: int = len(p)
|
|
23
|
-
n: int = 1
|
|
24
|
-
for i in range(d):
|
|
25
|
-
n = n * p[i]
|
|
26
|
-
|
|
27
|
-
# d dimensions and n leaves
|
|
28
|
-
|
|
29
|
-
A = numpy.zeros(n + 1, dtype=int)
|
|
30
|
-
B = numpy.zeros(n + 1, dtype=int)
|
|
31
|
-
count = numpy.zeros(n + 1, dtype=int)
|
|
32
|
-
gapter = numpy.zeros(n + 1, dtype=int)
|
|
33
|
-
gap = numpy.zeros(n * n + 1, dtype=int)
|
|
34
|
-
|
|
35
|
-
# B[m] is the leaf below leaf m in the current folding,
|
|
36
|
-
# A[m] the leaf above. count[m] is the no. of sections in which
|
|
37
|
-
# there is a gap for the new leaf l below leaf m,
|
|
38
|
-
# gap[gapter[l - 1] + j] is the j-th (possible or actual) gap for leaf l,
|
|
39
|
-
# and later gap[gapter[l]] is the gap where leaf l is currently inserted
|
|
40
|
-
|
|
41
|
-
P = numpy.ones(d + 1, dtype=int)
|
|
42
|
-
C = numpy.zeros((d + 1, n + 1), dtype=int)
|
|
43
|
-
D = numpy.zeros((d + 1, n + 1, n + 1), dtype=int)
|
|
44
|
-
|
|
45
|
-
for i in range(1, d + 1):
|
|
46
|
-
P[i] = P[i - 1] * p[i - 1]
|
|
47
|
-
|
|
48
|
-
for i in range(1, d + 1):
|
|
49
|
-
for m in range(1, n + 1):
|
|
50
|
-
C[i][m] = ((m - 1) // P[i - 1]) % p[i - 1] + 1 # NOTE Because modulo is available, this statement is simpler.
|
|
51
|
-
|
|
52
|
-
for i in range(1, d + 1):
|
|
53
|
-
for l in range(1, n + 1):
|
|
54
|
-
for m in range(1, l + 1):
|
|
55
|
-
if C[i][l] - C[i][m] == (C[i][l] - C[i][m]) // 2 * 2:
|
|
56
|
-
if C[i][m] == 1:
|
|
57
|
-
D[i][l][m] = m
|
|
58
|
-
else:
|
|
59
|
-
D[i][l][m] = m - P[i - 1]
|
|
60
|
-
else:
|
|
61
|
-
if C[i][m] == p[i - 1] or m + P[i - 1] > l:
|
|
62
|
-
D[i][l][m] = m
|
|
63
|
-
else:
|
|
64
|
-
D[i][l][m] = m + P[i - 1]
|
|
65
|
-
# P[i] = p[1] x ... x p[i], C[i][m] = i-th co-ordinate of leaf m,
|
|
66
|
-
# D[i][l][m] = leaf connected to m in section i when inserting l;
|
|
67
|
-
|
|
68
|
-
G: int = 0
|
|
69
|
-
l: int = 1
|
|
70
|
-
|
|
71
|
-
# kick off with null folding
|
|
72
|
-
while l > 0:
|
|
73
|
-
if l <= 1 or B[0] == 1: # NOTE This statement is part of a significant divergence from the 1971 paper. As a result, this version is greater than one order of magnitude faster.
|
|
74
|
-
if l > n:
|
|
75
|
-
G = G + n # NOTE Due to `B[0] == 1`, this implementation increments the counted foldings in batches of `n`-many foldings, rather than immediately incrementing when a folding is found, i.e. `G = G + 1`
|
|
76
|
-
else:
|
|
77
|
-
dd: int = 0
|
|
78
|
-
gg: int = gapter[l - 1]
|
|
79
|
-
g = gg
|
|
80
|
-
# dd is the no. of sections in which l is unconstrained,
|
|
81
|
-
# gg the no. of possible and g the no. of actual gaps for l, + gapter[l - 1]
|
|
82
|
-
|
|
83
|
-
# find the possible gaps for leaf l in each section,
|
|
84
|
-
# then discard those not common to all. All possible if dd = d
|
|
85
|
-
for i in range(1, d + 1):
|
|
86
|
-
if D[i][l][l] == l:
|
|
87
|
-
dd = dd + 1
|
|
88
|
-
else:
|
|
89
|
-
m: int = D[i][l][l]
|
|
90
|
-
while m != l:
|
|
91
|
-
gap[gg] = m
|
|
92
|
-
if count[m] == 0:
|
|
93
|
-
gg = gg + 1
|
|
94
|
-
count[m] += 1
|
|
95
|
-
m = D[i][l][B[m]]
|
|
96
|
-
|
|
97
|
-
if dd == d:
|
|
98
|
-
for m in range(l):
|
|
99
|
-
gap[gg] = m
|
|
100
|
-
gg = gg + 1
|
|
101
|
-
|
|
102
|
-
for j in range(g, gg):
|
|
103
|
-
gap[g] = gap[j]
|
|
104
|
-
if count[gap[j]] == d - dd:
|
|
105
|
-
g = g + 1
|
|
106
|
-
count[gap[j]] = 0
|
|
107
|
-
|
|
108
|
-
# for each gap insert leaf l, [the main while loop shall progress],
|
|
109
|
-
# remove leaf l
|
|
110
|
-
while l > 0 and g == gapter[l - 1]:
|
|
111
|
-
l = l - 1
|
|
112
|
-
B[A[l]] = B[l]
|
|
113
|
-
A[B[l]] = A[l]
|
|
114
|
-
|
|
115
|
-
if l > 0:
|
|
116
|
-
g = g - 1
|
|
117
|
-
A[l] = gap[g]
|
|
118
|
-
B[l] = B[A[l]]
|
|
119
|
-
B[A[l]] = l
|
|
120
|
-
A[B[l]] = l
|
|
121
|
-
gapter[l] = g
|
|
122
|
-
l = l + 1
|
|
123
|
-
return G
|
|
@@ -1,121 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
A largely faithful translation of the original Atlas Autocode code by W. F. Lunnon to Python using `while`.
|
|
3
|
-
W. F. Lunnon, Multi-dimensional map-folding, The Computer Journal, Volume 14, Issue 1, 1971, Pages 75-80, https://doi.org/10.1093/comjnl/14.1.75
|
|
4
|
-
"""
|
|
5
|
-
from typing import Sequence
|
|
6
|
-
|
|
7
|
-
def foldings(p: Sequence[int]) -> int:
|
|
8
|
-
"""
|
|
9
|
-
Run loop with (A, B) on each folding of a p[1] x ... x p[d] map, where A and B are the above and below vectors.
|
|
10
|
-
|
|
11
|
-
Parameters:
|
|
12
|
-
p: An array of integers representing the dimensions of the map.
|
|
13
|
-
|
|
14
|
-
Returns:
|
|
15
|
-
G: The number of distinct foldings for the given map dimensions.
|
|
16
|
-
|
|
17
|
-
NOTE If there are fewer than two dimensions, any dimensions are not positive, or any dimensions are not integers, the output will be unreliable.
|
|
18
|
-
"""
|
|
19
|
-
|
|
20
|
-
g: int = 0
|
|
21
|
-
d: int = len(p)
|
|
22
|
-
n: int = 1
|
|
23
|
-
for i in range(d):
|
|
24
|
-
n = n * p[i]
|
|
25
|
-
|
|
26
|
-
# d dimensions and n leaves
|
|
27
|
-
|
|
28
|
-
A = [0] * (n + 1)
|
|
29
|
-
B = [0] * (n + 1)
|
|
30
|
-
count = [0] * (n + 1)
|
|
31
|
-
gapter = [0] * (n + 1)
|
|
32
|
-
gap = [0] * (n * n + 1)
|
|
33
|
-
|
|
34
|
-
# B[m] is the leaf below leaf m in the current folding,
|
|
35
|
-
# A[m] the leaf above. count[m] is the no. of sections in which
|
|
36
|
-
# there is a gap for the new leaf l below leaf m,
|
|
37
|
-
# gap[gapter[l - 1] + j] is the j-th (possible or actual) gap for leaf l,
|
|
38
|
-
# and later gap[gapter[l]] is the gap where leaf l is currently inserted
|
|
39
|
-
|
|
40
|
-
P = [1] * (d + 1)
|
|
41
|
-
C = [[0] * (n + 1) for dimension1 in range(d + 1)]
|
|
42
|
-
D = [[[0] * (n + 1) for dimension2 in range(n + 1)] for dimension1 in range(d + 1)]
|
|
43
|
-
|
|
44
|
-
for i in range(1, d + 1):
|
|
45
|
-
P[i] = P[i - 1] * p[i - 1]
|
|
46
|
-
|
|
47
|
-
for i in range(1, d + 1):
|
|
48
|
-
for m in range(1, n + 1):
|
|
49
|
-
C[i][m] = ((m - 1) // P[i - 1]) - ((m - 1) // P[i]) * p[i - 1] + 1
|
|
50
|
-
|
|
51
|
-
for i in range(1, d + 1):
|
|
52
|
-
for l in range(1, n + 1):
|
|
53
|
-
for m in range(1, l + 1):
|
|
54
|
-
if C[i][l] - C[i][m] == (C[i][l] - C[i][m]) // 2 * 2: # !
|
|
55
|
-
if C[i][m] == 1:
|
|
56
|
-
D[i][l][m] = m
|
|
57
|
-
else:
|
|
58
|
-
D[i][l][m] = m - P[i - 1]
|
|
59
|
-
else:
|
|
60
|
-
if C[i][m] == p[i - 1] or m + P[i - 1] > l:
|
|
61
|
-
D[i][l][m] = m
|
|
62
|
-
else:
|
|
63
|
-
D[i][l][m] = m + P[i - 1]
|
|
64
|
-
# P[i] = p[1] x ... x p[i], C[i][m] = i-th co-ordinate of leaf m,
|
|
65
|
-
# D[i][l][m] = leaf connected to m in section i when inserting l;
|
|
66
|
-
|
|
67
|
-
G: int = 0
|
|
68
|
-
l: int = 1
|
|
69
|
-
|
|
70
|
-
# kick off with null folding
|
|
71
|
-
while l > 0:
|
|
72
|
-
if l > n:
|
|
73
|
-
G = G + 1
|
|
74
|
-
else:
|
|
75
|
-
dd: int = 0
|
|
76
|
-
gg: int = gapter[l - 1]
|
|
77
|
-
g = gg
|
|
78
|
-
# dd is the no. of sections in which l is unconstrained,
|
|
79
|
-
# gg the no. of possible and g the no. of actual gaps for l, + gapter[l - 1]
|
|
80
|
-
|
|
81
|
-
# find the possible gaps for leaf l in each section,
|
|
82
|
-
# then discard those not common to all. All possible if dd = d
|
|
83
|
-
for i in range(1, d + 1):
|
|
84
|
-
if D[i][l][l] == l:
|
|
85
|
-
dd = dd + 1
|
|
86
|
-
else:
|
|
87
|
-
m: int = D[i][l][l]
|
|
88
|
-
while m != l:
|
|
89
|
-
gap[gg] = m
|
|
90
|
-
if count[m] == 0:
|
|
91
|
-
gg = gg + 1
|
|
92
|
-
count[m] += 1
|
|
93
|
-
m = D[i][l][B[m]]
|
|
94
|
-
|
|
95
|
-
if dd == d:
|
|
96
|
-
for m in range(l):
|
|
97
|
-
gap[gg] = m
|
|
98
|
-
gg = gg + 1
|
|
99
|
-
|
|
100
|
-
for j in range(g, gg):
|
|
101
|
-
gap[g] = gap[j]
|
|
102
|
-
if count[gap[j]] == d - dd:
|
|
103
|
-
g = g + 1
|
|
104
|
-
count[gap[j]] = 0
|
|
105
|
-
|
|
106
|
-
# for each gap insert leaf l, [the main while loop shall progress],
|
|
107
|
-
# remove leaf l
|
|
108
|
-
while l > 0 and g == gapter[l - 1]:
|
|
109
|
-
l = l - 1
|
|
110
|
-
B[A[l]] = B[l]
|
|
111
|
-
A[B[l]] = A[l]
|
|
112
|
-
|
|
113
|
-
if l > 0:
|
|
114
|
-
g = g - 1
|
|
115
|
-
A[l] = gap[g]
|
|
116
|
-
B[l] = B[A[l]]
|
|
117
|
-
B[A[l]] = l
|
|
118
|
-
A[B[l]] = l
|
|
119
|
-
gapter[l] = g
|
|
120
|
-
l = l + 1
|
|
121
|
-
return G
|