Test-fstreamPointer.C 2.68 KB
Newer Older
shunbo's avatar
shunbo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     |
    \\  /    A nd           | www.openfoam.com
     \\/     M anipulation  |
-------------------------------------------------------------------------------
    Copyright (C) 2020-2021 OpenCFD Ltd.
-------------------------------------------------------------------------------
License
    This file is part of OpenFOAM, distributed under GPL-3.0-or-later.

Application
    Test-fstreamPointer

Description
    Low-level fstream tests

\*---------------------------------------------------------------------------*/

#include "argList.H"
#include "refPtr.H"
#include "fstreamPointer.H"

using namespace Foam;

// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
//  Main program:

int main(int argc, char *argv[])
{
    argList::noBanner();
    argList::noParallel();

    argList::addOption
    (
        "output",
        "file",
        "Output file name for cat"
    );

    argList::addArgument("file1");
    argList::addArgument("...");
    argList::addArgument("fileN");

    argList::noMandatoryArgs();

    #include "setRootCase.H"

    if (args.size() <= 1)
    {
        InfoErr<< "\nNo input files specified .. stopping\n" << endl;
        return 0;
    }

    refPtr<std::ostream> osRef;

    fileName outputName;
    if (args.readIfPresent("output", outputName))
    {
        InfoErr<< "output: " << outputName;

        IOstreamOption::compressionType comp(IOstreamOption::UNCOMPRESSED);
        if (outputName.hasExt("gz"))
        {
            comp = IOstreamOption::COMPRESSED;
            outputName.removeExt();

            InfoErr<< " [compress]";
        }
        InfoErr<< nl;

        osRef.reset(ofstreamPointer(outputName, comp).release());
    }
    else
    {
        osRef.ref(std::cout);
        InfoErr<< "output: stdout" << nl;
    }
    auto& os = osRef.ref();

    for (label argi = 1; argi < args.size(); ++argi)
    {
        const auto inputName = args.get<fileName>(argi);

        InfoErr<< "input: " << inputName;

        ifstreamPointer isPtr(inputName);

        if (!isPtr.get() || !isPtr->good())
        {
            InfoErr<< " (not good)" << nl;
            continue;
        }
        InfoErr<< nl;

        auto& is = *isPtr;

        // Loop getting single characters
        // - not efficient, but that is not the test here anyhow

        char c;
        while (is.get(c))
        {
            os << c;
        }
    }

    InfoErr<< "\nEnd\n" << endl;
    return 0;
}


// ************************************************************************* //