utilities/UploadApp.cxx (r4864/r3229)

Go to the documentation of this file.

//============================================================================
// Name        : UploadApp.cpp
// Author      : Stefan Mueller-Klieser / S.Linev
// Version     :
// Copyright   : Kirchhoff-Institut fuer Physik / GSI
// Description : Upload firmware/software/configuration files onto roc
//============================================================================

#include <string.h>
#include <time.h>
#include <sys/time.h>
#include <iostream>
#include <unistd.h>

#include "roc/Board.h"
#include "roc/UdpBoard.h"

int main(int argc, char* argv[])
{
   std::cout << "Data uploader for roc::Board" << std::endl;
   std::cout << "Can be used for firmware, software or configuration upload" << std::endl;

   std::cout << "Usage: rocupload boardIp -fw0|-fw1|-sw|-cfg fileName [rocFileName]" << std::endl;

   if (argc<4) {
      std::cout << "Not all parameters are specified!!!" << std::endl;
       return -1;
   }

   const char* boardIp = argv[1];
   int typ = -1;
   if (!strcmp(argv[2], "-fw0")) typ = 1; else
   if (!strcmp(argv[2], "-fw") || !strcmp(argv[2], "-fw1")) typ = 2; else
   if (!strcmp(argv[2], "-sw")) typ = 3; else
   if (!strcmp(argv[2], "-cfg")) typ = 4; else {
      std::cout << "Invalid type of upload " << argv[2] << std::endl;
      return -1;
   }
   const char* fileName = argv[3];
   const char* tgtFileName = 0;
   if ((argc>4) && (typ==4)) tgtFileName = argv[4];

   roc::Board* brd = roc::Board::Connect(boardIp);

   if(brd == 0) {
      std::cout << "-----   Setup could not connect to board. Exit!   -----" << std::endl;
      return -1;
   }

   roc::UdpBoard* udp = dynamic_cast<roc::UdpBoard*> (brd);
   if (udp==0) {
      std::cout << "Cannot use for non-udp connected board. Exit! ------" << std::endl;
      roc::Board::Close(brd);
      return -1;
   }

   udp->setConsoleOutput(true, true);

   int upload_tm = false;

   switch(typ) {
      case 1:
         std::cout << "-----   Uploading firmware on position 0: " << fileName << " -------- " << std::endl;
         upload_tm = udp->uploadBitfile(fileName, 0);
         break;
      case 2:
         std::cout << "-----   Uploading firmware on position 1: " << fileName << " -------- " << std::endl;
         upload_tm = udp->uploadBitfile(fileName, 1);
         break;
      case 3:
         std::cout << "-----   Uploading software: " << fileName << " -------- " << std::endl;
         upload_tm = udp->uploadSDfile(fileName, "image.bin");
         break;
      case 4:
         if (tgtFileName==0) tgtFileName = "roc1.cfg";
         std::cout << "-----   Uploading configuration: " << fileName << " to " << tgtFileName << std::endl;
         upload_tm = udp->uploadSDfile(fileName, tgtFileName);
         break;
      default:
         std::cout << "Error type !" << std::endl;
         break;
    }

   if (upload_tm<=0) {
      std::cout << "Error detected !" << std::endl;
      return 1;
   }

   bool upload_res = false;

   while (upload_tm>0) {
      sleep(1);
      upload_tm--;
      std::cout << ".";
      std::cout.flush();
      int res = udp->checkUpload();
      if (res==2) { upload_res = true; break; }
      if (res!=1) break;
   }
   std::cout << std::endl;

   udp->setConsoleOutput();

   if ((typ==3) && upload_res) {
      std::cout << "-----   Restart ROC -------- " << std::endl;
      udp->restartRoc();
   }

   roc::Board::Close(brd);

   return 0;
}