NdbEndfIO.cxx

Go to the documentation of this file.

#include <stdlib.h>

#include "NdbEndfIO.h"

#define NUMBER_SIZE     11

char    NdbEndfIO::_str[100];           // define static _str var

ClassImp(NdbEndfIO)

/* ============ NdbEndfIO ============== */
NdbEndfIO::NdbEndfIO( char *filename, Int_t mode )
{
        f = fopen(filename,mode==TENDF_READ?"r":"w");
        matStart = 0;
        mfStart = 0;
        mtStart = 0;
        iMAT = 0;
        iMF = 0;
        iMT = 0;
        lineNum = 0;
        lineTxt[0] = 0;
} // NdbEndfIO


/* -------- FindMAT --------- */
// Will search for the specific material
Bool_t
NdbEndfIO::FindMAT( Int_t mat, Bool_t rewind )
{
        if (rewind)
                fseek(f, 0L, SEEK_SET);

        while (ReadLine())
                if (iMAT == mat)
                        return TRUE;
        
        return FALSE;
} // FindMAT

/* -------- FindMATMF --------- */
// Will search for the specific material - mf
Bool_t
NdbEndfIO::FindMATMF( Int_t mat, Int_t mf, Bool_t rewind )
{
        FindMAT(mat,rewind);

        do {
                if (iMF == mf)
                        return TRUE;
        } while (ReadLine());
        
        return FALSE;
} // FindMATMF

/* -------- FindMATMFMT --------- */
// Will search for the specific material - mf - mt
Bool_t
NdbEndfIO::FindMATMFMT( Int_t mat, Int_t mf, Int_t mt, Bool_t rewind )
{
        FindMATMF(mat,mf,rewind);

        do {
                if (iMT == mt)
                        return TRUE;
        } while (ReadLine());
        
        return FALSE;
} // FindMATMFMT

/* ------------ FindMFMT ------------- */
Bool_t
NdbEndfIO::FindMFMT( Int_t mf, Int_t mt )
{
        // If current MFMT is higher than what we search go to the beggining
        if (mf*1000+mt >= iMF*1000+iMT) {
                RewindMAT();
                if (!ReadLine())
                        return FALSE;
        }

        while (mf*1000+mt > iMF*1000+iMT)
                if (!ReadLine())
                        return FALSE;
        return TRUE;
} // FindMFMT

/* -------- Substr ---------- */
char *
NdbEndfIO::Substr(Int_t start, Int_t length)
{
        if (start + length > lineLen) {
//              error(ERR_INVALID_RECORD);
                return NULL;
        }
        memcpy(_str, lineTxt+start, length);
        _str[length] = 0;

        return _str;
} // Substr

/* -------- NextNumber ---------- */
/* Advance lastNumber Point to the correct number
 * so to read 6 numbers of width 11 in each line
 */
Bool_t
NdbEndfIO::NextNumber(Int_t pos)
{
        if (pos<0 || pos>5*NUMBER_SIZE) {
                lastNumPos += NUMBER_SIZE;
                if (lineTxt[0]=='\0' || lastNumPos>5*NUMBER_SIZE) {
                        Int_t   mf = iMF;       // Remember current MF, MT
                        Int_t   mt = iMT;
                        if (!ReadLine() || (mf*1000+mt != iMF*1000+iMT))
                                return TRUE;
                        lastNumPos = 0;
                }
        } else {
                lastNumPos = pos * NUMBER_SIZE;
        }
        return FALSE;
} // NextNumber

/* -------- SubReadInt ---------- */
Int_t
NdbEndfIO::SubReadInt(Int_t start, Int_t length)
{
        NdbEndfIO::Substr(start,length);
        return atoi(_str);
} // SubReadInt

/* ---------- ReadInt ------------- */
/* Read one by one the int numbers with length 11 from the current line.
 * If the this is the last one then it prompts for an new line
 * only if it stays inside the same MF:MT type
 */
Int_t
NdbEndfIO::ReadInt(Bool_t *error, Int_t pos)
{
        if (NextNumber(pos)) {
                *error = TRUE;
                return 0;
        }
        *error = FALSE;
        return SubReadInt(lastNumPos,NUMBER_SIZE);
} // ReadInt

/* -------- SubReadReal ---------- */
Float_t
NdbEndfIO::SubReadReal(Int_t start, Int_t length)
{
        char    numstr[20];
        char    *dst, *src;
        
        NdbEndfIO::Substr(start,length);

        // Real numbers are in the following formats
        //      +/- N.NNNNNN+/-N
        //      +/- N.NNNNN+/-bN        (b=Blank)
        //      +/- N.NNNNN+/-NN

        dst = numstr;
        src = _str;

        /* skip blanks */
        while (*src==' ') src++;

        /* copy sign */
        if (*src=='+' || *src=='-') *dst++ = *src++;

        /* copy realpart */
        while (1) {
                if (memchr(".0123456789",*src,11))
                        *dst++ = *src++;
                else
                        break;
        }

        /* append the exponent */
        *dst++ = 'E';

        /* copy sign */
        if (*src=='+' || *src=='-') *dst++ = *src++;

        /* skip blanks */
        while (*src==' ') src++;

        /* copy exponent */
        while (1) {
                if (memchr("0123456789",*src,10))
                        *dst++ = *src++;
                else
                        break;
        }

        /* close the destination */
        *dst = 0;

        /* if something is still in source then we have an error number */
        if (*src) {
//              error(ERR_INVALID_REAL_NUMBER);
                return 0.0;
        }

        return atof(numstr);
} // SubReadReal

/* ---------- ReadReal ------------- */
/* Read one by one the real numbers with length 11 from the current line.
 * If the this is the last one then it prompts for an new line
 * only if it stays inside the same MF:MT type
 */
Float_t
NdbEndfIO::ReadReal(Bool_t *error, Int_t pos)
{
        if (NextNumber(pos)) {
                *error = TRUE;
                return 0.0;
        }
        *error = FALSE;
        return SubReadReal(lastNumPos,NUMBER_SIZE);
} // ReadReal

/* -------- ReadLine -------- */
Bool_t
NdbEndfIO::ReadLine()
{
        Long_t  current_pos;

        // --- Force the next number reading to be from the beggining of line
        lastNumPos = -NUMBER_SIZE;

        // --- Have we reached the end of file
        if (feof(f)) {
                lineTxt[0] = 0;
                return FALSE;
        }

        // --- Remember the position of start of line ---
        current_pos = ftell(f);

        // --- Read one line ---
        fgets(lineTxt,sizeof(lineTxt),f);

        lineLen = strlen(lineTxt);

        // Chop the trailing newline char
        if (lineTxt[lineLen-1] == '\n')
                lineTxt[--lineLen] = 0;

        // --- Remember previous values ---
        Int_t   oldMAT  = iMAT;
        Int_t   oldMF   = iMF;
        Int_t   oldMT   = iMT;

        // --- Read the material, mf, mt and line number ---
        iMAT    = SubReadInt(66, 4);
        iMF     = SubReadInt(70, 2);
        iMT     = SubReadInt(72, 3);
        lineNum = SubReadInt(75, 5);

        // --- Update pointers if necessary ---
        if (iMAT && iMAT != oldMAT)
                matStart = current_pos;

        if (iMF && iMF != oldMF)
                mfStart = current_pos;

        if (iMT && iMT != oldMT)
                mtStart = current_pos;

        return TRUE;
} // ReadLine

---