main_b.cpp

#include "header.h"
#include <iostream>
#include <string>
#include <vector>
#include <cstring>
#include <cstdio>
#include <algorithm>


using namespace std;
    
//Variables requeridas para que el sistema funcione
string url;
int amount_forward_processing;

vector<ForwardProcessing*> forward_processing;


    int main(int argc, char const *argv[])
    {
        if(argc<7){//No se cumple con la cantidad mínima de argumentos necesarios para el funcionamiento del Software
            cout << "This tools required minimum six parameters to work\n"
                 << "-URL of Picture\n"
                 << "-Forward Processing Methods\n"
                 << "-Type of Packetizer\n"
                 << "-Size of Packages\n"
                 << "-Channel of Simulation\n"
                 << "-Rebuilder Methods\n";
            return 0;
        }
        Images img(argv[1]);
        img.save("images/img_readed.bmp");
        /*
            list contains the elements of images
        */
        vector<DataType *> list = img.toList();
        string fm = argv[2];
        int i=0;
        if(!fm.compare(FORWARD_METHOD)){
            amount_forward_processing=stoi(argv[3]);
            int x=1;
            i=4;
            while(x<=amount_forward_processing){
                int nArguments = stoi(argv[i+1]);
                int args[nArguments];
                for(int c=1;c<=nArguments;c++){
                    args[c-1]=stoi(argv[i+1+c]);
                }
                string fpname = argv[i];
                ForwardProcessing* fp = ForwardProcessingCreator::create(fpname);
                if(fp==NULL){
                    cout << "The Forward Processing Method are not created\n";
                    return 0;
                }
                int result = fp->make(&list,img.getWidth(),img.getHeight(),args,nArguments);
                if(result==0){
                    forward_processing.push_back(fp);
                    Images mixed(list,img.getType(),img.getWidth(),img.getHeight());
                    replace( fpname.begin(), fpname.end(), '-', '_');
                    string saved = "images/img_mixed" + fpname+ ".bmp";
                    mixed.save(saved.c_str());
                }
                else{
                    if(result==1){
                        cout << "The numbers of Parameters are not valid\n";
                    }
                    if(result==2){
                        cout << "The Image need be squared (WxH)\n";
                    }
                    if(result==5){
                        cout << "the byteOffset or the packetOffset need be mayor than 0\n";
                    }
                }
                x=x+1;
                i=i+nArguments+2;
            }

        }
        Packetizer* p = NULL;
        vector<Package> pkgs;
        if(i<argc){
            string pkgname = argv[i];
            p = PacketizerCreator::create(pkgname);
            if(p!=NULL){
                pkgs = p->imgToPackageList(list,stoi(argv[i+1]));//Tamaño de cada Paquete
                cout << "N-PACKAGES: " << pkgs.size() << "\n";
            }
            else{
                cout << "Packetizer are not created\n";
                return 0;
            }
        }
        else{
            cout << "Packetizer are not defined\n";
            return 0;
        }
        i=i+2;
        RandomGenerator::generate(4,pkgs.size());//4000 Paquetes que se perderán

        
        
        vector<Package> rcved = Channel::lossFromFile("files/loss_random.loss", pkgs);
        cout << "N-PACKAGES-RECEIVED: " << rcved.size() << "\n";
        vector <DataType *> pkgs_rcved = p->packageListToImg(rcved);
        Images rbefore(pkgs_rcved,img.getType(),img.getWidth(),img.getHeight());
        rbefore.save("images/img_received_before_unmixed.bmp");

        for(int j=forward_processing.size()-1;j>=0;j--){
            forward_processing[j]->unmake(&pkgs_rcved);
        }

        Images received(pkgs_rcved,img.getType(),img.getWidth(),img.getHeight());
        received.save("images/img_received.bmp");


        if((i+1)<argc){
            string rbd = argv[i];
            if(rbd.compare(REBUILDER_METHOD)){
                cout << "Rebuilder Method has not defined\n";
                return 0;
            }
            rbd = argv[i+1];
            Rebuilder* rbdr = RebuilderCreator::create(rbd);
            if(rbdr!=NULL){
                rbdr->hidden(&received);
                received.save("images/img_received_rebuilded.bmp");
            }
            else{
                cout << "Rebuilder isn't created\n";
                return 0;
            }
        }
        else{
            cout << "Not Exists Rebuilder Method\n";
            return 0;
        }
        i=i+2;
        MetricPsnr* metric = new MetricPsnr();
        cout << "PSNR: " << metric->calculate(img,received) << "\n";

        return 0;
    }