@@ -54,9 +54,12 @@ This file is part of the iText (R) project.
54
54
import org .slf4j .LoggerFactory ;
55
55
56
56
import java .util .ArrayList ;
57
+ import java .util .Arrays ;
57
58
import java .util .HashMap ;
59
+ import java .util .HashSet ;
58
60
import java .util .List ;
59
61
import java .util .Map ;
62
+ import java .util .Set ;
60
63
61
64
public class PdfCatalog extends PdfObjectWrapper <PdfDictionary > {
62
65
@@ -74,6 +77,14 @@ public class PdfCatalog extends PdfObjectWrapper<PdfDictionary> {
74
77
//This flag determines if Outline tree of the document has been built via calling getOutlines method. If this flag is false all outline operations will be ignored
75
78
private boolean outlineMode ;
76
79
80
+ private static final Set <PdfName > PAGE_MODES = new HashSet <>(
81
+ Arrays .asList (PdfName .UseNone , PdfName .UseOutlines , PdfName .UseThumbs ,
82
+ PdfName .FullScreen , PdfName .UseOC , PdfName .UseAttachments ));
83
+
84
+ private static final Set <PdfName > PAGE_LAYOUTS = new HashSet <>(
85
+ Arrays .asList (PdfName .SinglePage , PdfName .OneColumn , PdfName .TwoColumnLeft ,
86
+ PdfName .TwoColumnRight , PdfName .TwoPageLeft , PdfName .TwoPageRight ));
87
+
77
88
protected PdfCatalog (PdfDictionary pdfObject ) {
78
89
super (pdfObject );
79
90
if (pdfObject == null ) {
@@ -161,9 +172,7 @@ public PdfCatalog setAdditionalAction(PdfName key, PdfAction action) {
161
172
* @return current instance of PdfCatalog
162
173
*/
163
174
public PdfCatalog setPageMode (PdfName pageMode ) {
164
- if (pageMode .equals (PdfName .UseNone ) || pageMode .equals (PdfName .UseOutlines ) ||
165
- pageMode .equals (PdfName .UseThumbs ) || pageMode .equals (PdfName .FullScreen ) ||
166
- pageMode .equals (PdfName .UseOC ) || pageMode .equals (PdfName .UseAttachments )) {
175
+ if (PAGE_MODES .contains (pageMode )) {
167
176
return put (PdfName .PageMode , pageMode );
168
177
}
169
178
return this ;
@@ -179,9 +188,7 @@ public PdfName getPageMode() {
179
188
* @param pageLayout
180
189
*/
181
190
public PdfCatalog setPageLayout (PdfName pageLayout ) {
182
- if (pageLayout .equals (PdfName .SinglePage ) || pageLayout .equals (PdfName .OneColumn ) ||
183
- pageLayout .equals (PdfName .TwoColumnLeft ) || pageLayout .equals (PdfName .TwoColumnRight ) ||
184
- pageLayout .equals (PdfName .TwoPageLeft ) || pageLayout .equals (PdfName .TwoPageRight )) {
191
+ if (PAGE_LAYOUTS .contains (pageLayout )) {
185
192
return put (PdfName .PageLayout , pageLayout );
186
193
}
187
194
return this ;
0 commit comments