Blender  V3.3
node_composite_trackpos.cc
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later
2  * Copyright 2011 Blender Foundation. All rights reserved. */
3 
8 #include "DNA_movieclip_types.h"
9 #include "DNA_tracking_types.h"
10 
11 #include "BKE_context.h"
12 #include "BKE_lib_id.h"
13 #include "BKE_tracking.h"
14 
15 #include "RNA_access.h"
16 #include "RNA_prototypes.h"
17 
18 #include "UI_interface.h"
19 #include "UI_resources.h"
20 
21 #include "node_composite_util.hh"
22 
24 
26 {
27  b.add_output<decl::Float>(N_("X"));
28  b.add_output<decl::Float>(N_("Y"));
29  b.add_output<decl::Vector>(N_("Speed")).subtype(PROP_VELOCITY);
30 }
31 
32 static void init(const bContext *C, PointerRNA *ptr)
33 {
34  bNode *node = (bNode *)ptr->data;
35 
36  NodeTrackPosData *data = MEM_cnew<NodeTrackPosData>(__func__);
37  node->storage = data;
38 
39  const Scene *scene = CTX_data_scene(C);
40  if (scene->clip) {
41  MovieClip *clip = scene->clip;
42  MovieTracking *tracking = &clip->tracking;
43 
44  node->id = &clip->id;
45  id_us_plus(&clip->id);
46 
47  const MovieTrackingObject *tracking_object = BKE_tracking_object_get_active(tracking);
48  BLI_strncpy(data->tracking_object, tracking_object->name, sizeof(data->tracking_object));
49 
50  const MovieTrackingTrack *active_track = BKE_tracking_track_get_active(tracking);
51  if (active_track) {
52  BLI_strncpy(data->track_name, active_track->name, sizeof(data->track_name));
53  }
54  }
55 }
56 
58 {
59  bNode *node = (bNode *)ptr->data;
60 
61  uiTemplateID(layout,
62  C,
63  ptr,
64  "clip",
65  nullptr,
66  "CLIP_OT_open",
67  nullptr,
69  false,
70  nullptr);
71 
72  if (node->id) {
73  MovieClip *clip = (MovieClip *)node->id;
74  MovieTracking *tracking = &clip->tracking;
75  MovieTrackingObject *object;
76  uiLayout *col;
77  PointerRNA tracking_ptr;
79 
80  RNA_pointer_create(&clip->id, &RNA_MovieTracking, tracking, &tracking_ptr);
81 
82  col = uiLayoutColumn(layout, false);
83  uiItemPointerR(col, ptr, "tracking_object", &tracking_ptr, "objects", "", ICON_OBJECT_DATA);
84 
85  object = BKE_tracking_object_get_named(tracking, data->tracking_object);
86  if (object) {
87  PointerRNA object_ptr;
88 
89  RNA_pointer_create(&clip->id, &RNA_MovieTrackingObject, object, &object_ptr);
90 
91  uiItemPointerR(col, ptr, "track_name", &object_ptr, "tracks", "", ICON_ANIM_DATA);
92  }
93  else {
94  uiItemR(layout, ptr, "track_name", UI_ITEM_R_SPLIT_EMPTY_NAME, "", ICON_ANIM_DATA);
95  }
96 
97  uiItemR(layout, ptr, "position", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
98 
100  uiItemR(layout, ptr, "frame_relative", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
101  }
102  }
103 }
104 
105 } // namespace blender::nodes::node_composite_trackpos_cc
106 
108 {
109  namespace file_ns = blender::nodes::node_composite_trackpos_cc;
110 
111  static bNodeType ntype;
112 
113  cmp_node_type_base(&ntype, CMP_NODE_TRACKPOS, "Track Position", NODE_CLASS_INPUT);
116  ntype.initfunc_api = file_ns::init;
118  &ntype, "NodeTrackPosData", node_free_standard_storage, node_copy_standard_storage);
119 
120  nodeRegisterType(&ntype);
121 }
struct Scene * CTX_data_scene(const bContext *C)
Definition: context.c:1090
void id_us_plus(struct ID *id)
Definition: lib_id.c:305
#define CMP_TRACKPOS_ABSOLUTE_FRAME
Definition: BKE_node.h:1328
#define CMP_TRACKPOS_RELATIVE_FRAME
Definition: BKE_node.h:1327
#define CMP_NODE_TRACKPOS
Definition: BKE_node.h:1267
void node_type_storage(struct bNodeType *ntype, const char *storagename, void(*freefunc)(struct bNode *node), void(*copyfunc)(struct bNodeTree *dest_ntree, struct bNode *dest_node, const struct bNode *src_node))
Definition: node.cc:4426
#define NODE_CLASS_INPUT
Definition: BKE_node.h:345
void nodeRegisterType(struct bNodeType *ntype)
Definition: node.cc:1357
struct MovieTrackingTrack * BKE_tracking_track_get_active(struct MovieTracking *tracking)
Definition: tracking.c:1089
struct MovieTrackingObject * BKE_tracking_object_get_named(struct MovieTracking *tracking, const char *name)
Definition: tracking.c:2077
struct MovieTrackingObject * BKE_tracking_object_get_active(struct MovieTracking *tracking)
Definition: tracking.c:2092
char * BLI_strncpy(char *__restrict dst, const char *__restrict src, size_t maxncpy) ATTR_NONNULL()
Definition: string.c:64
#define ELEM(...)
@ PROP_VELOCITY
Definition: RNA_types.h:156
#define C
Definition: RandGen.cpp:25
uiLayout * uiLayoutColumn(uiLayout *layout, bool align)
@ UI_ITEM_R_SPLIT_EMPTY_NAME
void uiItemR(uiLayout *layout, struct PointerRNA *ptr, const char *propname, int flag, const char *name, int icon)
void uiTemplateID(uiLayout *layout, const struct bContext *C, struct PointerRNA *ptr, const char *propname, const char *newop, const char *openop, const char *unlinkop, int filter, bool live_icon, const char *text)
void uiItemPointerR(uiLayout *layout, struct PointerRNA *ptr, const char *propname, struct PointerRNA *searchptr, const char *searchpropname, const char *name, int icon)
@ UI_TEMPLATE_ID_FILTER_ALL
OperationNode * node
Scene scene
uint col
static void init(const bContext *C, PointerRNA *ptr)
static void cmp_node_trackpos_declare(NodeDeclarationBuilder &b)
static void node_composit_buts_trackpos(uiLayout *layout, bContext *C, PointerRNA *ptr)
static const pxr::TfToken b("b", pxr::TfToken::Immortal)
void register_node_type_cmp_trackpos()
void cmp_node_type_base(bNodeType *ntype, int type, const char *name, short nclass)
void node_copy_standard_storage(bNodeTree *UNUSED(dest_ntree), bNode *dest_node, const bNode *src_node)
Definition: node_util.c:55
void node_free_standard_storage(bNode *node)
Definition: node_util.c:43
void RNA_pointer_create(ID *id, StructRNA *type, void *data, PointerRNA *r_ptr)
Definition: rna_access.c:136
struct MovieTracking tracking
void * data
Definition: RNA_types.h:38
struct MovieClip * clip
Defines a node type.
Definition: BKE_node.h:226
void(* initfunc_api)(const struct bContext *C, struct PointerRNA *ptr)
Definition: BKE_node.h:279
void(* draw_buttons)(struct uiLayout *, struct bContext *C, struct PointerRNA *ptr)
Definition: BKE_node.h:244
NodeDeclareFunction declare
Definition: BKE_node.h:324
#define N_(msgid)
PointerRNA * ptr
Definition: wm_files.c:3480