package com.iailab.netsdk.lib.enumeration;

/**
 * @author 47081
 * @version 1.0
 * @description
 * @date 2021/3/2
 */
public enum EM_REAL_DATA_TYPE {
  EM_REAL_DATA_TYPE_PRIVATE(0, "私有码流"),
  EM_REAL_DATA_TYPE_GBPS(1, "国标PS码流"),
  EM_REAL_DATA_TYPE_TS(2, "TS码流"),
  EM_REAL_DATA_TYPE_MP4(3, "MP4文件"),
  EM_REAL_DATA_TYPE_H264(4, "裸H264码流"),
  EM_REAL_DATA_TYPE_FLV_STREAM(5, "流式FLV");
  private int type;
  private String desc;

  public int getType() {
    return type;
  }

  public String getDesc() {
    return desc;
  }

  EM_REAL_DATA_TYPE(int type, String desc) {
    this.type = type;
    this.desc = desc;
  }

  /**
   * 根据type返回码流类型,默认返回私有码流
   *
   * @param type 码流类型
   * @return
   */
  public static EM_REAL_DATA_TYPE getRealDataType(int type) {
    for (EM_REAL_DATA_TYPE dataType : EM_REAL_DATA_TYPE.values()) {
      if (dataType.getType() == type) {
        return dataType;
      }
    }
    return EM_REAL_DATA_TYPE_PRIVATE;
  }

  /**
   * 根据码流类型返回枚举,如私有、ps、ts、mp4、h264、flv
   *
   * @param streamType
   * @return
   */
  public static EM_REAL_DATA_TYPE getRealDataType(String streamType) {
    for (EM_REAL_DATA_TYPE dataType : EM_REAL_DATA_TYPE.values()) {
      if (dataType.getDesc().toLowerCase().contains(streamType.toLowerCase().trim())) {
        return dataType;
      }
    }
    return EM_REAL_DATA_TYPE_PRIVATE;
  }
}