*/}}

la_tns_drivers.c 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. /*
  2. * LaGUI: A graphical application framework.
  3. * Copyright (C) 2022 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, laListHandle* l){
  33. if(LA_SRC_AND_PARENT(n->Mat)){ laBaseNode*bn=n->Mat->Source->Parent; LA_VISIT_NODE(bn); }
  34. n->Base.Eval=LA_DAG_FLAG_PERM;
  35. lstAppendPointer(l, n);
  36. return LA_DAG_FLAG_PERM;
  37. }
  38. int IDN_TransformEval(tnsTransformNode* n){
  39. if(!n->Target) return 0;
  40. if((!n->Mat->Source) || (n->Mat->Source->DataType!=(LA_PROP_FLOAT|LA_PROP_ARRAY)) || (n->Mat->Source->ArrLen!=16)){
  41. tnsLoadIdentity44d(&n->Target->DeltaTransform); tnsSelfMatrixChanged(n->Target, 1);
  42. }else{
  43. memcpy(n->Target->DeltaTransform, n->Mat->Source->Data, sizeof(tnsMatrix44d)); tnsSelfMatrixChanged(n->Target, 1);
  44. }
  45. laNotifyInstanceUsers(n->Target);
  46. return 1;
  47. }
  48. void tnsui_TransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  49. laColumn* c=laFirstColumn(uil); tnsTransformNode*n=This->EndInstance;
  50. LA_BASE_NODE_HEADER(uil,c,This);
  51. laColumn* cl,*cr; laSplitColumn(uil,c,0.3); cl=laLeftColumn(c,0); cr=laRightColumn(c,0);
  52. laUiItem* b=laBeginRow(uil,cl,0,0);
  53. laShowNodeSocket(uil,cl,This,"mat",0)->Flags|=LA_UI_SOCKET_LABEL_E;
  54. laEndRow(uil,b);
  55. laShowItem(uil,cr,This,"target");
  56. }
  57. void IDN_MakeTransformInit(tnsMakeTransformNode* n, int NoCreate){
  58. if(NoCreate){return;}
  59. n->Out=laCreateOutSocket(n,"MAT",0); strSafeSet(&n->Base.Name,"Make Transform");
  60. n->Loc=laCreateInSocket("LOC",0);
  61. n->Rot=laCreateInSocket("ROT",0); n->Angle=laCreateInSocket("ANGLE",0); n->UseRot[2]=1;
  62. n->Sca=laCreateInSocket("SCALE",0); n->UseSca=1;
  63. }
  64. void IDN_MakeTransformDestroy(tnsMakeTransformNode* n){
  65. laDestroyOutSocket(n->Out); strSafeDestroy(&n->Base.Name);
  66. laDestroyInSocket(n->Loc);
  67. laDestroyInSocket(n->Rot); laDestroyInSocket(n->Angle);
  68. laDestroyInSocket(n->Sca);
  69. }
  70. int IDN_MakeTransformVisit(tnsMakeTransformNode* n, laListHandle* l){
  71. LA_GUARD_THIS_NODE(n);
  72. if(LA_SRC_AND_PARENT(n->Loc)){ laBaseNode*bn=n->Loc->Source->Parent; LA_VISIT_NODE(bn); }
  73. if(LA_SRC_AND_PARENT(n->Rot)){ laBaseNode*bn=n->Rot->Source->Parent; LA_VISIT_NODE(bn); }
  74. if(LA_SRC_AND_PARENT(n->Sca)){ laBaseNode*bn=n->Sca->Source->Parent; LA_VISIT_NODE(bn); }
  75. if(LA_SRC_AND_PARENT(n->Angle)){ laBaseNode*bn=n->Angle->Source->Parent; LA_VISIT_NODE(bn); }
  76. n->Base.Eval=LA_DAG_FLAG_PERM; lstAppendPointer(l, n);
  77. return LA_DAG_FLAG_PERM;
  78. }
  79. int IDN_MakeTransformEval(tnsMakeTransformNode* n){
  80. tnsMatrix44d mm,mt,mr,ms; real* UseT,*UseR,*UseA,*UseS; tnsVector3d nr;
  81. 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; }
  82. 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; }
  83. if(LA_SRC_AND_PARENT(n->Angle) && n->Angle->Source->DataType&LA_PROP_FLOAT){ UseA=n->Angle->Source->Data; }else{ UseA=&n->UseAngle; }
  84. if(LA_SRC_AND_PARENT(n->Sca) && n->Sca->Source->DataType&LA_PROP_FLOAT){ UseS=n->Sca->Source->Data; }else{ UseS=&n->UseSca; }
  85. tnsMakeTranslationMatrix44d(mt,LA_COLOR3(UseT));
  86. tnsNormalize3d(nr, UseR); tnsMakeRotationMatrix44d(mr,*UseA, LA_COLOR3(nr));
  87. tnsMakeScaleMatrix44d(ms,*UseS,*UseS,*UseS);
  88. tnsMultiply44d(mm,mt,mr); tnsMultiply44d(n->Mat,mm,ms);
  89. n->Out->ArrLen=16; n->Out->Data=n->Mat; n->Out->DataType=LA_PROP_FLOAT|LA_PROP_ARRAY;
  90. return 1;
  91. }
  92. void tnsui_MakeTransformNode(laUiList *uil, laPropPack *This, laPropPack *Extra, laColumn *UNUSED, int context){
  93. laColumn* c=laFirstColumn(uil); tnsMakeTransformNode*n=This->EndInstance;
  94. laUiItem* b2, *rui;
  95. LA_BASE_NODE_HEADER(uil,c,This);
  96. laUiItem* b=laBeginRow(uil,c,0,0);
  97. laShowNodeSocket(uil,c,This,"loc",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"loc.source")));{
  98. laShowItem(uil,c,This,"use_loc")->Expand=1;
  99. }laEndCondition(uil,b2);
  100. laEndRow(uil,b);
  101. b=laBeginRow(uil,c,0,0);
  102. laShowNodeSocket(uil,c,This,"rot",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"rot.source")));{
  103. laShowItem(uil,c,This,"use_rot")->Expand=1;
  104. }laEndCondition(uil,b2);
  105. laEndRow(uil,b);
  106. b=laBeginRow(uil,c,0,0);
  107. laShowNodeSocket(uil,c,This,"angle",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"angle.source")));{
  108. laShowItem(uil,c,This,"use_angle")->Expand=1;
  109. }laEndCondition(uil,b2);
  110. laEndRow(uil,b);
  111. b=laBeginRow(uil,c,0,0);
  112. laShowNodeSocket(uil,c,This,"sca",0)->Flags|=LA_UI_SOCKET_LABEL_E; b2=laOnConditionThat(uil,c,laNot(laPropExpression(This,"sca.source")));{
  113. laShowItem(uil,c,This,"use_sca")->Expand=1;
  114. }laEndCondition(uil,b2);
  115. laEndRow(uil,b);
  116. b=laBeginRow(uil,c,0,0);
  117. laShowSeparator(uil,c)->Expand=1;;
  118. laShowNodeSocket(uil,c,This,"out",0)->Flags|=LA_UI_SOCKET_LABEL_W;
  119. laEndRow(uil,b);
  120. }
  121. int OPINV_AddDriverPage(laOperator* a, laEvent *e){
  122. laRackPage* dp=memAcquire(sizeof(laRackPage));
  123. strSafeSet(&dp->Name,"New Page");
  124. lstAppendItem(&MAIN.Drivers->Pages, dp); MAIN.Drivers->CurrentPage=dp; dp->RackType=LA_RACK_TYPE_DRIVER;
  125. laNotifyUsers("la.drivers.pages"); laRecordAndPush(0,"la.drivers","Add page", 0);
  126. return LA_FINISHED;
  127. }
  128. int OPINV_RebuildDrivers(laOperator* a, laEvent *e){
  129. laDriverRequestRebuild();
  130. return LA_FINISHED;
  131. }
  132. tnsObject* tnsget_FirstObject(void* unused, void* unused2){
  133. return T->World.AllObjects.pFirst;
  134. }
  135. void tns_RegisterNodes(){
  136. laPropContainer *pc; laProp *p;
  137. laOperatorType *at;
  138. laEnumProp *ep;
  139. laCreateOperatorType("LA_add_driver_page", "New Page", "Add a driver page", 0,0,0,OPINV_AddDriverPage,0,'+',0);
  140. laCreateOperatorType("LA_driver_rebuild", "Rebuild Drivers", "Rebuild drivers for evaluation",0,0,0,OPINV_RebuildDrivers,0,L'⭮',0);
  141. pc=laAddPropertyContainer("tns_transform_node", "Transform", "Transform objects",0,tnsui_TransformNode,sizeof(tnsTransformNode),lapost_Node,0,1);
  142. TNS_PC_IDN_TRANSFORM=pc; laPropContainerExtraFunctions(pc,0,0,0,0,laui_DefaultNodeOperationsPropUiDefine);
  143. laAddSubGroup(pc,"base","Base","Base node","la_base_node",0,0,0,0,0,0,0,0,0,0,0,LA_UDF_LOCAL);
  144. 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);
  145. 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);
  146. pc=laAddPropertyContainer("tns_make_transform_node", "Make Transform", "Make Transform matrix",0,tnsui_MakeTransformNode,sizeof(tnsMakeTransformNode),lapost_Node,0,1);
  147. TNS_PC_IDN_MAKE_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,"out", "Out","Output matrix","la_out_socket",0,0,0,offsetof(tnsMakeTransformNode,Out),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  150. laAddSubGroup(pc,"loc", "Location","Location","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Loc),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  151. laAddSubGroup(pc,"rot", "Rotation","Rotation","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Rot),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  152. laAddSubGroup(pc,"sca", "Scale","Scale","la_in_socket",0,0,0,offsetof(tnsMakeTransformNode,Sca),0,0,0,0,0,0,0,LA_UDF_SINGLE);
  153. 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);
  154. 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);
  155. 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);
  156. 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);
  157. 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);
  158. LA_IDN_REGISTER("Transform",0,TNS_IDN_TRANSFORM,TNS_PC_IDN_TRANSFORM, IDN_TransformInit, IDN_TransformDestroy, IDN_TransformVisit, IDN_TransformEval, tnsTransformNode);
  159. LA_IDN_REGISTER("Make Transform",0,TNS_IDN_MAKE_TRANSFORM,TNS_PC_IDN_MAKE_TRANSFORM, IDN_MakeTransformInit, IDN_MakeTransformDestroy, IDN_MakeTransformVisit, IDN_MakeTransformEval, tnsMakeTransformNode);
  160. LA_NODE_CATEGORY_DRIVER=laAddNodeCategory("Driver",0,LA_RACK_TYPE_DRIVER);
  161. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_MATH, &TNS_IDN_MAKE_TRANSFORM,0);
  162. laNodeCategoryAddNodeTypes(LA_NODE_CATEGORY_DRIVER, &TNS_IDN_TRANSFORM,0);
  163. }
  164. void laDriverRequestRebuild(){ MAIN.Drivers->NeedRebuild=1; }
  165. void laDriverRequestEval(){ MAIN.Drivers->NeedEval=1; }
  166. int la_RunDrivers(){
  167. MAIN.Drivers->NeedEval = 0;
  168. for(laListItemPointer*lip=MAIN.Drivers->Eval.pFirst;lip;lip=lip->pNext){
  169. laBaseNode* n=lip->p; if(!n->InitDone){ n->Type->Init(n,1); n->InitDone=1; } n->Type->Eval(n);
  170. }
  171. return 1;
  172. }
  173. int la_RebuildDriverEval(){
  174. MAIN.Drivers->NeedRebuild = 0;
  175. while(lstPopPointer(&MAIN.Drivers->Eval));
  176. laListHandle pending={0};
  177. for(laRackPage* dp=MAIN.Drivers->Pages.pFirst;dp;dp=dp->Item.pNext){
  178. for(laNodeRack* ir=dp->Racks.pFirst;ir;ir=ir->Item.pNext){
  179. for(laBaseNode*bn=ir->Nodes.pFirst;bn;bn=bn->Item.pNext){ if(!bn->InitDone){ bn->Type->Init(bn,1); bn->InitDone=1; } lstAppendPointer(&pending,bn); bn->Eval=0; }
  180. }
  181. }
  182. laBaseNode*n;int result=LA_DAG_FLAG_PERM; laListItemPointer*NextLip;
  183. for(laListItemPointer*lip=pending.pFirst;lip;lip=NextLip){ n=lip->p; NextLip=lip->pNext;
  184. if(n->Eval&LA_DAG_FLAG_PERM) continue;
  185. result=n->Type->Visit(n,&MAIN.Drivers->Eval); if(result==LA_DAG_FLAG_ERR){ while(lstPopPointer(&pending)); break; }
  186. }
  187. if(result==LA_DAG_FLAG_ERR){ while(lstPopPointer(&MAIN.Drivers->Eval)); return LA_DAG_FLAG_ERR; }
  188. return LA_DAG_FLAG_PERM;
  189. }