-
Notifications
You must be signed in to change notification settings - Fork 48
Expand file tree
/
Copy pathSnapshotTest.cpp
More file actions
248 lines (224 loc) · 8.8 KB
/
SnapshotTest.cpp
File metadata and controls
248 lines (224 loc) · 8.8 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
#include <imgui.h>
#include <application.h>
#include <imgui_helper.h>
#include <ImGuiFileDialog.h>
#include <string>
#include <sstream>
#include <MatUtilsImVecHelper.h>
#include "Overview.h"
#include "Snapshot.h"
#include "FFUtils.h"
#include "TextureManager.h"
#include "HwaccelManager.h"
#include "Logger.h"
#include "DebugHelper.h"
using namespace std;
using namespace MediaCore;
using namespace RenderUtils;
using namespace Logger;
static Overview::Holder g_movr;
static Snapshot::Generator::Holder g_ssgen;
static Snapshot::Viewer::Holder g_ssvw1;
static double g_windowPos = 0.f;
static double g_windowSize = 300.f;
static double g_windowFrames = 14.0f;
static MatUtils::Size2i g_v2SsDisplaySize;
static TextureManager::Holder g_txmgr;
static string g_snapTxPoolName = "SnapshotGridTexturePool";
const string c_imguiIniPath = "ms_test.ini";
const string c_bookmarkPath = "bookmark.ini";
static bool g_isImageSequence = false;
static MediaParser::Holder g_mediaParser;
// Application Framework Functions
static void MediaSnapshot_Initialize(void** handle)
{
GetDefaultLogger()
->SetShowLevels(DEBUG);
MediaParser::GetLogger()
->SetShowLevels(INFO);
Snapshot::GetLogger()
->SetShowLevels(INFO);
g_txmgr = TextureManager::CreateInstance();
#ifdef USE_PLACES_FEATURE
// load bookmarks
ifstream docFile(c_bookmarkPath, ios::in);
if (docFile.is_open())
{
stringstream strStream;
strStream << docFile.rdbuf(); //read the file
ImGuiFileDialog::Instance()->DeserializePlaces(strStream.str());
docFile.close();
}
#endif
ImGuiIO& io = ImGui::GetIO();
io.IniFilename = c_imguiIniPath.c_str();
size_t ssCnt = (size_t)ceil(g_windowFrames)+1;
g_movr = Overview::CreateInstance();
g_movr->SetSnapshotSize(320, 180);
g_ssgen = Snapshot::Generator::CreateInstance();
g_ssgen->SetLogLevel(DEBUG);
// g_ssgen->SetSnapshotResizeFactor(0.5f, 0.5f);
g_ssgen->SetCacheFactor(3);
g_ssvw1 = g_ssgen->CreateViewer(0);
HwaccelManager::GetDefaultInstance()->Init();
}
static void MediaSnapshot_Finalize(void** handle)
{
g_ssgen->ReleaseViewer(g_ssvw1);
g_ssgen = nullptr;
g_movr = nullptr;
#ifdef USE_PLACES_FEATURE
// save bookmarks
ofstream configFileWriter(c_bookmarkPath, ios::out);
if (!configFileWriter.bad())
{
configFileWriter << ImGuiFileDialog::Instance()->SerializePlaces();
configFileWriter.close();
}
#endif
g_txmgr->Release();
g_txmgr = nullptr;
}
static bool MediaSnapshot_Frame(void * handle, bool app_will_quit)
{
bool app_done = false;
auto& io = ImGui::GetIO();
ImGui::SetNextWindowPos({0, 0});
ImGui::SetNextWindowSize(io.DisplaySize);
if (ImGui::Begin("MainWindow", nullptr, ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoResize))
{
if (ImGui::Button((string(ICON_IGFD_FOLDER_OPEN)+" Open file").c_str()))
{
const char *filters = "视频文件(*.mp4 *.mov *.mkv *.webm *.avi *.mxf){.mp4,.mov,.mkv,.webm,.avi,.mxf,.MP4,.MOV,.MKV,.WEBM,.AVI,.MXF},.*";
IGFD::FileDialogConfig config;
config.path = "~/Videos/";
config.countSelectionMax = 1;
config.flags = ImGuiFileDialogFlags_OpenFile_Default;
ImGuiFileDialog::Instance()->OpenDialog("ChooseFileDlgKey", ICON_IGFD_FOLDER_OPEN " 打开视频文件",
filters,
config);
}
ImGui::SameLine();
ImGui::Checkbox("Open image sequence", &g_isImageSequence);
ImGui::SameLine();
if (ImGui::Button("Refresh snapwnd configuration"))
g_ssgen->ConfigSnapWindow(g_windowSize, g_windowFrames, true);
ImGui::Spacing();
float pos = g_windowPos;
float minPos = (float)g_ssgen->GetVideoMinPos()/1000.f;
float vidDur = (float)g_ssgen->GetVideoDuration()/1000.f;
if (ImGui::SliderFloat("Position", &pos, minPos, minPos+vidDur, "%.3f"))
{
g_windowPos = pos;
}
float wndSize = g_windowSize;
float minWndSize = (float)g_ssgen->GetMinWindowSize();
float maxWndSize = (float)g_ssgen->GetMaxWindowSize();
if (ImGui::SliderFloat("WindowSize", &wndSize, minWndSize, maxWndSize, "%.3f"))
g_windowSize = wndSize;
if (ImGui::IsItemDeactivated())
g_ssgen->ConfigSnapWindow(g_windowSize, g_windowFrames);
ImGui::Spacing();
vector<Snapshot::Image> snapshots;
// auto t0 = GetTimePoint();
bool ret = g_ssvw1->GetSnapshots(pos, snapshots);
// auto t1 = GetTimePoint();
// Log(WARN) << "<TimeCost> GetSnapshots() : " << CountElapsedMillisec(t0, t1) << endl;
if (!ret)
snapshots.clear();
else
g_ssvw1->UpdateSnapshotTexture(snapshots, g_txmgr, g_snapTxPoolName);
float startPos = minPos;
if (snapshots.size() > 0 && snapshots[0].hDispData && snapshots[0].hDispData->mTimestampMs != INT64_MIN)
startPos = (float)snapshots[0].hDispData->mTimestampMs/1000;
int snapshotCnt = (int)ceil(g_windowFrames);
for (int i = 0; i < snapshotCnt; i++)
{
const auto v2DispSize = MatUtils::ToImVec2(g_v2SsDisplaySize);
ImGui::BeginGroup();
if (i >= snapshots.size())
{
ImGui::Dummy(v2DispSize);
ImGui::TextUnformatted("No image");
}
else
{
auto& hDispData = snapshots[i].hDispData;
string tag = hDispData->mTimestampMs != INT64_MIN ? MillisecToString(hDispData->mTimestampMs) : "N/A";
auto hTx = hDispData->mTextureReady ? hDispData->mhTx : nullptr;
ImTextureID tid = hTx ? hTx->TextureID() : nullptr;
if (!tid)
{
ImGui::Dummy(v2DispSize);
tag += "(loading)";
}
else
{
auto roiRect = hTx->GetDisplayRoi();
ImGui::Image(tid, v2DispSize, MatUtils::ToImVec2(roiRect.leftTop), MatUtils::ToImVec2(roiRect.rightBottom()));
}
ImGui::TextUnformatted(tag.c_str());
}
ImGui::EndGroup();
ImGui::SameLine();
}
ImGui::End();
}
// open file dialog
ImVec2 modal_center(io.DisplaySize.x * 0.5f, io.DisplaySize.y * 0.5f);
ImVec2 maxSize = ImVec2((float)io.DisplaySize.x, (float)io.DisplaySize.y);
ImVec2 minSize = maxSize * 0.5f;
if (ImGuiFileDialog::Instance()->Display("ChooseFileDlgKey", ImGuiWindowFlags_NoCollapse, minSize, maxSize))
{
if (ImGuiFileDialog::Instance()->IsOk())
{
g_ssgen->Close();
g_txmgr->ReleaseTexturePool(g_snapTxPoolName);
string filePathName = ImGuiFileDialog::Instance()->GetFilePathName();
g_mediaParser = MediaParser::CreateInstance();
if (g_isImageSequence)
g_mediaParser->OpenImageSequence({25, 1}, filePathName, ".+_([[:digit:]]{1,})\\.png", false);
else
{
g_mediaParser->Open(filePathName);
g_mediaParser->EnableParseInfo(MediaParser::VIDEO_SEEK_POINTS);
}
g_movr->Open(g_mediaParser, 20);
g_ssgen->Open(g_mediaParser);
g_ssgen->SetOverview(g_movr);
const auto pVidstm = g_ssgen->GetVideoStream();
g_v2SsDisplaySize.x = (int32_t)(io.DisplaySize.x/(g_windowFrames+1));
g_v2SsDisplaySize.y = (int32_t)(g_v2SsDisplaySize.x*pVidstm->height/pVidstm->width);
g_ssgen->SetSnapshotSize(g_v2SsDisplaySize.x, g_v2SsDisplaySize.y);
g_txmgr->CreateGridTexturePool(g_snapTxPoolName, g_v2SsDisplaySize, IM_DT_INT8, {16, 9}, 0);
g_windowPos = (float)g_ssgen->GetVideoMinPos()/1000.f;
g_windowSize = (float)g_ssgen->GetVideoDuration()/10000.f;
g_ssgen->ConfigSnapWindow(g_windowSize, g_windowFrames);
}
ImGuiFileDialog::Instance()->Close();
}
if (!io.KeyCtrl && !io.KeyShift && !io.KeyAlt && ImGui::IsKeyPressed(ImGui::GetKeyIndex(ImGuiKey_Escape), false))
{
app_done = true;
}
if (app_will_quit)
{
app_done = true;
}
g_txmgr->UpdateTextureState();
// Log(DEBUG) << g_txmgr.get() << endl;
return app_done;
}
void Application_Setup(ApplicationWindowProperty& property)
{
property.name = "MediaSnapshotTest";
property.viewport = false;
property.docking = false;
property.auto_merge = false;
//property.power_save = false;
property.width = 1280;
property.height = 720;
property.application.Application_Initialize = MediaSnapshot_Initialize;
property.application.Application_Finalize = MediaSnapshot_Finalize;
property.application.Application_Frame = MediaSnapshot_Frame;
}