1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.supercsv.cellprocessor;
17
18 import org.supercsv.cellprocessor.ift.CellProcessor;
19 import org.supercsv.cellprocessor.ift.StringCellProcessor;
20 import org.supercsv.exception.SuperCsvCellProcessorException;
21 import org.supercsv.util.CsvContext;
22
23
24
25
26
27
28
29 public class ParseEnum extends CellProcessorAdaptor implements StringCellProcessor {
30
31 private final Class<? extends Enum<?>> enumClass;
32
33 private final boolean ignoreCase;
34
35
36
37
38
39
40
41
42
43
44
45 public <T extends Enum<?>> ParseEnum(final Class<T> enumClass) {
46 super();
47 checkPreconditions(enumClass);
48 this.enumClass = enumClass;
49 this.ignoreCase = false;
50 }
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65 public <T extends Enum<?>> ParseEnum(final Class<T> enumClass, final boolean ignoreCase) {
66 super();
67 checkPreconditions(enumClass);
68 this.enumClass = enumClass;
69 this.ignoreCase = ignoreCase;
70 }
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85 public <T extends Enum<?>> ParseEnum(final Class<T> enumClass, final CellProcessor next) {
86 super(next);
87 checkPreconditions(enumClass);
88 this.enumClass = enumClass;
89 this.ignoreCase = false;
90 }
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107 public <T extends Enum<?>> ParseEnum(final Class<T> enumClass, final boolean ignoreCase, final CellProcessor next) {
108 super(next);
109 checkPreconditions(enumClass);
110 this.enumClass = enumClass;
111 this.ignoreCase = ignoreCase;
112 }
113
114
115
116
117
118
119
120
121
122 private static void checkPreconditions(final Class<?> enumClass) {
123 if( enumClass == null ) {
124 throw new NullPointerException("enumClass should not be null");
125 }
126 }
127
128
129
130
131
132
133
134 public Object execute(final Object value, final CsvContext context) {
135 validateInputNotNull(value, context);
136
137 final String inputString = value.toString();
138
139 for( final Enum<?> enumConstant : enumClass.getEnumConstants() ) {
140 String constantName = enumConstant.name();
141 if( ignoreCase ? constantName.equalsIgnoreCase(inputString) : constantName.equals(inputString) ) {
142 return enumConstant;
143 }
144 }
145
146 throw new SuperCsvCellProcessorException(String.format("'%s' could not be parsed as a enum of type %s", value,
147 enumClass.getName()), context, this);
148
149 }
150 }