extract.cxx

Go to the documentation of this file.

/*
         Copyright 2002-2020 Pierre SMARS (smars@yuntech.edu.tw)
         This program is free software: you can redistribute it and/or modify
         it under the terms of the GNU General Public License as published by
         the Free Software Foundation, either version 2 of the License, or
         (at your option) any later version.
 
         This program is distributed in the hope that it will be useful,
         but WITHOUT ANY WARRANTY; without even the implied warranty of
         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
         GNU General Public License for more details.
 
         You should have received a copy of the GNU General Public License
         along with this program.  If not, see <http://www.gnu.org/licenses/>.
         */
#include <iostream>
#include <iomanip>
#include <vector>
#include <sstream>
#include <string>
#include <stdlib.h>
 
//*****************************************************
class s_pt
{
        public:
                s_pt(std::string ax, std::string ay, std::string az): x(ax), y(ay), z(az) {}
                std::string x,y,z;
};
//*****************************************************
int error(int code)
{
        std::cerr << "extract:" << std::endl;
        std::cerr << "      extract .pt files (x,y,z) from other .pt files or numbered pt files (i,x,y,z)" << std::endl;
        std::cerr << "syntax: extract (from to) < input_file > output_file" << std::endl;
        std::cerr << "        if 'from' and 'to' are omitted, all the points are extracted" << std::endl;
        std::cerr << "        'from' and 'to' can also take the values 'first' or 'last'" << std::endl;
        std::cerr << "author: P.Smars, 2002" << std::endl;
        return code;
}
//*****************************************************
int is_pt()
{
        char buf[256];
        std::cin.getline(buf,256);
        std::istringstream ff(buf);
        std::string x0,x;
        int i=0;
        ff >> x0;
        while (ff >> x)
        {
                i++;
        }
        if (i==0)
                return atoi(x0.c_str());
        else if (i==3)
                return 0;
        else
                return error(-2);
}
//*****************************************************
int main(int argc, char ** argv)
{
        if ((argc != 1)&&(argc != 3))
                return error(-1);
        bool code = false;
        std::vector<s_pt *> points;
        int nn = is_pt();
        int n = 0;
        if (argc == 1)
        {
                while (std::cin)
                {
                        int i;
                        std::string x,y,z;
                        if (nn==0)
                                std::cin >> i >> x >> y >> z;
                        else
                        {
                                std::cin >> x >> y >> z;
                                n++;
                        }
                        if (std::cin)
                        {
                                auto pt = new s_pt(x,y,z);
                                points.push_back(pt);
                        }
                        if ((nn!=0)&&(n==nn)) break;
                }
        }
        else
        {
                int imin,imax;
                std::ostringstream o;
                std::string a1(argv[1]);
                std::string a2(argv[2]);
                std::ostringstream aa;
                aa << nn;
                if (a1=="first")
                        a1 = "1";
                if (a2=="first")
                        a2 = "1";
                if (a1=="last")
                {
                        if (nn !=0)
                                a1 = aa.str();
                        else
                                a1 = "1";
                        code = true;
                }
                if (a2=="last")
                {
                        if (nn !=0)
                                a2 = aa.str();
                        else
                                a2 = "100000000";
                }
                o << a1 << " " << a2;
                std::istringstream i(o.str().c_str());
                i >> imin >> imax;
                if (imax<imin)
                        return error(-3);
                if ((nn!=0)&&((nn<imin)||(nn<imax)))
                        return error(-4);
                while (std::cin)
                {
                        static int i = 0;
                        std::string x,y,z;
                        if (nn==0)
                                std::cin >> i >> x >> y >> z;
                        else
                        {
                                i++;
                                std::cin >> x >> y >> z;
                        }
                        if ((std::cin)&&(i >= imin)&&(i <= imax))
                        {
                                auto pt = new s_pt(x,y,z);
                                points.push_back(pt);
                        }
                }
        }
        int n_pts = points.size();
        if (code)
        {
                std::cout << 1 << std::endl;
                s_pt * pt = *(points.end()-1);
                std::cout << pt->x << " " << pt->y << " " << pt->z << std::endl;
                std::cerr << std::endl << "1 point extracted." << std::endl;
        }
        else if (n_pts != 0)
        {
                std::cout << n_pts << std::endl;
                for (auto pt:points)
                        std::cout << pt->x << " " << pt->y << " " << pt->z << std::endl;
                for (auto pt:points)
                        delete pt;
                if (n_pts==1)
                        std::cerr << std::endl << "1 point extracted." << std::endl;
                else
                        std::cerr << std::endl << n_pts << " points extracted." << std::endl;
        }
        else
                std::cerr << std::endl << "no points extracted!" << std::endl;
        points.clear();
 
        return 0;
}