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
// Copyright (c) 2012-2017 VideoStitch SAS
// Copyright (c) 2018 stitchEm
#pragma once
#include "fakeReader.hpp"
#include <common/container.hpp>
#include <gpu/core1/transform.hpp>
#include "libvideostitch/ptv.hpp"
#include "libvideostitch/parse.hpp"
#include "libvideostitch/panoDef.hpp"
#include <fstream>
#include <iostream>
#include <atomic>
namespace VideoStitch {
namespace Testing {
/**
* Creates a Ptv value given a config. Dies on error.
*/
inline Ptv::Value* makePtvValue(const std::string& ptvString) {
Potential<Ptv::Parser> parser = Ptv::Parser::create();
ENSURE(parser->parseData(ptvString), "Cannot parse ptv data");
return parser->getRoot().clone();
}
inline Status prepareTransforms(const Core::PanoDefinition& panoDef,
std::map<int, std::unique_ptr<Core::Transform>>& transforms) {
std::atomic<int> readFrameCalls(0), readFrameExits(0);
std::unique_ptr<FakeReaderFactory> fakeReaderFactory =
std::unique_ptr<FakeReaderFactory>(new FakeReaderFactory(0, &readFrameCalls, &readFrameExits));
std::map<readerid_t, std::unique_ptr<Input::VideoReader>> readers;
for (readerid_t in = 0; in < panoDef.numInputs(); ++in) {
Potential<Input::Reader> reader = fakeReaderFactory->create(in, panoDef.getInput(in));
FAIL_RETURN(reader.status());
Input::VideoReader* videoReader = reader.release()->getVideoReader();
if (videoReader) {
readers[in] = std::unique_ptr<Input::VideoReader>(videoReader);
}
}
// Create transforms
transforms.clear();
for (auto& reader : readers) {
const Core::InputDefinition& inputDef = panoDef.getInput(reader.second->id);
Core::Transform* transform = Core::Transform::create(inputDef);
if (!transform) {
return {Origin::Stitcher, ErrType::SetupFailure,
"Cannot create v1 transformation for input " + std::to_string(reader.second->id)};
}
transforms[reader.second->id] = std::unique_ptr<Core::Transform>(transform);
}
return Status::OK();
}
inline Status prepareFakeReader(const Core::PanoDefinition& panoDef,
std::map<readerid_t, Input::VideoReader*>& readers) {
deleteAllValues(readers);
std::atomic<int> readFrameCalls(0), readFrameExits(0);
std::unique_ptr<FakeReaderFactory> fakeReaderFactory =
std::unique_ptr<FakeReaderFactory>(new FakeReaderFactory(0, &readFrameCalls, &readFrameExits));
for (readerid_t in = 0; in < panoDef.numInputs(); ++in) {
Potential<Input::Reader> reader = fakeReaderFactory->create(in, panoDef.getInput(in));
FAIL_RETURN(reader.status());
Input::VideoReader* videoReader = reader.release()->getVideoReader();
if (videoReader) {
readers[in] = videoReader;
}
}
return Status::OK();
}
} // namespace Testing
} // namespace VideoStitch