HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
extComputationUtils.h
Go to the documentation of this file.
1 //
2 // Copyright 2019 Pixar
3 //
4 // Licensed under the Apache License, Version 2.0 (the "Apache License")
5 // with the following modification; you may not use this file except in
6 // compliance with the Apache License and the following modification to it:
7 // Section 6. Trademarks. is deleted and replaced with:
8 //
9 // 6. Trademarks. This License does not grant permission to use the trade
10 // names, trademarks, service marks, or product names of the Licensor
11 // and its affiliates, except as required to comply with Section 4(c) of
12 // the License and to reproduce the content of the NOTICE file.
13 //
14 // You may obtain a copy of the Apache License at
15 //
16 // http://www.apache.org/licenses/LICENSE-2.0
17 //
18 // Unless required by applicable law or agreed to in writing, software
19 // distributed under the Apache License with the above modification is
20 // distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
21 // KIND, either express or implied. See the Apache License for the specific
22 // language governing permissions and limitations under the Apache License.
23 //
24 #ifndef PXR_IMAGING_HD_EXT_COMPUTATION_UTILS_H
25 #define PXR_IMAGING_HD_EXT_COMPUTATION_UTILS_H
26 
27 #include "pxr/pxr.h"
28 #include "pxr/imaging/hd/api.h"
30 
31 #include "pxr/base/tf/token.h"
32 #include "pxr/base/vt/value.h"
33 
34 #include <unordered_map>
35 
37 
39 using HdExtComputationConstPtrVector = std::vector<HdExtComputationConstPtr>;
40 
41 // This class contains utility methods to allow any Hydra backend to execute
42 // CPU computations via the Hydra ExtComputation framework.
43 //
44 // Note:
45 // The computation execution happens during Rprim sync. This precludes the
46 // use of computations shared by multiple Rprims, since the chain of
47 // computations for a computation primvar is executed for each Rprim.
49 public:
50  using ValueStore =
51  std::unordered_map<TfToken, VtValue, TfToken::HashFunctor>;
52 
53  // Returns a map containing the (token, value) pairs for each "computation
54  // primvar".
55  // The participating computations are ordered based on their dependency
56  // and then, the CPU kernel is executed for each computation.
57  HD_API
58  static ValueStore
60  HdExtComputationPrimvarDescriptorVector const& compPrimvars,
61  HdSceneDelegate* sceneDelegate);
62 
63  // Helper methods (these are public for testing purposes)
65  std::unordered_map<HdExtComputation const *,
67  // Returns true if an ordering of the computations wherein any dependencies
68  // of a given computation come before it is possible, and fills
69  // sortedComps with the ordering.
70  // Returns false otherwise.
71  // The directed graph of a computation (vertex) and its dependencies (edges)
72  // is represented via the ComputationDependencyMap.
73  HD_API
74  static bool
76  HdExtComputationConstPtrVector* sortedComps);
77 
78  HD_API
79  static void
81 };
82 
84 
85 #endif // PXR_IMAGING_HD_EXT_COMPUTATION_UTILS_H
#define HD_API
Definition: api.h:40
static HD_API void PrintDependencyMap(ComputationDependencyMap const &cdm)
std::unordered_map< HdExtComputation const *, HdExtComputationConstPtrVector > ComputationDependencyMap
static HD_API ValueStore GetComputedPrimvarValues(HdExtComputationPrimvarDescriptorVector const &compPrimvars, HdSceneDelegate *sceneDelegate)
PXR_NAMESPACE_CLOSE_SCOPE PXR_NAMESPACE_OPEN_SCOPE
Definition: path.h:1346
std::unordered_map< TfToken, VtValue, TfToken::HashFunctor > ValueStore
#define PXR_NAMESPACE_CLOSE_SCOPE
Definition: pxr.h:91
static HD_API bool DependencySort(ComputationDependencyMap cdm, HdExtComputationConstPtrVector *sortedComps)
std::vector< HdExtComputationConstPtr > HdExtComputationConstPtrVector
HdExtComputation const * HdExtComputationConstPtr
std::vector< HdExtComputationPrimvarDescriptor > HdExtComputationPrimvarDescriptorVector