*/}}

la_tns_drivers.c 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191
  1. /*
  2. * LaGUI: A graphical application framework.
  3. * Copyright (C) 2022-2023 Wu Yiming
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. */
  18. #include "../la_5.h"
  19. extern LA MAIN;
  20. extern struct _tnsMain *T;
  21. laBaseNodeType TNS_IDN_TRANSFORM;
  22. laBaseNodeType TNS_IDN_MAKE_TRANSFORM;
  23. laPropContainer* TNS_PC_IDN_TRANSFORM;
  24. laPropContainer* TNS_PC_IDN_MAKE_TRANSFORM;
  25. void IDN_TransformInit(tnsTransformNode* n, int NoCreate){
  26. if(NoCreate){return;}
  27. n->Mat=laCreateInSocket("MAT",0); strSafeSet(&n->Base.Name,"Transform");
  28. }
  29. void IDN_TransformDestroy(tnsTransformNode* n){
  30. laDestroyInSocket(n->Mat); strSafeDestroy(&n->Base.Name);
  31. }
  32. int IDN_TransformVisit(tnsTransformNode* n, laNodeVisitInfo* vi){
  33. if(LA_SRC_AND_PARENT(n->Mat)){ laBaseNode*bn=n->Mat->Source->Parent; LA_VISIT_NODE(bn,vi); }
  34. LA_ADD_THIS_NODE(n,vi);
  35. return LA_DAG_FLAG_PERM;
  36. }
  37. int IDN_TransformEval(tnsTransformNode* n){
  38. if(!n->Target) return 0;
  39. if((!n->Mat->Source) || (n->Mat->Source->DataType!=(LA_PROP_FLOAT|LA_PROP_ARRAY)) || (n->Mat->Source->ArrLen!=16)){
  40. tnsLoadIdentity44d(n->Target->DeltaTransform); tnsSelfMatrixChanged(n->Target, 1);
  41. }else{
  42. memcpy(n->Target->DeltaTransform, n->Mat->Source->Data, sizeof(tnsMatrix44d)); tnsSelfMatrixChanged(n->Target, 1);
  43. }
  44. laNotifyInstanceUsers(n->Target);
  45. return 1;
  46. }
  47. void tnsui_TransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  48. laColumn* c=laFirstColumn(uil); tnsTransformNode*n=This->EndInstance;
  49. LA_BASE_NODE_HEADER(uil,c,This);
  50. laColumn* cl,*cr; laSplitColumn(uil,c,0.3); cl=laLeftColumn(c,0); cr=laRightColumn(c,0);
  51. laUiItem* b=laBeginRow(uil,cl,0,0);
  52. laShowNodeSocket(uil,cl,This,"mat",0)->Flags|=LA_UI_SOCKET_LABEL_E;
  53. laEndRow(uil,b);
  54. laShowItem(uil,cr,This,"target");
  55. }
  56. void IDN_MakeTransformInit(tnsMakeTransformNode* n, int NoCreate){
  57. if(NoCreate){return;}
  58. n->Out=laCreateOutSocket(n,"MAT",0); strSafeSet(&n->Base.Name,"Make Transform");
  59. n->Loc=laCreateInSocket("LOC",0);
  60. n->Rot=laCreateInSocket("ROT",0); n->Angle=laCreateInSocket("ANGLE",0); n->UseRot[2]=1;
  61. n->Sca=laCreateInSocket("SCALE",0); n->UseSca=1;
  62. }
  63. void IDN_MakeTransformDestroy(tnsMakeTransformNode* n){
  64. laDestroyOutSocket(n->Out); strSafeDestroy(&n->Base.Name);
  65. laDestroyInSocket(n->Loc);
  66. laDestroyInSocket(n->Rot); laDestroyInSocket(n->Angle);
  67. laDestroyInSocket(n->Sca);
  68. }
  69. int IDN_MakeTransformVisit(tnsMakeTransformNode* n, laNodeVisitInfo* vi){
  70. LA_GUARD_THIS_NODE(n,vi);
  71. if(LA_SRC_AND_PARENT(n->Loc)){ laBaseNode*bn=n->Loc->Source->Parent; LA_VISIT_NODE(bn,vi); }
  72. if(LA_SRC_AND_PARENT(n->Rot)){ laBaseNode*bn=n->Rot->Source->Parent; LA_VISIT_NODE(bn,vi); }
  73. if(LA_SRC_AND_PARENT(n->Sca)){ laBaseNode*bn=n->Sca->Source->Parent; LA_VISIT_NODE(bn,vi); }
  74. if(LA_SRC_AND_PARENT(n->Angle)){ laBaseNode*bn=n->Angle->Source->Parent; LA_VISIT_NODE(bn,vi); }
  75. LA_ADD_THIS_NODE(n,vi);
  76. return LA_DAG_FLAG_PERM;
  77. }
  78. int IDN_MakeTransformEval(tnsMakeTransformNode* n){
  79. tnsMatrix44d mm,mt,mr,ms; real* UseT,*UseR,*UseA,*UseS; tnsVector3d nr;
  80. if(LA_SRC_AND_PARENT(n->Loc) && n->Loc->Source->ArrLen>=3 && n->Loc->Source->DataType==LA_PROP_FLOAT|LA_PROP_ARRAY){ UseT=n->Loc->Source->Data; }else{ UseT=n->UseLoc; }
  81. if(LA_SRC_AND_PARENT(n->Rot) && n->Rot->Source->ArrLen>=3 && n->Rot->Source->DataType==LA_PROP_FLOAT|LA_PROP_ARRAY){ UseR=n->Rot->Source->Data; }else{ UseR=n->UseRot; }
  82. if(LA_SRC_AND_PARENT(n->Angle) && n->Angle->Source->DataType&LA_PROP_FLOAT){ UseA=n->Angle->Source->Data; }else{ UseA=&n->UseAngle; }
  83. if(LA_SRC_AND_PARENT(n->Sca) && n->Sca->Source->DataType&LA_PROP_FLOAT){ UseS=n->Sca->Source->Data; }else{ UseS=&n->UseSca; }
  84. tnsMakeTranslationMatrix44d(mt,LA_COLOR3(UseT));
  85. tnsNormalize3d(nr, UseR); tnsMakeRotationMatrix44d(mr,*UseA, LA_COLOR3(nr));
  86. tnsMakeScaleMatrix44d(ms,*UseS,*UseS,*UseS);
  87. tnsMultiply44d(mm,mt,mr); tnsMultiply44d(n->Mat,mm,ms);
  88. n->Out->ArrLen=16; n->Out->Data=n->Mat; n->Out->DataType=LA_PROP_FLOAT|LA_PROP_ARRAY;
  89. return 1;
  90. }
  91. void tnsui_MakeTransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  92. laColumn* c=laFirstColumn(uil); tnsMakeTransformNode*n=This->EndInstance;
  93. laUiItem* b2, *rui;
  94. LA_BASE_NODE_HEADER(uil,c,This);
  95. laUiItem* b=laBeginRow(uil,c,0,0);
  96. laShowNodeSocket(uil,c,This,"loc",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"loc.source")));{
  97. laShowItem(uil,c,This,"use_loc")->Expand=1;
  98. }laEndCondition(uil,b2);
  99. laEndRow(uil,b);
  100. b=laBeginRow(uil,c,0,0);
  101. laShowNodeSocket(uil,c,This,"rot",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"rot.source")));{
  102. laShowItem(uil,c,This,"use_rot")->Expand=1;
  103. }laEndCondition(uil,b2);
  104. laEndRow(uil,b);
  105. b=laBeginRow(uil,c,0,0);
  106. laShowNodeSocket(uil,c,This,"angle",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"angle.source")));{
  107. laShowItem(uil,c,This,"use_angle")->Expand=1;
  108. }laEndCondition(uil,b2);
  109. laEndRow(uil,b);
  110. b=laBeginRow(uil,c,0,0);
  111. laShowNodeSocket(uil,c,This,"sca",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"sca.source")));{
  112. laShowItem(uil,c,This,"use_sca")->Expand=1;
  113. }laEndCondition(uil,b2);
  114. laEndRow(uil,b);
  115. b=laBeginRow(uil,c,0,0);
  116. laShowSeparator(uil,c)->Expand=1;;
  117. laShowNodeSocket(uil,c,This,"out",0)->Flags|=LA_UI_SOCKET_LABEL_W;
  118. laEndRow(uil,b);
  119. }
  120. int OPCHK_AddDriverPage(laPropPack *This, laStringSplitor *ss){
  121. if (This && la_EnsureSubTarget(This->LastPs->p,0) == TNS_PC_OBJECT_GENERIC) return 1;
  122. return 0;
  123. }
  124. int OPINV_AddDriverPage(laOperator* a, laEvent *e){
  125. tnsObject* ob=a->This?a->This->EndInstance:0; if(!ob || !ob->Drivers) return LA_CANCELED;
  126. laRackPage* dp=memAcquireHyper(sizeof(laRackPage));
  127. strSafeSet(&dp->Name,"New Page");
  128. lstAppendItem(&ob->Drivers->Pages, dp);
  129. memAssignRef(ob->Drivers,&ob->Drivers->CurrentPage,dp); dp->RackType=LA_RACK_TYPE_DRIVER;
  130. laNotifyInstanceUsers(ob->Drivers); laRecordAndPush(a->This,"","Add driver page", 0);
  131. return LA_FINISHED;
  132. }
  133. int OPINV_RebuildDrivers(laOperator* a, laEvent *e){
  134. laGraphRequestRebuild();
  135. return LA_FINISHED;
  136. }
  137. tnsObject* tnsget_FirstObject(void* unused, void* unused2){
  138. return T->World.AllObjects.pFirst;
  139. }
  140. void tns_RegisterNodes(){
  141. laPropContainer *pc; laProp *p;
  142. laOperatorType *at;
  143. laEnumProp *ep;
  144. laCreateOperatorType("LA_add_driver_page", "New Page", "Add a driver page",OPCHK_AddDriverPage,0,0,OPINV_AddDriverPage,0,'+',0);
  145. laCreateOperatorType("LA_driver_rebuild", "Rebuild Drivers", "Rebuild drivers for evaluation",0,0,0,OPINV_RebuildDrivers,0,U'⭮',0);
  146. pc=laAddPropertyContainer("tns_transform_node", "Transform", "Transform objects",0,tnsui_TransformNode,sizeof(tnsTransformNode),lapost_Node,0,1);
  147. TNS_PC_IDN_TRANSFORM=pc; laPropContainerExtraFunctions(pc,0,0,0,0,laui_DefaultNodeOperationsPropUiDefine);
  148. laAddSubGroup(pc,"base","Base","Base node","la_base_node",0,0,0,0,0,0,0,0,0,0,0,LA_UDF_LOCAL);
  149. laAddSubGroup(pc,"mat", "Mat","Input matrix","la_in_socket",0,0,0,offsetof(tnsTransformNode,Mat),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  150. laAddSubGroup(pc,"target", "Target","Target object","tns_object",0,LA_WIDGET_COLLECTION_SELECTOR,laui_IdentifierOnly,offsetof(tnsTransformNode,Target),tnsget_FirstObject,0,laget_ListNext,0,0,0,0,LA_UDF_REFER);
  151. pc=laAddPropertyContainer("tns_make_transform_node", "Make Transform", "Make Transform matrix",0,tnsui_MakeTransformNode,sizeof(tnsMakeTransformNode),lapost_Node,0,1);
  152. TNS_PC_IDN_MAKE_TRANSFORM=pc; laPropContainerExtraFunctions(pc,0,0,0,0,laui_DefaultNodeOperationsPropUiDefine);
  153. laAddSubGroup(pc,"base","Base","Base node","la_base_node",0,0,0,0,0,0,0,0,0,0,0,LA_UDF_LOCAL);
  154. laAddSubGroup(pc,"out", "Out","Output matrix","la_out_socket",0,0,0,offsetof(tnsMakeTransformNode,Out),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  155. laAddSubGroup(pc,"loc", "Location","Location","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Loc),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  156. laAddSubGroup(pc,"rot", "Rotation","Rotation","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Rot),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  157. laAddSubGroup(pc,"sca", "Scale","Scale","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Sca),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  158. laAddSubGroup(pc,"angle", "Angle","Rotation Angle","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Angle),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  159. laAddFloatProperty(pc,"use_loc", "Location", "Use Location",0,"X,Y,Z",0,0,0,0.05,0,0,offsetof(tnsMakeTransformNode, UseLoc),0,0,3,0,0,0,0,0,0,0,0);
  160. laAddFloatProperty(pc,"use_rot", "Rotation", "Use Rotation",0,"X,Y,Z",0,0,0,0.05,0,0,offsetof(tnsMakeTransformNode, UseRot),0,0,3,0,0,0,0,0,0,0,0);
  161. laAddFloatProperty(pc,"use_sca", "Scale", "Use Scale",0,0,0,0,0,0.05,0,0,offsetof(tnsMakeTransformNode, UseSca),0,0,0,0,0,0,0,0,0,0,0);
  162. laAddFloatProperty(pc,"use_angle", "Angle", "Use Angle",0,0,0,0,0,0.05,0,0,offsetof(tnsMakeTransformNode, UseAngle),0,0,0,0,0,0,0,0,0,0,0);
  163. LA_IDN_REGISTER("Transform",0,TNS_IDN_TRANSFORM,TNS_PC_IDN_TRANSFORM, IDN_TransformInit, IDN_TransformDestroy, IDN_TransformVisit, IDN_TransformEval, tnsTransformNode);
  164. LA_IDN_REGISTER("Make Transform",0,TNS_IDN_MAKE_TRANSFORM,TNS_PC_IDN_MAKE_TRANSFORM, IDN_MakeTransformInit, IDN_MakeTransformDestroy, IDN_MakeTransformVisit, IDN_MakeTransformEval, tnsMakeTransformNode);
  165. LA_NODE_CATEGORY_DRIVER=laAddNodeCategory("Driver",0,LA_RACK_TYPE_DRIVER);
  166. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_MATH, &TNS_IDN_MAKE_TRANSFORM,0);
  167. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_DRIVER, &TNS_IDN_TRANSFORM,0);
  168. }