| 49 | |
| 50 | template<typename Op, typename Src, typename Dst, typename Call> |
| 51 | auto runGuard(Op &op, Src &src, Dst &dst, std::optional<Tensor> &brightness, std::optional<Tensor> &contrast, |
| 52 | std::optional<Tensor> &brightnessShift, std::optional<Tensor> &contrastCenter, |
| 53 | std::optional<Stream> &pstream, Call &&call) |
| 54 | { |
| 55 | if (!pstream) |
| 56 | { |
| 57 | pstream = Stream::Current(); |
| 58 | } |
| 59 | |
| 60 | ResourceGuard guard(*pstream); |
| 61 | guard.add(LockMode::LOCK_MODE_READ, {src}); |
| 62 | for (auto &arg : {brightness, contrast, brightnessShift, contrastCenter}) |
| 63 | { |
| 64 | if (arg) |
| 65 | { |
| 66 | guard.add(LockMode::LOCK_MODE_READ, {*arg}); |
| 67 | } |
| 68 | } |
| 69 | guard.add(LockMode::LOCK_MODE_WRITE, {dst}); |
| 70 | guard.add(LockMode::LOCK_MODE_NONE, {*op}); |
| 71 | |
| 72 | call(*pstream, brightness ? *brightness : nvcv::Tensor{nullptr}, contrast ? *contrast : nvcv::Tensor{nullptr}, |
| 73 | brightnessShift ? *brightnessShift : nvcv::Tensor{nullptr}, |
| 74 | contrastCenter ? *contrastCenter : nvcv::Tensor{nullptr}); |
| 75 | } |
| 76 | |
| 77 | Tensor BrightnessContrastInto(Tensor &dst, Tensor &src, std::optional<Tensor> &brightness, |
| 78 | std::optional<Tensor> &contrast, std::optional<Tensor> &brightnessShift, |
no test coverage detected