xml/dom-impl/src/com/intellij/util/xml/reflect/DomExtensionImpl.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.util.xml.reflect;
  17
  18 import com.intellij.openapi.util.Key;
  19 import com.intellij.util.SmartList;
  20 import com.intellij.util.xml.Converter;
  21 import com.intellij.util.xml.XmlName;
  22 import com.intellij.util.xml.impl.ConvertAnnotationImpl;
  23 import com.intellij.util.xml.impl.DomChildDescriptionImpl;
  24 import gnu.trove.THashMap;
  25 import org.jetbrains.annotations.NotNull;
  26
  27 import java.lang.annotation.Annotation;
  28 import java.lang.reflect.Type;
  29 import java.util.List;
  30 import java.util.Map;
  31
  32 /**
  33  * @author peter
  34  */
  35 public class DomExtensionImpl implements DomExtension {
  36   public static final Key<List<DomExtender>> DOM_EXTENDER_KEY = Key.create("Dom.Extender");
  37   private final XmlName myXmlName;
  38   private final Type myType;
  39   private Converter myConverter;
  40   private final List<Annotation> myCustomAnnos = new SmartList<Annotation>();
  41   private boolean mySoft;
  42   private int myCount = 1;
  43   private Map myUserMap;
  44
  45   public DomExtensionImpl(final Type type, final XmlName xmlName) {
  46     myType = type;
  47     myXmlName = xmlName;
  48   }
  49
  50   @NotNull
  51   public XmlName getXmlName() {
  52     return myXmlName;
  53   }
  54
  55   @NotNull
  56   public Type getType() {
  57     return myType;
  58   }
  59
  60   public DomExtension setConverter(@NotNull Converter converter) {
  61     return setConverter(converter, false);
  62   }
  63
  64   public final DomExtension setConverter(@NotNull final Converter converter, final boolean soft) {
  65     myConverter = converter;
  66     mySoft = soft;
  67     return this;
  68   }
  69
  70   public DomExtension addCustomAnnotation(@NotNull final Annotation anno) {
  71     myCustomAnnos.add(anno);
  72     return this;
  73   }
  74
  75   public <T> void putUserData(final Key<T> key, final T value) {
  76     if (myUserMap == null) myUserMap = new THashMap();
  77     myUserMap.put(key, value);
  78   }
  79
  80   public DomExtension addExtender(final DomExtender extender) {
  81     if (myUserMap == null || !myUserMap.containsKey(DOM_EXTENDER_KEY)) {
  82       putUserData(DOM_EXTENDER_KEY, new SmartList<DomExtender>());
  83     }
  84     ((List<DomExtender>)myUserMap.get(DOM_EXTENDER_KEY)).add(extender);
  85     return this;
  86   }
  87
  88   public final DomExtensionImpl setCount(final int count) {
  89     myCount = count;
  90     return this;
  91   }
  92
  93   public final int getCount() {
  94     return myCount;
  95   }
  96
  97   public final <T extends DomChildDescriptionImpl> T addAnnotations(T t) {
  98     t.setUserMap(myUserMap);
  99     if (myConverter != null) {
 100       t.addCustomAnnotation(new ConvertAnnotationImpl(myConverter, mySoft));
 101     }
 102     for (final Annotation anno : myCustomAnnos) {
 103       t.addCustomAnnotation(anno);
 104     }
 105     return t;
 106   }
 107 }