logplus/Workflow/WFCrystal/SysUtility/utils/include/Utils.h

/**
 * @file Utils.h
 */
#ifndef PAI_FRAME_SYSUTILITY_UTILS_H
#define PAI_FRAME_SYSUTILITY_UTILS_H

#ifdef LLToString
#undef LLToString
#endif


#include <cstdlib>
#include <cstdio>
#include <vector>
#include <string>
#include <sstream>
#include <map>
#include <set>
#include <string.h>
#include <stdarg.h>
//#include<sys/time.h>
#include <cfloat>
#include <cmath>
#include "Turtle.h"
#ifdef LLToString
#undef LLToString
#endif

std::string PAI_UTILS_EXPORT GetDBPath();

typedef long long int64;
using namespace pai::turtle;
namespace pai {
namespace utils {

	static const int CONST_FORMAT_BUFFER_SIZE = 4096;

static const float F_EPSILON = static_cast<float>(1e-6);
static const double D_EPSILON = 1e-6;

class PAI_UTILS_EXPORT CUtils
{
public:
    /**
     * @brief 构造函数
     */
    CUtils();

    /**
     * @brief 析构函数
     */
    ~CUtils();

    /**
     * 获取当前时间,格式为"YYYY-MM-DD HH:MM:SS.MMM"
     */
    static std::string getCurrentDate();
    /**
	 * @brief Get C Format String
	 * @param[in] fmt format string
	 * @param[in] args
	 * @return formated string
	 */
    static std::string GetFormatStr(const std::string& fmt,...){
    	char buf[CONST_FORMAT_BUFFER_SIZE];
    	std::stringstream ss;
    	va_list ap;
    	va_start(ap, fmt);
    	vsprintf(buf, fmt.c_str(), ap);
    	va_end(ap);
    	ss << buf << std::endl;
    	return ss.str();
    }

    /**
     * @brief Get environment path of "PAI_HOME"
     * @return the environment path of "PAI_HOME"
     */
    static std::string GetPaiHome();

    /**
     * @brief string类型变量转换为int类型变量
     * @param[in] s string类型变量
     * @return int型变量
     *
     */
    static int StrToInt(const std::string& s)
    {
        return std::atoi(s.c_str());
    }

    /**
     * @brief int类型变量转换为string类型变量
     * @param[in] value int类型变量
     * @return string型变量
     *
     */
    static std::string IntToStr(long long value)
    {
        char buff[64];
        std::sprintf(buff, "%lld", value);
        return std::string(buff);
    }

    /**
     * @brief string类型变量转换为long long类型变量
     * @param[in] s string类型变量
     * @return long long型变量
     *
     */
    static long long StrToLL(const std::string& s)
    {
    	return GetTurtleAPI()->StringToLong(s);
    }

    /**
     * @brief long long类型变量转换为string类型变量
     * @param[in] value long long类型变量
     * @return string型变量
     *
     */
    static std::string LLToString(long long value)
    {
        char buff[64];
        std::sprintf(buff, "%lld", value);
        return std::string(buff);
    }

    /**
     * @brief string类型变量转换为float类型变量
     * @param[in] s string类型变量
     * @return float型变量
     *
     */
    static float StrToFloat(const std::string& s)
    {
        return static_cast<float>(std::atof(s.c_str()));
    }

    /**
     * @brief string类型变量转换为float类型变量
     * @param[in] s string类型变量
     * @return double型变量
     *
     */
    static double StrToDouble(const std::string& s)
    {
        return std::atof(s.c_str());
    }

    /**
     * @brief float类型变量转换为string类型变量
     * @param[in] value float类型变量
     * @return string型变量
     *
     */
    static std::string FloatToStr(double value)
    {
        char buff[128];
        std::sprintf(buff, "%f", value);
        std::string ret = std::string(buff);
        size_t size = ret.size();
        std::string s = ret;
        while(ret[size-1] == '0')
        {
            s.erase(size-1,1);
            size--;
        }
        if (ret[size-1] == '.')
        {
            s.erase(size-1,1);
            size--;
        }
        return s;
    }

    /**
     * @brief float类型变量转换为string类型变量
     * @param[in] value float类型变量
     * @return string型变量
     *
     */
    static std::string FloatToStr(double value,int digits)
    {
        char buff[128];
        std::stringstream ss;
        ss << "%0." << digits << "lf";
        std::sprintf(buff, ss.str().c_str(), value);
        return buff;
    }

    /**
     * @brief 比较两个float类型是否相等
     * @param[in] value float类型变量
     * @return 相等true 否则返回false
     */
    static bool CompareEqual(float value1,float value2)
    {
    	if(fabs(value1-value2) < F_EPSILON)
    		return true;
    	else
    		return false;
    }

    /**
     * @brief 比较两个double类型是否相等
     * @param[in] value double类型变量
     * @return 相等true 否则返回false
     */
    static bool CompareEqual(double value1,double value2)
    {
    	if(fabs(value1-value2) < D_EPSILON)
    		return true;
    	else
    		return false;
    }
    /**
     * @brief 设置classpath环境变量
     *
     * @return 成功返回0
     *         失败返回-1
     */
    static void SetHadoopClassPath();

    /**
     * @brief 将数字的容量大小装换为格式化的字符串
     *
     * 如：1=1KB  1024=1MB
     *
     * @param[in] size 文件大小
     */
    static std::string DigitSizeToString(long long size);

    /**
     * @brief 获取本地IP
     *
     * @param[out] 本地IP地址
     */
    // static std::string GetLocalIp();

    /**
     * @brief 判断指定进程ID是否存在
     * @param[in] pID 指定的进程ID
     * @param[out] true 存在正在执行 false 不存在
     */
    static bool ProcessExist(pid_t pID);

    /**
     * @brief 将经过的时间（以秒计算），转换为方便人类识别的字符串
     * @param[in] seconds 经过的时间，单位为秒
     * @return 方便识别的字符串
     */
    static std::string TimeToString(int64 seconds);

    /**
     * @brief 将友好格式的经过时间字符串，转换为秒
     * @param[in] timeStr 友好格式的经过时间字符串
     * @return 经过的时间（以秒计算）
     */
    static int64 TimeStringToSeconds(const std::string &timeStr);

    /**
     * 获取当前系统时间(精确到微秒)
    */
    static int64 getSystemTime();
    /**
     * @brief 获取当前系统时间(精确到纳秒)
     * @param[out] time 获取到的纳秒级计数
     * @return cpu频率支持纳秒精度 返回true ,否则返回false.
    */
    static bool getNanoTime(long long &time);

    //----------------SeisUtil-------------------------


    //----------------SegyUtil--------------------------
    enum EIBM2IEEE
    {
        IBM_TO_IEEE = 0, IEEE_TO_IBM = 1
    };

    static unsigned char ebasc(unsigned char ascii);
    static void ebasd(unsigned char* ascii, unsigned char* ebcd);
    static void asebd(char* ebcd, char* ascii);
    static char asebc(unsigned char ascii);

    static void SwapBhdr(unsigned char *buf);
    static void SwapTraceHead(unsigned char *buf);
    static void SwapTxtHead(unsigned char *buf);
    static void ChrToIEEE(unsigned char *src, float *dst, int len);
    static void SpiToIEEE(short *src, float *dst, int len);
    static void BgnToIEEE(short *src, float *dst, int len);
    static void IntToIEEE(int *src, float *dst, int len);
    static void IEEEToChr(float *src, char *dst, int len);
    static void IEEEToSpi(float *src, short *dst, int len, short &scalar);
    static void IEEEToInt(float *src, int *dst, int len);
    static void IEEEToBgn(float *src, short *dst, int len);
    static void IbmToIEEE(int *src, float *dst, int n);
    static void IEEEToIbm(float *src, int *dst, int n);
    static void SwapByte(unsigned char *buf, int len, int flag);


    /**
     * @brief 根据分割符合并字符串
     * @return 合并后的字符串
     */
    static std::string ConcatWithChar(std::vector<long>& source, char sep);
    static std::string ConcatWithChar(std::set<long>& source, char sep);
    static std::string ConcatWithChar(std::map<long, long>& source, char sep1, char sep2);
    static std::string ConcatWithChar(std::map<int, std::string>& source, char sep1, char sep2);
    static std::string ConcatWithChar(std::map<long long, std::string>& source, char sep1, char sep2);

    /**
     * @brief 根据传入的字符替换字符串中的所有匹配字符
     */
    static std::string Replace_all(const std::string & str, const char old_value, const char new_value);

    /**
     * @brief 去掉给定字符串前后的空格
     * @param[in] str 操作串
     *
     * @return 去掉空格后的字符串
     */
    static std::string Trim(std::string str);
private:

    static void AppendClassPath(const std::string& path, std::string & classPath);

    /**
     * @return Number of matching items.
     */
    static int ListFiles(const std::string& path, const std::string& filter, std::vector<std::string>& file);

    //-----------------SegyUtil------------------
    static void IbmIeee(float *src, float *dst, int n, int idirec);
    static void IeeeIbm(char *from, char *to,int len, int type);
};

/** @brief Handle class that supports RAII to wrap FILE pointers.
 * Can be used exactly as FILE pointers (or as if it were FILE pointer itself)
 * but users don't need to care about closing File_ptr as it will be closed
 * automatically when object will go out of it's scope. Also implicit
 * convertions from/to FILE pointers are provided.
 */
class PAI_UTILS_EXPORT File_ptr {
public:
    /** @brief constructor that opens file
     * @param[in] n file name to open a file
     * @param[in] a file opening mode
     */
    File_ptr(const char * n, const char * a);
    File_ptr(FILE * pp);
    ~File_ptr() ;
    operator FILE*() ;
private:
    File_ptr(File_ptr& fp);
    File_ptr& operator=(File_ptr & fp);
    FILE * p;
};

template <class C>
void SplitByChar(const std::string& s, char sep, C& result, bool allowEmpty = true)
{
    size_t len = s.size();
    std::string line;
    for (size_t i = 0; i < len; i++)
    {
        char c = s[i];
        if (c == sep)
        {
            if (!allowEmpty)
            {
                if (!line.empty())
                    result.push_back(line);
            } else
            {
                result.push_back(line);
            }
            line.clear();
        } else
        {
            line.push_back(c);
        }
    }
    if (!line.empty())
    {
        result.push_back(line);
    }
}

}
}
#endif