1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#!/usr/bin/env python3
# encoding: utf-8
"""
_UniformDimensionRepresentation.py
Created by Graham Dennis on 2008-07-30.
Copyright (c) 2008-2012, Graham Dennis
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
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
from xpdeint.Geometry.DimensionRepresentation import DimensionRepresentation
class _UniformDimensionRepresentation(DimensionRepresentation):
instanceAttributes = ['_minimum', '_maximum', '_stepSize']
@property
def stepSizeString(self):
if self._stepSize:
return self._stepSize
else:
return '((%(maximum)s - %(minimum)s)/%(globalLattice)s)' % dict(
minimum = self.minimum,
maximum = self.maximum,
globalLattice = self.globalLattice
)
def openLoop(self, loopingOrder):
if loopingOrder in [self.LoopingOrder.MemoryOrder, self.LoopingOrder.StrictlyAscendingOrder]:
return self.openLoopAscending()
elif loopingOrder == self.LoopingOrder.StrictlyDescendingOrder:
return self.openLoopDescending()
else:
raise NotImplemented
def closeLoop(self, loopingOrder):
if loopingOrder in [self.LoopingOrder.MemoryOrder, self.LoopingOrder.StrictlyAscendingOrder]:
return self.closeLoopAscending()
elif loopingOrder == self.LoopingOrder.StrictlyDescendingOrder:
return self.closeLoopDescending()
else:
raise NotImplemented
def nonlocalAccessIndexFromStringForFieldInBasis(self, accessString, field, basis):
result = super(_UniformDimensionRepresentation, self).nonlocalAccessIndexFromStringForFieldInBasis(accessString, field, basis)
if result: return result
# We only support non-local access for integer-valued dimensions, or access to the minimum value
minimum = self.minimum
# We shouldn't be called if we are distributed, but let's check just to be safe.
if self.hasLocalOffset: return
if not self.type == 'long':
if not self._stepSize and accessString == self._minimum:
return '0'
elif accessString == ('-' + self.name):
canAccessNegativeVariable = False
if self._minimum == ('-' + self._maximum):
canAccessNegativeVariable = True
elif 'Validation' in self.getVar('features'):
canAccessNegativeVariable = True
validationFeature = self.getVar('features')['Validation']
validationFeature.validationChecks.append("""
if (fabs(%(maximum)s + %(minimum)s) > 0.1 * %(stepSize)s)
_LOG(_ERROR_LOG_LEVEL, "ERROR: Nonlocal access of dimension '%(dimRepName)s' using '-%(dimRepName)s' is only valid for dimensions symmetric about zero.\\n");\n""" % {'maximum': self.maximum, 'minimum': self.minimum, 'dimRepName': self.name, 'stepSize': self.stepSize})
if canAccessNegativeVariable:
lattice = self.globalLattice
loopIndex = self.loopIndex
return '(%(lattice)s - %(loopIndex)s) %% %(lattice)s' % locals()
return
return '(%(accessString)s) - %(minimum)s' % locals()
|
