Home Research Papers Posters Manuals Talks Download People Help Internal

util/pup_cmialloc.C

Go to the documentation of this file.

#include "pup_cmialloc.h"

void PUP_cmiAllocSizer::bytes(void *,size_t n,size_t itemSize,PUP::dataType) {
    nBytes += n * itemSize;
}

void PUP_toCmiAllocMem::bytes(void *p, size_t n, size_t itemSize, 
                              PUP::dataType t) {

    n *= itemSize;
    memcpy((void *)buf, p, n);    
    buf += n;
}

void PUP_fromCmiAllocMem::bytes(void *p, size_t n, size_t itemSize, 
                                PUP::dataType t)
{
    n*=itemSize;
    memcpy(p,(const void *)buf,n);
    
    buf+= n;
}

void PUP_cmiAllocSizer::pupCmiAllocBuf(void **msg) {
    CmiChunkHeader chnk_hdr = *(BLKSTART(*msg));
    pupCmiAllocBuf(msg, chnk_hdr.size);
}

void PUP_cmiAllocSizer::pupCmiAllocBuf(void **msg, size_t msg_size) {

    //The cmialloced buf can only start at an aligned memory location
    //So nbytes has to be aligned
    nBytes = ALIGN8(nBytes);

    nBytes += sizeof(CmiChunkHeader);
    //Here the user buffer pointer will start, hence everything has to
    //be aligned till here
    nBytes += msg_size;  //The actual size of the user message
}


void PUP_toCmiAllocMem::pupCmiAllocBuf(void **msg) {
    pupCmiAllocBuf(msg, SIZEFIELD(msg));
}

void PUP_toCmiAllocMem::pupCmiAllocBuf(void **msg, size_t msg_size) {

    CmiChunkHeader chnk_hdr;

    buf = origBuf + ALIGN8_LONG(size());

    chnk_hdr.size = msg_size;
    chnk_hdr.setRef(origBuf - (buf + sizeof(CmiChunkHeader)));
    
    //Copy the Chunk header
    memcpy(buf, &chnk_hdr, sizeof(CmiChunkHeader));
    buf += sizeof(CmiChunkHeader);

    //Now buf is a memory aligned pointer
    //Copy the message
    //While unpacking, this aligned pointer will be returned
    memcpy(buf, *msg, msg_size);
    buf += msg_size;
}

void PUP_fromCmiAllocMem::pupCmiAllocBuf(void **msg) {
    //First align buf
    buf = (PUP::myByte *)(intptr_t)ALIGN8_LONG((intptr_t)buf);

    //Now get the chunk header
    CmiChunkHeader chnk_hdr;    
    //Get the Chunk header
    memcpy(&chnk_hdr, buf, sizeof(CmiChunkHeader));
    buf += sizeof(CmiChunkHeader);

    //Now we are at the begining of the user buffer
    *msg = buf;

    //Move the local buf forward by size bytes
    buf += chnk_hdr.size;
    
    //update the reference count of the original buf
    REFFIELDINC(origBuf);
}



/***** END CmiAlloc'ed buffer management functions ***********/

---