C++ MarSystemManager::getMarSystem方法代码示例

static MarSystem *load_network( const string system_filename )
{

  auto string_ends_with =
      []( const string & str, const string & ending ) -> bool
  {
    if (str.length() < ending.length())
      return false;
    return str.compare( str.length() - ending.length(), ending.length(), ending ) == 0;
  };

  MarSystem* system = 0;

  if (string_ends_with(system_filename, ".mrs"))
  {
#ifdef MARSYAS_HAS_SCRIPT
    system = system_from_script_file(system_filename);
#endif
  }
  else if (string_ends_with(system_filename, ".mpl"))
  {
    ifstream system_istream(system_filename);
    MarSystemManager mng;
    system = mng.getMarSystem(system_istream);
  }
  else if (string_ends_with(system_filename, ".json"))
  {
#ifdef MARSYAS_HAS_JSON
    system = system_from_json_file(system_filename);
#endif
  }

  return system;
}

int run( const string system_filename, const CommandLineOptions & opt )
{
    int ticks = 0;
    if (opt.has("count"))
    {
      ticks = opt.value<int>("count");
      if (ticks < 1)
      {
        cerr << "Invalid value for option 'count' (must be > 0)." << endl;
        return 1;
      }
    }

    ifstream system_istream(system_filename);
    MarSystemManager mng;
    MarSystem* system = mng.getMarSystem(system_istream);
    if (!system) {
        cerr << "Could not load filesystem file:" << system;
        return 1;
    }

    bool realtime = opt.value<bool>("realtime");
    mrs_real sr = opt.value<mrs_real>("samplerate");
    mrs_natural block = opt.value<mrs_natural>("block");

    if (sr > 0)
      system->setControl("mrs_real/israte", sr);
    if (block > 0)
      system->setControl("mrs_natural/inSamples", block);
    system->update();

    RealTime::Runner runner(system);
    runner.setRtPriorityEnabled(realtime);

    runner.start((unsigned int)ticks);
    runner.wait();

    return 0;
}

void
predict(mrs_string mode)
{

	MarSystemManager mng;

	cout << "Predicting using " << trainedclassifier_ << endl;

	ifstream pluginStream(trainedclassifier_.c_str());
    MRS_WARNINGS_OFF;
	MarSystem* net = mng.getMarSystem(pluginStream);
	MRS_WARNINGS_ON;

  if (!twekafname_Set()) return;

   vector<string> classNames;
   string s = net->getctrl("WekaSource/wsrc/mrs_string/classNames")->to<mrs_string>();
   char *str = (char *)s.c_str();
   char * pch;
   pch = strtok (str,",");
   classNames.push_back(pch);
   while (pch != NULL) {
 	pch = strtok (NULL, ",");
 	if (pch != NULL)
 	  classNames.push_back(pch);
   }





  ////////////////////////////////////////////////////////////
  //
  // Predict the classes of the test data
  //
  net->updControl("WekaSource/wsrc/mrs_string/filename", twekafname_);
  net->updControl("Classifier/cl/mrs_string/mode", "predict");
  ////////////////////////////////////////////////////////////
  //
  // Tick over the test WekaSource until all lines in the
  // test file have been read.
  //


  ofstream prout;
  prout.open(predictcollectionfname_.c_str());

  ofstream prtout;
  prtout.open(predicttimeline_.c_str());



  realvec data;
  int end=0;
  int start=0;

  mrs_string prev_name = "";
  mrs_string name;

  mrs_real srate;




  while (!net->getctrl("WekaSource/wsrc/mrs_bool/done")->to<mrs_bool>()) {
   	net->tick();
   	data = net->getctrl("mrs_realvec/processedData")->to<mrs_realvec>();
	srate = net->getctrl("WekaSource/wsrc/mrs_real/currentSrate")->to<mrs_real>();


	if (mode == "default")
	{
		cout << net->getctrl("WekaSource/wsrc/mrs_string/currentFilename")->to<mrs_string>() << "\t";
		cout << classNames[(int)data(0,0)] << endl;
		prout << net->getctrl("WekaSource/wsrc/mrs_string/currentFilename")->to<mrs_string>() << "\t";
		prout << classNames[(int)data(0,0)] << endl;
	}
	else if (mode == "timeline")
	{

	  name = classNames[(int)data(0,0)];

	  if (name != prev_name)
	  {
		if ((end * (1.0/srate)-start*(1.0 / srate) > minspan_))
		{
		  if (predicttimeline_ == EMPTYSTRING)
		  {
			cout << start*(1.0 / srate) << "\t" << end*(1.0 / srate) << "\t";
			cout << prev_name << endl;
		  }
		  else
		  {
			prtout << start*(1.0 / srate) << "\t" << end*(1.0 / srate) << "\t";
			prtout << prev_name << endl;
		  }
		  
		}
		start = end;
		
//.........这里部分代码省略.........

void
MarGrid::predict()
{
  MarSystemManager mng;

  // read trained som network from file som.mpl and normalization network norm.mpl
  ifstream iss;
  iss.open("som.mpl");
  som_ = mng.getMarSystem(iss);

  ifstream niss;
  niss.open("norm.mpl");
  norm_ = mng.getMarSystem(niss);

  resetPredict();
  cout << "Starting prediction" << endl;
  som_->updControl("mrs_string/mode", "predict");

  Collection l1;
  l1.read(predictFname.toStdString());
  cout << "Read collection" << endl;

  total_->updControl("mrs_natural/pos", 0);
  total_->updControl("mrs_natural/advance", 0);
  total_->updControl("mrs_string/filename", predictFname.toStdString());

  som_->updControl("mrs_natural/inSamples", 1);

  realvec predict_res(som_->getctrl("mrs_natural/onObservations")->to<mrs_natural>(),
                      som_->getctrl("mrs_natural/onSamples")->to<mrs_natural>());
  norm_->updControl("mrs_natural/inSamples", 1);


  realvec som_in;
  realvec som_res;
  realvec som_res1;

  realvec norm_som_res;

  mrs_natural inObs = total_->getctrl("mrs_natural/inObservations")->to<mrs_natural>();
  mrs_natural inSms = total_->getctrl("mrs_natural/inSamples")->to<mrs_natural>();

  mrs_natural onObs = total_->getctrl("mrs_natural/onObservations")->to<mrs_natural>();
  mrs_natural onSms = total_->getctrl("mrs_natural/onSamples")->to<mrs_natural>();

  som_in.create(inObs, inSms);
  som_res.create(onObs, onSms);

  som_res1.create(onObs+2, onSms);

  norm_som_res.create(onObs+2, onSms);

  for (unsigned int index = 0; index < l1.size(); index++)
  {
    total_->updControl("mrs_natural/label", index);
    total_->updControl("mrs_bool/memReset", true);
    total_->updControl("mrs_natural/cindex", index);


    total_->process(som_in, som_res);
    string current = total_->getctrl("mrs_string/currentlyPlaying")->to<mrs_string>();


    cout << "CURRENT = " << current << endl;

    for (int o=0; o < onObs; o++)
      som_res1(o, 0) = som_res(o, 0);

    norm_->process(som_res1, norm_som_res);
    som_->process(norm_som_res, predict_res);



    grid_x = predict_res(0);
    grid_y = predict_res(1);
    addFile(grid_x,grid_y, current);
    repaint();
    total_->updControl("mrs_natural/advance", 1);
  }



  cout << "end_prediction" << endl;


}

void classifyFile(vector<string> soundfiles, string pluginName)
{

  MRSDIAG("classifyFile.cpp - classifyFile");
  // read the plugin from the file
  ifstream pluginStream(pluginName.c_str());



  MarSystemManager mngr;
  MarSystem* msys = mngr.getMarSystem(pluginStream);



  if (msys == 0)
  {
    cout << "Manager does not support system " << endl;
    exit(1);
  }
  mrs_real srate;

  // run audio through the plugin for each file in the collection
  // assume there is a SoundFileSource in the plugin
  // and an audio sink
  string sfName;

  // output the plugin

  // cout << (*msys) << endl;



  vector<string>::iterator sfi;
  mrs_natural count = 0;





  for (sfi = soundfiles.begin(); sfi != soundfiles.end(); ++sfi)
  {
    count++;


    // udpate source filename
    sfName = *sfi;
    msys->updctrl("mrs_string/filename", sfName);



    mrs_natural size = msys->getctrl("SilenceRemove/srm/SoundFileSource/src/mrs_natural/size")->to<mrs_natural>();
    mrs_natural inSamples = msys->getctrl("SilenceRemove/srm/SoundFileSource/src/mrs_natural/inSamples")->to<mrs_natural>();

    mrs_natural memSize = (size / inSamples);
    memSize /= 2;

    // hardwire to approximately 15 seconds
    memSize = 600;


    msys->updctrl("Confidence/confidence/mrs_natural/memSize", memSize);

    if (verboseopt)
      cout << (*msys) << endl;



    srate = msys->getctrl("mrs_real/israte")->to<mrs_real>();


    mrs_natural samplesPlayed = 0;
    mrs_natural onSamples = msys->getctrl("mrs_natural/onSamples")->to<mrs_natural>();
    mrs_natural wc = 0;
    int i;




    if (onetick)
    {
      msys->tick();
    }
    else
    {
      // if (verboseopt)
      cout << sfName << " ";
      //

      // while (msys->getctrl("mrs_bool/hasData")->to<mrs_bool>())
      for (i=0; i < memSize-1; i++)
      {
        if (msys->getctrl("mrs_bool/hasData")->to<mrs_bool>() == false)
        {
          break;
        }
        msys->tick();
        wc++;
        samplesPlayed += onSamples;
      }

//.........这里部分代码省略.........