fr.inria.soctrace.tools.importer.pajedump.PajeDumpImporter.java Source code

Introduction

Here is the source code for fr.inria.soctrace.tools.importer.pajedump.PajeDumpImporter.java

Source

/*******************************************************************************
 * Copyright (c) 2012-2015 INRIA.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Generoso Pagano - initial API and implementation
 ******************************************************************************/
package fr.inria.soctrace.tools.importer.pajedump;

import java.io.File;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.commons.io.FilenameUtils;
import org.eclipse.core.runtime.IProgressMonitor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import fr.inria.soctrace.framesoc.core.FramesocManager;
import fr.inria.soctrace.framesoc.core.tools.management.PluginImporterJob;
import fr.inria.soctrace.framesoc.core.tools.model.FramesocTool;
import fr.inria.soctrace.framesoc.core.tools.model.IFramesocToolInput;
import fr.inria.soctrace.framesoc.core.tools.model.IPluginToolJobBody;
import fr.inria.soctrace.lib.model.utils.SoCTraceException;
import fr.inria.soctrace.lib.storage.DBObject;
import fr.inria.soctrace.lib.storage.DBObject.DBMode;
import fr.inria.soctrace.lib.storage.SystemDBObject;
import fr.inria.soctrace.lib.storage.TraceDBObject;
import fr.inria.soctrace.lib.utils.DeltaManager;
import fr.inria.soctrace.tools.importer.pajedump.core.PJDumpConstants;
import fr.inria.soctrace.tools.importer.pajedump.core.PJDumpParser;
import fr.inria.soctrace.tools.importer.pajedump.core.PJDumpTraceSizeMetadata;
import fr.inria.soctrace.tools.importer.pajedump.input.PajeDumpInput;

/**
 * Paje dump importer tool.
 * 
 * @author "Generoso Pagano <generoso.pagano@inria.fr>"
 */
public class PajeDumpImporter extends FramesocTool {

    private final static Logger logger = LoggerFactory.getLogger(PajeDumpImporter.class);

    /**
     * Plugin Tool Job body: we use a Job since we have to perform a long operation and we don't
     * want to freeze the UI.
     */
    public class PJDumpImporterPluginJobBody implements IPluginToolJobBody {

        private PajeDumpInput input;

        public PJDumpImporterPluginJobBody(IFramesocToolInput input) {
            this.input = (PajeDumpInput) input;
        }

        @Override
        public void run(IProgressMonitor monitor) throws SoCTraceException {
            DeltaManager delta = new DeltaManager();
            delta.start();

            List<String> traces = input.getFiles();
            int numberOfTraces = traces.size();
            int currentTrace = 1;
            Set<String> usedNames = new HashSet<>();
            DeltaManager traceDelta = new DeltaManager();
            for (String traceFile : traces) {

                logger.debug("Importing " + traceFile);

                if (monitor.isCanceled())
                    break;

                traceDelta.start();

                String traceDbName = getNewTraceDBName(usedNames, traceFile);

                SystemDBObject sysDB = null;
                TraceDBObject traceDB = null;

                try {
                    // open system DB
                    sysDB = SystemDBObject.openNewIstance();
                    // create new trace DB
                    traceDB = new TraceDBObject(traceDbName, DBMode.DB_CREATE);

                    // parsing
                    PJDumpParser parser = new PJDumpParser(sysDB, traceDB, traceFile, input);
                    parser.parseTrace(monitor, currentTrace, numberOfTraces);

                } catch (SoCTraceException e) {
                    e.printStackTrace();
                    PluginImporterJob.catchImporterException(e, sysDB, traceDB);
                } finally {
                    // close the trace DB and the system DB (commit)
                    DBObject.finalClose(traceDB);
                    DBObject.finalClose(sysDB);
                    if (monitor.isCanceled())
                        break;
                    new PJDumpTraceSizeMetadata(new File(traceFile), traceDbName);
                    traceDelta.end("Import trace");
                    currentTrace++;
                }
            }
            delta.end("All traces imported");
        }

    }

    private String getNewTraceDBName(Set<String> usedNames, String traceFile) {
        String basename = FilenameUtils.getBaseName(traceFile);
        String extension = FilenameUtils.getExtension(traceFile);
        if (extension.equals(PJDumpConstants.TRACE_EXT)) {
            basename = basename.replace(PJDumpConstants.TRACE_EXT, "");
        }
        final String traceDbName = FramesocManager.getInstance().getTraceDBName(basename);
        String realName = traceDbName;
        int n = 0;
        while (usedNames.contains(realName)) {
            System.out.println("tested " + realName);
            realName = traceDbName + "_" + n++;
        }
        usedNames.add(realName);
        return realName;
    }

    @Override
    public void launch(IFramesocToolInput input) {
        PluginImporterJob job = new PluginImporterJob("Paje Dump Importer", new PJDumpImporterPluginJobBody(input));
        job.setUser(true);
        job.schedule();
    }

    @Override
    public ParameterCheckStatus canLaunch(IFramesocToolInput input) {

        PajeDumpInput pjinput = (PajeDumpInput) input;
        // check if at least one trace file is specified
        if (pjinput.getFiles().size() < 1) {
            return new ParameterCheckStatus(false, "Specify at least one trace file.");
        }

        // check trace files
        for (String file : pjinput.getFiles()) {
            File f = new File(file);
            if (!f.isFile()) {
                return new ParameterCheckStatus(false, f.getName() + " does not exist.");
            }
        }

        return new ParameterCheckStatus(true, "");
    }

}