cc.recommenders.usages.Query.java Source code

Introduction

Here is the source code for cc.recommenders.usages.Query.java

Source

/**
 * Copyright (c) 2011-2013 Darmstadt University of Technology.
 * 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:
 *     Sebastian Proksch - initial API and implementation
 */
package cc.recommenders.usages;

import java.util.Set;

import com.google.common.collect.Sets;

import cc.recommenders.names.ICoReMethodName;
import cc.recommenders.names.ICoReTypeName;

public class Query extends AbstractUsage {

    // make sure the naming is consistent to the hardcoded names in
    // "UsageTypeAdapter"

    private ICoReTypeName type;
    private ICoReTypeName classCtx;
    private ICoReMethodName methodCtx;
    private DefinitionSite definition;
    private Set<CallSite> sites = Sets.newLinkedHashSet();

    public static Query createAsCopyFrom(Usage usage) {
        Query q = new Query();
        q.setType(usage.getType());
        q.setClassContext(usage.getClassContext());
        q.setMethodContext(usage.getMethodContext());
        q.setDefinition(usage.getDefinitionSite());
        for (CallSite s : usage.getAllCallsites()) {
            q.addCallSite(s);
        }
        return q;
    }

    public void setType(ICoReTypeName typeName) {
        this.type = typeName;
    }

    public ICoReTypeName getType() {
        return type;
    }

    public ICoReTypeName getClassContext() {
        return classCtx;
    }

    public ICoReMethodName getMethodContext() {
        return methodCtx;
    }

    public DefinitionSite getDefinitionSite() {
        return definition;
    }

    public void setAllCallsites(Set<CallSite> sites) {
        this.sites = sites;
    }

    public void resetCallsites() {
        Set<CallSite> sites = Sets.newLinkedHashSet();
        setAllCallsites(sites);
    }

    public boolean addCallSite(CallSite site) {
        if (!sites.contains(site)) {
            return sites.add(site);
        } else {
            return false;
        }
    }

    public Set<CallSite> getAllCallsites() {
        return sites;
    }

    public void setClassContext(ICoReTypeName typeName) {
        this.classCtx = typeName;
    }

    public void setMethodContext(ICoReMethodName methodName) {
        this.methodCtx = methodName;
    }

    public void setDefinition(DefinitionSite definition) {
        this.definition = definition;
    }
}