time.h

Go to the documentation of this file.

 
/*
Copyright (c) 2026 MrXie1109
 
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
 
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
 
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
*/
 
#pragma once
#include <chrono>
#include <cstdint>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <string>
#include <thread>
#include <utility>
 
#include "../core/csexc.h"
#include "../io/kb.h"
 
namespace console {
    class Time {
        intmax_t ns_; 
 
    public:
        Time(long long ns = 0) : ns_(ns) {}

        intmax_t count() const { return ns_; }

        operator intmax_t() const { return ns_; }

        double ns() const { return ns_; }
        double us() const { return ns_ / 1e3; }
        double ms() const { return ns_ / 1e6; }
        double s() const { return ns_ / 1e9; }
        double min() const { return ns_ / 6e10; }
        double hr() const { return ns_ / 36e11; }

        friend std::ostream &operator<<(std::ostream &os, const Time t) {
            if (t.ns_ >= 3600000000000LL) return os << t.hr() << "hr";
            if (t.ns_ >= 60000000000LL) return os << t.min() << "min";
            if (t.ns_ >= 1000000000LL) return os << t.s() << "s";
            if (t.ns_ >= 1000000LL) return os << t.ms() << "ms";
            if (t.ns_ >= 1000LL) return os << t.us() << "μs";
            return os << t.ns() << "ns";
        }

        std::string str() const {
            std::ostringstream os;
            os << *this;
            return os.str();
        }

        friend Time operator+(Time t1, Time t2) { return t1.ns_ + t2.ns_; }
        friend Time operator-(Time t1, Time t2) { return t1.ns_ - t2.ns_; }
        Time        operator*(double d) const { return ns_ * d; }
        Time        operator/(double d) const { return ns_ / d; }

        friend bool operator==(Time t1, Time t2) { return t1.ns_ == t2.ns_; }
        friend bool operator!=(Time t1, Time t2) { return t1.ns_ != t2.ns_; }
        friend bool operator<(Time t1, Time t2) { return t1.ns_ < t2.ns_; }
        friend bool operator>(Time t1, Time t2) { return t1.ns_ > t2.ns_; }
        friend bool operator<=(Time t1, Time t2) { return t1.ns_ <= t2.ns_; }
        friend bool operator>=(Time t1, Time t2) { return t1.ns_ >= t2.ns_; }
 
        template <class Rep, class Period>
        friend Time operator+(Time t, std::chrono::duration<Rep, Period> d) {
            return t
                   + Time(
                       std::chrono::duration_cast<std::chrono::nanoseconds>(d)
                           .count());
        }
 
        template <class Rep, class Period>
        friend Time operator-(Time t, std::chrono::duration<Rep, Period> d) {
            return t
                   - Time(
                       std::chrono::duration_cast<std::chrono::nanoseconds>(d)
                           .count());
        }
    };

    inline Time nanoseconds(double n) {
        return Time(n);
    }

    inline Time microseconds(double n) {
        return Time(n * 1e3);
    }

    inline Time milliseconds(double n) {
        return Time(n * 1e6);
    }

    inline Time seconds(double n) {
        return Time(n * 1e9);
    }

    inline Time minutes(double n) {
        return Time(n * 6e10);
    }

    inline Time hours(double n) {
        return Time(n * 3.6e12);
    }

    inline Time now() {
        return Time( //
            std::chrono::duration_cast<std::chrono::nanoseconds>(
                std::chrono::system_clock::now().time_since_epoch())
                .count());
    }

    inline Time operator""_ns(unsigned long long ns) {
        return Time(ns);
    }

    inline Time operator""_us(unsigned long long us) {
        return Time(us * 1e3);
    }

    inline Time operator""_ms(unsigned long long ms) {
        return Time(ms * 1e6);
    }

    inline Time operator""_s(unsigned long long s) {
        return Time(s * 1e9);
    }

    inline Time operator""_min(unsigned long long min) {
        return Time(min * 6e10);
    }

    inline Time operator""_hr(unsigned long long hr) {
        return Time(hr * 3.6e12);
    }

    inline Time operator""_ns(long double ns) {
        return Time(ns);
    }

    inline Time operator""_us(long double us) {
        return Time(us * 1e3);
    }

    inline Time operator""_ms(long double ms) {
        return Time(ms * 1e6);
    }

    inline Time operator""_s(long double s) {
        return Time(s * 1e9);
    }

    inline Time operator""_min(long double min) {
        return Time(min * 6e10);
    }

    inline Time operator""_hr(long double hr) {
        return Time(hr * 3.6e12);
    }

    template <class F, class... Args>
    inline Time timer(F &&f, Args &&...args) {
        Time start = now();
        f(args...);
        return now() - start;
    }

    inline void sleep(const Time &tr) {
        std::this_thread::sleep_for(std::chrono::duration<long double>(tr.s()));
    }

    inline std::string datetime(const std::string &fmt = "%Y-%m-%d %H:%M:%S") {
        std::stringstream ss;
        auto              now  = std::chrono::system_clock::now();
        auto              time = std::chrono::system_clock::to_time_t(now);
        std::tm           tm_buffer;
#ifdef _WIN32
        localtime_s(&tm_buffer, &time);
#else
        localtime_r(&time, &tm_buffer);
#endif
        ss << std::put_time(&tm_buffer, fmt.c_str());
        return ss.str();
    }

    inline double fps(double target) {
        using namespace std::chrono;
        thread_local auto last = steady_clock::now(); //于首次调用时初始化
        std::this_thread::sleep_until(last + duration<double>(1.0 / target));
        double actual
            = 1.0 / duration<double>(steady_clock::now() - last).count();
        last = steady_clock::now();
        return actual;
    }
 
    class DateTime {
        std::tm tm_{};
 
        void normalize() {
            auto time = std::mktime(&tm_);
            if (time != (std::time_t)-1) {
#ifdef _WIN32
                localtime_s(&tm_, &time);
#else
                localtime_r(&time, &tm_);
#endif
            }
        }
 
    public:
        DateTime() {
            auto        now  = std::chrono::system_clock::now();
            std::time_t time = std::chrono::system_clock::to_time_t(now);
#ifdef _WIN32
            localtime_s(&tm_, &time);
#else
            localtime_r(&time, &tm_);
#endif
        }
 
        DateTime(int year,
            int      month,
            int      day,
            int      hour   = 0,
            int      minute = 0,
            int      second = 0) {
            tm_.tm_year  = year - 1900;
            tm_.tm_mon   = month - 1;
            tm_.tm_mday  = day;
            tm_.tm_hour  = hour;
            tm_.tm_min   = minute;
            tm_.tm_sec   = second;
            tm_.tm_isdst = -1;
            normalize();
        }
 
        DateTime(const std::string &str,
            const std::string      &fmt = "%Y-%m-%d %H:%M:%S") {
            std::istringstream ss(str);
            ss >> std::get_time(&tm_, fmt.c_str());
            if (ss.fail()) throw BadFormat("Failed to parse DateTime");
        }
 
        DateTime(const std::wstring &str,
            const std::wstring      &fmt = L"%Y-%m-%d %H:%M:%S") {
            std::wistringstream ss(str);
            ss >> std::get_time(&tm_, fmt.c_str());
            if (ss.fail()) throw BadFormat("Failed to parse DateTime");
        }
 
        int  year() const { return tm_.tm_year + 1900; }
        int  month() const { return tm_.tm_mon + 1; }
        int  day() const { return tm_.tm_mday; }
        int  hour() const { return tm_.tm_hour; }
        int  minute() const { return tm_.tm_min; }
        int  second() const { return tm_.tm_sec; }
        int  weekday() const { return tm_.tm_wday; }
        int  yearday() const { return tm_.tm_yday; }
        bool isdst() const { return tm_.tm_isdst > 0; }

        static bool is_leap(int y) {
            return (y % 4 == 0 && y % 100 != 0) || y % 400 == 0;
        }
 
        bool is_leap() const { return is_leap(year()); }

        static int days_in_month(int y, int m) {
            static const int d[]
                = {31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
            if (m == 2 && is_leap(y)) return 29;
            return d[m - 1];
        }
        int days_in_month() const { return days_in_month(year(), month()); }

        int iso_weekday() const {
            int w = tm_.tm_wday;
            return w == 0 ? 7 : w;
        }

        int iso_week() const {
            // 取当年 1 月 4 日所在的周一为准
            DateTime jan4(year(), 1, 4);
            int      dow      = jan4.iso_weekday();
            int      week_num = (yearday() - jan4.yearday() + dow + 6) / 7;
            if (week_num < 1)
                week_num = DateTime(year() - 1, 12, 31).iso_week();
            if (week_num > 52 && days_in_month() == 31 && month() == 12
                && iso_weekday() <= 3)
                week_num = 1;
            return week_num;
        }

        DateTime begin_of_month() const { return DateTime(year(), month(), 1); }

        DateTime end_of_month() const {
            return DateTime(year(), month(), days_in_month(), 23, 59, 59);
        }

        DateTime begin_of_day() const {
            return DateTime(year(), month(), day());
        }

        DateTime end_of_day() const {
            return DateTime(year(), month(), day(), 23, 59, 59);
        }

        bool is_weekend() const { return tm_.tm_wday == 0 || tm_.tm_wday == 6; }

        int day_of_year() const { return tm_.tm_yday + 1; }

        std::string str(const std::string &fmt = "%Y-%m-%d %H:%M:%S") const {
            std::ostringstream ss;
            ss << std::put_time(&tm_, fmt.c_str());
            return ss.str();
        }

        std::wstring
        wstr(const std::wstring &fmt = L"%Y-%m-%d %H:%M:%S") const {
            std::wostringstream ss;
            ss << std::put_time(&tm_, fmt.c_str());
            return ss.str();
        }

        std::time_t to_time_t() const {
            return std::mktime(const_cast<std::tm *>(&tm_));
        }


        DateTime &operator+=(Time t) {
            std::time_t sec = static_cast<std::time_t>(t.s());
            auto        tt  = to_time_t() + sec;
#ifdef _WIN32
            localtime_s(&tm_, &tt);
#else
            localtime_r(&tt, &tm_);
#endif
            return *this;
        }

        DateTime &operator-=(Time t) { return *this += Time(-t.s() * 1e9); }
 
        friend DateTime operator+(DateTime dt, Time t) { return dt += t; }
 
        friend DateTime operator-(DateTime dt, Time t) { return dt -= t; }

        friend Time operator-(const DateTime &lhs, const DateTime &rhs) {
            return Time(static_cast<long long>(
                std::difftime(lhs.to_time_t(), rhs.to_time_t()) * 1e9));
        }


        friend bool operator==(const DateTime &lhs, const DateTime &rhs) {
            return std::difftime(lhs.to_time_t(), rhs.to_time_t()) == 0;
        }
        friend bool operator!=(const DateTime &lhs, const DateTime &rhs) {
            return !(lhs == rhs);
        }
        friend bool operator<(const DateTime &lhs, const DateTime &rhs) {
            return std::difftime(lhs.to_time_t(), rhs.to_time_t()) < 0;
        }
        friend bool operator>(const DateTime &lhs, const DateTime &rhs) {
            return rhs < lhs;
        }
        friend bool operator<=(const DateTime &lhs, const DateTime &rhs) {
            return !(lhs > rhs);
        }
        friend bool operator>=(const DateTime &lhs, const DateTime &rhs) {
            return !(lhs < rhs);
        }


        friend std::ostream &operator<<(std::ostream &os, const DateTime &dt) {
            return os << dt.str();
        }

        friend std::wostream &
        operator<<(std::wostream &os, const DateTime &dt) {
            return os << dt.wstr();
        }
    };
}