View Javadoc
1   package fr.ifremer.tutti.service.genericformat.producer;
2   
3   /*
4    * #%L
5    * Tutti :: Service
6    * $Id:$
7    * $HeadURL:$
8    * %%
9    * Copyright (C) 2012 - 2015 Ifremer
10   * %%
11   * This program is free software: you can redistribute it and/or modify
12   * it under the terms of the GNU General Public License as
13   * published by the Free Software Foundation, either version 3 of the
14   * License, or (at your option) any later version.
15   * 
16   * This program is distributed in the hope that it will be useful,
17   * but WITHOUT ANY WARRANTY; without even the implied warranty of
18   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19   * GNU General Public License for more details.
20   * 
21   * You should have received a copy of the GNU General Public
22   * License along with this program.  If not, see
23   * <http://www.gnu.org/licenses/gpl-3.0.html>.
24   * #L%
25   */
26  
27  import fr.ifremer.tutti.persistence.entities.CaracteristicMap;
28  import fr.ifremer.tutti.persistence.entities.data.Cruise;
29  import fr.ifremer.tutti.persistence.entities.data.FishingOperation;
30  import fr.ifremer.tutti.persistence.entities.referential.Caracteristic;
31  import fr.ifremer.tutti.service.csv.CsvProducer;
32  import fr.ifremer.tutti.service.genericformat.csv.ParameterModel;
33  import fr.ifremer.tutti.service.genericformat.csv.ParameterRow;
34  
35  import java.io.Serializable;
36  import java.nio.file.Path;
37  import java.util.ArrayList;
38  import java.util.List;
39  import java.util.Map;
40  
41  /**
42   * Created on 2/6/15.
43   *
44   * @author Tony Chemit - chemit@codelutin.com
45   * @since 3.13
46   */
47  public class CsvProducerForParameter extends CsvProducer<ParameterRow, ParameterModel> {
48  
49      public CsvProducerForParameter(Path file, ParameterModel model) {
50          super(file, model);
51      }
52  
53      public List<ParameterRow> getDataToExport(Cruise cruise, FishingOperation operation) {
54  
55          List<ParameterRow> rows = new ArrayList<>();
56  
57          CaracteristicMap gearUseFeatures = operation.getGearUseFeatures();
58          if (gearUseFeatures != null) {
59              addCaracteristicMap(rows, cruise, operation, ParameterRow.ParameterType.GEAR, gearUseFeatures);
60          }
61  
62          CaracteristicMap vesselUseFeatures = operation.getVesselUseFeatures();
63          if (vesselUseFeatures != null) {
64              addCaracteristicMap(rows, cruise, operation, ParameterRow.ParameterType.VESSEL, vesselUseFeatures);
65          }
66  
67          return rows;
68  
69      }
70  
71      protected void addCaracteristicMap(List<ParameterRow> rows,
72                                         Cruise cruise,
73                                         FishingOperation operation,
74                                         ParameterRow.ParameterType parameterType,
75                                         CaracteristicMap caracteristicMap) {
76  
77          for (Map.Entry<Caracteristic, Serializable> entry : caracteristicMap.entrySet()) {
78  
79              ParameterRow row = new ParameterRow();
80              row.setParameterType(parameterType);
81              row.setCruise(cruise);
82              row.setFishingOperation(operation);
83              row.setCaracteristic(entry.getKey());
84              row.setValue(entry.getValue());
85              rows.add(row);
86  
87          }
88  
89      }
90  
91  }