Commit: 205e7cdf584b86951dbd592c3b733e67f804f264 - fermisurfer (git)

fermisurfer Git

Commit MetaInfo

• Revision: 205e7cdf584b86951dbd592c3b733e67f804f264 (tree)
• Zeit: 2018-01-06 00:10:06
• Autor: mitsuaki1987 <kawamitsuaki@gmai...>
• Commiter: mitsuaki1987

Log Message

Add batch mode

Ändern Zusammenfassung

• modified: src/fermisurfer.c
• modified: src/menu.c
• modified: src/menu.h
• modified: src/read_file.c
• modified: src/read_file.h
• modified: src/variable.h

Diff

--- a/src/fermisurfer.c
+++ b/src/fermisurfer.c
@@ -63,6 +63,7 @@ with a color-plot of the arbitraly matrix element
 
 #include <stdlib.h>
 #include <stdio.h>
+#include <string.h>
 #include "variable.h"
 #include "read_file.h"
 #include "menu.h"
@@ -101,6 +102,27 @@ void init(void)
   FS_CreateMenu();
 } /* init */
 /**
+  @brief Glut Display function
+  called by glutDisplayFunc
+*/
+void timer(int it)
+{
+  int ibatch, ierr;
+  char command_name[256];
+
+  printf("\n  Batch mode.\n");
+  for (ibatch = 0; ibatch < nbatch; ibatch++) {
+    read_batch(batch_file[ibatch]);
+    refresh_patch_segment();
+    //glutPostRedisplay();
+    display();
+    glFlush();
+    sprintf(command_name, "import -window \"%s\" %s%d.png", window_name, window_name, ibatch);
+    ierr = system(command_name);
+  }/*for (ibatch = 0; ibatch < nbatch; ibatch++)*/
+  exit(0);
+}
+/**
  @brief Main routine of FermiSurfer
 
  Refer: ::query
@@ -112,6 +134,8 @@ int main(
   char *argv[] //!< [in] Input file name
 )
 {
+  int ibatch;
+
   printf("\n");
   printf("###########################################\n");
   printf("##                                       ##\n");
@@ -119,6 +143,7 @@ int main(
   printf("##                                       ##\n");
   printf("###########################################\n");
   printf("\n");
+  nbatch = argc - 4;
   if (argc < 2) {
     printf("\n");
     printf("  Input file is not specified !\n");
@@ -126,6 +151,14 @@ int main(
     getchar();
     exit(-1);
   }
+  if (nbatch > 0) {
+    strcpy(window_name, argv[1]);
+    batch_file = (char**)malloc(sizeof(char*) * nbatch);
+    for (ibatch = 0; ibatch < nbatch; ibatch++) {
+      batch_file[ibatch] = (char*)malloc(sizeof(char) * 256);
+      strcpy(batch_file[ibatch], argv[4 + ibatch]);
+    }
+  }
   /**/
 #if defined(_OPENMP)
 #pragma omp parallel 
@@ -166,6 +199,7 @@ int main(
   printf("\n");
   /**/
   glutInit(&argc, argv);
+  if (argc >= 4)glutInitWindowSize(atoi(argv[2]), atoi(argv[3]));
   glutInitDisplayMode(GLUT_RGBA | GLUT_DOUBLE  | GLUT_DEPTH);
   glutCreateWindow(argv[1]);
   glutDisplayFunc(display);
@@ -175,6 +209,7 @@ int main(
   glutKeyboardFunc(keyboard);
   glutSpecialFunc(special_key);
   glutMenuStateFunc(FS_ModifyMenu);
+  if (nbatch > 0) glutTimerFunc(10, timer, 0);
   init();
   glutMainLoop();
   return 0;

--- a/src/menu.c
+++ b/src/menu.c
@@ -57,7 +57,7 @@ void compute_patch_segment() {
   query = 0; equator();
 }
 
-static void refresh_patch_segment() {
+void refresh_patch_segment() {
   free_patch();
   compute_patch_segment();
 }

--- a/src/menu.h
+++ b/src/menu.h
@@ -25,3 +25,4 @@ THE SOFTWARE.
 void compute_patch_segment();
 void FS_CreateMenu();
 void FS_ModifyMenu(int status);
+void refresh_patch_segment();

--- a/src/read_file.c
+++ b/src/read_file.c
@@ -26,8 +26,17 @@ THE SOFTWARE.
 */
 #include <stdlib.h>
 #include <stdio.h>
+#include <math.h>
 #include "variable.h"
 #include "basic_math.h"
+#if defined(HAVE_CONFIG_H)
+#include <config.h>
+#endif
+#if defined(HAVE_GL_GLUT_H)
+#include <GL/glut.h>
+#elif defined(HAVE_GLUT_GLUT_H)
+#include <GLUT/glut.h>
+#endif
 /**
  @brief Input from Fermi surface file
 */
@@ -183,3 +192,58 @@ void read_file(
   }
   fclose(fp);
 } /* read_file */
+
+void read_batch(
+  char *fname//!<[in] Input file name
+)
+{
+  FILE *fp;
+  int ierr, ib;
+
+  printf("  Openning batch file %s ...\n", fname);
+  if ((fp = fopen(fname, "r")) == NULL) {
+    printf("file open error!!\n");
+    printf("  Press any key to exit.\n");
+    getchar();
+    exit(EXIT_FAILURE);
+  }
+  ierr = fscanf(fp, "%d", &blackback);
+  for (ib = 0; ib < nb; ib++) 
+    ierr = fscanf(fp, "%d", &draw_band[ib]);
+  ierr = fscanf(fp, "%d", &lcolorbar);
+  ierr = fscanf(fp, "%d", &fcscl);
+  ierr = fscanf(fp, "%d", &lequator);
+  if (lequator == 1) 
+    ierr = fscanf(fp, "%f %f %f", &eqvec[0], &eqvec[1], &eqvec[2]);
+  ierr = fscanf(fp, "%d", &interpol);
+  ierr = fscanf(fp, "%d", &lside);
+  ierr = fscanf(fp, "%f", &linewidth);
+  ierr = fscanf(fp, "%d", &nodeline);
+  ierr = fscanf(fp, "%d", &lsection);
+  if (lsection == 1) {
+    ierr = fscanf(fp, "%f", &secscale);
+    ierr = fscanf(fp, "%f %f %f", &secvec[0], &secvec[1], &secvec[2]);
+  }
+  ierr = fscanf(fp, "%f", &EF);
+  ierr = fscanf(fp, "%d", &lstereo);
+  ierr = fscanf(fp, "%f", &scl);
+  ierr = fscanf(fp, "%f %f %f", &trans[0], &trans[1], &trans[2]);
+  ierr = fscanf(fp, "%f %f %f", &thetax, &thetay, &thetaz);
+  fclose(fp);
+  if (blackback == 1) glClearColor(0.0, 0.0, 0.0, 0.0);
+  else glClearColor(1.0, 1.0, 1.0, 0.0);
+
+  thetax = 3.14159265f / 180.0f * thetax;
+  thetay = 3.14159265f / 180.0f * thetay;
+  thetaz = 3.14159265f / 180.0f * thetaz;
+
+  rot[0][0] = cosf(thetay)* cosf(thetaz);
+  rot[0][1] = -cosf(thetay)* sinf(thetaz);
+  rot[0][2] = sinf(thetay);
+  rot[1][0] = cosf(thetaz)* sinf(thetax)* sinf(thetay) + cosf(thetax)* sinf(thetaz);
+  rot[1][1] = cosf(thetax) * cosf(thetaz) - sinf(thetax)* sinf(thetay)* sinf(thetaz);
+  rot[1][2] = -cosf(thetay)* sinf(thetax);
+  rot[2][0] = -cosf(thetax)* cosf(thetaz)* sinf(thetay) + sinf(thetax)* sinf(thetaz);
+  rot[2][1] = cosf(thetaz)* sinf(thetax) + cosf(thetax)* sinf(thetay)* sinf(thetaz);
+  rot[2][2] = cosf(thetax)* cosf(thetay);
+}

--- a/src/read_file.h
+++ b/src/read_file.h
@@ -23,3 +23,4 @@ THE SOFTWARE.
 */
 
 void read_file(char *fname/**<[in] fname  Input file name*/);
+void read_batch(char *fname/**<[in] fname  Input file name*/);

--- a/src/variable.h
+++ b/src/variable.h
@@ -170,3 +170,9 @@ enum
   MOUSE_SCROLL_DOWN = 4 //!< Mouse wheel down
 };
 int nthreads;//!< Number of OpenMP threads
+/*
+ Batch mode
+*/
+int nbatch;
+char **batch_file;
+char window_name[256];