rdswitch.c
上传用户:wuyixingx
上传日期:2007-01-08
资源大小:745k
文件大小:10k
源码类别:

图形图象

开发平台:

C/C++

  1. /*
  2.  * rdswitch.c
  3.  *
  4.  * Copyright (C) 1991-1996, Thomas G. Lane.
  5.  * This file is part of the Independent JPEG Group's software.
  6.  * For conditions of distribution and use, see the accompanying README file.
  7.  *
  8.  * This file contains routines to process some of cjpeg's more complicated
  9.  * command-line switches.  Switches processed here are:
  10.  * -qtables file Read quantization tables from text file
  11.  * -scans file Read scan script from text file
  12.  * -qslots N[,N,...] Set component quantization table selectors
  13.  * -sample HxV[,HxV,...] Set component sampling factors
  14.  */
  15. #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */
  16. #include <ctype.h> /* to declare isdigit(), isspace() */
  17. LOCAL(int)
  18. text_getc (FILE * file)
  19. /* Read next char, skipping over any comments (# to end of line) */
  20. /* A comment/newline sequence is returned as a newline */
  21. {
  22.   register int ch;
  23.   
  24.   ch = getc(file);
  25.   if (ch == '#') {
  26.     do {
  27.       ch = getc(file);
  28.     } while (ch != 'n' && ch != EOF);
  29.   }
  30.   return ch;
  31. }
  32. LOCAL(boolean)
  33. read_text_integer (FILE * file, long * result, int * termchar)
  34. /* Read an unsigned decimal integer from a file, store it in result */
  35. /* Reads one trailing character after the integer; returns it in termchar */
  36. {
  37.   register int ch;
  38.   register long val;
  39.   
  40.   /* Skip any leading whitespace, detect EOF */
  41.   do {
  42.     ch = text_getc(file);
  43.     if (ch == EOF) {
  44.       *termchar = ch;
  45.       return FALSE;
  46.     }
  47.   } while (isspace(ch));
  48.   
  49.   if (! isdigit(ch)) {
  50.     *termchar = ch;
  51.     return FALSE;
  52.   }
  53.   val = ch - '0';
  54.   while ((ch = text_getc(file)) != EOF) {
  55.     if (! isdigit(ch))
  56.       break;
  57.     val *= 10;
  58.     val += ch - '0';
  59.   }
  60.   *result = val;
  61.   *termchar = ch;
  62.   return TRUE;
  63. }
  64. GLOBAL(boolean)
  65. read_quant_tables (j_compress_ptr cinfo, char * filename,
  66.    int scale_factor, boolean force_baseline)
  67. /* Read a set of quantization tables from the specified file.
  68.  * The file is plain ASCII text: decimal numbers with whitespace between.
  69.  * Comments preceded by '#' may be included in the file.
  70.  * There may be one to NUM_QUANT_TBLS tables in the file, each of 64 values.
  71.  * The tables are implicitly numbered 0,1,etc.
  72.  * NOTE: does not affect the qslots mapping, which will default to selecting
  73.  * table 0 for luminance (or primary) components, 1 for chrominance components.
  74.  * You must use -qslots if you want a different component->table mapping.
  75.  */
  76. {
  77.   FILE * fp;
  78.   int tblno, i, termchar;
  79.   long val;
  80.   unsigned int table[DCTSIZE2];
  81.   if ((fp = fopen(filename, "r")) == NULL) {
  82.     fprintf(stderr, "Can't open table file %sn", filename);
  83.     return FALSE;
  84.   }
  85.   tblno = 0;
  86.   while (read_text_integer(fp, &val, &termchar)) { /* read 1st element of table */
  87.     if (tblno >= NUM_QUANT_TBLS) {
  88.       fprintf(stderr, "Too many tables in file %sn", filename);
  89.       fclose(fp);
  90.       return FALSE;
  91.     }
  92.     table[0] = (unsigned int) val;
  93.     for (i = 1; i < DCTSIZE2; i++) {
  94.       if (! read_text_integer(fp, &val, &termchar)) {
  95. fprintf(stderr, "Invalid table data in file %sn", filename);
  96. fclose(fp);
  97. return FALSE;
  98.       }
  99.       table[i] = (unsigned int) val;
  100.     }
  101.     jpeg_add_quant_table(cinfo, tblno, table, scale_factor, force_baseline);
  102.     tblno++;
  103.   }
  104.   if (termchar != EOF) {
  105.     fprintf(stderr, "Non-numeric data in file %sn", filename);
  106.     fclose(fp);
  107.     return FALSE;
  108.   }
  109.   fclose(fp);
  110.   return TRUE;
  111. }
  112. #ifdef C_MULTISCAN_FILES_SUPPORTED
  113. LOCAL(boolean)
  114. read_scan_integer (FILE * file, long * result, int * termchar)
  115. /* Variant of read_text_integer that always looks for a non-space termchar;
  116.  * this simplifies parsing of punctuation in scan scripts.
  117.  */
  118. {
  119.   register int ch;
  120.   if (! read_text_integer(file, result, termchar))
  121.     return FALSE;
  122.   ch = *termchar;
  123.   while (ch != EOF && isspace(ch))
  124.     ch = text_getc(file);
  125.   if (isdigit(ch)) { /* oops, put it back */
  126.     if (ungetc(ch, file) == EOF)
  127.       return FALSE;
  128.     ch = ' ';
  129.   } else {
  130.     /* Any separators other than ';' and ':' are ignored;
  131.      * this allows user to insert commas, etc, if desired.
  132.      */
  133.     if (ch != EOF && ch != ';' && ch != ':')
  134.       ch = ' ';
  135.   }
  136.   *termchar = ch;
  137.   return TRUE;
  138. }
  139. GLOBAL(boolean)
  140. read_scan_script (j_compress_ptr cinfo, char * filename)
  141. /* Read a scan script from the specified text file.
  142.  * Each entry in the file defines one scan to be emitted.
  143.  * Entries are separated by semicolons ';'.
  144.  * An entry contains one to four component indexes,
  145.  * optionally followed by a colon ':' and four progressive-JPEG parameters.
  146.  * The component indexes denote which component(s) are to be transmitted
  147.  * in the current scan.  The first component has index 0.
  148.  * Sequential JPEG is used if the progressive-JPEG parameters are omitted.
  149.  * The file is free format text: any whitespace may appear between numbers
  150.  * and the ':' and ';' punctuation marks.  Also, other punctuation (such
  151.  * as commas or dashes) can be placed between numbers if desired.
  152.  * Comments preceded by '#' may be included in the file.
  153.  * Note: we do very little validity checking here;
  154.  * jcmaster.c will validate the script parameters.
  155.  */
  156. {
  157.   FILE * fp;
  158.   int scanno, ncomps, termchar;
  159.   long val;
  160.   jpeg_scan_info * scanptr;
  161. #define MAX_SCANS  100 /* quite arbitrary limit */
  162.   jpeg_scan_info scans[MAX_SCANS];
  163.   if ((fp = fopen(filename, "r")) == NULL) {
  164.     fprintf(stderr, "Can't open scan definition file %sn", filename);
  165.     return FALSE;
  166.   }
  167.   scanptr = scans;
  168.   scanno = 0;
  169.   while (read_scan_integer(fp, &val, &termchar)) {
  170.     if (scanno >= MAX_SCANS) {
  171.       fprintf(stderr, "Too many scans defined in file %sn", filename);
  172.       fclose(fp);
  173.       return FALSE;
  174.     }
  175.     scanptr->component_index[0] = (int) val;
  176.     ncomps = 1;
  177.     while (termchar == ' ') {
  178.       if (ncomps >= MAX_COMPS_IN_SCAN) {
  179. fprintf(stderr, "Too many components in one scan in file %sn",
  180. filename);
  181. fclose(fp);
  182. return FALSE;
  183.       }
  184.       if (! read_scan_integer(fp, &val, &termchar))
  185. goto bogus;
  186.       scanptr->component_index[ncomps] = (int) val;
  187.       ncomps++;
  188.     }
  189.     scanptr->comps_in_scan = ncomps;
  190.     if (termchar == ':') {
  191.       if (! read_scan_integer(fp, &val, &termchar) || termchar != ' ')
  192. goto bogus;
  193.       scanptr->Ss = (int) val;
  194.       if (! read_scan_integer(fp, &val, &termchar) || termchar != ' ')
  195. goto bogus;
  196.       scanptr->Se = (int) val;
  197.       if (! read_scan_integer(fp, &val, &termchar) || termchar != ' ')
  198. goto bogus;
  199.       scanptr->Ah = (int) val;
  200.       if (! read_scan_integer(fp, &val, &termchar))
  201. goto bogus;
  202.       scanptr->Al = (int) val;
  203.     } else {
  204.       /* set non-progressive parameters */
  205.       scanptr->Ss = 0;
  206.       scanptr->Se = DCTSIZE2-1;
  207.       scanptr->Ah = 0;
  208.       scanptr->Al = 0;
  209.     }
  210.     if (termchar != ';' && termchar != EOF) {
  211. bogus:
  212.       fprintf(stderr, "Invalid scan entry format in file %sn", filename);
  213.       fclose(fp);
  214.       return FALSE;
  215.     }
  216.     scanptr++, scanno++;
  217.   }
  218.   if (termchar != EOF) {
  219.     fprintf(stderr, "Non-numeric data in file %sn", filename);
  220.     fclose(fp);
  221.     return FALSE;
  222.   }
  223.   if (scanno > 0) {
  224.     /* Stash completed scan list in cinfo structure.
  225.      * NOTE: for cjpeg's use, JPOOL_IMAGE is the right lifetime for this data,
  226.      * but if you want to compress multiple images you'd want JPOOL_PERMANENT.
  227.      */
  228.     scanptr = (jpeg_scan_info *)
  229.       (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
  230.   scanno * SIZEOF(jpeg_scan_info));
  231.     MEMCOPY(scanptr, scans, scanno * SIZEOF(jpeg_scan_info));
  232.     cinfo->scan_info = scanptr;
  233.     cinfo->num_scans = scanno;
  234.   }
  235.   fclose(fp);
  236.   return TRUE;
  237. }
  238. #endif /* C_MULTISCAN_FILES_SUPPORTED */
  239. GLOBAL(boolean)
  240. set_quant_slots (j_compress_ptr cinfo, char *arg)
  241. /* Process a quantization-table-selectors parameter string, of the form
  242.  *     N[,N,...]
  243.  * If there are more components than parameters, the last value is replicated.
  244.  */
  245. {
  246.   int val = 0; /* default table # */
  247.   int ci;
  248.   char ch;
  249.   for (ci = 0; ci < MAX_COMPONENTS; ci++) {
  250.     if (*arg) {
  251.       ch = ','; /* if not set by sscanf, will be ',' */
  252.       if (sscanf(arg, "%d%c", &val, &ch) < 1)
  253. return FALSE;
  254.       if (ch != ',') /* syntax check */
  255. return FALSE;
  256.       if (val < 0 || val >= NUM_QUANT_TBLS) {
  257. fprintf(stderr, "JPEG quantization tables are numbered 0..%dn",
  258. NUM_QUANT_TBLS-1);
  259. return FALSE;
  260.       }
  261.       cinfo->comp_info[ci].quant_tbl_no = val;
  262.       while (*arg && *arg++ != ',') /* advance to next segment of arg string */
  263. ;
  264.     } else {
  265.       /* reached end of parameter, set remaining components to last table */
  266.       cinfo->comp_info[ci].quant_tbl_no = val;
  267.     }
  268.   }
  269.   return TRUE;
  270. }
  271. GLOBAL(boolean)
  272. set_sample_factors (j_compress_ptr cinfo, char *arg)
  273. /* Process a sample-factors parameter string, of the form
  274.  *     HxV[,HxV,...]
  275.  * If there are more components than parameters, "1x1" is assumed for the rest.
  276.  */
  277. {
  278.   int ci, val1, val2;
  279.   char ch1, ch2;
  280.   for (ci = 0; ci < MAX_COMPONENTS; ci++) {
  281.     if (*arg) {
  282.       ch2 = ','; /* if not set by sscanf, will be ',' */
  283.       if (sscanf(arg, "%d%c%d%c", &val1, &ch1, &val2, &ch2) < 3)
  284. return FALSE;
  285.       if ((ch1 != 'x' && ch1 != 'X') || ch2 != ',') /* syntax check */
  286. return FALSE;
  287.       if (val1 <= 0 || val1 > 4 || val2 <= 0 || val2 > 4) {
  288. fprintf(stderr, "JPEG sampling factors must be 1..4n");
  289. return FALSE;
  290.       }
  291.       cinfo->comp_info[ci].h_samp_factor = val1;
  292.       cinfo->comp_info[ci].v_samp_factor = val2;
  293.       while (*arg && *arg++ != ',') /* advance to next segment of arg string */
  294. ;
  295.     } else {
  296.       /* reached end of parameter, set remaining components to 1x1 sampling */
  297.       cinfo->comp_info[ci].h_samp_factor = 1;
  298.       cinfo->comp_info[ci].v_samp_factor = 1;
  299.     }
  300.   }
  301.   return TRUE;
  302. }