Refactoring flavours

veraPDF · May 30, 2024 · 7307c5d · 7307c5d
1 parent 7a8eaa6
commit 7307c5d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/verapdf/tools/TaggedPDFRoleMapHelper.java b/src/main/java/org/verapdf/tools/TaggedPDFRoleMapHelper.java
@@ -53,7 +53,7 @@ public String getStandardType(ASAtom type) {
 		}
 		Set<String> currentStandardTypes = getCurrentStandardTypes();
 		boolean isFastStop;
-		if (StaticResources.getFlavour().getSpecification() == PDFFlavour.Specification.ISO_19005_1) {
+		if (PDFFlavour.isFlavourPart(StaticResources.getFlavour(), PDFFlavour.Specification.ISO_19005_1)) {
 			isFastStop = true;
 		} else {
 			isFastStop = false;
@@ -62,10 +62,10 @@ public String getStandardType(ASAtom type) {
 	}
 
 	private static Set<String> getCurrentStandardTypes() {
-		if (StaticResources.getFlavour().getSpecification() == PDFFlavour.Specification.ISO_19005_1) {
+		if (PDFFlavour.isFlavourPart(StaticResources.getFlavour(), PDFFlavour.Specification.ISO_19005_1)) {
 			return TaggedPDFHelper.getPdf14StandardRoleTypes();
 		}
-		if (StaticResources.getFlavour().getSpecification().getFamily() == PDFFlavour.SpecificationFamily.WCAG) {
+		if (PDFFlavour.isFlavourFamily(StaticResources.getFlavour(), PDFFlavour.SpecificationFamily.WCAG)) {
 			return TaggedPDFHelper.getWcagStandardRoleTypes();
 		}
 		return TaggedPDFHelper.getPdf17StandardRoleTypes();