/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.carbondata.core.scan.partition;

import java.math.BigDecimal;
import java.text.DateFormat;
import java.util.BitSet;

import org.apache.carbondata.core.metadata.datatype.DataType;
import org.apache.carbondata.core.metadata.datatype.DataTypes;
import org.apache.carbondata.core.metadata.schema.PartitionInfo;

import org.apache.commons.lang.StringUtils;

public class PartitionUtil {

  public static Partitioner getPartitioner(PartitionInfo partitionInfo) {
    switch (partitionInfo.getPartitionType()) {
      case HASH:
        return new HashPartitioner(partitionInfo.getNumPartitions());
      case LIST:
        return new ListPartitioner(partitionInfo);
      case RANGE:
        return new RangePartitioner(partitionInfo);
      default:
        throw new UnsupportedOperationException(
            "unsupport partition type: " + partitionInfo.getPartitionType().name());
    }
  }

  public static Object getDataBasedOnDataType(String data, DataType actualDataType,
      DateFormat timestampFormatter, DateFormat dateFormatter) {
    if (data == null) {
      return null;
    }
    if (actualDataType != DataTypes.STRING && StringUtils.isEmpty(data)) {
      return null;
    }
    try {
      if (actualDataType == DataTypes.STRING) {
        return data;
      } else if (actualDataType == DataTypes.INT) {
        return Integer.parseInt(data);
      } else if (actualDataType == DataTypes.SHORT) {
        return Short.parseShort(data);
      } else if (actualDataType == DataTypes.DOUBLE) {
        return Double.parseDouble(data);
      } else if (actualDataType == DataTypes.LONG) {
        return Long.parseLong(data);
      } else if (actualDataType == DataTypes.DATE) {
        return dateFormatter.parse(data).getTime();
      } else if (actualDataType == DataTypes.TIMESTAMP) {
        return timestampFormatter.parse(data).getTime();
      } else if (DataTypes.isDecimal(actualDataType)) {
        return new BigDecimal(data);
      } else {
        return data;
      }
    } catch (Exception ex) {
      return null;
    }
  }

  /**
   * convert the string value of partition filter to the Object
   * @param data
   * @param actualDataType
   * @return
   */
  public static Object getDataBasedOnDataTypeForFilter(String data, DataType actualDataType) {
    if (data == null) {
      return null;
    }
    if (actualDataType != DataTypes.STRING && StringUtils.isEmpty(data)) {
      return null;
    }
    try {
      if (actualDataType == DataTypes.STRING) {
        return data;
      } else if (actualDataType == DataTypes.INT) {
        return Integer.parseInt(data);
      } else if (actualDataType == DataTypes.SHORT) {
        return Short.parseShort(data);
      } else if (actualDataType == DataTypes.DOUBLE) {
        return Double.parseDouble(data);
      } else if (actualDataType == DataTypes.LONG) {
        return Long.parseLong(data);
      } else if (actualDataType == DataTypes.DATE || actualDataType == DataTypes.TIMESTAMP) {
        return Long.parseLong(data) / 1000;
      } else if (DataTypes.isDecimal(actualDataType)) {
        return new BigDecimal(data);
      } else {
        return data;
      }
    } catch (Exception ex) {
      return null;
    }
  }

  /**
   * generate a BitSet by size
   * @param size
   * @param initValue true: initialize all bits to true
   * @return
   */
  public static BitSet generateBitSetBySize(int size, boolean initValue) {
    BitSet bitSet = new BitSet(size);
    if (initValue) {
      bitSet.set(0, size);
    }
    return bitSet;
  }

}
