Description:
my fault, fixing auto full score to properly handle subtask case
Commit status:
[Not Reviewed]
References:
Comments:
0 Commit comments 0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
Add another comment

r346:d02b70f6c474 - - 1 file changed: 7 inserted, 2 deleted

@@ -1,183 +1,188
1 1 require 'tmpdir'
2 2
3 3 class TestdataImporter
4 4
5 5 attr :log_msg
6 6
7 7 def initialize(problem)
8 8 @problem = problem
9 9 end
10 10
11 11 def import_from_file(tempfile,
12 12 time_limit,
13 13 memory_limit,
14 14 import_to_db=false)
15 15
16 16 dirname = extract(tempfile)
17 17 return false if not dirname
18 18 if not import_to_db
19 19 @log_msg = GraderScript.call_import_problem(@problem.name,
20 20 dirname,
21 21 time_limit,
22 22 memory_limit)
23 23 else
24 24 # Import test data to test pairs.
25 25
26 26 @problem.test_pairs.clear
27 27 if import_test_pairs(dirname)
28 28 test_pair_count = TestPair.count :conditions => "problem_id = #{@problem.id}"
29 29 @log_msg = "Importing test pair successful. (#{test_pair_count} test pairs imported)"
30 30 else
31 31 @log_msg = "Importing test pair failed. (0 test pairs imported)"
32 32 end
33 33 end
34 34
35 35 @log_msg << import_problem_description(dirname)
36 36 @log_msg << import_problem_pdf(dirname)
37 37 @log_msg << import_full_score(dirname)
38 38
39 39 return true
40 40 end
41 41
42 42 protected
43 43
44 44 def self.long_ext(filename)
45 45 i = filename.index('.')
46 46 len = filename.length
47 47 return filename.slice(i..len)
48 48 end
49 49
50 50 def extract(tempfile)
51 51 testdata_filename = save_testdata_file(tempfile)
52 52 ext = TestdataImporter.long_ext(tempfile.original_filename)
53 53
54 54 extract_dir = File.join(GraderScript.raw_dir, @problem.name)
55 55 if File.exists? extract_dir
56 56 backup_count = 0
57 57 begin
58 58 backup_count += 1
59 59 backup_dirname = "#{extract_dir}.backup.#{backup_count}"
60 60 end while File.exists? backup_dirname
61 61 File.rename(extract_dir, backup_dirname)
62 62 end
63 63 Dir.mkdir extract_dir
64 64
65 65 if ext=='.tar.gz' or ext=='.tgz'
66 66 cmd = "tar -zxvf #{testdata_filename} -C #{extract_dir}"
67 67 elsif ext=='.tar'
68 68 cmd = "tar -xvf #{testdata_filename} -C #{extract_dir}"
69 69 elsif ext=='.zip'
70 70 cmd = "unzip -o #{testdata_filename} -d #{extract_dir}"
71 71 else
72 72 return nil
73 73 end
74 74
75 75 system(cmd)
76 76
77 77 files = Dir["#{extract_dir}/**/*1*.in"]
78 78 return nil if files.length==0
79 79
80 80 File.delete(testdata_filename)
81 81
82 82 return File.dirname(files[0])
83 83 end
84 84
85 85 def save_testdata_file(tempfile)
86 86 ext = TestdataImporter.long_ext(tempfile.original_filename)
87 87 testdata_filename = File.join(Dir.tmpdir,"#{@problem.name}#{ext}")
88 88
89 89 return nil if tempfile==""
90 90
91 91 if tempfile.instance_of?(Tempfile)
92 92 tempfile.close
93 93 FileUtils.move(tempfile.path,testdata_filename)
94 94 else
95 95 File.open(testdata_filename, "wb") do |f|
96 96 f.write(tempfile.read)
97 97 end
98 98 end
99 99
100 100 return testdata_filename
101 101 end
102 102
103 103 def import_test_pairs(dirname)
104 104 test_num = 1
105 105 while FileTest.exists? "#{dirname}/#{test_num}.in"
106 106 in_filename = "#{dirname}/#{test_num}.in"
107 107 sol_filename = "#{dirname}/#{test_num}.sol"
108 108
109 109 break if not FileTest.exists? sol_filename
110 110
111 111 test_pair = TestPair.new(:input => open(in_filename).read,
112 112 :solution => open(sol_filename).read,
113 113 :problem => @problem)
114 114 break if not test_pair.save
115 115
116 116 test_num += 1
117 117 end
118 118 return test_num > 1
119 119 end
120 120
121 121 def import_problem_description(dirname)
122 122 html_files = Dir["#{dirname}/*.html"]
123 123 markdown_files = Dir["#{dirname}/*.md"] + Dir["#{dirname}/*.markdown"]
124 124 if (html_files.length != 0) or (markdown_files.length != 0)
125 125 description = @problem.description || Description.new
126 126
127 127 if html_files.length != 0
128 128 filename = html_files[0]
129 129 description.markdowned = false
130 130 else
131 131 filename = markdown_files[0]
132 132 description.markdowned = true
133 133 end
134 134
135 135 description.body = open(filename).read
136 136 description.save
137 137 @problem.description = description
138 138 @problem.save
139 139 return "\nProblem description imported from #{filename}."
140 140 else
141 141 return ''
142 142 end
143 143 end
144 144
145 145 def import_problem_pdf(dirname)
146 146 pdf_files = Dir["#{dirname}/*.pdf"]
147 147 puts "CHECKING... #{dirname}"
148 148 if pdf_files.length != 0
149 149 puts "HAS PDF FILE"
150 150 filename = pdf_files[0]
151 151
152 152 @problem.save if not @problem.id
153 153 out_dirname = "#{Problem.download_file_basedir}/#{@problem.id}"
154 154 if not FileTest.exists? out_dirname
155 155 Dir.mkdir out_dirname
156 156 end
157 157
158 158 out_filename = "#{out_dirname}/#{@problem.name}.pdf"
159 159
160 160 if FileTest.exists? out_filename
161 161 File.delete out_filename
162 162 end
163 163
164 164 File.rename(filename, out_filename)
165 165 @problem.description_filename = "#{@problem.name}.pdf"
166 166 @problem.save
167 167 return "\nProblem pdf imported from #{filename}."
168 168 else
169 169 return ""
170 170 end
171 171 end
172 172
173 173 #just set the full score to the total number of test case
174 174 #it is not perfect but works on most normal use case
175 175 def import_full_score(dirname)
176 - in_file = Dir["#{dirname}/*.in"]
177 - full_score =in_file.length * 10
176 + num = 0
177 + loop do
178 + num += 1
179 + in_file = Dir["#{dirname}/#{num}*.in"]
180 + break if in_file.length == 0
181 + end
182 + full_score = (num - 1) * 10
178 183 @problem.full_score = full_score
179 184 @problem.save
180 185 return "\nFull score is set to #{full_score}."
181 186 end
182 187
183 188 end
You need to be logged in to leave comments. Login now