*/}}

la_tns_drivers.c 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235
  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. tnsObject* ob=n->Target; if(ob->PlayDuplicate){ ob=ob->PlayDuplicate; }
  40. if((!n->Mat->Source) || (n->Mat->Source->DataType!=(LA_PROP_FLOAT|LA_PROP_ARRAY)) || (n->Mat->Source->ArrLen!=16)){
  41. tnsLoadIdentity44d(ob->DeltaTransform); tnsSelfMatrixChanged(ob, 1);
  42. }else{
  43. memcpy(ob->DeltaTransform, n->Mat->Source->Data, sizeof(tnsMatrix44d)); tnsSelfMatrixChanged(ob, 1);
  44. }
  45. laNotifyInstanceUsers(n->Target);
  46. return 1;
  47. }
  48. void IDN_TransformCopy(tnsTransformNode* new, tnsTransformNode* old, int DoRematch){
  49. if(DoRematch){ LA_IDN_NEW_LINK(Mat) return; }
  50. memAssignRef(new,&new->Target,old->Target);
  51. }
  52. void tnsui_TransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  53. laColumn* c=laFirstColumn(uil); tnsTransformNode*n=This->EndInstance;
  54. LA_BASE_NODE_HEADER(uil,c,This);
  55. laColumn* cl,*cr; laSplitColumn(uil,c,0.3); cl=laLeftColumn(c,0); cr=laRightColumn(c,0);
  56. laUiItem* b=laBeginRow(uil,cl,0,0);
  57. laShowNodeSocket(uil,cl,This,"mat",0)->Flags|=LA_UI_SOCKET_LABEL_E;
  58. laEndRow(uil,b);
  59. laShowItem(uil,cr,This,"target");
  60. }
  61. void IDN_MakeTransformInit(tnsMakeTransformNode* n, int NoCreate){
  62. if(NoCreate){return;}
  63. n->Out=laCreateOutSocket(n,"MAT",0); strSafeSet(&n->Base.Name,"Make Transform");
  64. n->Loc=laCreateInSocket("LOC",0);
  65. n->Rot=laCreateInSocket("ROT",0); n->Angle=laCreateInSocket("ANGLE",0); n->UseRot[2]=1;
  66. n->Sca=laCreateInSocket("SCALE",0); n->UseSca=1;
  67. }
  68. void IDN_MakeTransformDestroy(tnsMakeTransformNode* n){
  69. laDestroyOutSocket(n->Out); strSafeDestroy(&n->Base.Name);
  70. laDestroyInSocket(n->Loc);
  71. laDestroyInSocket(n->Rot); laDestroyInSocket(n->Angle);
  72. laDestroyInSocket(n->Sca);
  73. }
  74. int IDN_MakeTransformVisit(tnsMakeTransformNode* n, laNodeVisitInfo* vi){
  75. LA_GUARD_THIS_NODE(n,vi);
  76. if(LA_SRC_AND_PARENT(n->Loc)){ laBaseNode*bn=n->Loc->Source->Parent; LA_VISIT_NODE(bn,vi); }
  77. if(LA_SRC_AND_PARENT(n->Rot)){ laBaseNode*bn=n->Rot->Source->Parent; LA_VISIT_NODE(bn,vi); }
  78. if(LA_SRC_AND_PARENT(n->Sca)){ laBaseNode*bn=n->Sca->Source->Parent; LA_VISIT_NODE(bn,vi); }
  79. if(LA_SRC_AND_PARENT(n->Angle)){ laBaseNode*bn=n->Angle->Source->Parent; LA_VISIT_NODE(bn,vi); }
  80. LA_ADD_THIS_NODE(n,vi);
  81. return LA_DAG_FLAG_PERM;
  82. }
  83. int IDN_MakeTransformEval(tnsMakeTransformNode* n){
  84. tnsMatrix44d mm,mt,mr,ms; real* UseT,*UseR,*UseA,*UseS; tnsVector3d nr;
  85. 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; }
  86. 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; }
  87. if(LA_SRC_AND_PARENT(n->Angle) && n->Angle->Source->DataType&LA_PROP_FLOAT){ UseA=n->Angle->Source->Data; }else{ UseA=&n->UseAngle; }
  88. if(LA_SRC_AND_PARENT(n->Sca) && n->Sca->Source->DataType&LA_PROP_FLOAT){ UseS=n->Sca->Source->Data; }else{ UseS=&n->UseSca; }
  89. tnsMakeTranslationMatrix44d(mt,LA_COLOR3(UseT));
  90. tnsNormalize3d(nr, UseR); tnsMakeRotationMatrix44d(mr,*UseA, LA_COLOR3(nr));
  91. tnsMakeScaleMatrix44d(ms,*UseS,*UseS,*UseS);
  92. tnsMultiply44d(mm,mt,mr); tnsMultiply44d(n->Mat,mm,ms);
  93. n->Out->ArrLen=16; n->Out->Data=n->Mat; n->Out->DataType=LA_PROP_FLOAT|LA_PROP_ARRAY;
  94. return 1;
  95. }
  96. void IDN_MakeTransformCopy(tnsMakeTransformNode* new, tnsMakeTransformNode* old, int DoRematch){
  97. if(DoRematch){ LA_IDN_NEW_LINK(Loc) LA_IDN_NEW_LINK(Rot) LA_IDN_NEW_LINK(Sca) LA_IDN_NEW_LINK(Angle) return; }
  98. old->Out->Duplicated=new->Out; new->UseSca=old->UseSca; new->UseAngle=old->UseAngle;
  99. tnsVectorSet3v(new->UseLoc,old->UseLoc); tnsVectorSet3v(new->UseRot,old->UseRot);
  100. }
  101. void tnsui_MakeTransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  102. laColumn* c=laFirstColumn(uil); tnsMakeTransformNode*n=This->EndInstance;
  103. laUiItem* b2, *rui;
  104. LA_BASE_NODE_HEADER(uil,c,This);
  105. laUiItem* b=laBeginRow(uil,c,0,0);
  106. laShowNodeSocket(uil,c,This,"loc",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"loc.source")));{
  107. laShowItem(uil,c,This,"use_loc")->Expand=1;
  108. }laEndCondition(uil,b2);
  109. laEndRow(uil,b);
  110. b=laBeginRow(uil,c,0,0);
  111. laShowNodeSocket(uil,c,This,"rot",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"rot.source")));{
  112. laShowItem(uil,c,This,"use_rot")->Expand=1;
  113. }laEndCondition(uil,b2);
  114. laEndRow(uil,b);
  115. b=laBeginRow(uil,c,0,0);
  116. laShowNodeSocket(uil,c,This,"angle",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"angle.source")));{
  117. laShowItem(uil,c,This,"use_angle")->Expand=1;
  118. }laEndCondition(uil,b2);
  119. laEndRow(uil,b);
  120. b=laBeginRow(uil,c,0,0);
  121. laShowNodeSocket(uil,c,This,"sca",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"sca.source")));{
  122. laShowItem(uil,c,This,"use_sca")->Expand=1;
  123. }laEndCondition(uil,b2);
  124. laEndRow(uil,b);
  125. b=laBeginRow(uil,c,0,0);
  126. laShowSeparator(uil,c)->Expand=1;;
  127. laShowNodeSocket(uil,c,This,"out",0)->Flags|=LA_UI_SOCKET_LABEL_W;
  128. laEndRow(uil,b);
  129. }
  130. int OPCHK_AddDriverPage(laPropPack *This, laStringSplitor *ss){
  131. if (This && la_EnsureSubTarget(This->LastPs->p,0) == TNS_PC_OBJECT_GENERIC) return 1;
  132. return 0;
  133. }
  134. int OPINV_AddDriverPage(laOperator* a, laEvent *e){
  135. tnsObject* ob=a->This?a->This->EndInstance:0; if(!ob || !ob->Drivers) return LA_CANCELED;
  136. laRackPage* dp=memAcquire(sizeof(laRackPage));
  137. strSafeSet(&dp->Name,"New Page");
  138. lstAppendItem(&ob->Drivers->Pages, dp);
  139. memAssignRef(dp,&dp->ParentObject,ob); memAssignRef(ob->Drivers,&ob->Drivers->CurrentPage,dp);
  140. dp->RackType=LA_RACK_TYPE_DRIVER;
  141. laNotifyInstanceUsers(ob->Drivers); laRecordAndPush(a->This,"","Add driver page", 0);
  142. return LA_FINISHED;
  143. }
  144. int OPCHK_RemoveDriverPage(laPropPack *This, laStringSplitor *ss){
  145. if ((!This) || (la_EnsureSubTarget(This->LastPs->p,0) != LA_PC_RACK_PAGE)) return 0;
  146. laRackPage* dp=This->EndInstance; if(!dp->ParentObject) return 0;
  147. return 1;
  148. }
  149. int OPINV_RemoveDriverPage(laOperator* a, laEvent *e){
  150. laRackPage* dp=a->This?a->This->EndInstance:0; if(!dp || !dp->ParentObject) return LA_CANCELED;
  151. tnsObject* ob=dp->ParentObject;
  152. if(strSame(strGetArgumentString(a->ExtraInstructionsP,"confirm"),"true")){
  153. strSafeDestroy(&dp->Name);
  154. while(dp->Racks.pFirst){ laDestroyRack(dp->Racks.pFirst); }
  155. while(lstPopPointer(&dp->Eval)); while(lstPopPointer(&dp->AlwaysBranchers));
  156. if(dp->ParentObject->Drivers->CurrentPage==dp){
  157. laRackPage* adp=dp->Item.pNext?dp->Item.pNext:dp->Item.pPrev;
  158. memAssignRef(dp->ParentObject->Drivers,&dp->ParentObject->Drivers->CurrentPage,adp);
  159. }laNotifyInstanceUsers(dp);
  160. lstRemoveItem(&dp->ParentObject->Drivers->Pages, dp); memLeave(dp);
  161. laNotifyInstanceUsers(ob->Drivers); laRecordAndPush(a->This,"","Remove driver page", 0);
  162. return LA_FINISHED;
  163. }
  164. laEnableOperatorPanel(a,a->This,e->x,e->y,200,200,0,0,0,0,0,0,0,0,e);
  165. return LA_RUNNING;
  166. }
  167. void laui_RemoveDriverPage(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  168. laColumn* c=laFirstColumn(uil);
  169. laShowItemFull(uil,c,This,"remove_driver_page",0,"confirm=true;text=Confirm",0,0);
  170. }
  171. int OPINV_RebuildDrivers(laOperator* a, laEvent *e){
  172. laGraphRequestRebuild();
  173. return LA_FINISHED;
  174. }
  175. tnsObject* tnsget_FirstObject(void* unused, void* unused2){
  176. return T->World->AllObjects.pFirst;
  177. }
  178. void tns_RegisterNodes(){
  179. laPropContainer *pc; laProp *p;
  180. laOperatorType *at;
  181. laEnumProp *ep;
  182. laCreateOperatorType("LA_add_driver_page", "New Page", "Add a driver page",OPCHK_AddDriverPage,0,0,OPINV_AddDriverPage,0,'+',0);
  183. laCreateOperatorType("LA_remove_driver_page", "Remove Page", "Remove a driver page",OPCHK_RemoveDriverPage,0,0,OPINV_RemoveDriverPage,OPMOD_FinishOnData,L'🗴',0)
  184. ->UiDefine=laui_RemoveDriverPage;
  185. laCreateOperatorType("LA_driver_rebuild", "Rebuild Drivers", "Rebuild drivers for evaluation",0,0,0,OPINV_RebuildDrivers,0,U'⭮',0);
  186. LA_NODE_CATEGORY_DRIVER=laAddNodeCategory("Driver",0,LA_RACK_TYPE_DRIVER);
  187. if(!MAIN.InitArgs.HasWorldObjects) return;
  188. pc=laAddPropertyContainer("tns_transform_node", "Transform", "Transform objects",0,tnsui_TransformNode,sizeof(tnsTransformNode),lapost_Node,0,1);
  189. TNS_PC_IDN_TRANSFORM=pc; laPropContainerExtraFunctions(pc,0,0,0,0,laui_DefaultNodeOperationsPropUiDefine);
  190. laAddSubGroup(pc,"base","Base","Base node","la_base_node",0,0,0,0,0,0,0,0,0,0,0,LA_UDF_LOCAL);
  191. 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);
  192. 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);
  193. pc=laAddPropertyContainer("tns_make_transform_node", "Make Transform", "Make Transform matrix",0,tnsui_MakeTransformNode,sizeof(tnsMakeTransformNode),lapost_Node,0,1);
  194. TNS_PC_IDN_MAKE_TRANSFORM=pc; laPropContainerExtraFunctions(pc,0,0,0,0,laui_DefaultNodeOperationsPropUiDefine);
  195. laAddSubGroup(pc,"base","Base","Base node","la_base_node",0,0,0,0,0,0,0,0,0,0,0,LA_UDF_LOCAL);
  196. 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);
  197. laAddSubGroup(pc,"loc", "Location","Location","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Loc),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  198. laAddSubGroup(pc,"rot", "Rotation","Rotation","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Rot),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  199. laAddSubGroup(pc,"sca", "Scale","Scale","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Sca),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  200. 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);
  201. 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);
  202. 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);
  203. 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);
  204. 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);
  205. LA_IDN_REGISTER("Transform",0,TNS_IDN_TRANSFORM,TNS_PC_IDN_TRANSFORM, IDN_Transform, tnsTransformNode);
  206. LA_IDN_REGISTER("Make Transform",0,TNS_IDN_MAKE_TRANSFORM,TNS_PC_IDN_MAKE_TRANSFORM, IDN_MakeTransform, tnsMakeTransformNode);
  207. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_MATH, &TNS_IDN_MAKE_TRANSFORM,0);
  208. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_DRIVER, &TNS_IDN_TRANSFORM,0);
  209. }