001/*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017package org.jetbrains.jet;
018
019import com.intellij.lang.ASTNode;
020import com.intellij.psi.tree.IElementType;
021import org.jetbrains.annotations.NonNls;
022import org.jetbrains.annotations.NotNull;
023import org.jetbrains.jet.lang.psi.JetElement;
024import org.jetbrains.jet.lang.psi.JetElementImpl;
025import org.jetbrains.jet.plugin.JetLanguage;
026
027import java.lang.reflect.Constructor;
028
029public class JetNodeType extends IElementType {
030    private Constructor<? extends JetElement> myPsiFactory;
031
032    public JetNodeType(@NotNull @NonNls String debugName) {
033        this(debugName, null);
034    }
035
036    public JetNodeType(@NotNull @NonNls String debugName, Class<? extends JetElement> psiClass) {
037        super(debugName, JetLanguage.INSTANCE);
038        try {
039            myPsiFactory = psiClass != null ? psiClass.getConstructor(ASTNode.class) : null;
040        } catch (NoSuchMethodException e) {
041            throw new RuntimeException("Must have a constructor with ASTNode");
042        }
043    }
044
045    public JetElement createPsi(ASTNode node) {
046        assert node.getElementType() == this;
047
048        try {
049            if (myPsiFactory == null) {
050                return new JetElementImpl(node);
051            }
052            return myPsiFactory.newInstance(node);
053        } catch (Exception e) {
054            throw new RuntimeException("Error creating psi element for node", e);
055        }
056    }
057}