benchmarks: Added CSV benchmark to bench suite

[nit.git] / benchmarks / csv / scripts / csv_gen.nit

diff --git a/benchmarks/csv/scripts/csv_gen.nit b/benchmarks/csv/scripts/csv_gen.nit
new file mode 100644 (file)
index 0000000..123dbb6
--- /dev/null
+++ b/benchmarks/csv/scripts/csv_gen.nit
@@ -0,0 +1,61 @@
+# This file is part of NIT ( http://www.nitlanguage.org ).
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import csv
+
+if args.length < 3 then
+       print "Usage ./csv_gen record_length record_nb out_filepath [--unicode]"
+       exit 1
+end
+
+var record_length = args[0].to_i
+var record_nb = args[1].to_i
+var outpath = args[2]
+var unicode = false
+
+if args.length == 4 then
+       if not args[3] == "--unicode" then
+               print "Usage ./csv_gen record_length record_nb [--unicode]"
+               exit 1
+       end
+       unicode = true
+end
+
+var ocsv = new CsvDocument
+ocsv.eol = "\r\n"
+
+var sep = ocsv.separator.to_s
+var eol = ocsv.eol
+var del = ocsv.delimiter.to_s
+
+for i in [0 .. record_length[ do ocsv.header.add "Col{i}"
+
+var c = if unicode then "á" else "a"
+for i in [0 .. record_nb[ do
+       var line = new Array[String].with_capacity(record_length)
+       for j in [0 .. record_length[ do
+               var add_sep = 100.rand > 70
+               var add_del = 100.rand > 70
+               var add_eol = 100.rand > 70
+               var ln = 10.rand
+               var s = c * ln
+               if add_sep then s = sep + s
+               if add_del then s += del
+               if add_eol then s += eol
+               line.add s
+       end
+       ocsv.records.add line
+end
+
+ocsv.write_to_file(outpath)