001    /*
002     * Copyright 2010-2015 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    
017    package org.jetbrains.kotlin.js.translate.operation;
018    
019    import com.google.common.collect.ImmutableBiMap;
020    import org.jetbrains.kotlin.js.backend.ast.JsBinaryOperator;
021    import org.jetbrains.kotlin.js.backend.ast.JsUnaryOperator;
022    import org.jetbrains.annotations.NotNull;
023    import org.jetbrains.kotlin.lexer.KtToken;
024    import org.jetbrains.kotlin.lexer.KtTokens;
025    
026    import java.util.Map;
027    
028    public final class OperatorTable {
029    
030        //TODO : not all operators , add and test bit operators
031        private static final Map<KtToken, JsBinaryOperator> binaryOperatorsMap = ImmutableBiMap.<KtToken, JsBinaryOperator>builder()
032                .put(KtTokens.PLUS, JsBinaryOperator.ADD)
033                .put(KtTokens.MINUS, JsBinaryOperator.SUB)
034                .put(KtTokens.MUL, JsBinaryOperator.MUL)
035                .put(KtTokens.DIV, JsBinaryOperator.DIV)
036                .put(KtTokens.EQ, JsBinaryOperator.ASG)
037                .put(KtTokens.GT, JsBinaryOperator.GT)
038                .put(KtTokens.GTEQ, JsBinaryOperator.GTE)
039                .put(KtTokens.LT, JsBinaryOperator.LT)
040                .put(KtTokens.LTEQ, JsBinaryOperator.LTE)
041                .put(KtTokens.ANDAND, JsBinaryOperator.AND)
042                .put(KtTokens.OROR, JsBinaryOperator.OR)
043                .put(KtTokens.PERC, JsBinaryOperator.MOD)
044                .put(KtTokens.PLUSEQ, JsBinaryOperator.ASG_ADD)
045                .put(KtTokens.MINUSEQ, JsBinaryOperator.ASG_SUB)
046                .put(KtTokens.DIVEQ, JsBinaryOperator.ASG_DIV)
047                .put(KtTokens.MULTEQ, JsBinaryOperator.ASG_MUL)
048                .put(KtTokens.PERCEQ, JsBinaryOperator.ASG_MOD)
049                .put(KtTokens.IN_KEYWORD, JsBinaryOperator.INOP)
050                .put(KtTokens.EQEQEQ, JsBinaryOperator.REF_EQ)
051                .put(KtTokens.EXCLEQEQEQ, JsBinaryOperator.REF_NEQ)
052                .build();
053    
054        private static final ImmutableBiMap<KtToken, JsUnaryOperator> unaryOperatorsMap = ImmutableBiMap.<KtToken, JsUnaryOperator>builder()
055                .put(KtTokens.PLUSPLUS, JsUnaryOperator.INC)
056                .put(KtTokens.MINUSMINUS, JsUnaryOperator.DEC)
057                .put(KtTokens.EXCL, JsUnaryOperator.NOT)
058                .put(KtTokens.MINUS, JsUnaryOperator.NEG)
059                .put(KtTokens.PLUS, JsUnaryOperator.POS)
060                .build();
061    
062        private OperatorTable() {
063        }
064    
065        public static boolean hasCorrespondingOperator(@NotNull KtToken token) {
066            return binaryOperatorsMap.containsKey(token) || unaryOperatorsMap.containsKey(token);
067        }
068    
069        public static boolean hasCorrespondingBinaryOperator(@NotNull KtToken token) {
070            return binaryOperatorsMap.containsKey(token);
071        }
072    
073        @NotNull
074        static public JsBinaryOperator getBinaryOperator(@NotNull KtToken token) {
075            assert KtTokens.OPERATIONS.contains(token) : "Token should represent an operation!";
076            return binaryOperatorsMap.get(token);
077        }
078    
079        @NotNull
080        static public JsUnaryOperator getUnaryOperator(@NotNull KtToken token) {
081            assert KtTokens.OPERATIONS.contains(token) : "Token should represent an operation!";
082            return unaryOperatorsMap.get(token);
083        }
084    }