DataBlock.h

Go to the documentation of this file.

#ifndef CLASS_DATABLOCK
#define CLASS_DATABLOCK

#include <iostream>
#include <vector>
#include "DataType.h"
#include "Pixel.h"

 using namespace std;

class DataBlock: public DataType{

    vector<Pixel> block; 
    int width; 
    int height; 
    int amount_channels; 

public:

    DataBlock(vector<Pixel> nBlock, int w, int h, int nAmount){
         block = nBlock;
         width = w;
         height = h;
         amount_channels = nAmount;
         this->setValid(true);
    }

    DataBlock(int nAmount, vector<int> chns, int w, int h){
        width = w;
        height = h;
        for (int i = 0; i < width*height; i=i+1)
        {
            Pixel aux(chns,nAmount);
            block.push_back(aux);
        }
        this->setValid(true);
    }

    DataBlock(){
        width=1;
        height=1;
        amount_channels=0;  
        this->setValid(false);
    }

    ~DataBlock(){}

    double getSize(){
    /*
    *  @brief Método que devuelve el tamaño del Elemento
    *  @return result
    */
        double result = 0;
        for (int i = 0; i < (width*height); i=i+1)
        {
            result+= block[i].getSize();
        }
        return result;
    }

    int* getContent(){
        int* array = (int*)malloc(sizeof(int)*3);
        if(this->isValid()){
            array[0]=0;
        }
        else{
            array[0]=1;
        }
        array[1]=height;
        array[2]=width;
        /*for (int i = 0; i < (width*height); i++)
        {
            for (int j = 0; j < amount_channels; j++)
            {
                array[3+(i*amount_channels)+j]=block[i].getChannels()[j];
            }
        }*/
        return array;
    }

    int getIntensity(){
         int local_max = 0;
         for (int i = 0; i < (width*height); ++i)
         {
            local_max = local_max + block[i].getIntensity();         
         }
         
         local_max = (int)(local_max/(width*height));
         return local_max;
    }

    void setAmountChannels(int aChannels){
        amount_channels = aChannels;
        for(int i=0;i<(height*width);i++){
            block[i].setAmountChannels(amount_channels);
        }
    }

    void setChannels(vector<int> nChannels){
        for(int i=0;i<(height*width);i++){
            block[i].setChannels(nChannels);
        }
    }

    void* getExtras(){
        return (void*)&block;
    }

};

#endif