org.bimserver.tools.ifcloader.IfcLoader.java Source code

Introduction

Here is the source code for org.bimserver.tools.ifcloader.IfcLoader.java

Source

package org.bimserver.tools.ifcloader;

/******************************************************************************
 * Copyright (C) 2009-2019  BIMserver.org
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see {@literal<http://www.gnu.org/licenses/>}.
 *****************************************************************************/

import java.io.IOException;
import java.nio.file.DirectoryStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.bimserver.client.BimServerClient;
import org.bimserver.client.json.JsonBimServerClientFactory;
import org.bimserver.interfaces.objects.SDeserializerPluginConfiguration;
import org.bimserver.interfaces.objects.SProject;
import org.bimserver.plugins.services.Flow;
import org.bimserver.shared.ChannelConnectionException;
import org.bimserver.shared.UsernamePasswordAuthenticationInfo;
import org.bimserver.shared.exceptions.BimServerClientException;
import org.bimserver.shared.exceptions.PublicInterfaceNotFoundException;
import org.bimserver.shared.exceptions.ServerException;
import org.bimserver.shared.exceptions.ServiceException;
import org.bimserver.shared.exceptions.UserException;

public class IfcLoader {
    private BimServerClient client;

    public static void main(String[] args) {
        Options options = new Options();
        options.addOption("path", true, "Directory with IFC files (can have subdirectories)");
        options.addOption("address", true, "Address of BIMserver");
        options.addOption("username", true, "Username");
        options.addOption("password", true, "Password");
        options.addOption("mainproject", true, "Main project name");
        CommandLineParser parser = new DefaultParser();
        try {
            CommandLine cmd = parser.parse(options, args);
            new IfcLoader().start(cmd);
        } catch (ParseException e) {
            e.printStackTrace();
        }
    }

    private void start(CommandLine commandLine) {
        try {
            Path baseDirectory = Paths.get(commandLine.getOptionValue("path"));

            JsonBimServerClientFactory factory = new JsonBimServerClientFactory(
                    commandLine.getOptionValue("address"));
            client = factory.create(new UsernamePasswordAuthenticationInfo(commandLine.getOptionValue("username"),
                    commandLine.getOptionValue("password")));

            String mainProjectName = commandLine.getOptionValue("mainproject");

            if (Files.exists(baseDirectory)) {
                createRecursive(baseDirectory, null, mainProjectName, 0);
            }
        } catch (BimServerClientException e) {
            e.printStackTrace();
        } catch (ServiceException e) {
            e.printStackTrace();
        } catch (ChannelConnectionException e) {
            e.printStackTrace();
        }
    }

    private void createRecursive(Path baseDirectory, SProject parent, String mainProjectName, int nrChildren) {
        try {
            if (Files.isDirectory(baseDirectory)) {
                SProject project = null;
                if (parent == null) {
                    project = client.getServiceInterface().addProject(mainProjectName, "ifc2x3tc1");
                } else {
                    project = client.getServiceInterface().addProjectAsSubProject(
                            baseDirectory.getFileName().toString(), parent.getOid(), "ifc2x3tc1");
                }
                List<Path> paths = new ArrayList<>();
                try (DirectoryStream<Path> stream = Files.newDirectoryStream(baseDirectory)) {
                    for (Path entry : stream) {
                        paths.add(entry);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
                for (Path path : paths) {
                    createRecursive(path, project, null, paths.size());
                }
            } else {
                SDeserializerPluginConfiguration deserializer = client.getServiceInterface()
                        .getSuggestedDeserializerForExtension("ifc", parent.getOid());
                if (nrChildren == 1) {
                    try {
                        client.checkinSync(parent.getOid(), "Initial", deserializer.getOid(), false, baseDirectory);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                } else {
                    String folderName = baseDirectory.getFileName().toString();
                    folderName = folderName.substring(0, folderName.lastIndexOf(".") - 1);
                    SProject project = client.getServiceInterface().addProjectAsSubProject(folderName,
                            parent.getOid(), "ifc2x3tc1");
                    try {
                        client.checkinSync(project.getOid(), "Initial", deserializer.getOid(), false,
                                baseDirectory);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        } catch (ServerException e) {
            e.printStackTrace();
        } catch (UserException e) {
            e.printStackTrace();
        } catch (PublicInterfaceNotFoundException e) {
            e.printStackTrace();
        }
    }
}