/*====================================================================*\
FILE    : TestRelation.c
PURPOSE : Test of the Relations functionality.

HISTORY:
DATE       BUILD    AUTHOR   MODIFICATIONS
16-May-97  H-03-1   Pavel    $$1 - Created.
03-Jun-97  H-03-13  Pavel    $$2 - fixed some bugs
01-Jul-97  H-03-16  Pavel    $$3 - fixed some bugs
22-Sep-97  H-03-23  Pavel    $$4 - More includes
17-Oct-97  H-03-27  Pavel    $$5 Replaced Visit functions by UtilCollect
10-Dec-98  I-01-28  mka      $$6 Correct TEST_CALL_REPORT
\*====================================================================*/

/*--------------------------------------------------------------------*\
Pro/Toolkit includes -- include this first
\*--------------------------------------------------------------------*/
#include "ProToolkit.h"
#include "ProMdl.h"
#include "ProMenu.h"
#include "ProMessage.h"
#include "ProRelSet.h"
#include "ProModelitem.h"
#include "ProSelection.h"

/*--------------------------------------------------------------------*\
Pro/Develop includes
\*--------------------------------------------------------------------*/

/*--------------------------------------------------------------------*\
Application includes
\*--------------------------------------------------------------------*/
#include "TestError.h"
#include "TestSetup.h"
#include "UtilMessage.h"
#include "UtilString.h"
#include "TestFiletypes.h"
#include "UtilTypes.h"
#include "UtilCollect.h"
/*--------------------------------------------------------------------*\
Application macros
\*--------------------------------------------------------------------*/

#define USER_RL_EDIT  0 
#define USER_RL_DELETE 1

/*====================================================================*\
FUNCTION : ProTestRelationsMenu
PURPOSE  : Top level menu for the Relations Menu testing.
\*====================================================================*/
int ProTestRelationsMenu(ProMdl *model)
{
  ProMdlType   model_type;
  int          action;
  ProError     status;

    int ProTestModelRelsMenu();
    int ProTestModelRelsetInfo();

    status = ProMdlTypeGet(*model, &model_type);
    TEST_CALL_REPORT("ProMdlTypeGet()", "ProTestRelationsMnu()", status,
						status != PRO_TK_NO_ERROR);
    ProMenuPush();
    ProMenuFileRegister("TkModelRel","tkmodelrel.mnu", NULL);
    ProMenubuttonActionSet("TkModelRel","TkModelRel",
                           (ProMenubuttonAction)ProMenuDelete, NULL,  0);
    ProMenubuttonActionSet("TkModelRel","-Done",
                           (ProMenubuttonAction)ProMenuDelete, NULL,  0);
    ProMenubuttonActionSet("TkModelRel","-Assem Rel",
                           (ProMenubuttonAction)ProTestModelRelsMenu,
                                               model, PRO_RELOBJ_ASSEMBLY);
    ProMenubuttonActionSet("TkModelRel","-Part Rel",
                           (ProMenubuttonAction)ProTestModelRelsMenu,
                                               model, PRO_RELOBJ_PART);
    ProMenubuttonActionSet("TkModelRel","-Feat Rel",
                           (ProMenubuttonAction)ProTestModelRelsMenu,
                                               model, PRO_RELOBJ_FEATURE);
    ProMenubuttonActionSet("TkModelRel","-Surface Rel",
                           (ProMenubuttonAction)ProTestModelRelsMenu,
                                               model, PRO_RELOBJ_SURFACE);
    ProMenubuttonActionSet("TkModelRel","-Edge Rel",
                           (ProMenubuttonAction)ProTestModelRelsMenu,
                                               model, PRO_RELOBJ_EDGE);
    ProMenubuttonActionSet("TkModelRel","-Info",
        (ProMenubuttonAction)ProTestModelRelsetInfo, model, 0);
    ProMenuCreate(PROMENUTYPE_MAIN, "TkModelRel", NULL);
    if ( model_type == PRO_ASSEMBLY)
        ProMenubuttonActivate("TkModelRel","-Assem Rel");
    else
        ProMenubuttonDeactivate("TkModelRel","-Assem Rel");

    ProMenuProcess("", &action);
    ProMenuPop();

  return(0);
}


/*====================================================================*\
    FUNCTION :	ProTestModelRelsMenu()
    PURPOSE  :  General action function for the Note testing.
\*====================================================================*/
int ProTestModelRelsMenu(
    ProMdl *model,
    int action)
{
  ProError status;
  ProMdlType   model_type;
  ProModelitem modelitem;
  int select = 0, n_sel;
  char *opt = NULL;
  ProSelection *p_sel;
  ProCharLine str;
  ProName w_name;

  int ProTestAddModelitemRel();
  int ProTestEditModelitemRel();
  int ProTestRegenerateModelitemRel();

    status = ProMdlTypeGet(*model, &model_type);
    TEST_CALL_REPORT("ProMdlTypeGet()", "ProTestModelRelsMenu()", status,
						status != PRO_TK_NO_ERROR);

    if (((action == PRO_RELOBJ_PART) && ( model_type == PRO_PART)) ||
	(action == PRO_RELOBJ_ASSEMBLY))
    {
	status = ProMdlToModelitem(*model, &modelitem);
	TEST_CALL_REPORT("ProMdlToModelitem()", "ProTestModelRelsMenu()", 
	    status, status != PRO_TK_NO_ERROR);
	select = 1;
    }
    else
    {
    	switch(action)
    	{
	case PRO_RELOBJ_PART :
	    opt = "part";
            break;
    	case PRO_RELOBJ_FEATURE :
	    opt = "feature";
       	    break;
        case PRO_RELOBJ_SURFACE :
	    opt = "surface";
            break;
        case PRO_RELOBJ_EDGE :
	    opt = "edge";
            break;
        case PRO_RELOBJ_PAT_DIR1 :
            break;
	}

	if (opt != NULL)
	{
	    sprintf(str, "Select %s", opt);
    	    ProUtilMsgPrint("gen", "TEST %0s", str);

	    status = ProSelect(opt, 1, NULL, NULL, NULL, NULL, &p_sel, &n_sel);
	    if (status == PRO_TK_NO_ERROR && n_sel == 1)
	    {
		status = ProSelectionModelitemGet(p_sel[0], &modelitem);
	        TEST_CALL_REPORT("ProSelectionModelitemGet()", 
		    "ProTestModelRelsMenu()", status, 
		    status != PRO_TK_NO_ERROR);
		
		if (modelitem.type==PRO_EDGE || modelitem.type==PRO_SURFACE)
		{
/*--------------------------------------------------------------------------*\
   Only named edges and surfaces can have relations 
\*--------------------------------------------------------------------------*/
		    status = ProModelitemNameGet(&modelitem, w_name);
	            TEST_CALL_REPORT("ProModelitemNameGet()", 
		    	"ProTestModelRelsMenu()", status, 
			status != PRO_TK_NO_ERROR);

		    if (status != PRO_TK_NO_ERROR)
		    {
			sprintf(str, "Please name selected %s first [-QUIT-]:",
			    opt);
    	    		ProUtilMsgPrint("gen", "TEST %0s", str);

			status = ProMessageStringRead(PRO_NAME_SIZE, w_name);
	            	TEST_CALL_REPORT("ProMessageStringRead()", 
		    	    "ProTestModelRelsMenu()", status, 
			     status != PRO_TK_NO_ERROR);
			
			if (status == PRO_TK_NO_ERROR)
			{
			    status = ProModelitemNameSet(&modelitem, w_name);
	            	    TEST_CALL_REPORT("ProModelitemNameSet()", 
		    	    	"ProTestModelRelsMenu()", status, 
			     	status != PRO_TK_NO_ERROR);

			    if (status == PRO_TK_NO_ERROR)
				select = 1;				
			}
		    }
		}
		else
		    select = 1;
	    }
        }
    }

    ProMenuFileRegister("TkRelations","tkrelations.mnu", NULL);
    ProMenubuttonActionSet("TkRelations","TkRelations",
                           (ProMenubuttonAction)ProMenuDelete, NULL,  0);
    ProMenubuttonActionSet("TkRelations","-Add Rel",
        (ProMenubuttonAction)ProTestAddModelitemRel, &modelitem, 0);
    ProMenubuttonActionSet("TkRelations","-Edit Rel",
        (ProMenubuttonAction)ProTestEditModelitemRel, &modelitem, USER_RL_EDIT);
    ProMenubuttonActionSet("TkRelations","-Delete Rel",
        (ProMenubuttonAction)ProTestEditModelitemRel, &modelitem, 
	USER_RL_DELETE);
    ProMenubuttonActionSet("TkRelations","-Regenerate",
        (ProMenubuttonAction)ProTestRegenerateModelitemRel, &modelitem, 0);
    ProMenubuttonActionSet("TkRelations","-Done/Return",
        (ProMenubuttonAction)ProMenuDelete, NULL, 0);

    ProMenuCreate(PROMENUTYPE_MAIN, "TkRelations", NULL);
    ProMenuProcess("", &action);

  return(0);
}


/*====================================================================*\
    FUNCTION :	ProTestAddModelitemRel()
    PURPOSE  :  Add new relation to the modelitem
\*====================================================================*/
int ProTestAddModelitemRel(
    ProModelitem *modelitem)
{
  ProError status;
  ProLine w_line;
  ProWstring *w_array;
  int n_lines, cont=1;
  ProRelset relset;
  wchar_t *w_ptr;

    while (cont)
    {
    ProUtilMsgPrint("gen", "TEST %0s", "Enter Relation [QUIT]:");

    status = ProMessageStringRead(PRO_LINE_SIZE, w_line);
    if (status == PRO_TK_NO_ERROR)
    {   
	status = ProArrayAlloc(0, sizeof(ProWstring), 1, (ProArray*)&w_array);
	TEST_CALL_REPORT("ProArrayAlloc()", "ProTestAddModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
                                        
	status = ProModelitemToRelset(modelitem, &relset);
	TEST_CALL_REPORT("ProModelitemToRelset()", "ProTestAddModelitemRel()", 
	    status, status != PRO_TK_NO_ERROR && status != PRO_TK_E_NOT_FOUND);
	
	if (status == PRO_TK_NO_ERROR)
	{
	    status = ProRelsetRelationsGet(&relset, &w_array);
	    TEST_CALL_REPORT("ProRelsetRelationsGet()", 
		"ProTestAddModelitemRel()", status, (status != PRO_TK_NO_ERROR)&&
                        (status!=PRO_TK_BAD_INPUTS));
	}
	else
	{
	    status = ProRelsetCreate(modelitem, &relset);
	    TEST_CALL_REPORT("ProRelsetCreate()", 
		"ProTestAddModelitemRel()", status, status != PRO_TK_NO_ERROR);
	}
	w_ptr = w_line;
	status = ProArrayObjectAdd((ProArray*)&w_array, PRO_VALUE_UNUSED, 1,
	    &w_ptr);
	TEST_CALL_REPORT("ProArrayObjectAdd()", "ProTestAddModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
	status = ProArraySizeGet((ProArray)w_array, &n_lines);
	TEST_CALL_REPORT("ProArraySizeGet()", "ProTestAddModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
	status = ProRelsetRelationsSet(&relset, w_array, n_lines);
	TEST_CALL_REPORT("ProRelsetRelationsSet()", 
	    "ProTestAddModelitemRel()", status, status != PRO_TK_NO_ERROR);
	if (status == PRO_TK_NO_ERROR)
	    ProUtilMsgPrint("gen", "TEST %0s", "Relation added successfully");
	
	status = ProArrayFree((ProArray*)&w_array);
	TEST_CALL_REPORT("ProArrayFree()", "ProTestAddModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    }
    else
	cont = 0;
    }
    return (0);
}

/*====================================================================*\
    FUNCTION :	ProTestEditModelitemRel()
    PURPOSE  :  Edit/Delete existing relations for the modelitem
\*====================================================================*/
int ProTestEditModelitemRel(
    ProModelitem *modelitem,
    int action)
{
  ProError status;
  ProLine w_line;
  ProCharLine str, line;
  ProWstring *w_array;
  int i, n_lines;
  wchar_t w_empty = 0, **w_select, *w_ptr;
  ProRelset relset;
  ProName w_name;

    status = ProModelitemToRelset(modelitem, &relset);
    TEST_CALL_REPORT("ProModelitemToRelset()", "ProTestEditModelitemRel()", 
	    status, status != PRO_TK_NO_ERROR && status != PRO_TK_E_NOT_FOUND);
	
    if (status != PRO_TK_NO_ERROR)
    {
    	ProUtilMsgPrint("gen", "TEST %0s", "No relations found");
	return (0);
    }

    status = ProArrayAlloc(0, sizeof(ProWstring), 1, (ProArray*)&w_array);
    TEST_CALL_REPORT("ProArrayAlloc()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);

    status = ProRelsetRelationsGet(&relset, &w_array);
    TEST_CALL_REPORT("ProRelsetRelationGet()", 
	"ProTestAddModelitemRel()", status, status != PRO_TK_NO_ERROR);

    w_ptr = &w_empty;
    status = ProArrayObjectAdd((ProArray*)&w_array, PRO_VALUE_UNUSED, 1,
	    &w_ptr);
    TEST_CALL_REPORT("ProArrayObjectAdd()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    ProStringToWstring(w_name, "Item Rels");
    status = ProMenuStringsSelect(w_name, w_array, 1, w_array, &w_select, 
	&n_lines);

    if ((status != PRO_TK_NO_ERROR) || (n_lines<1))
	return (0); 
    status = ProArraySizeGet((ProArray)w_array, &n_lines);
    TEST_CALL_REPORT("ProArraySizeGet()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    for (i=0; i<n_lines-1; i++)
    {
       if (ProUtilWstrCmp(w_select[0], w_array[i])==0)
	   break;

    }
    status = ProArrayObjectRemove((ProArray*)&w_array, n_lines-1, 1);
    TEST_CALL_REPORT("ProArrayFree()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);

    if (action == USER_RL_EDIT)
    {
	ProWstringToString(line, w_select[0]);
	sprintf(str, "Enter Relation [%s]:", line);
    	ProUtilMsgPrint("gen", "TEST %0s", str);

    	status = ProMessageStringRead(PRO_LINE_SIZE, w_line);
	if (status == PRO_TK_NO_ERROR)
	    w_array[i] = w_line;
    }
    else
    {
	status = ProArrayObjectRemove((ProArray*)&w_array, i, 1);
	TEST_CALL_REPORT("ProArrayFree()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);

	n_lines--;
    }
	
    status = ProArraySizeGet((ProArray)w_array, &n_lines);
    TEST_CALL_REPORT("ProArraySizeGet()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    if (n_lines<1)
    {
	status =  ProRelsetDelete(&relset);
	TEST_CALL_REPORT("ProRelsetDelete()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    }
    else
    {
    	status = ProRelsetRelationsSet(&relset, w_array, n_lines);
    	TEST_CALL_REPORT("ProRelsetRelationsSet()", 
	    "ProTestEditModelitemRel()", status, status != PRO_TK_NO_ERROR);
    }
	
    status = ProArrayFree((ProArray*)&w_array);
    TEST_CALL_REPORT("ProArrayFree()", "ProTestEditModelitemRel()", 
					status, status != PRO_TK_NO_ERROR);
    return (0);
}


/*====================================================================*\
    FUNCTION :	ProTestRegenerateModelitemRel()
    PURPOSE  :  Regenerate relset 
\*====================================================================*/
int ProTestRegenerateModelitemRel(
    ProModelitem *modelitem,
    int action)
{
  ProError status;
  wchar_t w_empty = 0;
  ProRelset relset;

    status = ProModelitemToRelset(modelitem, &relset);
    TEST_CALL_REPORT("ProModelitemToRelset()", 
	"ProTestRegeneratModelitemRel()", 
	    status, status != PRO_TK_NO_ERROR && status != PRO_TK_E_NOT_FOUND);
	
    if (status != PRO_TK_NO_ERROR)
    {
    	ProUtilMsgPrint("gen", "TEST %0s", "No relations found");
	return (0);
    }

    status = ProRelsetRegenerate(&relset);
    TEST_CALL_REPORT("ProRelsetRegenerate()", 
	"ProTestRegeneratModelitemRel()", status, status != PRO_TK_NO_ERROR);

    if (status == PRO_TK_NO_ERROR)
	ProUtilMsgPrint("gen", "TEST %0s", "Relset regenerated successfully");

    return (0);
}

/*====================================================================*\
    FUNCTION :	ProTestRelsetVisitAction()
    PURPOSE  :  Write info about one relset to the file
\*====================================================================*/
int ProTestRelsetVisitAction(
    ProRelset *p_relset,
    FILE *fp)
{
  ProError status;
  ProLine w_line;
  ProModelitem modelitem;
  ProCharLine str_v, line;
  ProParamvalue value;
  int i, n_lines;
  ProWstring *w_array;

    status = ProRelsetToModelitem(p_relset, &modelitem);
    TEST_CALL_REPORT("ProRelsetToModelitem()", "ProTestrelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
    fprintf(fp, "Id %4d  Type %4d\n", modelitem.id, modelitem.type);

    status = ProArrayAlloc(0, sizeof(ProWstring), 1, (ProArray*)&w_array);
    TEST_CALL_REPORT("ProArrayAlloc()", "ProTestRelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
    
    status = ProRelsetRelationsGet(p_relset, &w_array);
    TEST_CALL_REPORT("ProRelsetRelationsGet()", "ProTestRelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
    
    status = ProArraySizeGet((ProArray)w_array, &n_lines);
    TEST_CALL_REPORT("ProArraySizeGet()", "ProTestRelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
    for (i=0; i<n_lines; i++)
    {
	ProWstringToString(line, w_array[i]);
	ProStringToWstring(w_line, line);

	if (strchr(line, '=')!=NULL)
	{
	    w_line[strchr(line, '=') - line] = 0; 

    	    status = ProRelationEval(p_relset, w_line, &value);
    	    TEST_CALL_REPORT("ProRelationEval()", "ProTestRelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
	    if (status != PRO_TK_NO_ERROR)
	    	value.type = PRO_PARAM_VOID;
            switch (value.type)
	    {
	    case PRO_PARAM_DOUBLE:
	    	sprintf(str_v, "%8.2f", value.value.d_val);
	    	break;
	    case PRO_PARAM_INTEGER:
	    case PRO_PARAM_NOTE_ID:
	    	sprintf(str_v, "%d", value.value.i_val);
	        break;
	    case PRO_PARAM_STRING:
	    	ProWstringToString(str_v, value.value.s_val);
	    	break;
	    case PRO_PARAM_BOOLEAN:
	    	strcpy(str_v, value.value.i_val ? "TRUE" : "FALSE");
	    	break;
   	    default:
	    	strcpy(str_v, "ERROR");
	    	break;
	    }
	}
	else
	   str_v[0]='\0'; 
	fprintf(fp, "%10s%25s  %10s\n", " ", line, str_v);
    }

    status = ProArrayFree((ProArray*)&w_array);
    TEST_CALL_REPORT("ProArrayFree()", "ProTestRelsetVisitAction()", 
					status, status != PRO_TK_NO_ERROR);
    return (0);
}

/*====================================================================*\
    FUNCTION :	ProTestModelRelsetInfo()
    PURPOSE  :  Collect info information for all rels in the model
\*====================================================================*/
int ProTestModelRelsetInfo(
    ProMdl *model)
{
  ProError status;
  char fname[PRO_FILE_NAME_SIZE];
  FILE *fp;
  ProPath w_path;
  ProRelset	    *relset;
  int relset_num, i;

    ProTestQcrName(model, ".rls", fname);
    fp = fopen(fname, "w");

    fprintf(fp, "Owner                      Relation           Value\n");           

    status = ProUtilCollectRelset (*model, &relset);
    if (status == PRO_TK_NO_ERROR)
    {
        status = ProArraySizeGet ((ProArray)relset, &relset_num);
        TEST_CALL_REPORT( "ProArraySizeGet()", "ProTestModelRelsetInfo()", 
            status, status != PRO_TK_NO_ERROR );
        for (i = 0; i < relset_num; i++)
        {
            status = ProTestRelsetVisitAction (relset+i,
	        (ProAppData)fp);
        }
        status = ProArrayFree ((ProArray*)&relset);
        TEST_CALL_REPORT( "ProArrayFree()", "ProTestModelRelsetInfo()", 
            status, status != PRO_TK_NO_ERROR );
    }
    
    fclose(fp);
    ProStringToWstring(w_path, fname);

    ProInfoWindowDisplay(w_path, NULL, NULL);
    return (0);
}