def realMinList(stn):
origList = getList(stn)
newList = getList(
stnConverter.matrixToDict(test.minimal(
stnConverter.dictToMatrix(stn))))
finalList = []
for ineq in origList:
if ineq in newList:
finalList.append(ineq)
return finalList
def earlyBird(ineqDict):
n = len(ineqDict.keys())
earlyPoint = [0] * n
matrix = stnConverter.dictToMatrix(ineqDict)
matrix = test.minimal(matrix)
ineqDict = stnConverter.matrixToDict(matrix)
for i in range(n - 1, 0, -1):
earlyPoint[i] = -50000
for ineq in ineqDict[i]:
start = ineq[0]
end = ineq[1]
constraint = ineq[2]
if i == start and (start < end or end == 0):
earlyPoint[i] = max(earlyPoint[end] - constraint,
earlyPoint[i])
return earlyPoint
def rigidPoint(ineqDict, metric):
minimalSTN = test.minimal(stnConverter.dictToMatrix(ineqDict))
indices = range(0, len(minimalSTN))
flexSchedule = [0] * (len(minimalSTN))
while indices:
maxFlex = 10000000
maxIndex = -1
time = -1
for index in indices:
timepoint = ((minimalSTN[0][index] + minimalSTN[index][0]) /
2) - minimalSTN[index][0]
newMat = copy.deepcopy(minimalSTN)
newMat[index][0] = -timepoint
newMat[0][index] = timepoint
if len(indices) > 1:
currFlex = metricFunctions.sphericalMetric(
stnConverter.matrixToDict(newMat),
len(indices) - 1, metric)
else:
currFlex = 1
if currFlex < maxFlex:
# print 'currFlex: ' + str(currFlex)
# print 'maxFlex: ' + str(maxFlex)
maxFlex = currFlex
maxIndex = index
time = timepoint
indexToAssign = maxIndex
# delete the chosen index from the list of indices still to be assigned
# print "THIS INDEX"
# print indexToAssign
indices.remove(indexToAssign)
# assign the index a random time that can satisfy the STN
flexSchedule[indexToAssign] = time
# recalculate the minimal STN with this event nailed down
minimalSTN[indexToAssign][0] = -time
minimalSTN[0][indexToAssign] = time
minimalSTN = test.minimal(minimalSTN)
return flexSchedule
def closestBoundary(S, p):
S = stnConverter.matrixToDict(test.minimal(stnConverter.dictToMatrix(S)))
# throw out the zeroeth coordinate of the point because that is
# the zero timepoint and isn't part of a given schedule
p = [0] + [coord - p[0] for coord in p[1:]]
r = sys.maxint
# Loop through all the constraints and adjust r, the radius,
# effectively finding the distance to the closest boundary of
# the solution space for the STN
for l in S.values():
for inequal in l:
start = inequal[0]
end = inequal[1]
constraint = inequal[2]
temp = constraint - p[end] + p[start]
if start != 0 and end != 0:
temp /= math.sqrt(2)
r = min(temp, r)
return r
def pointBrawl(numTrials,
dim,
manWalk=False,
posTime=False,
randDim=False,
multi=False,
ray=False):
chebMargin = 0
centMargin = 0
birdMargin = 0
chebVCentMargin = 0
chebVBirdMargin = 0
centVBirdMargin = 0
# Matric to store all the different quantities we may want to compute in a
# simulation. The numbers on the right refer to indices of the matrix
lol = [
[], # sphereMetric 0
[], # burgerMetric 1
[], # wilsonMetric 2
[], # radRatio 3
[], # naiveMetric 4
[], # chebMargin 5
[], # centMargin 6
[], # birdMargin 7
[], # chebVCentMargin 8
[], # chebVBirdMargin 9
[], # centVBirdMargin 10
[], # randomWalkCheb 11
[], # randomWalkCent 12
[], # randomWalkBird 13
[], # randomWalkManyPoints 14
[], # dimension 15
[], # STNs. 16
]
# Run the point arena on numTrials number of STNs
procPool = mp.Pool(2)
for i in range(numTrials):
print "\nTrial " + str(i)
chebAvg = 0
centAvg = 0
birdAvg = 0
randAvg = 0
numWalks = 100
# Randomly choose the dimension between 2 and 10 events
if randDim:
dim = random.randint(2, 10)
# generate STN with dim events, each constrained by 0 to 50
genstn = stnConverter.matrixToDict(test.generateRandomSTN(dim, 0, 50))
# minimize the STN because why not
genstn = stnConverter.matrixToDict(
test.minimal(stnConverter.dictToMatrix(genstn)))
lol[15].append(dim)
lol[16].append(genstn)
# Metrics for analyzing an STN depending on what we are looking for.
# This could be how "spherelike" the solution space is,
sphereMet = metricFunctions.sphericalMetric(genstn, dim, "spherical")
burgerMet = metricFunctions.sphericalMetric(genstn, dim, "burgers")
wilsonMet = metricFunctions.sphericalMetric(genstn, dim, "wilson")
radRatio = metricFunctions.sphericalMetric(genstn, dim, "radRatio")
naiveMet = metricFunctions.sphericalMetric(genstn, dim, "hell")
# Compute the possible "centers", or points we think may be optimal
cheb = randomWalk.chebyshev(genstn)[0]
cent = centroid(genstn)
birdie = earlyBird(genstn)
# Append the computed metric values to their respective rows
lol[0].append(sphereMet)
lol[1].append(burgerMet)
lol[2].append(wilsonMet)
lol[3].append(radRatio)
lol[4].append(naiveMet)
# Take the average of 100 randomWalks using 100 different points
if not multi:
for i in range(100):
curr = simpleRandSched(genstn)
for i in range(numWalks):
randAvg += randomWalk.perturb(genstn, curr, not ray,
posTime, ray)
else:
tuplist = [(genstn, numWalks, posTime, ray)] * 100
results = procPool.map(mpRandoRandomWalker, tuplist)
randAvg = sum(results)
randAvg /= 100 * float(numWalks)
# Take the average of 100 randomWalks starting at each "center"
if not multi:
for i in range(numWalks):
chebAvg += randomWalk.perturb(genstn, cheb, not ray, posTime,
ray)
centAvg += randomWalk.perturb(genstn, cent, not ray, posTime,
ray)
birdAvg += randomWalk.perturb(genstn, birdie, not ray, posTime,
ray)
else:
tup = [(genstn, cheb, cent, birdie, posTime, ray)] * numWalks
results = procPool.map(mpCenterRandomWalker, tup)
chebAvg = sum([res[0] for res in results])
centAvg = sum([res[1] for res in results])
birdAvg = sum([res[2] for res in results])
chebAvg /= float(numWalks)
centAvg /= float(numWalks)
birdAvg /= float(numWalks)
print "cheb " + "Average: " + str(chebAvg)
print "cent " + "Average: " + str(centAvg)
print "bird " + "Average: " + str(birdAvg)
print "rand " + "Average: " + str(randAvg)
# Compute the different margins of the centers vs each other
if chebAvg == 0 and centAvg == 0:
chebVCentMargin = 0
else:
chebVCentMargin = 2 * (chebAvg - centAvg) / (chebAvg + centAvg)
if chebAvg == 0 and birdAvg == 0:
chebVBirdMargin = 0
else:
chebVBirdMargin = 2 * (chebAvg - birdAvg) / (chebAvg + birdAvg)
if centAvg == 0 and centAvg == 0:
centVBirdMargin = 0
else:
centVBirdMargin = 2 * (centAvg - birdAvg) / (centAvg + birdAvg)
# Compute the different margins of the centers vs random points
if chebAvg == 0 and randAvg == 0:
chebMargin = 0
else:
chebMargin = 2 * (chebAvg - randAvg) / (chebAvg + randAvg)
if centAvg == 0 and randAvg == 0:
centMargin = 0
else:
centMargin = 2 * (centAvg - randAvg) / (centAvg + randAvg)
if birdAvg == 0 and randAvg == 0:
birdMargin = 0
else:
birdMargin = 2 * (birdAvg - randAvg) / (birdAvg + randAvg)
# Append the computed margins to their respective rows
lol[5].append(chebMargin)
lol[6].append(centMargin)
lol[7].append(birdMargin)
lol[8].append(chebVCentMargin)
lol[9].append(chebVBirdMargin)
lol[10].append(centVBirdMargin)
# Append the computed randomWalk lengths starting at each center to
# their respective rows
lol[11].append(chebAvg)
lol[12].append(centAvg)
lol[13].append(birdAvg)
lol[14].append(randAvg)
print "SPHEERE " + str(sphereMet)
print "BURGERS " + str(burgerMet)
print "WILSOON " + str(wilsonMet)
print "RAAAADD " + str(radRatio)
print "NAIIIVE " + str(naiveMet)
title = "THE_GRAND_BATTLE_ON_"
if randDim:
title += "MANY"
else:
title += str(dim)
title += "_DIMENSIONS_"
title += "AND_" + str(numTrials) + "_MILLION_TRIALS"
if posTime:
title += "_AS_TIME_MOVES_FORWARD!"
if ray:
title += "...IN A RAY!"
# Code to store data in a json file
with open(os.path.abspath('./Data/' + title + '.json'), 'w') as fp:
json.dump(lol, fp)
return lol