C++博客-Cpper-随笔分类-Python脚本

KDJ指标代码

ccsdu2009 — Sat, 01 Apr 2017 06:57:00 GMT

class KDJ:
    history = []
    code = ''
    min = 99999.99
    max = 0.0
    k = 0.0
    d = 0.0
    rsv = 0.0

    def __init__(self,code):
        self.code = code

    def checkMinAndMax(self,n):
        min = 99999.9
        max = 0.0
        for item in self.history:
            if item[3] < min:
                min = item[3]
            if item[4] > max:
                max = item[4]
        return [min,max]

    def update(self,date,start,close,low,high,n):
        self.history.append([date,start,close,low,high])
        if len(self.history) < n:
            self.min = min(self.min,low)
            self.max = max(self.max,high)

            return [date,self.k,self.d,0,start,close,low,high]
        else:
            if len(self.history) > n:
                del self.history[0]

            p = self.checkMinAndMax(n)

            rsv = (close - p[0])/(p[1]-p[0])*100.0

            oldk = self.k
            oldd = self.d

            self.k = (rsv + 2*oldk)/3.0
            self.d = (self.k + 2*oldd)/3.0

            self.k = min(100,max(self.k,0))
            self.d = min(100,max(self.d,0))
            j = 3*self.k - 2*self.d
            j = min(100,max(j,0))

            return [date,self.k,self.d,j,start,close,low,high]

ccsdu2009 2017-04-01 14:57 发表评论

C语言调用python脚本3

ccsdu2009 — Sat, 07 Dec 2013 07:48:00 GMT

脚本如下:

from bs4 import BeautifulSoup

def list_get(file):
    soup = BeautifulSoup(open(file))
    alist = soup.find_all('a',class_ = 'link')
    list = []
    for i in alist:
        list.append(i.get('href'));
    #for i in list:
    #    print(i)
    return list

if __name__=="__main__":
    list_get('List.htm')

list_get函数返回的是list字符串对象
其c语言调用的代码如下:

#include <stdio.h>
#include <stdlib.h>
#include <Python.h>

char* GDALPythonObjectToCStr(PyObject* pyObject);

int main(int argc, char *argv[])
{
    Py_Initialize();
    if(!Py_IsInitialized())
    {
        return -1;
    }

    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('./script')");
    PyObject* pModule;
    PyObject* pDict;
    PyObject* pFunc;

    pModule = PyImport_ImportModule("list");
    if(!pModule)
    {
        printf("can't find list.py");
        system("PAUSE");
        getchar();
        return -1;
    }

    pDict = PyModule_GetDict(pModule);
    if(!pDict)
    {
        return -1;
    }

    pFunc = PyDict_GetItemString(pDict,"list_get");
    if(!pFunc || !PyCallable_Check(pFunc))
    {
        printf("can't find function [list_get]");
        getchar();
        return -1;
    }

    PyObject* args = PyTuple_New(1);
    PyTuple_SetItem(args,0,Py_BuildValue("s","List.htm"));
    PyObject* value = PyObject_CallObject(pFunc,args);
    int ret = PySequence_Check(value);
    printf("check:%d\n",ret);
    int length = PySequence_Size(value);
    printf("length:%d\n",length);

    int i = 0;
    for(;i<length;i++)
    {
        PyObject* obj = PySequence_GetItem(value,i);
        //char* str = PyBytes_AS_STRING(obj);
        char* str = GDALPythonObjectToCStr(obj);
        printf("link:%s\n",str);
        free(str);
    }

    Py_DECREF(args);
    Py_DECREF(pModule);
    Py_Finalize();
    system("PAUSE");
    return 0;
}

/* Return a NULL terminated c String from a PyObject */
/* Result must be freed with GDALPythonFreeCStr */
char* GDALPythonObjectToCStr(PyObject* pyObject)
{
#if PY_VERSION_HEX >= 0x03000000
    if(PyUnicode_Check(pyObject))
    {
        char *pszStr;
        char *pszNewStr;
        Py_ssize_t nLen;
        PyObject* pyUTF8Str = PyUnicode_AsUTF8String(pyObject);
        PyBytes_AsStringAndSize(pyUTF8Str,&pszStr,&nLen);
        pszNewStr = (char*)malloc(nLen+1);
        memcpy(pszNewStr,pszStr,nLen+1);
        Py_XDECREF(pyUTF8Str);
        return pszNewStr;
    }
    else if(PyBytes_Check(pyObject))
    {
        char *pszStr;
        char *pszNewStr;
        Py_ssize_t nLen;
        PyBytes_AsStringAndSize(pyObject,&pszStr,&nLen);
        pszNewStr = (char*)malloc(nLen+1);
        memcpy(pszNewStr,pszStr,nLen+1);
        return pszNewStr;
    }
    else
    {
        char *pszStr = (char*)malloc(1);
        pszStr[0] = '\0';
        return pszStr;
    }
#else
    return PyString_AsString(pyObject);
#endif
}

ccsdu2009 2013-12-07 15:48 发表评论

使用Beautiful Soup 解析html文档

ccsdu2009 — Sat, 07 Dec 2013 03:17:00 GMT

不得不承认，有时候使用python比c++方便很多，
就拿解析html来说，使用 Beautiful Soup 就比使用libtidy方便很多 - 当然也有可能是

Beautiful Soup封装的很厉害吧

使用Beautiful Soup的一个例子如下:

from bs4 import BeautifulSoup

soup = BeautifulSoup(open('List.htm'))
for a in soup.find_all('a',class_ = 'link'):
print (a.get('href'))

目的是找出html中class属性为link的a节点对应的href属性字符串
如果是使用c++ libtidy的话
对应的代码如下:

Bool TIDY_CALL tidyFilterCb(TidyDoc tdoc,TidyReportLevel lvl,uint line,uint col,ctmbstr mssg)
{
    return no;
}

void extractContent(TidyNode node,TidyDoc doc);

void parseContent(TidyNode node,TidyDoc doc)
{
    TidyNode child;
    for(child = tidyGetChild(node);child;child = tidyGetNext(child))
    {
        if(tidyNodeIsA(child))
            extractContent(child,doc);
        else
            parseContent(child,doc);
    }
}

void extractContent(TidyNode node,TidyDoc doc)
{
    if(yes == tidyNodeIsA(node))
    {
        TidyAttr cls = tidyAttrGetCLASS(node);
        if(cls != NULL)
        {
            char* value = (char*)tidyAttrValue(cls);
            if(!strcmp(value,"link"))
            {
                TidyAttr href = tidyAttrGetHREF(node);
                if(href != NULL)
                {
                    char* link = (char*)tidyAttrValue(href);
                    printf("link:%s\n",link);
                    return;
                }
            }
        }
    }
    parseContent(node,doc);
}

void tidyParseHtml(char* file)
{
    TidyDoc doc = tidyCreate();
    tidySetReportFilter(doc,tidyFilterCb);
    tidyParseFile(doc,file);
    TidyNode body = tidyGetBody(doc);
    TidyNode child;
    for(child = tidyGetChild(body);child;child = tidyGetNext(child))
    {
        parseContent(child,doc);
    }
    tidyRelease(doc);
}

还是很啰嗦的

当然下面的python代码也能完成任务:

from bs4 import BeautifulSoup

soup = BeautifulSoup(open('List.htm'))
list = soup.select('a[class="link"]')
for a in list:
if a.has_attr('href'):
print (a.get('href'))

如果想分析网页我觉得BeatifulSoup绝对是一个利器
链接:

http://www.crummy.com/software/BeautifulSoup/bs4/doc/

ccsdu2009 2013-12-07 11:17 发表评论

c语言调用Python脚本-2

ccsdu2009 — Tue, 15 Jan 2013 13:39:00 GMT

python代码二段:

call.py

def test():
print 'hello world'

def add(a,b):
return a + b

api.py

import io

def load_test():
    fp = open('call.py','r')
    buffer = ''
    if fp:
        buffer = fp.read()
    fp.close()
    return buffer

cpp代码:

#include <stdio.h>
#include <stdlib.h>
#include <Python.h>

int main(int argc, char *argv[])
{
    Py_Initialize();
    if(!Py_IsInitialized())
    {
        return -1;
    }

    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('./')");
    PyObject* pName;
    PyObject* pModule;
    PyObject* pDict;
    PyObject* pFunc;

    pName = PyString_FromString("api");
    pModule = PyImport_Import(pName);
    if(!pModule)
    {
        printf("can't find call.py");
        getchar();
        return -1;
    }

    pDict = PyModule_GetDict(pModule);
    if(!pDict)
    {
        return -1;
    }

    {
        pFunc = PyDict_GetItemString(pDict,"load_test");
        if(!pFunc || !PyCallable_Check(pFunc))
        {
            printf("can't find function [test]");
            getchar();
            return -1;
        }

        PyObject *pFn = PyObject_CallObject(pFunc,0);
        char* buffer = PyString_AsString(pFn);
        printf("%s\n",buffer);

        PyObject* o = Py_CompileString(buffer,"none",Py_file_input);
        PyObject* m = PyImport_ExecCodeModule("a.a",o);
        PyObject* d = PyModule_GetDict(m);
        pFunc = PyDict_GetItemString(d,"add");
        if(!pFunc || !PyCallable_Check(pFunc))
        {
            printf("can't find function [add]");
            getchar();
            return -1;
        }

        PyObject* args = PyTuple_New(2);
        PyTuple_SetItem(args,0,Py_BuildValue("l",3));
        PyTuple_SetItem(args,1,Py_BuildValue("l",4));
        PyObject *pAdded = PyObject_CallObject(pFunc,args);
        int ret = PyInt_AsLong(pAdded);
        printf("add value:%d\n",ret);
    }

    Py_Finalize();
    system("PAUSE");
    return 0;
}

这段代码和上一篇有点区别
主要区别是从从内存载入python模块然后调用函数
主要部分是这块:

        PyObject* o = Py_CompileString(buffer,"none",Py_file_input);
        PyObject* m = PyImport_ExecCodeModule("a.a",o);
        PyObject* d = PyModule_GetDict(m);

buffer是python源码字符串

在python2.7中执行正常

ccsdu2009 2013-01-15 21:39 发表评论

c语言调用Python脚本

ccsdu2009 — Mon, 14 Jan 2013 13:44:00 GMT

先上python代码:

# call.py

def test():
print 'hello world'

def add(a,b):
return a + b

再上c代码

#include <stdio.h>
#include <stdlib.h>
#include <Python.h>

int main(int argc, char *argv[])
{
    Py_Initialize();
    if(!Py_IsInitialized())
    {
        return -1;
    }

    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('./')");
    PyObject* pName;
    PyObject* pModule;
    PyObject* pDict;
    PyObject* pFunc;

    pName = PyString_FromString("call");
    pModule = PyImport_Import(pName);
    if(!pModule)
    {
        printf("can't find call.py");
        getchar();
        return -1;
    }

    pDict = PyModule_GetDict(pModule);
    if(!pDict)
    {
        return -1;
    }

    {
        pFunc = PyDict_GetItemString(pDict,"test");
        if(!pFunc || !PyCallable_Check(pFunc))
        {
            printf("can't find function [test]");
            getchar();
            return -1;
        }

        PyObject_CallObject(pFunc,0);
    }

    {
        pFunc = PyDict_GetItemString(pDict,"add");
        if(!pFunc || !PyCallable_Check(pFunc))
        {
            printf("can't find function [test]");
            getchar();
            return -1;
        }

        PyObject* args = PyTuple_New(2);
        PyTuple_SetItem(args,0,Py_BuildValue("l",3));
        PyTuple_SetItem(args,1,Py_BuildValue("l",4));
        PyObject *pAdded = PyObject_CallObject(pFunc,args);
        int ret = PyInt_AsLong(pAdded);
        printf("add value:%d\n",ret);
        Py_DECREF(args);
    }

    Py_DECREF(pName);
    Py_DECREF(pDict);
    Py_DECREF(pModule);
    Py_Finalize();
    system("PAUSE");
    return 0;
}

就不做解释了
不过如何从字符串中载入模块？

ccsdu2009 2013-01-14 21:44 发表评论

使用python获取文件夹下文件名字并保存.

ccsdu2009 — Sat, 10 Nov 2012 10:53:00 GMT

import glob
import io

file = open('out.txt','w')

for filename in glob.glob("E:\\*.cpp"):
    str = '@'
    str += filename
    str += '@\n'
    file.write(str)

file.close()

ccsdu2009 2012-11-10 18:53 发表评论

QT学习笔记-6.QApplication和基本控件的使用

ccsdu2009 — Wed, 16 Mar 2011 02:41:00 GMT

接上文，这篇学学QT中基本控件的使用和QApplication对象
1.什么是QApplication?
文档说明:The QApplication class manages the GUI application's control flow and main settings.
Application类管理GUI程序控制流和主要参数设置
QApplication继承于QCoreApplication。后者提供了控制台程序的事件流
2.基本控件的使用例子:

#include <QApplication>
#include <QLabel>
#include <QPalette>

#define QT_HTML

QLabel* label = NULL;

void initlabel()
{
#ifndef QT_HTML
label = new QLabel("Hello Qt!");
#else
label = new QLabel("

HelloQt!

");
#endif
    //! set size
    label->setBaseSize(64,48);
    //! set alignment
    label->setAlignment(Qt::AlignHCenter);

    //! sht background color
    QColor bk(100,100,125);
    QPalette palette(bk);
    label->setPalette(palette);
}

int main(int argc, char *argv[])
{
    QApplication app(argc, argv);
    app.setApplicationName("QT Test");
    initlabel();
    label->show();
    return app.exec();
}

QLabel是QT中的标签控件它具有控件的一般属性比如设置大小setBaseSite,设置对齐格式，当然也可以设置背景色或者图片-这都是通过QPalette调色板来实现的
需要说明的是QT中的控件文本可以使用Html语法的文本来操作具体如上。
那觉这个功能比较给力！
3.那么什么是QPalette?
QPalette负责控制控件状态的颜色组-注意是控件状态。
那么对一个控件每个状态的颜色都可以是不一样的咯
至于QPalette的详细功能和使用方法以后需要的时候再看吧
4.基本的信号链接使用例子

#include <QApplication>
#include <QPushButton>

int main(int argc, char *argv[])
{
    QApplication app(argc, argv);
    QPushButton *button = new QPushButton("Quit");
    //! when click button, app exit.
    QObject::connect(button, SIGNAL(clicked()),&app, SLOT(quit()));
    button->show();
    return app.exec();
}

5.一个复杂点的例子

编译运行可以看出QWidget中默认的布局管理器是竖直向下排列的
在QT中可以通过setWindowIcon来设置窗体图标
通过setWindowTitle设置窗体标题
6.加上布局管理器和信号连接的话代码大致应该是这个样子

#include <QApplication>
#include <QHBoxLayout>
#include <QSlider>
#include <QSpinBox>
#include <QIcon>

int main(int argc, char *argv[])
{
    QApplication app(argc, argv);
    QWidget* widget = new QWidget;
    QIcon icon("config.png");
    widget->setWindowIcon(icon);
    widget->setWindowTitle("Using QT");

    QSlider* slider = new QSlider(widget);
    slider->setRange(0,99);

    QSpinBox* spinbox = new QSpinBox(widget);
    spinbox->setRange(0,99);

    QHBoxLayout* layout = new QHBoxLayout;
    layout->addWidget(spinbox);
    //! adjust slider's direction
    slider->setOrientation(Qt::Horizontal);
    layout->addWidget(slider);

    spinbox->setValue(28);

    //! connect signals and slots
    QObject::connect(spinbox, SIGNAL(valueChanged(int)),slider,SLOT(setValue(int)));
    QObject::connect(slider,SIGNAL(valueChanged(int)),spinbox,SLOT(setValue(int)));

    widget->setLayout(layout);
    widget->show();
    return app.exec();
}

需要说明的是在这里QSlider,QPinBox控件是互动
编译程序并运行界面如下:

这是关于QT的第六篇笔记
总结下吧
QT功能还是很强大贴心的
比较容易上手
不过有2点我感觉不大舒服的地方是对这个变量命名格式有点不大喜欢
比如setValue我喜欢写成SetValue.
仅此而已

ccsdu2009 2011-03-16 10:41 发表评论