RuleSet.java

/*
 * Copyright 2003-2011 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package jetbrains.mps.lang.typesystem.runtime;

import gnu.trove.THashSet;
import jetbrains.mps.lang.smodel.generator.smodelAdapter.SConceptOperations;
import jetbrains.mps.languageScope.LanguageScope;
import jetbrains.mps.newTypesystem.rules.SingleTermRules;
import org.apache.log4j.Logger;
import org.jetbrains.mps.openapi.language.SAbstractConcept;
import org.jetbrains.mps.openapi.model.SNode;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

/*
 *  Synchronized.
 */
public class RuleSet<T extends IApplicableToConcept> {

  private Logger LOG = Logger.getLogger(RuleSet.class);

  private static final String TYPESYSTEM_SUFFIX = ".typesystem";
  private ConcurrentMap<SAbstractConcept, Set<T>> myRules = new ConcurrentHashMap<SAbstractConcept, /* synchronized */ Set<T>>();

  private SingleTermRules<T> mySingleTermRules = new SingleTermRules<T>() {

    @Override
    protected List<SAbstractConcept> getParents(SAbstractConcept nextConcept) {
      return SConceptOperations.getDirectSuperConcepts(nextConcept, false);
    }

    @Override
    protected Iterable<T> allForConcept(SAbstractConcept concept, LanguageScope langScope) {
      return getAllApplicableTo(concept, langScope);
    }

    @Override
    protected boolean isOverriding(T rule) {
      return rule instanceof ICheckingRule_Runtime && ((ICheckingRule_Runtime) rule).overrides();
    }
  };


  public void addRuleSetItem(Set<T> rules) {
    for (T rule : rules) {
      try {
        addRule_internal(rule);
      }
      catch (Throwable ex) {
        LOG.error("Error initializing rule '"+String.valueOf(rule)+"'", ex);
      }
    }
    mySingleTermRules.purgeCache();
  }

  @Deprecated
  public void addRule(T rule) {
    addRule_internal(rule);
    mySingleTermRules.purgeCache();
  }

  private void addRule_internal(T rule) {
    SAbstractConcept concept = rule.getApplicableConcept();
    Set<T> existingRules = myRules.get(concept);
    while (existingRules == null) {
      myRules.putIfAbsent(concept, Collections.synchronizedSet(new THashSet<T>(2)));
      existingRules = myRules.get(concept);
    }
    existingRules.add(rule);
  }

  /**
   * Returns a set of rules with predictable iteration order: on the node concept, from most specific to most generic.
   * @param term
   * @return
   */
  public Set<T> getRules(SNode term) {
    return mySingleTermRules.lookupRules(term);
  }

  private Iterable<T> getAllApplicableTo(SAbstractConcept concept, LanguageScope scope) {
    if (!myRules.containsKey(concept)) return Collections.emptyList();

    List<T> result = new ArrayList<T>(4);
    Set<T> rules = myRules.get(concept);
    synchronized (rules) {
      for (T rule : rules) {
        if (scope.containsNamespace(getNamespace(rule))) {
          result.add(rule);
        }
      }
    }
    return Collections.unmodifiableList(result);
  }

  private String getNamespace(T rule) {
    String pkg = rule.getClass().getPackage().getName();
    if (pkg.endsWith(TYPESYSTEM_SUFFIX)) {
      return pkg.substring(0, pkg.length() - TYPESYSTEM_SUFFIX.length());
    }
    return pkg;
  }

  public void clear() {
    myRules.clear();
    mySingleTermRules.purgeCache();
  }
}