source: libcfa/src/containers/string.cfa @ 7e1dbd7

//
// Cforall Version 1.0.0 Copyright (C) 2016 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// string -- variable-length, mutable run of text, with value semantics
//
// Author           : Michael L. Brooks
// Created On       : Fri Sep 03 11:00:00 2021
// Last Modified By : Peter A. Buhr
// Last Modified On : Mon Aug 28 19:00:45 2023
// Update Count     : 146
//

#include "string.hfa"
#include "string_res.hfa"
#include <stdlib.hfa>

#pragma GCC visibility push(default)

/*
Implementation Principle: typical operation translates to the equivalent
operation on `inner`.  Exceptions are implementing new RAII pattern for value
semantics and some const-hell handling.
*/

////////////////////////////////////////////////////////
// string RAII


void ?{}( string & this ) {
    (this.inner) { malloc() };
    ?{}( *this.inner );
}

// private (not in header)
static void ?{}( string & this, string_res & src, size_t start, size_t end ) {
    (this.inner) { malloc() };
    ?{}( *this.inner, src, SHARE_EDITS, start, end );
}

void ?{}( string & this, const string & other ) {
    (this.inner) { malloc() };
    ?{}( *this.inner, *other.inner, COPY_VALUE );
}

void ?{}( string & this, string & other ) {
    ?{}( this, (const string &) other );
}

void ?{}( string & this, const char * val ) {
    (this.inner) { malloc() };
    ?{}( *this.inner, val );
}

void ?{}( string & this, const char * buffer, size_t bsize) {
    (this.inner) { malloc() };
    ?{}( *this.inner, buffer, bsize );
}

void ^?{}( string & this ) {
    ^(*this.inner){};
    free( this.inner );
    this.inner = 0p;
}

////////////////////////////////////////////////////////
// Alternate construction: request shared edits

string_WithSharedEdits ?`shareEdits( string & this ) {
    string_WithSharedEdits ret = { &this };
    return ret;
}

void ?{}( string & this, string_WithSharedEdits src ) {
    ?{}( this, *src.s->inner, 0, src.s->inner->Handle.lnth);
}

////////////////////////////////////////////////////////
// Assignment

void ?=?( string & this, const char * val ) {
    (*this.inner) = val;
}

void ?=?(string & this, const string & other) {
    (*this.inner) = (*other.inner);
}

void ?=?( string & this, char val ) {
    (*this.inner) = val;
}

string & ?=?(string & this, string & other) { //// <---- straw man change
    (*this.inner) = (*other.inner);
    return this;
}


////////////////////////////////////////////////////////
// Input-Output

ofstream & ?|?( ofstream & out, const string & this ) {
    return out | (*this.inner); // print internal string_res
}

void ?|?( ofstream & out, const string & this ) {
    (ofstream &)(out | (*this.inner)); ends( out );
}

ifstream & ?|?(ifstream & in, string & this) {
    return in | (*this.inner); // read to internal string_res
}

void ?|?( ifstream & in, string & this ) {
    (ifstream &)(in | this); ends( in );
}

static void readstr( ifstream & is, _Istream_str f, char fmtstr[], char cstr[] ) {
        int check = f.rwd - 1;

        if ( ! f.flags.rwd ) cstr[check] = '\0';                        // insert sentinel
        int len = fmt( is, fmtstr, cstr );
        // fprintf( stderr, "KK %s %zd %d %c %s\n", fmtstr, len, check, cstr[check], cstr );

        if ( ! f.flags.rwd && cstr[check] != '\0' )                     // sentinel overwritten ?
                throw (cstring_length){ &cstring_length_vt };

        if ( f.flags.delimit ) {                                                        // getline ?
                if ( len == 0 ) cstr[0] = '\0';                                 // empty read => argument unchanged => set empty
                if ( ! eof( is ) ) fmt( is, "%*c" );                    // ignore delimiter
        } //if
} // readstr

ifstream & ?|?( ifstream & is, _Istream_str f ) {
        // skip, same as for char *
        if ( ! &f.s ) {
                // fprintf( stderr,  "skip %s %d\n", f.scanset, f.wd );
                if ( f.rwd == -1 ) fmt( is, f.scanset, "" ); // no input arguments
                else for ( f.rwd ) fmt( is, "%*c" );
                return is;
        } // if

        enum { gwd = 16 + 2, wd = gwd - 1 };                            // guarded and unguarded width
        char cstr[gwd];                                                                         // read in chunks
        bool cont = false;;

        if ( f.rwd == -1 ) f.rwd = wd;
        const char * scanset = f.scanset;;
        if ( f.flags.delimit ) scanset = f.delimit;                     // getline ?

        size_t len = 0;
        if ( scanset ) len = strlen( scanset );
        char fmtstr[len + 16];
        int start = 1;
        fmtstr[0] = '%';
        if ( f.flags.ignore ) { fmtstr[1] = '*'; start += 1; }
        if ( f.rwd != -1 ) { start += sprintf( &fmtstr[start], "%d", f.rwd ); }

        if ( ! scanset ) {
                // %s, %*s, %ws, %*ws
                fmtstr[start] = 's'; fmtstr[start + 1] = '\0';
                // printf( "cstr %s\n", fmtstr );
        } else {
                // incl %[xxx],  %*[xxx],  %w[xxx],  %*w[xxx]
                // excl %[^xxx], %*[^xxx], %w[^xxx], %*w[^xxx]
                fmtstr[start] = '['; start += 1;
                if ( f.flags.inex ) { fmtstr[start] = '^'; start += 1; }
                strcpy( &fmtstr[start], scanset );                              // copy includes '\0'
                len += start;
                fmtstr[len] = ']'; fmtstr[len + 1] = '\0';
                // printf( "incl/excl %s\n", fmtstr );
        } // if

        cstr[wd] = '\0';                                                                        // guard null terminate string
        try {
                readstr( is, f, fmtstr, cstr );
        } catch( cstring_length * ) {
                cont = true;
        } finally {
                f.s = cstr;                                                                             // ok to initialize string
        } // try
        for ( ; cont; )  {                                                                      // overflow read ?
                cont = false;
                try {
                        readstr( is, f, fmtstr, cstr );
                } catch( cstring_length * ) {
                        cont = true;                                                            // continue not allowed
                } finally {
                        f.s += cstr;                                                            // build string chunk at a time
                } // try
        } // for
        return is;
} // ?|?

void ?|?( ifstream & in, _Istream_str f ) {
    (ifstream &)(in | f); ends( in );
}

// void getline( ifstream & in, string & str, const char delimit = '\n' ) {
//      // .---------------,
//      // | | | | |...|0|0| check and guard
//      // `---------------'
//      enum { gwd = 128 + 2, wd = gwd - 1 };                           // guarded and unguarded width
//      char cstr[gwd];                                                                         // read in chunks
//      bool cont = false;;

//      cstr[wd] = '\0';                                                                        // guard null terminate string
//      try {
//              in | getline( wdi( wd, cstr ), delimit );               // must have room for string terminator
//              if ( eof( in  ) ) return;                                               // do not change argument
//      } catch( cstring_length * ) {
//              cont = true;
//      } finally {
//              str = cstr;                                                                             // ok to initialize string
//      } // try
//      for ( ; cont; )  {                                                                      // overflow read ?
//              cont = false;
//              try {
//                      in | getline( wdi( wd, cstr ), delimit );       // must have room for string terminator
//                      if ( eof( in  ) ) return;
//              } catch( cstring_length * ) {
//                      cont = true;                                                            // continue not allowed
//              } finally {
//                      str += cstr;                                                            // build string chunk at a time
//              } // try
//      } // for
// }

////////////////////////////////////////////////////////
// Slicing

string ?()( string & this, size_t start, size_t end ) {
    string ret = { *this.inner, start, end };
    return ret`shareEdits;
}

string ?()( string & this, size_t start ) {
    string ret = { *this.inner, start, size( this ) };
    return ret`shareEdits;
}

////////////////////////////////////////////////////////
// Comparison

bool ?==?(const string & s, const string & other) {
    return *s.inner == *other.inner;
}

bool ?!=?(const string & s, const string & other) {
    return *s.inner != *other.inner;
}

bool ?==?(const string & s, const char * other) {
    return *s.inner == other;
}

bool ?!=?(const string & s, const char * other) {
    return *s.inner != other;
}

////////////////////////////////////////////////////////
// Getter

size_t size(const string & s) {
    return size( * s.inner );
}

////////////////////////////////////////////////////////
// Concatenation

void ?+=?(string & s, char other) {
    (*s.inner) += other;
}

void ?+=?(string & s, const string & s2) {
    (*s.inner) += (*s2.inner);
}

void ?+=?(string & s, const char * other) {
    (*s.inner) += other;
}

string ?+?(const string & s, char other) {
    string ret = s;
    ret += other;
    return ret;
}

string ?+?(const string & s, const string & s2) {
    string ret = s;
    ret += s2;
    return ret;
}

string ?+?(const char * s1, const char * s2) {
    string ret = s1;
    ret += s2;
    return ret;
}

string ?+?(const string & s, const char * other) {
    string ret = s;
    ret += other;
    return ret;
}

////////////////////////////////////////////////////////
// Repetition

string ?*?(const string & s, size_t factor) {
    string ret = "";
    for (factor) ret += s;
    return ret;
}

string ?*?(char c, size_t size) {
    string ret = "";
    for ((size_t)size) ret += c;
    return ret;
}

string ?*?(const char *s, size_t factor) {
    string ss = s;
    return ss * factor;
}

////////////////////////////////////////////////////////
// Character access

char ?[?](const string & s, size_t index) {
    return (*s.inner)[index];
}

string ?[?](string & s, size_t index) {
    string ret = { *s.inner, index, index + 1 };
    return ret`shareEdits;
}

////////////////////////////////////////////////////////
// Search

bool contains(const string & s, char ch) {
    return contains( *s.inner, ch );
}

int find(const string & s, char search) {
    return find( *s.inner, search );
}

int find(const string & s, const string & search) {
    return find( *s.inner, *search.inner );
}

int find(const string & s, const char * search) {
    return find( *s.inner, search);
}

int find(const string & s, const char * search, size_t searchsize) {
    return find( *s.inner, search, searchsize);
}

int findFrom(const string & s, size_t fromPos, char search) {
    return findFrom( *s.inner, fromPos, search );
}

int findFrom(const string & s, size_t fromPos, const string & search) {
    return findFrom( *s.inner, fromPos, *search.inner );
}

int findFrom(const string & s, size_t fromPos, const char * search) {
    return findFrom( *s.inner, fromPos, search );
}

int findFrom(const string & s, size_t fromPos, const char * search, size_t searchsize) {
    return findFrom( *s.inner, fromPos, search, searchsize );
}

bool includes(const string & s, const string & search) {
    return includes( *s.inner, *search.inner );
}

bool includes(const string & s, const char * search) {
    return includes( *s.inner, search );
}

bool includes(const string & s, const char * search, size_t searchsize) {
    return includes( *s.inner, search, searchsize );
}

bool startsWith(const string & s, const string & prefix) {
    return startsWith( *s.inner, *prefix.inner );
}

bool startsWith(const string & s, const char * prefix) {
    return startsWith( *s.inner, prefix );
}

bool startsWith(const string & s, const char * prefix, size_t prefixsize) {
    return startsWith( *s.inner, prefix, prefixsize );
}

bool endsWith(const string & s, const string & suffix) {
    return endsWith( *s.inner, *suffix.inner );
}

bool endsWith(const string & s, const char * suffix) {
    return endsWith( *s.inner, suffix );
}

bool endsWith(const string & s, const char * suffix, size_t suffixsize) {
    return endsWith( *s.inner, suffix, suffixsize );
}


///////////////////////////////////////////////////////////////////////////
// charclass, include, exclude

void ?{}( charclass & this, const string & chars) {
    (this.inner) { malloc() };
    ?{}( *this.inner, *(const string_res *)chars.inner );
}

void ?{}( charclass & this, const char * chars ) {
    (this.inner) { malloc() };
    ?{}( *this.inner, chars );
}

void ?{}( charclass & this, const char * chars, size_t charssize ) {
    (this.inner) { malloc() };
    ?{}( *this.inner, chars, charssize );
}

void ^?{}( charclass & this ) {
    ^(*this.inner){};
    free( this.inner );
    this.inner = 0p;
}


int exclude(const string & s, const charclass & mask) {
    return exclude( *s.inner, *mask.inner );
}
/*
StrSlice exclude(string & s, const charclass & mask) {
}
*/

int include(const string & s, const charclass & mask) {
    return include( *s.inner, *mask.inner );
}

/*
StrSlice include(string & s, const charclass & mask) {
}
*/