#include "DriverMED_W_SMDS_Mesh.h"

#include "SMDS_MeshElement.hxx"
#include "SMDS_MeshNode.hxx"
#include "SMDS_MeshFacesIterator.hxx"
#include "SMDS_MeshNodesIterator.hxx"
#include "SMDS_MeshVolumesIterator.hxx"

#include "utilities.h"
#include <vector>

DriverMED_W_SMDS_Mesh::DriverMED_W_SMDS_Mesh() {
;
}

DriverMED_W_SMDS_Mesh::~DriverMED_W_SMDS_Mesh() {
;
}

void DriverMED_W_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
  myMesh = aMesh;
}

void DriverMED_W_SMDS_Mesh::SetFile(string aFile) {
  myFile = aFile;
}

void DriverMED_W_SMDS_Mesh::SetFileId(med_idt aFileId) {
  myFileId = aFileId;
}

void DriverMED_W_SMDS_Mesh::SetMeshId(int aMeshId) {
  myMeshId = aMeshId;
}

void DriverMED_W_SMDS_Mesh::Add() {
  ;
}

void DriverMED_W_SMDS_Mesh::Write() {

  med_err ret = 0;
  int i,j,k,l;
  int numero;
  char message[200];
  Standard_Boolean ok;
  /* nombre d'objets MED */
  char nom_universel[MED_TAILLE_LNOM+1];
  med_int long_fichier_en_tete; 
  char *fichier_en_tete;
  char version_hdf[10];
  char version_med[10];
  med_int nmaa,mdim,nnoe;
  med_int nmai[MED_NBR_GEOMETRIE_MAILLE],nfac[MED_NBR_GEOMETRIE_FACE];
  med_int nare[MED_NBR_GEOMETRIE_ARETE];
  /* nom du maillage */
  char nommaa[MED_TAILLE_NOM+1];
  /* noeuds */
  med_float *coo;
  char nomcoo[3*MED_TAILLE_PNOM+1];
  char unicoo[3*MED_TAILLE_PNOM+1];
  char *nomnoe;
  med_int *numnoe;
  med_int *nufano; 
  med_repere rep;
  med_booleen inonoe,inunoe;
  med_mode_switch mode_coo;
  char str[MED_TAILLE_PNOM+1];
  med_int nbNodes;
  /* elements */
  med_int nsup;
  med_int edim;
  med_int taille;
  med_int elem_id,myId;
  med_int *connectivite;
  char *nomele;
  med_int *numele;
  med_int *nufael;
  med_booleen inoele, inuele;
  med_connectivite typ_con;
  med_geometrie_element typgeo;
  med_geometrie_element typmai[MED_NBR_GEOMETRIE_MAILLE] = {MED_POINT1,MED_SEG2, 
						   MED_SEG3,MED_TRIA3,
						   MED_TRIA6,MED_QUAD4,
						   MED_QUAD8,MED_TETRA4,
						   MED_TETRA10,MED_HEXA8,
						   MED_HEXA20,MED_PENTA6,
						   MED_PENTA15,MED_PYRA5,
						   MED_PYRA13};
  med_int desmai[MED_NBR_GEOMETRIE_MAILLE] = {0,2,3,3,3,4,4,4,4,6,6,5,5,5,5};
  med_int nmailles[MED_NBR_GEOMETRIE_MAILLE];
  char nommai[MED_NBR_GEOMETRIE_MAILLE] [MED_TAILLE_NOM+1] = {"MED_POINT1",
							  "MED_SEG2", 
							  "MED_SEG3",
							  "MED_TRIA3",
							  "MED_TRIA6",
							  "MED_QUAD4",
							  "MED_QUAD8",
							  "MED_TETRA4",
							  "MED_TETRA10",
							  "MED_HEXA8",
							  "MED_HEXA20",
							  "MED_PENTA6",
							  "MED_PENTA15",
							  "MED_PYRA5",
							  "MED_PYRA13"};
  med_geometrie_element typfac[MED_NBR_GEOMETRIE_FACE] = {MED_TRIA3,MED_TRIA6,
						 MED_QUAD4,MED_QUAD8};
  med_int desfac[MED_NBR_GEOMETRIE_FACE] = {3,3,4,4};
  med_int nfaces[MED_NBR_GEOMETRIE_FACE];
  char nomfac[MED_NBR_GEOMETRIE_FACE][MED_TAILLE_NOM+1] = {"MED_TRIA3","MED_TRIA6",
						       "MED_QUAD4","MED_QUAD8"};
  med_geometrie_element typare[MED_NBR_GEOMETRIE_ARETE] = {MED_SEG2,MED_SEG3};
  med_int desare[MED_NBR_GEOMETRIE_ARETE] = {2,3};
  med_int naretes[MED_NBR_GEOMETRIE_ARETE];
  char nomare[MED_NBR_GEOMETRIE_ARETE] [MED_TAILLE_NOM+1] = {"MED_SEG2","MED_SEG3"};
  /* familles */
  med_int nfam;
  med_int natt,ngro;
  char *attdes,*gro;
  med_int *attval,*attide;
  char nomfam[MED_TAILLE_NOM+1];
  med_int numfam;
  char str1[MED_TAILLE_DESC+1];
  char str2[MED_TAILLE_LNOM+1];

  /****************************************************************************
  *                      OUVERTURE DU FICHIER EN ECRITURE                      *
  ****************************************************************************/
  /*!  char* file2Read = (char*)myFile.c_str();
  myFileId = MEDouvrir(file2Read,MED_REMP);
  if (myFileId < 0)
    {
      fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
      exit(EXIT_FAILURE);
    }
  */
  typ_con = MED_NOD;
  mode_coo = MED_FULL_INTERLACE;
  numero = myMeshId;

  /****************************************************************************
  *                       NOMBRES D'OBJETS MED                                *
  ****************************************************************************/
  fprintf(stdout,"\n(****************************)\n");
  fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
  fprintf(stdout,"(****************************)\n");

  /* creation du maillage */
  mdim=3;
  //nommaa = QString("Mesh "+myMeshId).latin1();
  //nommaa = "";
  //plutot recuperer le nom du maillage dans l'etude
  if (ret == 0)
    ret = MEDmaaCr(myFileId,nommaa,mdim);
  printf("%d\n",ret);

      
  /* Combien de noeuds ? */
  nnoe = myMesh->NbNodes();
  //SCRUTE(nnoe);
  /* Combien de mailles, faces ou aretes ? */
  for (i=0;i<MED_NBR_GEOMETRIE_MAILLE;i++)
    nmailles[i]=0;

  SMDS_MeshFacesIterator itFaces(myMesh);
  Standard_Integer nb_of_nodes, nb_of_faces;
  nb_of_faces = myMesh->NbFaces();
  //SCRUTE(nb_of_faces);

  //med_int elem_Id[MED_NBR_GEOMETRIE_FACE][nb_of_faces];
  vector<int> elem_Id[MED_NBR_GEOMETRIE_MAILLE];
  //typedef vector<int> Integer_vector;
  //vector<Integer_vector> elem_Id;
  //elem_Id.resize(MED_NBR_GEOMETRIE_MAILLE);
   
  for (;itFaces.More();itFaces.Next()) {
    const Handle(SMDS_MeshElement)& elem = itFaces.Value();

    nb_of_nodes = elem->NbNodes();

    switch (nb_of_nodes) {
    case 3 : {
      //elem_Id[3][nmailles[3]] = elem->GetID();
      elem_Id[3].push_back(elem->GetID());
      nmailles[3]++;
      break;
    }
    case 4 : {
      //elem_Id[5][nmailles[5]] = elem->GetID();
      elem_Id[5].push_back(elem->GetID());
      nmailles[5]++;
      break;
    }
    case 6 : {
      //elem_Id[4][nmailles[4]] = elem->GetID();
      elem_Id[4].push_back(elem->GetID());
      nmailles[4]++;
      break;
    }
    }

  }

  SMDS_MeshVolumesIterator itVolumes(myMesh);
   
  for (;itVolumes.More();itVolumes.Next()) {
    const Handle(SMDS_MeshElement)& elem = itVolumes.Value();

    nb_of_nodes = elem->NbNodes();
    switch (nb_of_nodes) {
    case 8 : {
      //elem_Id[9][nmailles[9]] = elem->GetID();
      elem_Id[9].push_back(elem->GetID());
      nmailles[9]++;
      break;
    }
    }
  }

  /****************************************************************************
  *                       ECRITURE DES NOEUDS                                  *
  ****************************************************************************/
  fprintf(stdout,"\n(************************)\n");
  fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
  fprintf(stdout,"(************************)\n");

  /* Allocations memoires */
  /* table des coordonnees 
     profil : (dimension * nombre de noeuds ) */
  coo = (med_float*) malloc(sizeof(med_float)*nnoe*mdim);
  /* table  des numeros, des numeros de familles des noeuds
     profil : (nombre de noeuds) */
  numnoe = (med_int*) malloc(sizeof(med_int)*nnoe);
  nufano = (med_int*) malloc(sizeof(med_int)*nnoe);
  /* table des noms des noeuds 
     profil : (nnoe*MED_TAILLE_PNOM+1) */
  nomnoe ="";

  i = 0;
  SMDS_MeshNodesIterator itNodes(myMesh);
  for (;itNodes.More();itNodes.Next()) {
    const Handle(SMDS_MeshElement)& elem = itNodes.Value();
    const Handle(SMDS_MeshNode)& node = myMesh->GetNode(1,elem);
 
    coo[i*3]=node->X();
    coo[i*3+1]=node->Y();
    coo[i*3+2]=node->Z();
    numnoe[i]=node->GetID();
    nufano[i]=0;
    i++;
  }
    
  /* ecriture des noeuds : 
     - coordonnees
     - noms (optionnel dans un fichier MED) 
     - numeros (optionnel dans un fichier MED) 
     - numeros des familles */
  ret = MEDnoeudsEcr(myFileId,nommaa,mdim,coo,mode_coo,MED_CART,
		       nomcoo,unicoo,nomnoe,MED_FAUX,numnoe,MED_VRAI,
		       nufano,nnoe,MED_REMP);

  /* liberation memoire */
  free(coo);
  //free(nomnoe);
  free(numnoe);
  free(nufano);

  /****************************************************************************
  *                       ECRITURE DES ELEMENTS                                *
  ****************************************************************************/
  fprintf(stdout,"\n(**************************)\n");
  fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
  fprintf(stdout,"(**************************)");
  //fprintf(Out,"CELLS\n");
  /* Ecriture des connectivites, noms, numeros des mailles */

  if (ret == 0)
    for (i=0;i<MED_NBR_GEOMETRIE_MAILLE;i++)
      {
	if (nmailles[i] > 0 && ret == 0)
	  {
	    MESSAGE ( " Start "<<typmai[i]);
	    /* dimension de la maille */
	    edim = typmai[i] / 100;
	    nsup = 0;
	    if (mdim  == 2 || mdim == 3)
	      if (edim == 1)
		nsup = 1;
	    if (mdim == 3)
	      if (edim == 2)
		nsup = 1;

	    taille = nsup+typmai[i]%100;
	    //taille = typmai[i]%100;
	    
	    /* allocation memoire */
	    connectivite = (med_int*)malloc(sizeof(med_int)*
					    taille*nmailles[i]);
	    nomele = (char*)malloc(sizeof(char)*MED_TAILLE_PNOM*
				   nmailles[i]+1);
	    numele = (med_int*)malloc(sizeof(med_int)*
				      nmailles[i]);
	    nufael = (med_int*)malloc(sizeof(med_int)*
				      nmailles[i]);
	    nomele="";
	    nbNodes=typmai[i]%100;

	    //penser a mater les (taille-nsup)
	    for (j=0;j<nmailles[i];j++) {
	      myId = elem_Id[i][j];
	      const Handle(SMDS_MeshElement)& elem = myMesh->FindElement(myId);
	      *(numele+j) = myId;
	      //elem_id=*(numele+j);
	      //fprintf(stdout,"%d \n",myId);

	      for (k=0; k<nbNodes; k++) {
		//*(connectivite+j*(taille-1)+k)=cnx[k];
		*(connectivite+j*(taille-nsup)+k) = elem->GetConnection(k+1);
		//fprintf(stdout,"%d ",*(connectivite+j*(taille-nsup)+k));
	      }
	      nufael[j]=0;
	      //fprintf(stdout,"\n");
	    }

	    /* ecriture des données */

	    ret=MEDelementsEcr(myFileId,nommaa,mdim,connectivite,mode_coo,nomele,MED_FAUX,numele,MED_VRAI,nufael,nmailles[i],MED_MAILLE,typmai[i],typ_con,MED_REMP);

	    if (ret < 0)
	      MESSAGE(">> ERREUR : lecture des mailles \n");

	    /* liberation memoire */
	    free(connectivite);
	    //free(nomele);
	    free(numele);
	    free(nufael);
	    MESSAGE ( " End "<<typmai[i]);
	  }
      }

  /****************************************************************************
  *                      FERMETURE DU FICHIER                                 *
  ****************************************************************************/

	    /*!  ret = MEDfermer(myFileId);

  if (ret != 0)
    fprintf(stderr,">> ERREUR : erreur a la fermeture du fichier %s\n",file2Read);
	    */

}