Skip to content

Commit 60628ad

Browse files
committed
Fixed bug where all organizationes where tagged as companies
1 parent cce7b38 commit 60628ad

File tree

4 files changed

+10
-3
lines changed

4 files changed

+10
-3
lines changed

src/main/java/ldbc/socialnet/dbgen/serializer/CSVMergeForeign.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,7 @@ public void serialize(StudyAt studyAt) {
527527
public void serialize(Organization organization) {
528528
Vector<String> arguments = new Vector<String>();
529529
arguments.add(Long.toString(organization.id));
530-
arguments.add(ScalableGenerator.OrganisationType.company.toString());
530+
arguments.add(organization.type.toString());
531531
arguments.add(organization.name);
532532
arguments.add(DBP.getUrl(organization.name));
533533
arguments.add(Integer.toString(organization.location));

src/main/java/ldbc/socialnet/dbgen/serializer/CSVOriginal.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -600,7 +600,7 @@ public void serialize(StudyAt studyAt) {
600600
public void serialize(Organization organization) {
601601
Vector<String> arguments = new Vector<String>();
602602
arguments.add(Long.toString(organization.id));
603-
arguments.add(ScalableGenerator.OrganisationType.company.toString());
603+
arguments.add(organization.type.toString());
604604
arguments.add(organization.name);
605605
arguments.add(DBP.getUrl(organization.name));
606606
ToCSV(arguments, Files.ORGANISATION.ordinal());

src/main/java/ldbc/socialnet/dbgen/serializer/Turtle.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@
4949

5050
import ldbc.socialnet.dbgen.dictionary.*;
5151
import ldbc.socialnet.dbgen.generator.DateGenerator;
52+
import ldbc.socialnet.dbgen.generator.ScalableGenerator;
5253
import ldbc.socialnet.dbgen.objects.*;
5354
import ldbc.socialnet.dbgen.vocabulary.DBP;
5455
import ldbc.socialnet.dbgen.vocabulary.DBPOWL;
@@ -528,7 +529,11 @@ public void serialize(StudyAt studyAt) {
528529
public void serialize(Organization organization) {
529530

530531
StringBuffer result = new StringBuffer(19000);
531-
writeDBPData(DBP.fullPrefixed(organization.name), RDF.type, DBPOWL.Organisation);
532+
if( organization.type == ScalableGenerator.OrganisationType.company ) {
533+
writeDBPData(DBP.fullPrefixed(organization.name), RDF.type, DBPOWL.Company);
534+
} else {
535+
writeDBPData(DBP.fullPrefixed(organization.name), RDF.type, DBPOWL.University);
536+
}
532537
writeDBPData(DBP.fullPrefixed(organization.name), FOAF.Name, createLiteral(organization.name));
533538
createTripleSPO(result, DBP.fullPrefixed(organization.name),
534539
SNVOC.locatedIn, DBP.fullPrefixed(locationDic.getLocationName(organization.location)));

src/main/java/ldbc/socialnet/dbgen/vocabulary/DBPOWL.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ public class DBPOWL {
4949
public static final String Country = PREFIX+"Country";
5050
public static final String Continent = PREFIX+"Continent";
5151
public static final String Organisation = PREFIX+"Organisation";
52+
public static final String University = PREFIX+"University";
53+
public static final String Company = PREFIX+"Company";
5254
public static final String partOf = PREFIX+"isPartOf";
5355

5456
/**

0 commit comments

Comments
 (0)