[1] | 1 | package agents.ai2014.group2;
|
---|
| 2 | import java.io.File;
|
---|
| 3 | import java.io.FileWriter;
|
---|
| 4 | import java.io.IOException;
|
---|
| 5 | import java.io.PrintWriter;
|
---|
| 6 | import java.util.ArrayList;
|
---|
| 7 |
|
---|
| 8 | public class G2CSVLogger {
|
---|
| 9 |
|
---|
| 10 | FileWriter fileWriter = null;
|
---|
| 11 | PrintWriter printWriter;
|
---|
| 12 | String eol = System.getProperty("line.separator");
|
---|
| 13 |
|
---|
| 14 | ArrayList<ArrayList<Double>> Buffer = new ArrayList<ArrayList<Double>>();
|
---|
| 15 |
|
---|
| 16 | G2CSVLogger () {
|
---|
| 17 |
|
---|
| 18 | //Empty previous logs
|
---|
| 19 | File dir = new File("./logs/model/");
|
---|
| 20 | if(dir != null){
|
---|
| 21 | for(File file: dir.listFiles()) file.delete();
|
---|
| 22 | }
|
---|
| 23 | dir = new File("./logs/real/");
|
---|
| 24 | if(dir != null){
|
---|
| 25 | for(File file: dir.listFiles()) file.delete();
|
---|
| 26 | }
|
---|
| 27 |
|
---|
| 28 | }
|
---|
| 29 |
|
---|
| 30 | void init (int partyNumber, String name) {
|
---|
| 31 | try {
|
---|
| 32 | fileWriter = new FileWriter("./logs/"+name+"/party"+partyNumber+".csv");
|
---|
| 33 | //printWriter = new PrintWriter(fileWriter);
|
---|
| 34 |
|
---|
| 35 | for(ArrayList<Double> list: Buffer) {
|
---|
| 36 | for(double s:list){
|
---|
| 37 | fileWriter.append(String.valueOf(s));
|
---|
| 38 | fileWriter.append(',');
|
---|
| 39 | }
|
---|
| 40 | fileWriter.append(eol);
|
---|
| 41 | }
|
---|
| 42 | Buffer.clear();
|
---|
| 43 | fileWriter.flush();
|
---|
| 44 | } catch (IOException e) {
|
---|
| 45 | System.out.println("Could not create log for party " + partyNumber);
|
---|
| 46 | }
|
---|
| 47 |
|
---|
| 48 |
|
---|
| 49 | }
|
---|
| 50 |
|
---|
| 51 | void log(ArrayList<Double> list) {
|
---|
| 52 | if(fileWriter == null) {
|
---|
| 53 | Buffer.add(list);
|
---|
| 54 | } else {
|
---|
| 55 | try {
|
---|
| 56 | for(double s:list){
|
---|
| 57 | fileWriter.append(String.valueOf(s));
|
---|
| 58 | fileWriter.append(',');
|
---|
| 59 | }
|
---|
| 60 | fileWriter.append(eol);
|
---|
| 61 | fileWriter.flush();
|
---|
| 62 | }catch (IOException e) {
|
---|
| 63 | System.out.println("Could not add to log");
|
---|
| 64 | }
|
---|
| 65 | }
|
---|
| 66 | }
|
---|
| 67 | void log(Double d) {
|
---|
| 68 | if(fileWriter == null) {
|
---|
| 69 | ArrayList<Double> list = new ArrayList<Double>();
|
---|
| 70 | list.add(d);
|
---|
| 71 | Buffer.add(list);
|
---|
| 72 | } else {
|
---|
| 73 | try {
|
---|
| 74 |
|
---|
| 75 | fileWriter.append(String.valueOf(d));
|
---|
| 76 | fileWriter.append(eol);
|
---|
| 77 | fileWriter.flush();
|
---|
| 78 | }catch (IOException e) {
|
---|
| 79 | System.out.println("Could not add to log");
|
---|
| 80 | }
|
---|
| 81 | }
|
---|
| 82 | }
|
---|
| 83 |
|
---|
| 84 | boolean isInitialized() {
|
---|
| 85 | return fileWriter != null;
|
---|
| 86 | }
|
---|
| 87 | void close(){
|
---|
| 88 | try {
|
---|
| 89 | fileWriter.close();
|
---|
| 90 | }catch (IOException e) {
|
---|
| 91 | System.out.println("Could not close file");
|
---|
| 92 | }
|
---|
| 93 | }
|
---|
| 94 | }
|
---|