sources/apps/sample_apps/deepstream-transfer-learning-app/capture_time_rules.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed 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.
 */
 
#pragma once
 
#include <string>
#include <iostream>
#include <chrono>
#include <vector>
 
class CaptureTimeRules {
    typedef std::chrono::time_point<std::chrono::system_clock> t_time_pt;
    typedef std::chrono::duration<unsigned long long> t_duration;
public:
    void init(const std::string &path, unsigned default_second_interval);
 
    t_duration getCurrentTimeInterval();
 
    bool is_init_();
 
private:
    struct TimeRule {
        unsigned begin_time_hour;
        unsigned begin_time_minute;
        unsigned end_time_hour;
        unsigned end_time_minute;
        unsigned interval_between_frame_capture_seconds;
        bool end_time_is_next_day;
    };
 
    static bool isInTimeRule(const TimeRule &t, const tm &now);
 
    enum ParseResult{
        PARSE_RESULT_OK,
        PARSE_RESULT_BAD_CHARS,
        PARSE_RESULT_OUT_OF_BOUND,
        PARSE_RESULT_EMPTY
    };
 
    static ParseResult stoi_err_handling(unsigned& dst, const std::string &src, unsigned max_bound);
    static bool parsing_contains_error(const std::vector<ParseResult>& parse_res_list,
            const std::vector<std::string>& str_list, const std::string& curr_line,
                             unsigned line_number);
    bool single_time_rule_parser(const std::string &path, const std::string &line,
                                 unsigned line_number);
 
    std::chrono::seconds default_duration_;
    t_time_pt end_of_current_time_interval_;
    t_duration current_time_interval_;
    std::vector<TimeRule> rules_;
    bool init_ = false;
};