git.cworth.org Git - apitrace/blob - cli/cli_dump_images.cpp

   1 /*********************************************************************
   2  *
   3  * Copyright 2011 Jose Fonseca
   4  * Copyright 2012 Intel Corporation
   5  * All Rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person
   8  * obtaining a copy of this software and associated documentation
   9  * files (the "Software"), to deal in the Software without
  10  * restriction, including without limitation the rights to use, copy,
  11  * modify, merge, publish, distribute, sublicense, and/or sell copies
  12  * of the Software, and to permit persons to whom the Software is
  13  * furnished to do so, subject to the following conditions:
  14  *
  15  * The above copyright notice and this permission notice shall be
  16  * included in all copies or substantial portions of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  19  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  21  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  22  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  23  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  24  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  25  * SOFTWARE.
  26  *
  27  *********************************************************************/
  28
  29 #include <string.h>
  30 #include <limits.h> // for CHAR_MAX
  31 #include <getopt.h>
  32 #include <iostream>
  33
  34 #include "os_string.hpp"
  35
  36 #include "cli.hpp"
  37 #include "cli_retrace.hpp"
  38
  39 static const char *synopsis = "Dump frame images obtained from a trace.";
  40
  41 static void
  42 usage(void)
  43 {
  44     std::cout << "usage apitrace dump-images [OPTIONS] TRACE_FILE\n"
  45               << synopsis << "\n"
  46         "\n"
  47         "    -h, --help           show this help message and exit\n"
  48         "        --calls=CALLSET  dump images only for specified calls\n"
  49         "                         (default value is \"*/frame\" which\n"
  50         "                          which dumps an image for each frame)\n"
  51         "    -o, --output=PREFIX  prefix to use in naming output files\n"
  52         "                         (default is trace filename without extension)\n"
  53         "\n";
  54 }
  55
  56 enum {
  57     CALLS_OPT = CHAR_MAX + 1,
  58 };
  59
  60 const static char *
  61 shortOptions = "ho:";
  62
  63 const static struct option
  64 longOptions[] = {
  65     {"help", no_argument, 0, 'h'},
  66     {"calls", required_argument, 0, CALLS_OPT},
  67     {"output", required_argument, 0, 'o'},
  68     {0, 0, 0, 0}
  69 };
  70
  71 static int
  72 command(int argc, char *argv[])
  73 {
  74     os::String prefix;
  75     const char *calls = NULL;
  76     const char *traceName = NULL;
  77     const char *output = NULL;
  78
  79     int opt;
  80     while ((opt = getopt_long(argc, argv, shortOptions, longOptions, NULL)) != -1) {
  81         switch (opt) {
  82         case 'h':
  83             usage();
  84             return 0;
  85         case CALLS_OPT:
  86             calls = optarg;
  87             break;
  88         case 'o':
  89             output = optarg;
  90             break;
  91         default:
  92             std::cerr << "error: unexpected option `" << opt << "`\n";
  93             usage();
  94             return 1;
  95         }
  96     }
  97
  98     if (optind >= argc) {
  99         std::cerr << "error: apitrace dump-images requires a trace file as an argument.\n";
 100         usage();
 101         return 1;
 102     }
 103
 104     if (optind < argc - 1) {
 105         std::cerr << "error: apitrace dump-images can accept only a single trace file argument.\n";
 106         usage();
 107         return 1;
 108     }
 109
 110     traceName = argv[optind];
 111
 112     if (output == NULL) {
 113         prefix = traceName;
 114         prefix.trimDirectory();
 115         prefix.trimExtension();
 116         prefix.append('.');
 117         output = prefix.str();
 118     }
 119
 120     std::vector<const char *> opts;
 121
 122     opts.push_back("-s");
 123     opts.push_back(output);
 124     opts.push_back("-S");
 125     if (calls)
 126         opts.push_back(calls);
 127     else
 128         opts.push_back("*/frame");
 129
 130     return executeRetrace(opts, traceName);
 131 }
 132
 133 const Command dump_images_command = {
 134     "dump-images",
 135     synopsis,
 136     usage,
 137     command
 138 };