summaryrefslogtreecommitdiffstats
path: root/base/config.py
blob: 09eb0fbb5bf32a005f2796a30b302d843ba91fd7 (plain)
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
#!/usr/bin/env python2
#
#	config.py:	parse problem config
#	Copyright (C) 2011  Hamed Saleh and Mahrud Sayrafi

#	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 3 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/>.

import os

PROBCONFADDR = os.getenv('PROBCONFADDR')
INPUT_DIR = os.getenv('INPUT_DIR')

class Config:
	def __init__(self, address):
		self.data = []
		self.address = address
	
	def parse(self):
		PROBCONF = open (self.address, 'r')

		self.count = int (PROBCONF.readline().strip())
		CONFMOD = PROBCONF.readline().strip()

		if CONFMOD == 'default':
			confline = PROBCONF.readline().strip()
			for i in range(self.count):
				self.data.append (confline.replace ('%n', str(i)).split())
		else:
			for i in range(self.count):
				confline = PROBCONF.readline()
				self.data.append (confline.replace ('%n', str(i)).split())

		PROBCONF.close()

	def input_command(self, test):
		input_command = INPUT_DIR + '/' + self.data[test][3];

		if (self.data[test][0] == 'normal'):
			input_command = 'cat ' + input_command
		return input_command
	
	def get_limits(self, test):
		return [int(self.data[test][1]), int(self.data[test][2])]

def parse():
	CONFIG = Config (PROBCONFADDR)
	CONFIG.parse()
	return CONFIG