source/datatool.cpp

   1 #include <stdexcept>
   2 #include <msp/fs/utils.h>
   3 #include "builder.h"
   4 #include "component.h"
   5 #include "datacollection.h"
   6 #include "datapack.h"
   7 #include "datatool.h"
   8 #include "datatransform.h"
   9 #include "externaltask.h"
  10 #include "sourcepackage.h"
  11
  12 using namespace std;
  13 using namespace Msp;
  14
  15 DataTool::DataTool(Builder &b):
  16         Tool(b, "DATA")
  17 {
  18         set_command("mspdatatool");
  19         input_suffixes.push_back(".mdt");
  20 }
  21
  22 Target *DataTool::create_source(const Component &comp, const FS::Path &path) const
  23 {
  24         return new DataTransform(builder, comp, path);
  25 }
  26
  27 Target *DataTool::create_target(const list<Target *> &sources, const string &arg)
  28 {
  29         if(arg=="collection")
  30         {
  31                 if(sources.size()!=1)
  32                         throw invalid_argument("DataTool::create_target");
  33                 DataTransform &source = dynamic_cast<DataTransform &>(*sources.front());
  34                 DataCollection *coll = new DataCollection(builder, *source.get_component(), source);
  35                 coll->set_tool(*this);
  36                 return coll;
  37         }
  38         else if(arg=="pack")
  39         {
  40                 list<FileTarget *> files;
  41                 for(list<Target *>::const_iterator i=sources.begin(); i!=sources.end(); ++i)
  42                         files.push_back(&dynamic_cast<FileTarget &>(**i));
  43                 DataPack *pack = new DataPack(builder, *files.front()->get_component(), files);
  44                 pack->set_tool(*this);
  45                 return pack;
  46         }
  47         else
  48                 throw invalid_argument("DataTool::create_target");
  49 }
  50
  51 string DataTool::create_build_signature(const BuildInfo &binfo) const
  52 {
  53         string result;
  54         if(binfo.debug)
  55                 result += 'g';
  56         if(binfo.optimize>0)
  57         {
  58                 result += 'b';
  59                 if(binfo.optimize>1)
  60                         result += 'z';
  61         }
  62         return result;
  63 }
  64
  65 Task *DataTool::run(const Target &tgt) const
  66 {
  67         const Component &comp = *tgt.get_component();
  68         FS::Path work_dir = comp.get_package().get_source_directory();
  69
  70         vector<string> argv;
  71         argv.push_back(executable->get_path().str());
  72
  73         argv.push_back("-o");
  74         argv.push_back(FS::relative(dynamic_cast<const FileTarget &>(tgt).get_path(), work_dir).str());
  75
  76         BuildInfo binfo;
  77         tgt.collect_build_info(binfo);
  78         if(binfo.debug)
  79                 argv.push_back("-g");
  80         if(binfo.optimize>0)
  81         {
  82                 argv.push_back("-b");
  83                 if(binfo.optimize>1)
  84                         argv.push_back("-z");
  85         }
  86
  87         if(const DataCollection *coll = dynamic_cast<const DataCollection *>(&tgt))
  88         {
  89                 argv.push_back("-c");
  90                 argv.push_back(FS::relative(coll->get_source().get_path(), work_dir).str());
  91         }
  92         else if(const DataPack *pack = dynamic_cast<const DataPack *>(&tgt))
  93         {
  94                 argv.push_back("-p");
  95                 const DataPack::FileList &files = pack->get_files();
  96                 for(DataPack::FileList::const_iterator i=files.begin(); i!=files.end(); ++i)
  97                         argv.push_back(FS::relative((*i)->get_path(), work_dir).str());
  98         }
  99
 100         return new ExternalTask(argv, work_dir);
 101 }