Added locate features

Author: elly-funique

src/master/GoProMaster.cpp

@@ -301,6 +301,25 @@ void GoProMaster::presetSwitch(const std::string server, const std::string targe
     }).detach();
 }
 
+void GoProMaster::locate(const std::string server, const std::string target){
+    std::lock_guard<std::mutex> lock(locate_mtx);
+    int32_t index = haslocate(server, target);
+    if(index == -1){
+        locates.push_back(std::pair<std::string, std::string>(server, target));
+    }else{
+        locates.erase(locates.begin() + index);
+    }
+}
+
+int32_t GoProMaster::haslocate(const std::string server, const std::string target){
+    std::lock_guard<std::mutex> lock(locate_mtx);
+    for(int32_t i = 0; i < locates.size(); i++){
+        auto& s = locates.at(i);
+        if(s.first == server && s.second == target) return i;
+    }
+    return -1;
+}
+
 void GoProMaster::apply(const std::string& ip, const int32_t id, const int32_t value){
     std::thread([=](){    
         for (auto& s : servers) {
@@ -392,8 +411,13 @@ void GoProMaster::update(){
             mediaQueryFinish.insert_or_assign(s->ip, true);
             s->client.send(get_status.dump());
         }
+
+        std::lock_guard<std::mutex> lock(locate_mtx);
+        for (auto& s : locates) {
+            presetSwitch(s.first, s.second, 0);
+        }
 
-        std::this_thread::sleep_for(std::chrono::seconds(3));
+        std::this_thread::sleep_for(std::chrono::seconds(2));
     }
 }
 

src/master/GoProMaster.h

@@ -89,6 +89,8 @@ class GoProMaster {
     void download_last_media(const std::string dir, bool put_finish);
 
     void presetSwitch(const std::string server, const std::string target, int32_t mode);
+    void locate(const std::string server, const std::string target);
+    int32_t haslocate(const std::string server, const std::string target);
     void apply(const std::string& ip, const int32_t id, const int32_t value);
     void applyAll(const std::string& ip, const json& res);
 
@@ -111,6 +113,7 @@ class GoProMaster {
      * This will prevent race condition
      */
     std::mutex camera_mtx;
+    std::mutex locate_mtx;
 
     // ----------------------------------------------------------
     //
@@ -134,6 +137,10 @@ class GoProMaster {
      * All websocket servers record for master
     */
     std::vector<std::shared_ptr<ServerConnection>> servers;
+    /**
+     * All the locate records
+     */
+    std::vector<std::pair<std::string, std::string>> locates;
     /**
      * Basically a update thread, THere is a while true loop in it,
      * And when done is true, it automatically escape the loop

src/master/windows/commands.cpp

@@ -102,14 +102,17 @@ void CommandWindow::render_local(){
         if(ImGui::Button("Connect", button_4size)) master->command_only(info.server, "usb_on", info.ip); ImGui::SameLine();
         if(ImGui::Button("Disconnect", button_4size)) master->command_only(info.server, "usb_off", info.ip); ImGui::SameLine();
         if(ImGui::Button("Shutdown", button_4size)) master->command_only(info.server, "usb_on", info.ip); ImGui::SameLine();
-        if(ImGui::Button("Locate", button_4size)) master->presetSwitch(info.server, info.ip, 0);
+        int32_t lo = master->haslocate(info.server, info.ip);
+        if(lo == -1){
+            if(ImGui::Button("Locate", button_4size)) master->locate(info.server, info.ip);
+        }else{
+            if(ImGui::Button("UnLocate", button_4size)) master->locate(info.server, info.ip);
+        }
 
         ImGui::Dummy(ImVec2(0, 20));
         ImGui::Separator();
         ImGui::Dummy(ImVec2(0, 20));
 
-    
-
         if(state->current_status_items[std::to_string(PRESET_ID)].is_number_integer()){
 
             int32_t cuccrent_select = state->current_status_items[std::to_string(PRESET_ID)].get<int32_t>();