force/rcparser.py at master · TACCProjects/force · GitHub

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
from rcproblem import RCProblem
import itertools

class RCParser:
  def __init__(self):
    pass

  def parse(self, filename):
    codesegment = False
    with open(filename) as fp:
      for line in fp:
        words = line.strip().split(' ')
        if words[0] == "end":
          if words[1] == "result" or words[1] == "codelet":
            problem.addCodelet(name, result, params, deps, output, code)
          elif words[1] == "problem":
            #We are all done with this problem!
            break
          elif words[1] == "code":
            codesegment = False
        if codesegment:
          code += line.strip()+"\n  "
        else:
          if words[0] == "problem":
            problem = RCProblem(words[1])
          if words[0] == "result" or words[0] == "codelet":
            if words[0] == "result":
              result = True
            else:
              result = False
            params = []
            deps = []
            output = []
            code = ""
            name = words[1]
          if words[0] == "parameter":
            foo = words[1].split('=')
            params.append((foo[0],foo[1]))

          if words[0] == "depends-on":
	    #Perform range expansion on constant ranges in each dependency
	    splitdeps = words[1].split('[')
	    evalstrng = "itertools.product("
	    basename = splitdeps[0]
	    for sdep in splitdeps[1:]:
	      sdep = sdep.rstrip(']')
	      if not sdep.find(',') == -1 or not sdep.find(':') == -1:
		for dep in sdep.rstrip(']').split(','):
		  prange = dep.split(':')
		  if len(prange) == 2:
		    evalstrng+="range("+prange[0]+","+prange[1]+"+1),"
	      else:
		evalstrng+="['"+sdep+"'],"
	    evalstrng = evalstrng.rstrip(",") + ")"

	    for prod in eval(evalstrng):
	      toadd = basename
	      for dep in prod:
		toadd+="["+str(dep)+"]"
	      deps.append(toadd)

          if words[0] == "output":
            output.append(words[1])

          if words[0] == "code":
            code = " ".join(words[1:]).strip()
            codesegment = True

          if words[0] == "data":
            problem.addData(words[1:])

    return problem